mirror of
https://github.com/zeromq/libzmq.git
synced 2024-12-14 02:57:47 +01:00
da31917f4f
Relicense permission collected from all relevant authors as tallied at: https://github.com/rlenferink/libzmq-relicense/blob/master/checklist.md The relicense grants are collected under RELICENSE/ and will be moved to the above repository in a later commit. Fixes https://github.com/zeromq/libzmq/issues/2376
280 lines
8.4 KiB
C++
280 lines
8.4 KiB
C++
/* SPDX-License-Identifier: MPL-2.0 */
|
|
|
|
#include "testutil.hpp"
|
|
#include "testutil_unity.hpp"
|
|
|
|
SETUP_TEARDOWN_TESTCONTEXT
|
|
|
|
const int MAX_SENDS = 10000;
|
|
|
|
enum TestType
|
|
{
|
|
BIND_FIRST,
|
|
CONNECT_FIRST
|
|
};
|
|
|
|
void test_defaults ()
|
|
{
|
|
// Set up bind socket
|
|
void *bind_socket = test_context_socket (ZMQ_PULL);
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_bind (bind_socket, "inproc://a"));
|
|
|
|
// Set up connect socket
|
|
void *connect_socket = test_context_socket (ZMQ_PUSH);
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (connect_socket, "inproc://a"));
|
|
|
|
// Send until we block
|
|
int send_count = 0;
|
|
while (send_count < MAX_SENDS
|
|
&& zmq_send (connect_socket, NULL, 0, ZMQ_DONTWAIT) == 0)
|
|
++send_count;
|
|
|
|
msleep (SETTLE_TIME);
|
|
|
|
// Now receive all sent messages
|
|
int recv_count = 0;
|
|
while (zmq_recv (bind_socket, NULL, 0, ZMQ_DONTWAIT) == 0)
|
|
++recv_count;
|
|
|
|
TEST_ASSERT_EQUAL_INT (send_count, recv_count);
|
|
|
|
// Clean up
|
|
test_context_socket_close (connect_socket);
|
|
test_context_socket_close (bind_socket);
|
|
|
|
// Default values are 1000 on send and 1000 one receive, so 2000 total
|
|
TEST_ASSERT_EQUAL_INT (2000, send_count);
|
|
}
|
|
|
|
int count_msg (int send_hwm_, int recv_hwm_, TestType test_type_)
|
|
{
|
|
void *bind_socket;
|
|
void *connect_socket;
|
|
if (test_type_ == BIND_FIRST) {
|
|
// Set up bind socket
|
|
bind_socket = test_context_socket (ZMQ_PULL);
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_setsockopt (
|
|
bind_socket, ZMQ_RCVHWM, &recv_hwm_, sizeof (recv_hwm_)));
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_bind (bind_socket, "inproc://a"));
|
|
|
|
// Set up connect socket
|
|
connect_socket = test_context_socket (ZMQ_PUSH);
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_setsockopt (
|
|
connect_socket, ZMQ_SNDHWM, &send_hwm_, sizeof (send_hwm_)));
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (connect_socket, "inproc://a"));
|
|
|
|
// we must wait for the connect to succeed here, unfortunately we don't
|
|
// have monitoring events for inproc, so we just hope SETTLE_TIME suffices
|
|
msleep (SETTLE_TIME);
|
|
} else {
|
|
// Set up connect socket
|
|
connect_socket = test_context_socket (ZMQ_PUSH);
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_setsockopt (
|
|
connect_socket, ZMQ_SNDHWM, &send_hwm_, sizeof (send_hwm_)));
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (connect_socket, "inproc://a"));
|
|
|
|
// Set up bind socket
|
|
bind_socket = test_context_socket (ZMQ_PULL);
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_setsockopt (
|
|
bind_socket, ZMQ_RCVHWM, &recv_hwm_, sizeof (recv_hwm_)));
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_bind (bind_socket, "inproc://a"));
|
|
}
|
|
|
|
// Send until we block
|
|
int send_count = 0;
|
|
while (send_count < MAX_SENDS
|
|
&& zmq_send (connect_socket, NULL, 0, ZMQ_DONTWAIT) == 0)
|
|
++send_count;
|
|
|
|
// Now receive all sent messages
|
|
int recv_count = 0;
|
|
while (zmq_recv (bind_socket, NULL, 0, ZMQ_DONTWAIT) == 0)
|
|
++recv_count;
|
|
|
|
TEST_ASSERT_EQUAL_INT (send_count, recv_count);
|
|
|
|
// Now it should be possible to send one more.
|
|
send_string_expect_success (connect_socket, NULL, 0);
|
|
|
|
// Consume the remaining message.
|
|
recv_string_expect_success (bind_socket, NULL, 0);
|
|
|
|
// Clean up
|
|
test_context_socket_close (connect_socket);
|
|
test_context_socket_close (bind_socket);
|
|
|
|
return send_count;
|
|
}
|
|
|
|
int test_inproc_bind_first (int send_hwm_, int recv_hwm_)
|
|
{
|
|
return count_msg (send_hwm_, recv_hwm_, BIND_FIRST);
|
|
}
|
|
|
|
int test_inproc_connect_first (int send_hwm_, int recv_hwm_)
|
|
{
|
|
return count_msg (send_hwm_, recv_hwm_, CONNECT_FIRST);
|
|
}
|
|
|
|
int test_inproc_connect_and_close_first (int send_hwm_, int recv_hwm_)
|
|
{
|
|
// Set up connect socket
|
|
void *connect_socket = test_context_socket (ZMQ_PUSH);
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_setsockopt (connect_socket, ZMQ_SNDHWM,
|
|
&send_hwm_, sizeof (send_hwm_)));
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (connect_socket, "inproc://a"));
|
|
|
|
// Send until we block
|
|
int send_count = 0;
|
|
while (send_count < MAX_SENDS
|
|
&& zmq_send (connect_socket, NULL, 0, ZMQ_DONTWAIT) == 0)
|
|
++send_count;
|
|
|
|
// Close connect
|
|
test_context_socket_close (connect_socket);
|
|
|
|
// Set up bind socket
|
|
void *bind_socket = test_context_socket (ZMQ_PULL);
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
zmq_setsockopt (bind_socket, ZMQ_RCVHWM, &recv_hwm_, sizeof (recv_hwm_)));
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_bind (bind_socket, "inproc://a"));
|
|
|
|
// Now receive all sent messages
|
|
int recv_count = 0;
|
|
while (zmq_recv (bind_socket, NULL, 0, ZMQ_DONTWAIT) == 0)
|
|
++recv_count;
|
|
|
|
TEST_ASSERT_EQUAL_INT (send_count, recv_count);
|
|
|
|
// Clean up
|
|
test_context_socket_close (bind_socket);
|
|
|
|
return send_count;
|
|
}
|
|
|
|
int test_inproc_bind_and_close_first (int send_hwm_, int /* recv_hwm */)
|
|
{
|
|
// Set up bind socket
|
|
void *bind_socket = test_context_socket (ZMQ_PUSH);
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
zmq_setsockopt (bind_socket, ZMQ_SNDHWM, &send_hwm_, sizeof (send_hwm_)));
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_bind (bind_socket, "inproc://a"));
|
|
|
|
// Send until we block
|
|
int send_count = 0;
|
|
while (send_count < MAX_SENDS
|
|
&& zmq_send (bind_socket, NULL, 0, ZMQ_DONTWAIT) == 0)
|
|
++send_count;
|
|
|
|
// Close bind
|
|
test_context_socket_close (bind_socket);
|
|
|
|
/* TODO Can't currently do connect without then wiring up a bind as things hang, this needs top be fixed.
|
|
// Set up connect socket
|
|
void *connect_socket = test_context_socket (ZMQ_PULL);
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_setsockopt (connect_socket, ZMQ_RCVHWM, &recv_hwm, sizeof (recv_hwm)));
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (connect_socket, "inproc://a"));
|
|
|
|
// Now receive all sent messages
|
|
int recv_count = 0;
|
|
while (zmq_recv (connect_socket, NULL, 0, ZMQ_DONTWAIT) == 0)
|
|
++recv_count;
|
|
|
|
TEST_ASSERT_EQUAL_INT(send_count, recv_count);
|
|
*/
|
|
|
|
// Clean up
|
|
//test_context_socket_close (connect_socket);
|
|
|
|
return send_count;
|
|
}
|
|
|
|
void test_infinite_both_inproc_bind_first ()
|
|
{
|
|
int count = test_inproc_bind_first (0, 0);
|
|
TEST_ASSERT_EQUAL_INT (MAX_SENDS, count);
|
|
}
|
|
|
|
void test_infinite_both_inproc_connect_first ()
|
|
{
|
|
int count = test_inproc_connect_first (0, 0);
|
|
TEST_ASSERT_EQUAL_INT (MAX_SENDS, count);
|
|
}
|
|
|
|
void test_infinite_receive_inproc_bind_first ()
|
|
{
|
|
int count = test_inproc_bind_first (1, 0);
|
|
TEST_ASSERT_EQUAL_INT (MAX_SENDS, count);
|
|
}
|
|
|
|
void test_infinite_receive_inproc_connect_first ()
|
|
{
|
|
int count = test_inproc_connect_first (1, 0);
|
|
TEST_ASSERT_EQUAL_INT (MAX_SENDS, count);
|
|
}
|
|
|
|
void test_infinite_send_inproc_bind_first ()
|
|
{
|
|
int count = test_inproc_bind_first (0, 1);
|
|
TEST_ASSERT_EQUAL_INT (MAX_SENDS, count);
|
|
}
|
|
|
|
void test_infinite_send_inproc_connect_first ()
|
|
{
|
|
int count = test_inproc_connect_first (0, 1);
|
|
TEST_ASSERT_EQUAL_INT (MAX_SENDS, count);
|
|
}
|
|
|
|
void test_finite_both_bind_first ()
|
|
{
|
|
// Send and recv buffers hwm 1, so total that can be queued is 2
|
|
int count = test_inproc_bind_first (1, 1);
|
|
TEST_ASSERT_EQUAL_INT (2, count);
|
|
}
|
|
void test_finite_both_connect_first ()
|
|
{
|
|
// Send and recv buffers hwm 1, so total that can be queued is 2
|
|
int count = test_inproc_connect_first (1, 1);
|
|
TEST_ASSERT_EQUAL_INT (2, count);
|
|
}
|
|
|
|
void test_infinite_recv_connect_and_close_first ()
|
|
{
|
|
// Send hwm of 1, send before bind so total that can be queued is 1
|
|
int count = test_inproc_connect_and_close_first (1, 0);
|
|
TEST_ASSERT_EQUAL_INT (1, count);
|
|
}
|
|
|
|
void test_infinite_recv_bind_and_close_first ()
|
|
{
|
|
// Send hwm of 1, send from bind side before connect so total that can be queued should be 1,
|
|
// however currently all messages get thrown away before the connect. BUG?
|
|
/*int count = */ test_inproc_bind_and_close_first (1, 0);
|
|
// TEST_ASSERT_EQUAL_INT (1, count);
|
|
}
|
|
|
|
int main (void)
|
|
{
|
|
setup_test_environment ();
|
|
|
|
UNITY_BEGIN ();
|
|
RUN_TEST (test_defaults);
|
|
|
|
RUN_TEST (test_infinite_both_inproc_bind_first);
|
|
RUN_TEST (test_infinite_both_inproc_connect_first);
|
|
|
|
RUN_TEST (test_infinite_receive_inproc_bind_first);
|
|
RUN_TEST (test_infinite_receive_inproc_connect_first);
|
|
|
|
RUN_TEST (test_infinite_send_inproc_bind_first);
|
|
RUN_TEST (test_infinite_send_inproc_connect_first);
|
|
|
|
RUN_TEST (test_finite_both_bind_first);
|
|
RUN_TEST (test_finite_both_connect_first);
|
|
|
|
RUN_TEST (test_infinite_recv_connect_and_close_first);
|
|
RUN_TEST (test_infinite_recv_bind_and_close_first);
|
|
|
|
return UNITY_END ();
|
|
}
|