mirror of
https://github.com/zeromq/libzmq.git
synced 2024-12-26 23:01:04 +08: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
87 lines
2.2 KiB
C++
87 lines
2.2 KiB
C++
/* SPDX-License-Identifier: MPL-2.0 */
|
|
|
|
#include "testutil.hpp"
|
|
#include "testutil_unity.hpp"
|
|
|
|
SETUP_TEARDOWN_TESTCONTEXT
|
|
|
|
void test_peer ()
|
|
{
|
|
size_t len = MAX_SOCKET_STRING;
|
|
char my_endpoint[MAX_SOCKET_STRING];
|
|
|
|
void *peer1 = test_context_socket (ZMQ_PEER);
|
|
bind_loopback (peer1, false, my_endpoint, len);
|
|
|
|
void *peer2 = test_context_socket (ZMQ_PEER);
|
|
uint32_t peer1_routing_id = zmq_connect_peer (peer2, my_endpoint);
|
|
TEST_ASSERT_NOT_EQUAL (0, peer1_routing_id);
|
|
|
|
{
|
|
zmq_msg_t msg;
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_msg_init_size (&msg, 1));
|
|
|
|
char *data = static_cast<char *> (zmq_msg_data (&msg));
|
|
data[0] = 1;
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
zmq_msg_set_routing_id (&msg, peer1_routing_id));
|
|
|
|
int rc = zmq_msg_send (&msg, peer2, 0);
|
|
TEST_ASSERT_EQUAL_INT (1, rc);
|
|
}
|
|
|
|
uint32_t peer2_routing_id;
|
|
{
|
|
zmq_msg_t msg;
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_msg_init (&msg));
|
|
|
|
int rc = TEST_ASSERT_SUCCESS_ERRNO (zmq_msg_recv (&msg, peer1, 0));
|
|
TEST_ASSERT_EQUAL_INT (1, rc);
|
|
|
|
peer2_routing_id = zmq_msg_routing_id (&msg);
|
|
TEST_ASSERT_NOT_EQUAL (0, peer2_routing_id);
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_msg_close (&msg));
|
|
}
|
|
|
|
{
|
|
zmq_msg_t msg;
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_msg_init_size (&msg, 1));
|
|
|
|
char *data = static_cast<char *> (zmq_msg_data (&msg));
|
|
data[0] = 2;
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (
|
|
zmq_msg_set_routing_id (&msg, peer2_routing_id));
|
|
|
|
int rc = zmq_msg_send (&msg, peer1, 0);
|
|
TEST_ASSERT_EQUAL_INT (1, rc);
|
|
}
|
|
|
|
{
|
|
zmq_msg_t msg;
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_msg_init (&msg));
|
|
|
|
int rc = zmq_msg_recv (&msg, peer2, 0);
|
|
TEST_ASSERT_EQUAL_INT (1, rc);
|
|
|
|
uint32_t routing_id = zmq_msg_routing_id (&msg);
|
|
TEST_ASSERT_EQUAL_UINT32 (peer1_routing_id, routing_id);
|
|
|
|
TEST_ASSERT_SUCCESS_ERRNO (zmq_msg_close (&msg));
|
|
}
|
|
|
|
test_context_socket_close (peer1);
|
|
test_context_socket_close (peer2);
|
|
}
|
|
|
|
int main (void)
|
|
{
|
|
setup_test_environment ();
|
|
|
|
UNITY_BEGIN ();
|
|
RUN_TEST (test_peer);
|
|
return UNITY_END ();
|
|
}
|