mirror of
https://github.com/zeromq/libzmq.git
synced 2025-10-24 17:30:47 +02:00
Make pipeline/reqrep tests try tcp:// endpoints.
The inproc:// endpoints sometimes use different code paths so testing with regular tcp:// endpoints as well can show different issues.
This commit is contained in:
@@ -21,6 +21,9 @@
|
||||
#include <stdlib.h>
|
||||
#include "testutil.hpp"
|
||||
|
||||
const char *bind_address = 0;
|
||||
const char *connect_address = 0;
|
||||
|
||||
void test_fair_queue_in (void *ctx)
|
||||
{
|
||||
void *receiver = zmq_socket (ctx, ZMQ_ROUTER);
|
||||
@@ -30,7 +33,7 @@ void test_fair_queue_in (void *ctx)
|
||||
int rc = zmq_setsockopt (receiver, ZMQ_RCVTIMEO, &timeout, sizeof(int));
|
||||
assert (rc == 0);
|
||||
|
||||
rc = zmq_bind (receiver, "inproc://a");
|
||||
rc = zmq_bind (receiver, bind_address);
|
||||
assert (rc == 0);
|
||||
|
||||
const size_t N = 5;
|
||||
@@ -49,7 +52,7 @@ void test_fair_queue_in (void *ctx)
|
||||
assert (rc == 0);
|
||||
free (str);
|
||||
|
||||
rc = zmq_connect (senders[i], "inproc://a");
|
||||
rc = zmq_connect (senders[i], connect_address);
|
||||
assert (rc == 0);
|
||||
}
|
||||
|
||||
@@ -63,32 +66,43 @@ void test_fair_queue_in (void *ctx)
|
||||
s_send_seq (senders[0], "M", SEQ_END);
|
||||
s_recv_seq (receiver, "A", "M", SEQ_END);
|
||||
|
||||
int sum = 0;
|
||||
|
||||
// send N requests
|
||||
for (size_t i = 0; i < N; ++i)
|
||||
{
|
||||
s_send_seq (senders[i], "M", SEQ_END);
|
||||
sum += 'A' + i;
|
||||
}
|
||||
|
||||
assert (sum == N*'A' + N*(N-1)/2);
|
||||
|
||||
// handle N requests
|
||||
for (size_t i = 0; i < N; ++i)
|
||||
{
|
||||
char *str = strdup("A");
|
||||
str[0] += i;
|
||||
s_recv_seq (receiver, str, "M", SEQ_END);
|
||||
free (str);
|
||||
rc = zmq_msg_recv (&msg, receiver, 0);
|
||||
assert (rc == 2);
|
||||
const char *id = (const char *)zmq_msg_data (&msg);
|
||||
sum -= id[0];
|
||||
|
||||
s_recv_seq (receiver, "M", SEQ_END);
|
||||
}
|
||||
|
||||
assert (sum == 0);
|
||||
|
||||
rc = zmq_msg_close (&msg);
|
||||
assert (rc == 0);
|
||||
|
||||
rc = zmq_close (receiver);
|
||||
assert (rc == 0);
|
||||
close_zero_linger (receiver);
|
||||
|
||||
for (size_t i = 0; i < N; ++i)
|
||||
{
|
||||
rc = zmq_close (senders[i]);
|
||||
assert (rc == 0);
|
||||
close_zero_linger (senders[i]);
|
||||
}
|
||||
|
||||
// Wait for disconnects.
|
||||
rc = zmq_poll (0, 0, 100);
|
||||
assert (rc == 0);
|
||||
}
|
||||
|
||||
void test_destroy_queue_on_disconnect (void *ctx)
|
||||
@@ -100,7 +114,7 @@ void test_destroy_queue_on_disconnect (void *ctx)
|
||||
int rc = zmq_setsockopt (A, ZMQ_ROUTER_MANDATORY, &enabled, sizeof(enabled));
|
||||
assert (rc == 0);
|
||||
|
||||
rc = zmq_bind (A, "inproc://d");
|
||||
rc = zmq_bind (A, bind_address);
|
||||
assert (rc == 0);
|
||||
|
||||
void *B = zmq_socket (ctx, ZMQ_DEALER);
|
||||
@@ -109,20 +123,26 @@ void test_destroy_queue_on_disconnect (void *ctx)
|
||||
rc = zmq_setsockopt (B, ZMQ_IDENTITY, "B", 2);
|
||||
assert (rc == 0);
|
||||
|
||||
rc = zmq_connect (B, "inproc://d");
|
||||
rc = zmq_connect (B, connect_address);
|
||||
assert (rc == 0);
|
||||
|
||||
// Wait for connection.
|
||||
rc = zmq_poll (0, 0, 100);
|
||||
assert (rc == 0);
|
||||
|
||||
// Send a message in both directions
|
||||
s_send_seq (A, "B", "ABC", SEQ_END);
|
||||
s_send_seq (B, "DEF", SEQ_END);
|
||||
|
||||
rc = zmq_disconnect (B, "inproc://d");
|
||||
rc = zmq_disconnect (B, connect_address);
|
||||
assert (rc == 0);
|
||||
|
||||
// Disconnect may take time and need command processing.
|
||||
zmq_pollitem_t poller[2] = { { A, 0, 0, 0 }, { B, 0, 0, 0 } };
|
||||
rc = zmq_poll (poller, 2, 100);
|
||||
assert (rc == 0);
|
||||
rc = zmq_poll (poller, 2, 100);
|
||||
assert (rc == 0);
|
||||
|
||||
// No messages should be available, sending should fail.
|
||||
zmq_msg_t msg;
|
||||
@@ -137,7 +157,7 @@ void test_destroy_queue_on_disconnect (void *ctx)
|
||||
assert (errno == EAGAIN);
|
||||
|
||||
// After a reconnect of B, the messages should still be gone
|
||||
rc = zmq_connect (B, "inproc://d");
|
||||
rc = zmq_connect (B, connect_address);
|
||||
assert (rc == 0);
|
||||
|
||||
rc = zmq_msg_recv (&msg, A, ZMQ_DONTWAIT);
|
||||
@@ -151,10 +171,11 @@ void test_destroy_queue_on_disconnect (void *ctx)
|
||||
rc = zmq_msg_close (&msg);
|
||||
assert (rc == 0);
|
||||
|
||||
rc = zmq_close (A);
|
||||
assert (rc == 0);
|
||||
close_zero_linger (A);
|
||||
close_zero_linger (B);
|
||||
|
||||
rc = zmq_close (B);
|
||||
// Wait for disconnects.
|
||||
rc = zmq_poll (0, 0, 100);
|
||||
assert (rc == 0);
|
||||
}
|
||||
|
||||
@@ -164,14 +185,22 @@ int main ()
|
||||
void *ctx = zmq_ctx_new ();
|
||||
assert (ctx);
|
||||
|
||||
// SHALL receive incoming messages from its peers using a fair-queuing
|
||||
// strategy.
|
||||
test_fair_queue_in (ctx);
|
||||
const char *binds[] = { "inproc://a", "tcp://*:5555" };
|
||||
const char *connects[] = { "inproc://a", "tcp://localhost:5555" };
|
||||
|
||||
// SHALL create a double queue when a peer connects to it. If this peer
|
||||
// disconnects, the ROUTER socket SHALL destroy its double queue and SHALL
|
||||
// discard any messages it contains.
|
||||
test_destroy_queue_on_disconnect (ctx);
|
||||
for (int i = 0; i < 2; ++i) {
|
||||
bind_address = binds[i];
|
||||
connect_address = connects[i];
|
||||
|
||||
// SHALL receive incoming messages from its peers using a fair-queuing
|
||||
// strategy.
|
||||
test_fair_queue_in (ctx);
|
||||
|
||||
// SHALL create a double queue when a peer connects to it. If this peer
|
||||
// disconnects, the ROUTER socket SHALL destroy its double queue and SHALL
|
||||
// discard any messages it contains.
|
||||
test_destroy_queue_on_disconnect (ctx);
|
||||
}
|
||||
|
||||
int rc = zmq_ctx_term (ctx);
|
||||
assert (rc == 0);
|
||||
|
||||
Reference in New Issue
Block a user