test_probe_router.cpp
Go to the documentation of this file.
1 /* SPDX-License-Identifier: MPL-2.0 */
2 
3 #include "testutil.hpp"
4 #include "testutil_unity.hpp"
5 
6 SETUP_TEARDOWN_TESTCONTEXT
7 
8 void test_probe_router_router ()
9 {
10  // Create server and bind to endpoint
11  void *server = test_context_socket (ZMQ_ROUTER);
12 
13  char my_endpoint[MAX_SOCKET_STRING];
14  bind_loopback_ipv4 (server, my_endpoint, sizeof (my_endpoint));
15 
16  // Create client and connect to server, doing a probe
17  void *client = test_context_socket (ZMQ_ROUTER);
18  TEST_ASSERT_SUCCESS_ERRNO (zmq_setsockopt (client, ZMQ_ROUTING_ID, "X", 1));
19  int probe = 1;
20  TEST_ASSERT_SUCCESS_ERRNO (
21  zmq_setsockopt (client, ZMQ_PROBE_ROUTER, &probe, sizeof (probe)));
22  TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (client, my_endpoint));
23 
24  // We expect a routing id=X + empty message from client
25  recv_string_expect_success (server, "X", 0);
26  unsigned char buffer[255];
27  TEST_ASSERT_EQUAL_INT (
28  0, TEST_ASSERT_SUCCESS_ERRNO (zmq_recv (server, buffer, 255, 0)));
29 
30  // Send a message to client now
31  send_string_expect_success (server, "X", ZMQ_SNDMORE);
32  send_string_expect_success (server, "Hello", 0);
33 
34  // receive the routing ID, which is auto-generated in this case, since the
35  // peer did not set one explicitly
36  TEST_ASSERT_EQUAL_INT (
37  5, TEST_ASSERT_SUCCESS_ERRNO (zmq_recv (client, buffer, 255, 0)));
38 
39  recv_string_expect_success (client, "Hello", 0);
40 
41  test_context_socket_close (server);
42  test_context_socket_close (client);
43 }
44 
45 void test_probe_router_dealer ()
46 {
47  // Create server and bind to endpoint
48  void *server = test_context_socket (ZMQ_ROUTER);
49 
50  char my_endpoint[MAX_SOCKET_STRING];
51  bind_loopback_ipv4 (server, my_endpoint, sizeof (my_endpoint));
52 
53  // Create client and connect to server, doing a probe
54  void *client = test_context_socket (ZMQ_DEALER);
55  TEST_ASSERT_SUCCESS_ERRNO (zmq_setsockopt (client, ZMQ_ROUTING_ID, "X", 1));
56  int probe = 1;
57  TEST_ASSERT_SUCCESS_ERRNO (
58  zmq_setsockopt (client, ZMQ_PROBE_ROUTER, &probe, sizeof (probe)));
59  TEST_ASSERT_SUCCESS_ERRNO (zmq_connect (client, my_endpoint));
60 
61  // We expect a routing id=X + empty message from client
62  recv_string_expect_success (server, "X", 0);
63  unsigned char buffer[255];
64  TEST_ASSERT_SUCCESS_ERRNO (zmq_recv (server, buffer, 255, 0));
65 
66  // Send a message to client now
67  send_string_expect_success (server, "X", ZMQ_SNDMORE);
68  send_string_expect_success (server, "Hello", 0);
69 
70  recv_string_expect_success (client, "Hello", 0);
71 
72  test_context_socket_close (server);
73  test_context_socket_close (client);
74 }
75 
76 int main ()
77 {
78  setup_test_environment ();
79 
80  UNITY_BEGIN ();
81 
82  RUN_TEST (test_probe_router_router);
83  RUN_TEST (test_probe_router_dealer);
84 
85  return UNITY_END ();
86 }
UNITY_END
return UNITY_END()
RUN_TEST
#define RUN_TEST(func)
Definition: unity_internals.h:615
main
int main()
Definition: test_probe_router.cpp:76
SETUP_TEARDOWN_TESTCONTEXT
#define SETUP_TEARDOWN_TESTCONTEXT
Definition: testutil_unity.hpp:172
bind_loopback_ipv4
void bind_loopback_ipv4(void *socket_, char *my_endpoint_, size_t len_)
Definition: testutil_unity.cpp:246
client
void client(int num)
Definition: test_multithread.cpp:134
zmq_connect
ZMQ_EXPORT int zmq_connect(void *s_, const char *addr_)
Definition: zmq.cpp:307
testutil_unity.hpp
ZMQ_DEALER
#define ZMQ_DEALER
Definition: zmq.h:263
zmq_setsockopt
ZMQ_EXPORT int zmq_setsockopt(void *s_, int option_, const void *optval_, size_t optvallen_)
Definition: zmq.cpp:250
testutil.hpp
ZMQ_ROUTER
#define ZMQ_ROUTER
Definition: zmq.h:264
test_probe_router_router
SETUP_TEARDOWN_TESTCONTEXT void test_probe_router_router()
Definition: test_probe_router.cpp:8
my_endpoint
char my_endpoint[MAX_SOCKET_STRING]
Definition: test_security_curve.cpp:31
MAX_SOCKET_STRING
#define MAX_SOCKET_STRING
Definition: libzmq/tests/testutil.hpp:35
buffer
Definition: buffer_processor.h:43
test_context_socket
void * test_context_socket(int type_)
Definition: testutil_unity.cpp:200
server
void * server
Definition: test_security_curve.cpp:29
TEST_ASSERT_EQUAL_INT
#define TEST_ASSERT_EQUAL_INT(expected, actual)
Definition: unity.h:128
send_string_expect_success
void send_string_expect_success(void *socket_, const char *str_, int flags_)
Definition: testutil_unity.cpp:94
zmq_recv
ZMQ_EXPORT int zmq_recv(void *s_, void *buf_, size_t len_, int flags_)
Definition: zmq.cpp:487
recv_string_expect_success
void recv_string_expect_success(void *socket_, const char *str_, int flags_)
Definition: testutil_unity.cpp:101
ZMQ_PROBE_ROUTER
#define ZMQ_PROBE_ROUTER
Definition: zmq.h:316
setup_test_environment
void setup_test_environment(int timeout_seconds_)
Definition: testutil.cpp:201
UNITY_BEGIN
UNITY_BEGIN()
ZMQ_SNDMORE
#define ZMQ_SNDMORE
Definition: zmq.h:359
ZMQ_ROUTING_ID
#define ZMQ_ROUTING_ID
Definition: zmq.h:277
test_context_socket_close
void * test_context_socket_close(void *socket_)
Definition: testutil_unity.cpp:208
test_probe_router_dealer
void test_probe_router_dealer()
Definition: test_probe_router.cpp:45
TEST_ASSERT_SUCCESS_ERRNO
#define TEST_ASSERT_SUCCESS_ERRNO(expr)
Definition: proxy_thr.cpp:47

libaditof
Author(s):
autogenerated on Wed May 21 2025 02:06:59