diff options
Diffstat (limited to 'src/protocol/reqrep0')
| -rw-r--r-- | src/protocol/reqrep0/rep_test.c | 489 | ||||
| -rw-r--r-- | src/protocol/reqrep0/req_test.c | 815 | ||||
| -rw-r--r-- | src/protocol/reqrep0/xrep_test.c | 397 | ||||
| -rw-r--r-- | src/protocol/reqrep0/xreq_test.c | 329 |
4 files changed, 971 insertions, 1059 deletions
diff --git a/src/protocol/reqrep0/rep_test.c b/src/protocol/reqrep0/rep_test.c index a79a342b..5a47e67a 100644 --- a/src/protocol/reqrep0/rep_test.c +++ b/src/protocol/reqrep0/rep_test.c @@ -7,15 +7,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/reqrep0/rep.h> -#include <nng/protocol/reqrep0/req.h> -#include <nng/supplemental/util/platform.h> - -#include <acutest.h> -#include <testutil.h> +#include <nuts.h> static void test_rep_identity(void) @@ -25,16 +17,16 @@ test_rep_identity(void) char * n1; char * n2; - TEST_NNG_PASS(nng_rep0_open(&s)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PROTO, &p1)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PEER, &p2)); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PROTONAME, &n1)); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PEERNAME, &n2)); - TEST_NNG_PASS(nng_close(s)); - TEST_CHECK(p1 == NNG_REP0_SELF); - TEST_CHECK(p2 == NNG_REP0_PEER); - TEST_CHECK(strcmp(n1, NNG_REP0_SELF_NAME) == 0); - TEST_CHECK(strcmp(n2, NNG_REP0_PEER_NAME) == 0); + NUTS_PASS(nng_rep0_open(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p1)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p2)); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n1)); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n2)); + NUTS_CLOSE(s); + NUTS_TRUE(p1 == NNG_REP0_SELF); + NUTS_TRUE(p2 == NNG_REP0_PEER); + NUTS_MATCH(n1, NNG_REP0_SELF_NAME); + NUTS_MATCH(n2, NNG_REP0_PEER_NAME); nng_strfree(n1); nng_strfree(n2); } @@ -45,11 +37,11 @@ test_rep_send_bad_state(void) nng_socket rep; nng_msg * msg = NULL; - TEST_CHECK(nng_rep0_open(&rep) == 0); - TEST_CHECK(nng_msg_alloc(&msg, 0) == 0); - TEST_CHECK(nng_sendmsg(rep, msg, 0) == NNG_ESTATE); + NUTS_TRUE(nng_rep0_open(&rep) == 0); + NUTS_TRUE(nng_msg_alloc(&msg, 0) == 0); + NUTS_TRUE(nng_sendmsg(rep, msg, 0) == NNG_ESTATE); nng_msg_free(msg); - TEST_CHECK(nng_close(rep) == 0); + NUTS_CLOSE(rep); } void @@ -59,33 +51,33 @@ test_rep_poll_writeable(void) nng_socket req; nng_socket rep; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_getopt_int(rep, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_get_int(rep, NNG_OPT_SENDFD, &fd)); + NUTS_TRUE(fd >= 0); // Not writable before connect. - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); // Still not writable. - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); // If we get a job, *then* we become writable - TEST_NNG_SEND_STR(req, "abc"); - TEST_NNG_RECV_STR(rep, "abc"); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_SEND(req, "abc"); + NUTS_RECV(rep, "abc"); + NUTS_TRUE(nuts_poll_fd(fd) == true); // And is no longer writable once we send a message - TEST_NNG_SEND_STR(rep, "def"); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_SEND(rep, "def"); + NUTS_TRUE(nuts_poll_fd(fd) == false); // Even after receiving it - TEST_NNG_RECV_STR(req, "def"); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_RECV(req, "def"); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -96,34 +88,34 @@ test_rep_poll_readable(void) nng_socket rep; nng_msg * msg; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_getopt_int(rep, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_get_int(rep, NNG_OPT_RECVFD, &fd)); + NUTS_TRUE(fd >= 0); // Not readable if not connected! - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); // Even after connect (no message yet) - TEST_NNG_PASS(testutil_marry(req, rep)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(req, rep); + NUTS_TRUE(nuts_poll_fd(fd) == false); // But once we send messages, it is. // We have to send a request, in order to send a reply. - TEST_NNG_SEND_STR(req, "abc"); - testutil_sleep(100); + NUTS_SEND(req, "abc"); + NUTS_SLEEP(100); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); // and receiving makes it no longer ready - TEST_NNG_PASS(nng_recvmsg(rep, &msg, 0)); + NUTS_PASS(nng_recvmsg(rep, &msg, 0)); nng_msg_free(msg); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); // TODO verify unsolicited response - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -133,14 +125,12 @@ test_rep_context_no_poll(void) nng_socket req; nng_ctx ctx; - TEST_NNG_PASS(nng_rep0_open(&req)); - TEST_NNG_PASS(nng_ctx_open(&ctx, req)); - TEST_NNG_FAIL( - nng_ctx_getopt_int(ctx, NNG_OPT_SENDFD, &fd), NNG_ENOTSUP); - TEST_NNG_FAIL( - nng_ctx_getopt_int(ctx, NNG_OPT_RECVFD, &fd), NNG_ENOTSUP); - TEST_NNG_PASS(nng_ctx_close(ctx)); - TEST_NNG_PASS(nng_close(req)); + NUTS_PASS(nng_rep0_open(&req)); + NUTS_PASS(nng_ctx_open(&ctx, req)); + NUTS_FAIL(nng_ctx_get_int(ctx, NNG_OPT_SENDFD, &fd), NNG_ENOTSUP); + NUTS_FAIL(nng_ctx_get_int(ctx, NNG_OPT_RECVFD, &fd), NNG_ENOTSUP); + NUTS_PASS(nng_ctx_close(ctx)); + NUTS_CLOSE(req); } void @@ -149,28 +139,27 @@ test_rep_validate_peer(void) nng_socket s1, s2; nng_stat * stats; nng_stat * reject; - char addr[64]; - - testutil_scratch_addr("inproc", sizeof(addr), addr); + char * addr; - TEST_NNG_PASS(nng_rep0_open(&s1)); - TEST_NNG_PASS(nng_rep0_open(&s2)); + NUTS_ADDR(addr, "inproc"); + NUTS_PASS(nng_rep0_open(&s1)); + NUTS_PASS(nng_rep0_open(&s2)); - TEST_NNG_PASS(nng_listen(s1, addr, NULL, 0)); - TEST_NNG_PASS(nng_dial(s2, addr, NULL, NNG_FLAG_NONBLOCK)); + NUTS_PASS(nng_listen(s1, addr, NULL, 0)); + NUTS_PASS(nng_dial(s2, addr, NULL, NNG_FLAG_NONBLOCK)); - testutil_sleep(100); - TEST_NNG_PASS(nng_stats_get(&stats)); + NUTS_SLEEP(100); + NUTS_PASS(nng_stats_get(&stats)); - TEST_CHECK(stats != NULL); - TEST_CHECK((reject = nng_stat_find_socket(stats, s1)) != NULL); - TEST_CHECK((reject = nng_stat_find(reject, "reject")) != NULL); + NUTS_TRUE(stats != NULL); + NUTS_TRUE((reject = nng_stat_find_socket(stats, s1)) != NULL); + NUTS_TRUE((reject = nng_stat_find(reject, "reject")) != NULL); - TEST_CHECK(nng_stat_type(reject) == NNG_STAT_COUNTER); - TEST_CHECK(nng_stat_value(reject) > 0); + NUTS_TRUE(nng_stat_type(reject) == NNG_STAT_COUNTER); + NUTS_TRUE(nng_stat_value(reject) > 0); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(s2)); + NUTS_CLOSE(s1); + NUTS_CLOSE(s2); nng_stats_free(stats); } @@ -181,17 +170,17 @@ test_rep_double_recv(void) nng_aio * aio1; nng_aio * aio2; - TEST_NNG_PASS(nng_rep0_open(&s1)); - TEST_NNG_PASS(nng_aio_alloc(&aio1, NULL, NULL)); - TEST_NNG_PASS(nng_aio_alloc(&aio2, NULL, NULL)); + NUTS_PASS(nng_rep0_open(&s1)); + NUTS_PASS(nng_aio_alloc(&aio1, NULL, NULL)); + NUTS_PASS(nng_aio_alloc(&aio2, NULL, NULL)); nng_recv_aio(s1, aio1); nng_recv_aio(s1, aio2); nng_aio_wait(aio2); - TEST_NNG_FAIL(nng_aio_result(aio2), NNG_ESTATE); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_FAIL(nng_aio_result(aio1), NNG_ECLOSED); + NUTS_FAIL(nng_aio_result(aio2), NNG_ESTATE); + NUTS_CLOSE(s1); + NUTS_FAIL(nng_aio_result(aio1), NNG_ECLOSED); nng_aio_free(aio1); nng_aio_free(aio2); } @@ -205,26 +194,26 @@ test_rep_close_pipe_before_send(void) nng_aio * aio1; nng_msg * m; - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_aio_alloc(&aio1, NULL, NULL)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_aio_alloc(&aio1, NULL, NULL)); - TEST_NNG_PASS(testutil_marry(req, rep)); - TEST_NNG_SEND_STR(req, "test"); + NUTS_MARRY(req, rep); + NUTS_SEND(req, "test"); nng_recv_aio(rep, aio1); nng_aio_wait(aio1); - TEST_NNG_PASS(nng_aio_result(aio1)); - TEST_CHECK((m = nng_aio_get_msg(aio1)) != NULL); + NUTS_PASS(nng_aio_result(aio1)); + NUTS_TRUE((m = nng_aio_get_msg(aio1)) != NULL); p = nng_msg_get_pipe(m); - TEST_NNG_PASS(nng_pipe_close(p)); - TEST_NNG_PASS(nng_sendmsg(rep, m, 0)); + NUTS_PASS(nng_pipe_close(p)); + NUTS_PASS(nng_sendmsg(rep, m, 0)); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); nng_aio_free(aio1); } @@ -236,25 +225,24 @@ test_rep_close_pipe_during_send(void) nng_pipe p = NNG_PIPE_INITIALIZER; nng_msg * m; - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_RECVBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_RECVBUF, 20)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); for (int i = 0; i < 100; i++) { int rv; - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS( - nng_msg_append_u32(m, (unsigned) i | 0x80000000u)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); - TEST_NNG_PASS(nng_recvmsg(rep, &m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, (unsigned) i | 0x80000000u)); + NUTS_PASS(nng_sendmsg(req, m, 0)); + NUTS_PASS(nng_recvmsg(rep, &m, 0)); p = nng_msg_get_pipe(m); rv = nng_sendmsg(rep, m, 0); if (rv == NNG_ETIMEDOUT) { @@ -262,12 +250,12 @@ test_rep_close_pipe_during_send(void) nng_msg_free(m); break; } - TEST_NNG_PASS(rv); + NUTS_PASS(rv); } - TEST_NNG_PASS(nng_pipe_close(p)); + NUTS_PASS(nng_pipe_close(p)); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -277,16 +265,16 @@ test_rep_ctx_recv_aio_stopped(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_ctx_open(&ctx, rep)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_ctx_open(&ctx, rep)); nng_aio_stop(aio); nng_ctx_recv(ctx, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECANCELED); - TEST_NNG_PASS(nng_ctx_close(ctx)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_PASS(nng_ctx_close(ctx)); + NUTS_CLOSE(rep); nng_aio_free(aio); } @@ -301,54 +289,53 @@ test_rep_close_pipe_context_send(void) nng_aio * aio[100]; int i; - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); for (i = 0; i < 100; i++) { - TEST_NNG_PASS(nng_ctx_open(&ctx[i], rep)); - TEST_NNG_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); + NUTS_PASS(nng_ctx_open(&ctx[i], rep)); + NUTS_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); } - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); for (i = 0; i < 100; i++) { - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS( - nng_msg_append_u32(m, (unsigned) i | 0x80000000u)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, (unsigned) i | 0x80000000u)); + NUTS_PASS(nng_sendmsg(req, m, 0)); nng_ctx_recv(ctx[i], aio[i]); } for (i = 0; i < 100; i++) { nng_aio_wait(aio[i]); - TEST_NNG_PASS(nng_aio_result(aio[i])); - TEST_CHECK((m = nng_aio_get_msg(aio[i])) != NULL); + NUTS_PASS(nng_aio_result(aio[i])); + NUTS_TRUE((m = nng_aio_get_msg(aio[i])) != NULL); p = nng_msg_get_pipe(m); nng_aio_set_msg(aio[i], m); nng_ctx_send(ctx[i], aio[i]); } // Note that REQ socket is not reading the results. - TEST_NNG_PASS(nng_pipe_close(p)); + NUTS_PASS(nng_pipe_close(p)); for (i = 0; i < 100; i++) { int rv; nng_aio_wait(aio[i]); rv = nng_aio_result(aio[i]); if (rv != 0) { - TEST_NNG_FAIL(rv, NNG_ECLOSED); + NUTS_FAIL(rv, NNG_ECLOSED); nng_msg_free(nng_aio_get_msg(aio[i])); } nng_aio_free(aio[i]); - TEST_NNG_PASS(nng_ctx_close(ctx[i])); + NUTS_PASS(nng_ctx_close(ctx[i])); } - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -361,33 +348,32 @@ test_rep_close_context_send(void) nng_aio * aio[100]; int i; - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); for (i = 0; i < 100; i++) { - TEST_NNG_PASS(nng_ctx_open(&ctx[i], rep)); - TEST_NNG_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); + NUTS_PASS(nng_ctx_open(&ctx[i], rep)); + NUTS_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); } - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); for (i = 0; i < 100; i++) { - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS( - nng_msg_append_u32(m, (unsigned) i | 0x80000000u)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, (unsigned) i | 0x80000000u)); + NUTS_PASS(nng_sendmsg(req, m, 0)); nng_ctx_recv(ctx[i], aio[i]); } for (i = 0; i < 100; i++) { nng_aio_wait(aio[i]); - TEST_NNG_PASS(nng_aio_result(aio[i])); - TEST_CHECK((m = nng_aio_get_msg(aio[i])) != NULL); + NUTS_PASS(nng_aio_result(aio[i])); + NUTS_TRUE((m = nng_aio_get_msg(aio[i])) != NULL); nng_aio_set_msg(aio[i], m); nng_ctx_send(ctx[i], aio[i]); } @@ -395,17 +381,17 @@ test_rep_close_context_send(void) // Note that REQ socket is not reading the results. for (i = 0; i < 100; i++) { int rv; - TEST_NNG_PASS(nng_ctx_close(ctx[i])); + NUTS_PASS(nng_ctx_close(ctx[i])); nng_aio_wait(aio[i]); rv = nng_aio_result(aio[i]); if (rv != 0) { - TEST_NNG_FAIL(rv, NNG_ECLOSED); + NUTS_FAIL(rv, NNG_ECLOSED); nng_msg_free(nng_aio_get_msg(aio[i])); } nng_aio_free(aio[i]); } - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -415,19 +401,19 @@ test_rep_close_recv(void) nng_socket req; nng_aio * aio; - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_MARRY(req, rep); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_recv_aio(rep, aio); - TEST_NNG_PASS(nng_close(rep)); - TEST_NNG_PASS(nng_close(req)); + NUTS_CLOSE(rep); + NUTS_CLOSE(req); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECLOSED); + NUTS_FAIL(nng_aio_result(aio), NNG_ECLOSED); nng_aio_free(aio); } @@ -461,28 +447,28 @@ test_rep_close_recv_cb(void) struct rep_close_recv_cb_state state; memset(&state, 0, sizeof(state)); - TEST_NNG_PASS(nng_mtx_alloc(&state.mtx)); - TEST_NNG_PASS(nng_cv_alloc(&state.cv, state.mtx)); + NUTS_PASS(nng_mtx_alloc(&state.mtx)); + NUTS_PASS(nng_cv_alloc(&state.cv, state.mtx)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); - TEST_NNG_PASS(nng_aio_alloc(&state.aio, rep_close_recv_cb, &state)); + NUTS_MARRY(req, rep); + NUTS_PASS(nng_aio_alloc(&state.aio, rep_close_recv_cb, &state)); nng_recv_aio(rep, state.aio); - TEST_NNG_PASS(nng_close(rep)); - TEST_NNG_PASS(nng_close(req)); + NUTS_CLOSE(rep); + NUTS_CLOSE(req); nng_mtx_lock(state.mtx); while (!state.done) { - TEST_NNG_PASS(nng_cv_until(state.cv, nng_clock() + 1000)); + NUTS_PASS(nng_cv_until(state.cv, nng_clock() + 1000)); } nng_mtx_unlock(state.mtx); - TEST_CHECK(state.done != 0); - TEST_NNG_FAIL(nng_aio_result(state.aio), NNG_ECLOSED); - TEST_CHECK(nng_aio_get_msg(state.aio) == NULL); + NUTS_TRUE(state.done != 0); + NUTS_FAIL(nng_aio_result(state.aio), NNG_ECLOSED); + NUTS_TRUE(nng_aio_get_msg(state.aio) == NULL); nng_aio_free(state.aio); nng_cv_free(state.cv); nng_mtx_free(state.mtx); @@ -495,16 +481,16 @@ test_rep_ctx_recv_nonblock(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_ctx_open(&ctx, rep)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_ctx_open(&ctx, rep)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_set_timeout(aio, 0); // Instant timeout nng_ctx_recv(ctx, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(rep)); + NUTS_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); + NUTS_CLOSE(rep); nng_aio_free(aio); } @@ -517,19 +503,19 @@ test_rep_ctx_send_nonblock(void) nng_aio * aio; nng_msg * msg; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 2000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_ctx_open(&ctx, rep)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 2000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_ctx_open(&ctx, rep)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_MARRY(req, rep); - TEST_NNG_SEND_STR(req, "SEND"); + NUTS_SEND(req, "SEND"); nng_ctx_recv(ctx, aio); nng_aio_wait(aio); - TEST_NNG_PASS(nng_aio_result(aio)); + NUTS_PASS(nng_aio_result(aio)); // message carries over msg = nng_aio_get_msg(aio); nng_aio_set_msg(aio, msg); @@ -537,9 +523,9 @@ test_rep_ctx_send_nonblock(void) nng_ctx_send(ctx, aio); nng_aio_wait(aio); - TEST_NNG_PASS(nng_aio_result(aio)); - TEST_NNG_PASS(nng_close(rep)); - TEST_NNG_PASS(nng_close(req)); + NUTS_PASS(nng_aio_result(aio)); + NUTS_CLOSE(rep); + NUTS_CLOSE(req); nng_aio_free(aio); } @@ -556,29 +542,28 @@ test_rep_ctx_send_nonblock2(void) // We are going to send a bunch of requests, receive them, // but then see that non-block pressure exerts for some, but // that at least one non-blocking send works. - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); for (int i = 0; i < 10; i++) { - TEST_NNG_PASS(nng_ctx_open(&rep_ctx[i], rep)); - TEST_NNG_PASS(nng_aio_alloc(&rep_aio[i], NULL, NULL)); + NUTS_PASS(nng_ctx_open(&rep_ctx[i], rep)); + NUTS_PASS(nng_aio_alloc(&rep_aio[i], NULL, NULL)); } - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); for (int i = 0; i < 10; i++) { nng_msg *msg; - TEST_NNG_PASS(nng_msg_alloc(&msg, 4)); - TEST_NNG_PASS( - nng_msg_append_u32(msg, (unsigned) i | 0x80000000u)); + NUTS_PASS(nng_msg_alloc(&msg, 4)); + NUTS_PASS(nng_msg_append_u32(msg, (unsigned) i | 0x80000000u)); nng_ctx_recv(rep_ctx[i], rep_aio[i]); - TEST_NNG_PASS(nng_sendmsg(req, msg, 0)); + NUTS_PASS(nng_sendmsg(req, msg, 0)); } for (int i = 0; i < 10; i++) { nng_msg *msg; nng_aio_wait(rep_aio[i]); - TEST_NNG_PASS(nng_aio_result(rep_aio[i])); + NUTS_PASS(nng_aio_result(rep_aio[i])); msg = nng_aio_get_msg(rep_aio[i]); nng_aio_set_timeout(rep_aio[i], 0); nng_aio_set_msg(rep_aio[i], msg); @@ -592,7 +577,7 @@ test_rep_ctx_send_nonblock2(void) if (rv == 0) { num_good++; } else { - TEST_NNG_FAIL(rv, NNG_ETIMEDOUT); + NUTS_FAIL(rv, NNG_ETIMEDOUT); nng_msg_free(nng_aio_get_msg(rep_aio[i])); num_fail++; } @@ -605,8 +590,8 @@ test_rep_ctx_send_nonblock2(void) nng_aio_free(rep_aio[i]); nng_ctx_close(rep_ctx[i]); } - TEST_NNG_PASS(nng_close(rep)); - TEST_NNG_PASS(nng_close(req)); + NUTS_CLOSE(rep); + NUTS_CLOSE(req); } static void @@ -616,24 +601,24 @@ test_rep_send_nonblock(void) nng_socket req; int rv; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_MARRY(req, rep); - TEST_NNG_SEND_STR(req, "SEND"); - TEST_NNG_RECV_STR(rep, "SEND"); + NUTS_SEND(req, "SEND"); + NUTS_RECV(rep, "SEND"); // Use the nonblock flag rv = nng_send(rep, "RECV", 5, NNG_FLAG_NONBLOCK); - TEST_NNG_PASS(rv); - TEST_NNG_RECV_STR(req, "RECV"); - TEST_NNG_PASS(nng_close(rep)); - TEST_NNG_PASS(nng_close(req)); + NUTS_PASS(rv); + NUTS_RECV(req, "RECV"); + NUTS_CLOSE(rep); + NUTS_CLOSE(req); } void @@ -643,24 +628,24 @@ test_rep_recv_garbage(void) nng_socket req; nng_msg * m; - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 200)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_append_u32(m, 1u)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(rep, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, 1u)); + NUTS_PASS(nng_sendmsg(req, m, 0)); + NUTS_FAIL(nng_recvmsg(rep, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } -TEST_LIST = { +NUTS_TESTS = { { "rep identity", test_rep_identity }, { "rep send bad state", test_rep_send_bad_state }, { "rep poll readable", test_rep_poll_readable }, diff --git a/src/protocol/reqrep0/req_test.c b/src/protocol/reqrep0/req_test.c index add420de..fb78efa0 100644 --- a/src/protocol/reqrep0/req_test.c +++ b/src/protocol/reqrep0/req_test.c @@ -8,14 +8,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/reqrep0/rep.h> -#include <nng/protocol/reqrep0/req.h> - -#include <acutest.h> -#include <testutil.h> +#include <nuts.h> static void test_req_identity(void) @@ -24,18 +17,18 @@ test_req_identity(void) int p; char * n; - TEST_NNG_PASS(nng_req0_open(&s)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PROTO, &p)); - TEST_CHECK(p == NNG_REQ0_SELF); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PEER, &p)); - TEST_CHECK(p == NNG_REQ0_PEER); // 49 - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PROTONAME, &n)); - TEST_CHECK(strcmp(n, NNG_REQ0_SELF_NAME) == 0); + NUTS_PASS(nng_req0_open(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p)); + NUTS_TRUE(p == NNG_REQ0_SELF); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p)); + NUTS_TRUE(p == NNG_REQ0_PEER); // 49 + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n)); + NUTS_MATCH(n, NNG_REQ0_SELF_NAME); nng_strfree(n); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PEERNAME, &n)); - TEST_CHECK(strcmp(n, NNG_REQ0_PEER_NAME) == 0); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n)); + NUTS_MATCH(n, NNG_REQ0_PEER_NAME); nng_strfree(n); - TEST_NNG_PASS(nng_close(s)); + NUTS_CLOSE(s); } static void @@ -47,31 +40,31 @@ test_req_ttl_option(void) size_t sz; const char *opt = NNG_OPT_MAXTTL; - TEST_NNG_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_setopt_int(req, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(req, opt, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(req, opt, -1), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(req, opt, 1)); + NUTS_FAIL(nng_socket_set_int(req, opt, 0), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(req, opt, -1), NNG_EINVAL); // This test will fail if the NNI_MAX_MAX_TTL is changed from the // builtin default of 15. - TEST_NNG_FAIL(nng_setopt_int(req, opt, 16), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(req, opt, 256), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(req, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(req, opt, &v)); - TEST_CHECK(v == 3); + NUTS_FAIL(nng_socket_set_int(req, opt, 16), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(req, opt, 256), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(req, opt, 3)); + NUTS_PASS(nng_socket_get_int(req, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(req, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(req, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_NNG_FAIL(nng_setopt(req, opt, "", 1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set(req, opt, "", 1), NNG_EINVAL); sz = 1; - TEST_NNG_FAIL(nng_getopt(req, opt, &v, &sz), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(req, opt, true), NNG_EBADTYPE); - TEST_NNG_FAIL(nng_getopt_bool(req, opt, &b), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get(req, opt, &v, &sz), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_bool(req, opt, true), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(req, opt, &b), NNG_EBADTYPE); - TEST_NNG_PASS(nng_close(req)); + NUTS_CLOSE(req); } static void @@ -83,17 +76,17 @@ test_req_resend_option(void) size_t sz = sizeof(b); const char * opt = NNG_OPT_REQ_RESENDTIME; - TEST_NNG_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_req0_open(&req)); - TEST_CHECK(nng_setopt_ms(req, opt, 10) == 0); - TEST_NNG_FAIL(nng_setopt(req, opt, "", 1), NNG_EINVAL); - TEST_NNG_FAIL(nng_getopt(req, opt, &b, &sz), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(req, opt, true), NNG_EBADTYPE); - TEST_NNG_FAIL(nng_getopt_bool(req, opt, &b), NNG_EBADTYPE); + NUTS_TRUE(nng_socket_set_ms(req, opt, 10) == 0); + NUTS_FAIL(nng_socket_set(req, opt, "", 1), NNG_EINVAL); + NUTS_FAIL(nng_socket_get(req, opt, &b, &sz), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_bool(req, opt, true), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(req, opt, &b), NNG_EBADTYPE); - TEST_NNG_PASS(nng_getopt_ms(req, opt, &d)); - TEST_CHECK(d == 10); - TEST_NNG_PASS(nng_close(req)); + NUTS_PASS(nng_socket_get_ms(req, opt, &d)); + NUTS_TRUE(d == 10); + NUTS_CLOSE(req); } void @@ -102,10 +95,10 @@ test_req_recv_bad_state(void) nng_socket req; nng_msg * msg = NULL; - TEST_CHECK(nng_req0_open(&req) == 0); - TEST_CHECK(nng_recvmsg(req, &msg, 0) == NNG_ESTATE); - TEST_CHECK(msg == NULL); - TEST_CHECK(nng_close(req) == 0); + NUTS_TRUE(nng_req0_open(&req) == 0); + NUTS_TRUE(nng_recvmsg(req, &msg, 0) == NNG_ESTATE); + NUTS_NULL(msg); + NUTS_CLOSE(req); } static void @@ -116,31 +109,31 @@ test_req_recv_garbage(void) nng_msg * m; uint32_t req_id; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_sendmsg(req, m, 0)); - TEST_NNG_PASS(nng_recvmsg(rep, &m, 0)); + NUTS_PASS(nng_recvmsg(rep, &m, 0)); // The message will have a header that contains the 32-bit pipe ID, // followed by the 32-bit request ID. We will discard the request // ID before sending it out. - TEST_CHECK(nng_msg_header_len(m) == 8); - TEST_NNG_PASS(nng_msg_header_chop_u32(m, &req_id)); + NUTS_TRUE(nng_msg_header_len(m) == 8); + NUTS_PASS(nng_msg_header_chop_u32(m, &req_id)); - TEST_NNG_PASS(nng_sendmsg(rep, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(req, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_sendmsg(rep, m, 0)); + NUTS_FAIL(nng_recvmsg(req, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } #define SECOND 1000 @@ -150,40 +143,24 @@ test_req_rep_exchange(void) { nng_socket req; nng_socket rep; - nng_msg * msg = NULL; - TEST_CHECK(nng_req0_open(&req) == 0); - TEST_CHECK(nng_rep0_open(&rep) == 0); - - TEST_CHECK(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, SECOND) == 0); - TEST_CHECK(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, SECOND) == 0); - TEST_CHECK(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, SECOND) == 0); - TEST_CHECK(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, SECOND) == 0); - - TEST_CHECK(testutil_marry(rep, req) == 0); - - TEST_CHECK(nng_msg_alloc(&msg, 0) == 0); - TEST_CHECK(nng_msg_append(msg, "ping", 5) == 0); - TEST_CHECK(nng_msg_len(msg) == 5); - TEST_CHECK(strcmp(nng_msg_body(msg), "ping") == 0); - TEST_CHECK(nng_sendmsg(req, msg, 0) == 0); - msg = NULL; - TEST_CHECK(nng_recvmsg(rep, &msg, 0) == 0); - TEST_CHECK(msg != NULL); - TEST_CHECK(nng_msg_len(msg) == 5); - TEST_CHECK(strcmp(nng_msg_body(msg), "ping") == 0); - nng_msg_trim(msg, 5); - TEST_CHECK(nng_msg_append(msg, "pong", 5) == 0); - TEST_CHECK(nng_sendmsg(rep, msg, 0) == 0); - msg = NULL; - TEST_CHECK(nng_recvmsg(req, &msg, 0) == 0); - TEST_CHECK(msg != NULL); - TEST_CHECK(nng_msg_len(msg) == 5); - TEST_CHECK(strcmp(nng_msg_body(msg), "pong") == 0); - nng_msg_free(msg); + NUTS_TRUE(nng_req0_open(&req) == 0); + NUTS_TRUE(nng_rep0_open(&rep) == 0); + + NUTS_TRUE(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, SECOND) == 0); + NUTS_TRUE(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, SECOND) == 0); + NUTS_TRUE(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, SECOND) == 0); + NUTS_TRUE(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, SECOND) == 0); + + NUTS_MARRY(rep, req); - TEST_CHECK(nng_close(req) == 0); - TEST_CHECK(nng_close(rep) == 0); + NUTS_SEND(req, "ping"); + NUTS_RECV(rep, "ping"); + NUTS_SEND(rep, "pong"); + NUTS_RECV(req, "pong"); + + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -192,24 +169,24 @@ test_req_resend(void) nng_socket req; nng_socket rep; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_REQ_RESENDTIME, 10)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_REQ_RESENDTIME, 10)); - TEST_NNG_PASS(testutil_marry(rep, req)); + NUTS_MARRY(rep, req); - TEST_NNG_SEND_STR(req, "ping"); - TEST_NNG_RECV_STR(rep, "ping"); - TEST_NNG_RECV_STR(rep, "ping"); - TEST_NNG_RECV_STR(rep, "ping"); + NUTS_SEND(req, "ping"); + NUTS_RECV(rep, "ping"); + NUTS_RECV(rep, "ping"); + NUTS_RECV(rep, "ping"); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -219,34 +196,34 @@ test_req_resend_reconnect(void) nng_socket rep1; nng_socket rep2; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep1)); - TEST_NNG_PASS(nng_rep0_open(&rep2)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep1)); + NUTS_PASS(nng_rep0_open(&rep2)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep1, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep2, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep1, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep2, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep1, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep2, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep1, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep2, NNG_OPT_SENDTIMEO, SECOND)); // We intentionally set the retry time long; that way we only see // the retry from loss of our original peer. - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_REQ_RESENDTIME, 60 * SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_REQ_RESENDTIME, 60 * SECOND)); - TEST_NNG_PASS(testutil_marry(rep1, req)); + NUTS_MARRY(rep1, req); - TEST_NNG_SEND_STR(req, "ping"); - TEST_NNG_RECV_STR(rep1, "ping"); + NUTS_SEND(req, "ping"); + NUTS_RECV(rep1, "ping"); - TEST_NNG_PASS(nng_close(rep1)); - TEST_NNG_PASS(testutil_marry(rep2, req)); + NUTS_CLOSE(rep1); + NUTS_MARRY(rep2, req); - TEST_NNG_RECV_STR(rep2, "ping"); - TEST_NNG_SEND_STR(rep2, "rep2"); - TEST_NNG_RECV_STR(req, "rep2"); + NUTS_RECV(rep2, "ping"); + NUTS_SEND(rep2, "rep2"); + NUTS_RECV(req, "rep2"); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep2)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep2); } void @@ -256,33 +233,33 @@ test_req_resend_disconnect(void) nng_socket rep1; nng_socket rep2; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep1)); - TEST_NNG_PASS(nng_rep0_open(&rep2)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep1)); + NUTS_PASS(nng_rep0_open(&rep2)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep1, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep2, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep1, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep2, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep1, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep2, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep1, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep2, NNG_OPT_SENDTIMEO, SECOND)); // We intentionally set the retry time long; that way we only see // the retry from loss of our original peer. - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_REQ_RESENDTIME, 60 * SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_REQ_RESENDTIME, 60 * SECOND)); - TEST_NNG_PASS(testutil_marry(rep1, req)); - TEST_NNG_SEND_STR(req, "ping"); - TEST_NNG_RECV_STR(rep1, "ping"); + NUTS_MARRY(rep1, req); + NUTS_SEND(req, "ping"); + NUTS_RECV(rep1, "ping"); - TEST_NNG_PASS(testutil_marry(rep2, req)); - TEST_NNG_PASS(nng_close(rep1)); + NUTS_MARRY(rep2, req); + NUTS_CLOSE(rep1); - TEST_NNG_RECV_STR(rep2, "ping"); - TEST_NNG_SEND_STR(rep2, "rep2"); - TEST_NNG_RECV_STR(req, "rep2"); + NUTS_RECV(rep2, "ping"); + NUTS_SEND(rep2, "rep2"); + NUTS_RECV(req, "rep2"); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep2)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep2); } void @@ -292,32 +269,32 @@ test_req_disconnect_no_retry(void) nng_socket rep1; nng_socket rep2; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep1)); - TEST_NNG_PASS(nng_rep0_open(&rep2)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep1)); + NUTS_PASS(nng_rep0_open(&rep2)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep1, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep2, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep1, NNG_OPT_SENDTIMEO, SECOND / 10)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep1, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep2, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep1, NNG_OPT_SENDTIMEO, SECOND / 10)); // Setting the resend time to zero so we will force an error // if the peer disconnects without sending us an answer. - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_REQ_RESENDTIME, 0)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_REQ_RESENDTIME, 0)); - TEST_NNG_PASS(testutil_marry(rep1, req)); - TEST_NNG_SEND_STR(req, "ping"); - TEST_NNG_RECV_STR(rep1, "ping"); + NUTS_MARRY(rep1, req); + NUTS_SEND(req, "ping"); + NUTS_RECV(rep1, "ping"); - TEST_NNG_PASS(testutil_marry(rep2, req)); - TEST_NNG_PASS(nng_close(rep1)); + NUTS_MARRY(rep2, req); + NUTS_CLOSE(rep1); nng_msg *msg = NULL; - TEST_NNG_FAIL(nng_recvmsg(req, &msg, 0), NNG_ECONNRESET); - TEST_NNG_FAIL(nng_recvmsg(rep2, &msg, 0), NNG_ETIMEDOUT); + NUTS_FAIL(nng_recvmsg(req, &msg, 0), NNG_ECONNRESET); + NUTS_FAIL(nng_recvmsg(rep2, &msg, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep2)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep2); } void @@ -328,186 +305,152 @@ test_req_disconnect_abort(void) nng_socket rep2; nng_aio * aio; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep1)); - TEST_NNG_PASS(nng_rep0_open(&rep2)); - TEST_NNG_PASS(nng_aio_alloc(&aio, 0, 0)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep1)); + NUTS_PASS(nng_rep0_open(&rep2)); + NUTS_PASS(nng_aio_alloc(&aio, 0, 0)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep1, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep2, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep1, NNG_OPT_SENDTIMEO, SECOND / 10)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep1, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep2, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep1, NNG_OPT_SENDTIMEO, SECOND / 10)); // Setting the resend time to zero so we will force an error // if the peer disconnects without sending us an answer. - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_REQ_RESENDTIME, 0)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_REQ_RESENDTIME, 0)); - TEST_NNG_PASS(testutil_marry(rep1, req)); - TEST_NNG_SEND_STR(req, "ping"); - TEST_NNG_RECV_STR(rep1, "ping"); + NUTS_MARRY(rep1, req); + NUTS_SEND(req, "ping"); + NUTS_RECV(rep1, "ping"); nng_recv_aio(req, aio); - TEST_NNG_PASS(testutil_marry(rep2, req)); - TEST_NNG_PASS(nng_close(rep1)); + NUTS_MARRY(rep2, req); + NUTS_CLOSE(rep1); nng_msg *msg = NULL; nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECONNRESET); - TEST_NNG_FAIL(nng_recvmsg(rep2, &msg, 0), NNG_ETIMEDOUT); + NUTS_FAIL(nng_aio_result(aio), NNG_ECONNRESET); + NUTS_FAIL(nng_recvmsg(rep2, &msg, 0), NNG_ETIMEDOUT); nng_aio_free(aio); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep2)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep2); } void test_req_cancel(void) { - nng_msg * abc; - nng_msg * def; - nng_msg * cmd; nng_duration retry = SECOND; nng_socket req; nng_socket rep; - TEST_NNG_PASS(nng_rep_open(&rep)); - TEST_NNG_PASS(nng_req_open(&req)); - - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 5 * SECOND)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 5 * SECOND)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_REQ_RESENDTIME, retry)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_SENDBUF, 16)); + NUTS_PASS(nng_rep_open(&rep)); + NUTS_PASS(nng_req_open(&req)); - TEST_NNG_PASS(nng_msg_alloc(&abc, 0)); - TEST_NNG_PASS(nng_msg_append(abc, "abc", 4)); - TEST_NNG_PASS(nng_msg_alloc(&def, 0)); - TEST_NNG_PASS(nng_msg_append(def, "def", 4)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_REQ_RESENDTIME, retry)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 16)); - TEST_NNG_PASS(testutil_marry(rep, req)); + NUTS_MARRY(rep, req); // Send req #1 (abc). - TEST_CHECK(nng_sendmsg(req, abc, 0) == 0); + NUTS_SEND(req, "abc"); // Sleep a bit. This is so that we ensure that our request gets // to the far side. (If we cancel too fast, then our outgoing send // will be canceled before it gets to the peer.) - testutil_sleep(100); + NUTS_SLEEP(100); // Send the next next request ("def"). Note that // the REP side server will have already buffered the receive // request, and should simply be waiting for us to reply to abc. - TEST_NNG_PASS(nng_sendmsg(req, def, 0)); + NUTS_SEND(req, "def"); // Receive the first request (should be abc) on the REP server. - TEST_NNG_PASS(nng_recvmsg(rep, &cmd, 0)); - TEST_ASSERT(cmd != NULL); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "abc") == 0); + NUTS_RECV(rep, "abc"); // REP sends the reply to first command. This will be discarded // by the REQ socket. - TEST_NNG_PASS(nng_sendmsg(rep, cmd, 0)); + NUTS_SEND(rep, "abc"); // Now get the next command from the REP; should be "def". - TEST_NNG_PASS(nng_recvmsg(rep, &cmd, 0)); - TEST_ASSERT(cmd != NULL); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "def") == 0); - TEST_MSG("Received body was %s", nng_msg_body(cmd)); + NUTS_RECV(rep, "def"); // And send it back to REQ. - TEST_NNG_PASS(nng_sendmsg(rep, cmd, 0)); + NUTS_SEND(rep, "def"); - // Try a req command. This should give back "def" - TEST_NNG_PASS(nng_recvmsg(req, &cmd, 0)); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "def") == 0); - nng_msg_free(cmd); + // And we got back only the second result. + NUTS_RECV(req, "def"); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void test_req_cancel_abort_recv(void) { - - nng_msg * abc; - nng_msg * def; - nng_msg * cmd; nng_aio * aio; nng_duration retry = SECOND * 10; // 10s (kind of never) nng_socket req; nng_socket rep; - TEST_CHECK(nng_rep_open(&rep) == 0); - TEST_CHECK(nng_req_open(&req) == 0); - TEST_CHECK(nng_aio_alloc(&aio, NULL, NULL) == 0); - - TEST_CHECK(nng_setopt_ms(req, NNG_OPT_REQ_RESENDTIME, retry) == 0); - TEST_CHECK(nng_setopt_int(req, NNG_OPT_SENDBUF, 16) == 0); - TEST_CHECK(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, 5 * SECOND) == 0); - TEST_CHECK(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 5 * SECOND) == 0); - TEST_CHECK(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 5 * SECOND) == 0); - TEST_CHECK(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 5 * SECOND) == 0); + NUTS_PASS(nng_rep_open(&rep)); + NUTS_PASS(nng_req_open(&req)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_CHECK(nng_msg_alloc(&abc, 0) == 0); - TEST_CHECK(nng_msg_append(abc, "abc", 4) == 0); - TEST_CHECK(nng_msg_alloc(&def, 0) == 0); - TEST_CHECK(nng_msg_append(def, "def", 4) == 0); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_REQ_RESENDTIME, retry)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 16)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 5 * SECOND)); - TEST_CHECK(testutil_marry(rep, req) == 0); + NUTS_MARRY(rep, req); // Send req #1 (abc). - TEST_CHECK(nng_sendmsg(req, abc, 0) == 0); + NUTS_SEND(req, "abc"); // Wait for it to get ot the other side. - testutil_sleep(100); + NUTS_SLEEP(100); nng_aio_set_timeout(aio, 5 * SECOND); nng_recv_aio(req, aio); // Give time for this recv to post properly. - testutil_sleep(100); + NUTS_SLEEP(100); // Send the next next request ("def"). Note that // the REP side server will have already buffered the receive // request, and should simply be waiting for us to reply to // abc. - TEST_CHECK(nng_sendmsg(req, def, 0) == 0); + NUTS_SEND(req, "def"); // Our pending I/O should have been canceled. nng_aio_wait(aio); - TEST_CHECK(nng_aio_result(aio) == NNG_ECANCELED); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); // Receive the first request (should be abc) on the REP server. - TEST_CHECK(nng_recvmsg(rep, &cmd, 0) == 0); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "abc") == 0); + NUTS_RECV(rep, "abc"); // REP sends the reply to first command. This will be // discarded by the REQ socket. - TEST_CHECK(nng_sendmsg(rep, cmd, 0) == 0); + NUTS_SEND(rep, "abc"); // Now get the next command from the REP; should be "def". - TEST_CHECK(nng_recvmsg(rep, &cmd, 0) == 0); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "def") == 0); + NUTS_RECV(rep, "def"); // And send it back to REQ. - TEST_CHECK(nng_sendmsg(rep, cmd, 0) == 0); + NUTS_SEND(rep, "def"); // Try a req command. This should give back "def" - TEST_CHECK(nng_recvmsg(req, &cmd, 0) == 0); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "def") == 0); - nng_msg_free(cmd); + NUTS_RECV(req, "def"); nng_aio_free(aio); - TEST_CHECK(nng_close(req) == 0); - TEST_CHECK(nng_close(rep) == 0); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -516,25 +459,25 @@ test_req_cancel_post_recv(void) nng_socket req; nng_socket rep; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); - - TEST_NNG_SEND_STR(req, "ONE"); - TEST_NNG_RECV_STR(rep, "ONE"); - TEST_NNG_SEND_STR(rep, "one"); - testutil_sleep(100); // Make sure reply arrives! - TEST_NNG_SEND_STR(req, "TWO"); - TEST_NNG_RECV_STR(rep, "TWO"); - TEST_NNG_SEND_STR(rep, "two"); - TEST_NNG_RECV_STR(req, "two"); - - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_MARRY(req, rep); + + NUTS_SEND(req, "ONE"); + NUTS_RECV(rep, "ONE"); + NUTS_SEND(rep, "one"); + NUTS_SLEEP(100); // Make sure reply arrives! + NUTS_SEND(req, "TWO"); + NUTS_RECV(rep, "TWO"); + NUTS_SEND(rep, "two"); + NUTS_RECV(req, "two"); + + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -544,18 +487,18 @@ test_req_poll_writeable(void) nng_socket req; nng_socket rep; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_getopt_int(req, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_get_int(req, NNG_OPT_SENDFD, &fd)); + NUTS_TRUE(fd >= 0); // Not writable before connect. - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); // It should be writable now. - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd)); // Submit a bunch of jobs. Note that we have to stall a bit // between each message to let it queue up. @@ -564,12 +507,12 @@ test_req_poll_writeable(void) if (rv == NNG_EAGAIN) { break; } - TEST_NNG_PASS(rv); - testutil_sleep(50); + NUTS_PASS(rv); + NUTS_SLEEP(50); } - TEST_CHECK(testutil_pollfd(fd) == 0); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_TRUE(nuts_poll_fd(fd) == false); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -584,26 +527,26 @@ test_req_poll_contention(void) nng_msg * ctx_msg[5]; nng_msg * msg; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); for (int i = 0; i < 5; i++) { - TEST_NNG_PASS(nng_ctx_open(&ctx[i], req)); - TEST_NNG_PASS(nng_aio_alloc(&ctx_aio[i], NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&ctx_msg[i], 0)); + NUTS_PASS(nng_ctx_open(&ctx[i], req)); + NUTS_PASS(nng_aio_alloc(&ctx_aio[i], NULL, NULL)); + NUTS_PASS(nng_msg_alloc(&ctx_msg[i], 0)); } - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_getopt_int(req, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_socket_get_int(req, NNG_OPT_SENDFD, &fd)); + NUTS_TRUE(fd >= 0); // Not writable before connect. - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); nng_aio_set_msg(aio, msg); nng_send_aio(req, aio); @@ -611,35 +554,35 @@ test_req_poll_contention(void) nng_aio_set_msg(ctx_aio[i], ctx_msg[i]); nng_ctx_send(ctx[i], ctx_aio[i]); } - testutil_sleep(50); // so everything is queued steady state + NUTS_SLEEP(50); // so everything is queued steady state - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); // It should not be writable now. - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(nng_recvmsg(rep, &msg, 0)); + NUTS_PASS(nng_recvmsg(rep, &msg, 0)); nng_msg_free(msg); // Still not writeable... - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); for (int i = 0; i < 5; i++) { - TEST_NNG_PASS(nng_recvmsg(rep, &msg, 0)); + NUTS_PASS(nng_recvmsg(rep, &msg, 0)); nng_msg_free(msg); } // It can take a little bit of time for the eased back-pressure // to reflect across the network. - testutil_sleep(100); + NUTS_SLEEP(100); // Should be come writeable now... - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); for (int i = 0; i < 5; i++) { nng_aio_free(ctx_aio[i]); } nng_aio_free(aio); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } void @@ -650,28 +593,28 @@ test_req_poll_multi_pipe(void) nng_socket rep1; nng_socket rep2; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep1)); - TEST_NNG_PASS(nng_rep0_open(&rep2)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep1)); + NUTS_PASS(nng_rep0_open(&rep2)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_getopt_int(req, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_socket_get_int(req, NNG_OPT_SENDFD, &fd)); + NUTS_TRUE(fd >= 0); // Not writable before connect. - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(testutil_marry(req, rep1)); - TEST_NNG_PASS(testutil_marry(req, rep2)); + NUTS_MARRY(req, rep1); + NUTS_MARRY(req, rep2); - TEST_CHECK(testutil_pollfd(fd) == true); - TEST_NNG_SEND_STR(req, "ONE"); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); + NUTS_SEND(req, "ONE"); + NUTS_TRUE(nuts_poll_fd(fd) == true); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep1)); - TEST_NNG_PASS(nng_close(rep2)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep1); + NUTS_CLOSE(rep2); } void @@ -682,39 +625,39 @@ test_req_poll_readable(void) nng_socket rep; nng_msg * msg; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_getopt_int(req, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_get_int(req, NNG_OPT_RECVFD, &fd)); + NUTS_TRUE(fd >= 0); // Not readable if not connected! - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); // Even after connect (no message yet) - TEST_NNG_PASS(testutil_marry(req, rep)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(req, rep); + NUTS_TRUE(nuts_poll_fd(fd) == false); // But once we send messages, it is. // We have to send a request, in order to send a reply. - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append(msg, "xyz", 3)); - TEST_NNG_PASS(nng_sendmsg(req, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(rep, &msg, 0)); // recv on rep - TEST_NNG_PASS(nng_sendmsg(rep, msg, 0)); // echo it back - testutil_sleep(200); // give time for message to arrive + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_append(msg, "xyz", 3)); + NUTS_PASS(nng_sendmsg(req, msg, 0)); + NUTS_PASS(nng_recvmsg(rep, &msg, 0)); // recv on rep + NUTS_PASS(nng_sendmsg(rep, msg, 0)); // echo it back + NUTS_SLEEP(200); // give time for message to arrive - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); // and receiving makes it no longer ready - TEST_NNG_PASS(nng_recvmsg(req, &msg, 0)); + NUTS_PASS(nng_recvmsg(req, &msg, 0)); nng_msg_free(msg); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); // TODO verify unsolicited response - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -724,14 +667,12 @@ test_req_ctx_no_poll(void) nng_socket req; nng_ctx ctx; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_ctx_open(&ctx, req)); - TEST_NNG_FAIL( - nng_ctx_getopt_int(ctx, NNG_OPT_SENDFD, &fd), NNG_ENOTSUP); - TEST_NNG_FAIL( - nng_ctx_getopt_int(ctx, NNG_OPT_RECVFD, &fd), NNG_ENOTSUP); - TEST_NNG_PASS(nng_ctx_close(ctx)); - TEST_NNG_PASS(nng_close(req)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_ctx_open(&ctx, req)); + NUTS_FAIL(nng_ctx_getopt_int(ctx, NNG_OPT_SENDFD, &fd), NNG_ENOTSUP); + NUTS_FAIL(nng_ctx_getopt_int(ctx, NNG_OPT_RECVFD, &fd), NNG_ENOTSUP); + NUTS_PASS(nng_ctx_close(ctx)); + NUTS_CLOSE(req); } static void @@ -743,15 +684,15 @@ test_req_ctx_send_queued(void) nng_aio * aio[3]; nng_msg * msg[3]; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 100)); for (int i = 0; i < 3; i++) { - TEST_NNG_PASS(nng_ctx_open(&ctx[i], req)); - TEST_NNG_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&msg[i], 0)); + NUTS_PASS(nng_ctx_open(&ctx[i], req)); + NUTS_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); + NUTS_PASS(nng_msg_alloc(&msg[i], 0)); } for (int i = 0; i < 3; i++) { @@ -759,20 +700,20 @@ test_req_ctx_send_queued(void) nng_ctx_send(ctx[i], aio[i]); } - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); - testutil_sleep(50); // Only to ensure stuff queues up + NUTS_SLEEP(50); // Only to ensure stuff queues up for (int i = 0; i < 3; i++) { nng_msg *m; - TEST_NNG_PASS(nng_recvmsg(rep, &m, 0)); + NUTS_PASS(nng_recvmsg(rep, &m, 0)); nng_msg_free(m); } - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); for (int i = 0; i < 3; i++) { nng_aio_wait(aio[i]); - TEST_NNG_PASS(nng_aio_result(aio[i])); + NUTS_PASS(nng_aio_result(aio[i])); nng_aio_free(aio[i]); } } @@ -785,13 +726,13 @@ test_req_ctx_send_close(void) nng_aio * aio[3]; nng_msg * msg[3]; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); for (int i = 0; i < 3; i++) { - TEST_NNG_PASS(nng_ctx_open(&ctx[i], req)); - TEST_NNG_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&msg[i], 0)); + NUTS_PASS(nng_ctx_open(&ctx[i], req)); + NUTS_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); + NUTS_PASS(nng_msg_alloc(&msg[i], 0)); } for (int i = 0; i < 3; i++) { @@ -805,11 +746,11 @@ test_req_ctx_send_close(void) for (int i = 0; i < 3; i++) { nng_aio_wait(aio[i]); - TEST_NNG_FAIL(nng_aio_result(aio[i]), NNG_ECLOSED); + NUTS_FAIL(nng_aio_result(aio[i]), NNG_ECLOSED); nng_aio_free(aio[i]); nng_msg_free(msg[i]); } - TEST_NNG_PASS(nng_close(req)); + NUTS_CLOSE(req); } static void @@ -820,13 +761,13 @@ test_req_ctx_send_abort(void) nng_aio * aio[3]; nng_msg * msg[3]; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); for (int i = 0; i < 3; i++) { - TEST_NNG_PASS(nng_ctx_open(&ctx[i], req)); - TEST_NNG_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&msg[i], 0)); + NUTS_PASS(nng_ctx_open(&ctx[i], req)); + NUTS_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); + NUTS_PASS(nng_msg_alloc(&msg[i], 0)); } for (int i = 0; i < 3; i++) { @@ -840,11 +781,11 @@ test_req_ctx_send_abort(void) for (int i = 0; i < 3; i++) { nng_aio_wait(aio[i]); - TEST_NNG_FAIL(nng_aio_result(aio[i]), NNG_ECANCELED); + NUTS_FAIL(nng_aio_result(aio[i]), NNG_ECANCELED); nng_aio_free(aio[i]); nng_msg_free(msg[i]); } - TEST_NNG_PASS(nng_close(req)); + NUTS_CLOSE(req); } static void @@ -855,26 +796,26 @@ test_req_ctx_send_twice(void) nng_aio * aio[2]; nng_msg * msg[2]; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_ctx_open(&ctx, req)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_ctx_open(&ctx, req)); for (int i = 0; i < 2; i++) { - TEST_NNG_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&msg[i], 0)); + NUTS_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); + NUTS_PASS(nng_msg_alloc(&msg[i], 0)); } for (int i = 0; i < 2; i++) { nng_aio_set_msg(aio[i], msg[i]); nng_ctx_send(ctx, aio[i]); - testutil_sleep(50); + NUTS_SLEEP(50); } - TEST_NNG_PASS(nng_close(req)); + NUTS_CLOSE(req); nng_aio_wait(aio[0]); nng_aio_wait(aio[1]); - TEST_NNG_FAIL(nng_aio_result(aio[0]), NNG_ECANCELED); - TEST_NNG_FAIL(nng_aio_result(aio[1]), NNG_ECLOSED); + NUTS_FAIL(nng_aio_result(aio[0]), NNG_ECANCELED); + NUTS_FAIL(nng_aio_result(aio[1]), NNG_ECLOSED); for (int i = 0; i < 2; i++) { nng_aio_free(aio[i]); @@ -891,25 +832,25 @@ test_req_ctx_recv_nonblock(void) nng_aio * aio; nng_msg * msg; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_ctx_open(&ctx, req)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_ctx_open(&ctx, req)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); nng_aio_set_msg(aio, msg); nng_ctx_send(ctx, aio); nng_aio_wait(aio); - TEST_NNG_PASS(nng_aio_result(aio)); + NUTS_PASS(nng_aio_result(aio)); nng_aio_set_timeout(aio, 0); // Instant timeout nng_ctx_recv(ctx, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); nng_aio_free(aio); } @@ -921,17 +862,17 @@ test_req_ctx_send_nonblock(void) nng_aio * aio; nng_msg * msg; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_ctx_open(&ctx, req)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_ctx_open(&ctx, req)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); nng_aio_set_msg(aio, msg); nng_aio_set_timeout(aio, 0); // Instant timeout nng_ctx_send(ctx, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(req)); + NUTS_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); + NUTS_CLOSE(req); nng_aio_free(aio); nng_msg_free(msg); } @@ -945,23 +886,23 @@ test_req_ctx_recv_close_socket(void) nng_aio * aio; nng_msg * m; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_ctx_open(&ctx, req)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(testutil_marry(req, rep)); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_ctx_open(&ctx, req)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_MARRY(req, rep); + NUTS_PASS(nng_msg_alloc(&m, 0)); nng_aio_set_msg(aio, m); nng_ctx_send(ctx, aio); nng_aio_wait(aio); - TEST_NNG_PASS(nng_aio_result(aio)); + NUTS_PASS(nng_aio_result(aio)); nng_ctx_recv(ctx, aio); nng_close(req); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECLOSED); + NUTS_FAIL(nng_aio_result(aio), NNG_ECLOSED); nng_aio_free(aio); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(rep); } static void @@ -970,32 +911,32 @@ test_req_validate_peer(void) nng_socket s1, s2; nng_stat * stats; nng_stat * reject; - char addr[64]; + char * addr; - testutil_scratch_addr("inproc", sizeof(addr), addr); + NUTS_ADDR(addr, "inproc"); - TEST_NNG_PASS(nng_req0_open(&s1)); - TEST_NNG_PASS(nng_req0_open(&s2)); + NUTS_PASS(nng_req0_open(&s1)); + NUTS_PASS(nng_req0_open(&s2)); - TEST_NNG_PASS(nng_listen(s1, addr, NULL, 0)); - TEST_NNG_PASS(nng_dial(s2, addr, NULL, NNG_FLAG_NONBLOCK)); + NUTS_PASS(nng_listen(s1, addr, NULL, 0)); + NUTS_PASS(nng_dial(s2, addr, NULL, NNG_FLAG_NONBLOCK)); - testutil_sleep(100); - TEST_NNG_PASS(nng_stats_get(&stats)); + NUTS_SLEEP(100); + NUTS_PASS(nng_stats_get(&stats)); - TEST_CHECK(stats != NULL); - TEST_CHECK((reject = nng_stat_find_socket(stats, s1)) != NULL); - TEST_CHECK((reject = nng_stat_find(reject, "reject")) != NULL); + NUTS_TRUE(stats != NULL); + NUTS_TRUE((reject = nng_stat_find_socket(stats, s1)) != NULL); + NUTS_TRUE((reject = nng_stat_find(reject, "reject")) != NULL); - TEST_CHECK(nng_stat_type(reject) == NNG_STAT_COUNTER); - TEST_CHECK(nng_stat_value(reject) > 0); + NUTS_TRUE(nng_stat_type(reject) == NNG_STAT_COUNTER); + NUTS_TRUE(nng_stat_value(reject) > 0); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(s2)); + NUTS_CLOSE(s1); + NUTS_CLOSE(s2); nng_stats_free(stats); } -TEST_LIST = { +NUTS_TESTS = { { "req identity", test_req_identity }, { "req ttl option", test_req_ttl_option }, { "req resend option", test_req_resend_option }, diff --git a/src/protocol/reqrep0/xrep_test.c b/src/protocol/reqrep0/xrep_test.c index 8fc36964..6f1564eb 100644 --- a/src/protocol/reqrep0/xrep_test.c +++ b/src/protocol/reqrep0/xrep_test.c @@ -7,14 +7,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/reqrep0/rep.h> -#include <nng/protocol/reqrep0/req.h> - -#include <acutest.h> -#include <testutil.h> +#include <nuts.h> static void test_xrep_identity(void) @@ -24,16 +17,16 @@ test_xrep_identity(void) char * n1; char * n2; - TEST_NNG_PASS(nng_rep0_open_raw(&s)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PROTO, &p1)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PEER, &p2)); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PROTONAME, &n1)); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PEERNAME, &n2)); - TEST_NNG_PASS(nng_close(s)); - TEST_CHECK(p1 == NNG_REP0_SELF); - TEST_CHECK(p2 == NNG_REP0_PEER); - TEST_CHECK(strcmp(n1, NNG_REP0_SELF_NAME) == 0); - TEST_CHECK(strcmp(n2, NNG_REP0_PEER_NAME) == 0); + NUTS_PASS(nng_rep0_open_raw(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p1)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p2)); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n1)); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n2)); + NUTS_CLOSE(s); + NUTS_TRUE(p1 == NNG_REP0_SELF); + NUTS_TRUE(p2 == NNG_REP0_PEER); + NUTS_MATCH(n1, NNG_REP0_SELF_NAME); + NUTS_MATCH(n2, NNG_REP0_PEER_NAME); nng_strfree(n1); nng_strfree(n2); } @@ -44,10 +37,10 @@ test_xrep_raw(void) nng_socket s; bool b; - TEST_NNG_PASS(nng_rep0_open_raw(&s)); - TEST_NNG_PASS(nng_getopt_bool(s, NNG_OPT_RAW, &b)); - TEST_CHECK(b); - TEST_NNG_PASS(nng_close(s)); + NUTS_PASS(nng_rep0_open_raw(&s)); + NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_RAW, &b)); + NUTS_TRUE(b); + NUTS_CLOSE(s); } static void @@ -56,9 +49,9 @@ test_xrep_no_context(void) nng_socket s; nng_ctx ctx; - TEST_NNG_PASS(nng_rep0_open_raw(&s)); - TEST_NNG_FAIL(nng_ctx_open(&ctx, s), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(s)); + NUTS_PASS(nng_rep0_open_raw(&s)); + NUTS_FAIL(nng_ctx_open(&ctx, s), NNG_ENOTSUP); + NUTS_CLOSE(s); } static void @@ -68,24 +61,24 @@ test_xrep_poll_writeable(void) nng_socket req; nng_socket rep; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_getopt_int(rep, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_socket_get_int(rep, NNG_OPT_SENDFD, &fd)); + NUTS_TRUE(fd >= 0); // We are always writeable, even before connect. This is so that // back-pressure from a bad peer can't trash others. We assume // that peers won't send us requests faster than they can consume // the answers. If they do, they will lose their answers. - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); // Now it's writable. - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -96,32 +89,32 @@ test_xrep_poll_readable(void) nng_socket rep; nng_msg * msg; - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_getopt_int(rep, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_socket_get_int(rep, NNG_OPT_RECVFD, &fd)); + NUTS_TRUE(fd >= 0); // Not readable if not connected! - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); // Even after connect (no message yet) - TEST_NNG_PASS(testutil_marry(req, rep)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(req, rep); + NUTS_TRUE(nuts_poll_fd(fd) == false); // But once we send messages, it is. // We have to send a request, in order to send a reply. - TEST_NNG_SEND_STR(req, "abc"); - testutil_sleep(100); + NUTS_SEND(req, "abc"); + NUTS_SLEEP(100); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); // and receiving makes it no longer ready - TEST_NNG_PASS(nng_recvmsg(rep, &msg, 0)); + NUTS_PASS(nng_recvmsg(rep, &msg, 0)); nng_msg_free(msg); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -130,28 +123,28 @@ test_xrep_validate_peer(void) nng_socket s1, s2; nng_stat * stats; nng_stat * reject; - char addr[64]; + char *addr; - testutil_scratch_addr("inproc", sizeof(addr), addr); + NUTS_ADDR(addr, "inproc"); - TEST_NNG_PASS(nng_rep0_open_raw(&s1)); - TEST_NNG_PASS(nng_rep0_open(&s2)); + NUTS_PASS(nng_rep0_open_raw(&s1)); + NUTS_PASS(nng_rep0_open(&s2)); - TEST_NNG_PASS(nng_listen(s1, addr, NULL, 0)); - TEST_NNG_PASS(nng_dial(s2, addr, NULL, NNG_FLAG_NONBLOCK)); + NUTS_PASS(nng_listen(s1, addr, NULL, 0)); + NUTS_PASS(nng_dial(s2, addr, NULL, NNG_FLAG_NONBLOCK)); - testutil_sleep(100); - TEST_NNG_PASS(nng_stats_get(&stats)); + NUTS_SLEEP(100); + NUTS_PASS(nng_stats_get(&stats)); - TEST_CHECK(stats != NULL); - TEST_CHECK((reject = nng_stat_find_socket(stats, s1)) != NULL); - TEST_CHECK((reject = nng_stat_find(reject, "reject")) != NULL); + NUTS_TRUE(stats != NULL); + NUTS_TRUE((reject = nng_stat_find_socket(stats, s1)) != NULL); + NUTS_TRUE((reject = nng_stat_find(reject, "reject")) != NULL); - TEST_CHECK(nng_stat_type(reject) == NNG_STAT_COUNTER); - TEST_CHECK(nng_stat_value(reject) > 0); + NUTS_TRUE(nng_stat_type(reject) == NNG_STAT_COUNTER); + NUTS_TRUE(nng_stat_value(reject) > 0); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(s2)); + NUTS_PASS(nng_close(s1)); + NUTS_PASS(nng_close(s2)); nng_stats_free(stats); } @@ -164,26 +157,26 @@ test_xrep_close_pipe_before_send(void) nng_aio * aio1; nng_msg * m; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_aio_alloc(&aio1, NULL, NULL)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_aio_alloc(&aio1, NULL, NULL)); - TEST_NNG_PASS(testutil_marry(req, rep)); - TEST_NNG_SEND_STR(req, "test"); + NUTS_MARRY(req, rep); + NUTS_SEND(req, "test"); nng_recv_aio(rep, aio1); nng_aio_wait(aio1); - TEST_NNG_PASS(nng_aio_result(aio1)); - TEST_CHECK((m = nng_aio_get_msg(aio1)) != NULL); + NUTS_PASS(nng_aio_result(aio1)); + NUTS_TRUE((m = nng_aio_get_msg(aio1)) != NULL); p = nng_msg_get_pipe(m); - TEST_NNG_PASS(nng_pipe_close(p)); - TEST_NNG_PASS(nng_sendmsg(rep, m, 0)); + NUTS_PASS(nng_pipe_close(p)); + NUTS_PASS(nng_sendmsg(rep, m, 0)); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); nng_aio_free(aio1); } @@ -195,37 +188,37 @@ test_xrep_close_pipe_during_send(void) nng_pipe p; nng_msg * m; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_RECVBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_RECVBUF, 1)); - - TEST_NNG_PASS(testutil_marry(req, rep)); - - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_append_u32(m, (unsigned) 0x81000000u)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); - TEST_NNG_PASS(nng_recvmsg(rep, &m, 0)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_RECVBUF, 20)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_RECVBUF, 1)); + + NUTS_MARRY(req, rep); + + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, (unsigned) 0x81000000u)); + NUTS_PASS(nng_sendmsg(req, m, 0)); + NUTS_PASS(nng_recvmsg(rep, &m, 0)); p = nng_msg_get_pipe(m); nng_msg_free(m); for (int i = 0; i < 100; i++) { - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_header_append_u32(m, nng_pipe_id(p))); - TEST_NNG_PASS( + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_header_append_u32(m, nng_pipe_id(p))); + NUTS_PASS( nng_msg_header_append_u32(m, (unsigned) i | 0x80000000u)); // xrep does not exert back-pressure - TEST_NNG_PASS(nng_sendmsg(rep, m, 0)); + NUTS_PASS(nng_sendmsg(rep, m, 0)); } - TEST_NNG_PASS(nng_pipe_close(p)); + NUTS_PASS(nng_pipe_close(p)); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -235,27 +228,27 @@ test_xrep_close_during_recv(void) nng_socket req; nng_msg * m; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_RECVBUF, 5)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_RECVBUF, 5)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); for (unsigned i = 0; i < 100; i++) { int rv; - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_header_append_u32(m, i | 0x80000000u)); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_header_append_u32(m, i | 0x80000000u)); rv = nng_sendmsg(req, m, 0); if (rv == NNG_ETIMEDOUT) { nng_msg_free(m); break; } } - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -264,14 +257,14 @@ test_xrep_recv_aio_stopped(void) nng_socket rep; nng_aio * aio; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_stop(aio); nng_recv_aio(rep, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECANCELED); - TEST_NNG_PASS(nng_close(rep)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_CLOSE(rep); nng_aio_free(aio); } @@ -282,21 +275,21 @@ test_xrep_send_no_header(void) nng_socket req; nng_msg * m; - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_sendmsg(rep, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(rep, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_sendmsg(rep, m, 0)); + NUTS_FAIL(nng_recvmsg(rep, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -306,21 +299,21 @@ test_xrep_recv_garbage(void) nng_socket req; nng_msg * m; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_append_u32(m, 1u)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(rep, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, 1u)); + NUTS_PASS(nng_sendmsg(req, m, 0)); + NUTS_FAIL(nng_recvmsg(rep, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -332,29 +325,29 @@ test_xrep_ttl_option(void) size_t sz; const char *opt = NNG_OPT_MAXTTL; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_setopt_int(rep, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(rep, opt, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(rep, opt, -1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(rep, opt, 16), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(rep, opt, 256), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(rep, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(rep, opt, &v)); - TEST_CHECK(v == 3); + NUTS_PASS(nng_socket_set_int(rep, opt, 1)); + NUTS_FAIL(nng_socket_set_int(rep, opt, 0), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(rep, opt, -1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(rep, opt, 16), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(rep, opt, 256), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(rep, opt, 3)); + NUTS_PASS(nng_socket_get_int(rep, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(rep, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(rep, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_CHECK(nng_setopt(rep, opt, "", 1) == NNG_EINVAL); + NUTS_TRUE(nng_socket_set(rep, opt, "", 1) == NNG_EINVAL); sz = 1; - TEST_CHECK(nng_getopt(rep, opt, &v, &sz) == NNG_EINVAL); - TEST_CHECK(nng_setopt_bool(rep, opt, true) == NNG_EBADTYPE); - TEST_CHECK(nng_getopt_bool(rep, opt, &b) == NNG_EBADTYPE); + NUTS_TRUE(nng_socket_get(rep, opt, &v, &sz) == NNG_EINVAL); + NUTS_TRUE(nng_socket_set_bool(rep, opt, true) == NNG_EBADTYPE); + NUTS_TRUE(nng_socket_get_bool(rep, opt, &b) == NNG_EBADTYPE); - TEST_CHECK(nng_close(rep) == 0); + NUTS_TRUE(nng_close(rep) == 0); } static void @@ -364,65 +357,65 @@ test_xrep_ttl_drop(void) nng_socket req; nng_msg * m; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_MAXTTL, 3)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_MAXTTL, 3)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 200)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); // Send messages. Note that xrep implicitly adds a hop on receive. - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_append_u32(m, 1u)); // 2 hops - TEST_NNG_PASS(nng_msg_append_u32(m, 0x80000001u)); - TEST_NNG_PASS(nng_msg_append(m, "PASS1", 6)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); - - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_append_u32(m, 1u)); // 4 hops -- discard! - TEST_NNG_PASS(nng_msg_append_u32(m, 2u)); - TEST_NNG_PASS(nng_msg_append_u32(m, 3u)); - TEST_NNG_PASS(nng_msg_append_u32(m, 0x80000002u)); - TEST_NNG_PASS(nng_msg_append(m, "FAIL2", 6)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); - - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_append_u32(m, 1u)); // 3 hops - passes - TEST_NNG_PASS(nng_msg_append_u32(m, 2u)); - TEST_NNG_PASS(nng_msg_append_u32(m, 0x80000003u)); - TEST_NNG_PASS(nng_msg_append(m, "PASS3", 6)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); - - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_append_u32(m, 1u)); // 4 hops -- discard! - TEST_NNG_PASS(nng_msg_append_u32(m, 2u)); - TEST_NNG_PASS(nng_msg_append_u32(m, 3u)); - TEST_NNG_PASS(nng_msg_append_u32(m, 0x80000003u)); - TEST_NNG_PASS(nng_msg_append(m, "FAIL4", 6)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_append_u32(m, 1u)); // 2 hops + NUTS_PASS(nng_msg_append_u32(m, 0x80000001u)); + NUTS_PASS(nng_msg_append(m, "PASS1", 6)); + NUTS_PASS(nng_sendmsg(req, m, 0)); + + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_append_u32(m, 1u)); // 4 hops -- discard! + NUTS_PASS(nng_msg_append_u32(m, 2u)); + NUTS_PASS(nng_msg_append_u32(m, 3u)); + NUTS_PASS(nng_msg_append_u32(m, 0x80000002u)); + NUTS_PASS(nng_msg_append(m, "FAIL2", 6)); + NUTS_PASS(nng_sendmsg(req, m, 0)); + + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_append_u32(m, 1u)); // 3 hops - passes + NUTS_PASS(nng_msg_append_u32(m, 2u)); + NUTS_PASS(nng_msg_append_u32(m, 0x80000003u)); + NUTS_PASS(nng_msg_append(m, "PASS3", 6)); + NUTS_PASS(nng_sendmsg(req, m, 0)); + + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_append_u32(m, 1u)); // 4 hops -- discard! + NUTS_PASS(nng_msg_append_u32(m, 2u)); + NUTS_PASS(nng_msg_append_u32(m, 3u)); + NUTS_PASS(nng_msg_append_u32(m, 0x80000003u)); + NUTS_PASS(nng_msg_append(m, "FAIL4", 6)); + NUTS_PASS(nng_sendmsg(req, m, 0)); // So on receive we should see 80000001 and 80000003. - TEST_NNG_PASS(nng_recvmsg(rep, &m, 0)); - TEST_CHECK(nng_msg_header_len(m) == 12); - TEST_CHECK(nng_msg_len(m) == 6); - TEST_CHECK(strcmp(nng_msg_body(m), "PASS1") == 0); + NUTS_PASS(nng_recvmsg(rep, &m, 0)); + NUTS_TRUE(nng_msg_header_len(m) == 12); + NUTS_TRUE(nng_msg_len(m) == 6); + NUTS_TRUE(strcmp(nng_msg_body(m), "PASS1") == 0); nng_msg_free(m); - TEST_NNG_PASS(nng_recvmsg(rep, &m, 0)); - TEST_CHECK(nng_msg_header_len(m) == 16); // 3 hops + ID - TEST_CHECK(nng_msg_len(m) == 6); - TEST_CHECK(strcmp(nng_msg_body(m), "PASS3") == 0); + NUTS_PASS(nng_recvmsg(rep, &m, 0)); + NUTS_TRUE(nng_msg_header_len(m) == 16); // 3 hops + ID + NUTS_TRUE(nng_msg_len(m) == 6); + NUTS_TRUE(strcmp(nng_msg_body(m), "PASS3") == 0); nng_msg_free(m); - TEST_NNG_FAIL(nng_recvmsg(rep, &m, 0), NNG_ETIMEDOUT); + NUTS_FAIL(nng_recvmsg(rep, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } -TEST_LIST = { +NUTS_TESTS = { { "xrep identity", test_xrep_identity }, { "xrep raw", test_xrep_raw }, { "xrep no context", test_xrep_no_context }, diff --git a/src/protocol/reqrep0/xreq_test.c b/src/protocol/reqrep0/xreq_test.c index 68a7c7f5..8c850cba 100644 --- a/src/protocol/reqrep0/xreq_test.c +++ b/src/protocol/reqrep0/xreq_test.c @@ -7,14 +7,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/reqrep0/rep.h> -#include <nng/protocol/reqrep0/req.h> - -#include <acutest.h> -#include <testutil.h> +#include <nuts.h> static void test_xreq_identity(void) @@ -24,16 +17,16 @@ test_xreq_identity(void) char * n1; char * n2; - TEST_NNG_PASS(nng_req0_open_raw(&s)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PROTO, &p1)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PEER, &p2)); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PROTONAME, &n1)); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PEERNAME, &n2)); - TEST_NNG_PASS(nng_close(s)); - TEST_CHECK(p1 == NNG_REQ0_SELF); - TEST_CHECK(p2 == NNG_REQ0_PEER); - TEST_CHECK(strcmp(n1, NNG_REQ0_SELF_NAME) == 0); - TEST_CHECK(strcmp(n2, NNG_REQ0_PEER_NAME) == 0); + NUTS_PASS(nng_req0_open_raw(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p1)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p2)); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n1)); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n2)); + NUTS_CLOSE(s); + NUTS_TRUE(p1 == NNG_REQ0_SELF); + NUTS_TRUE(p2 == NNG_REQ0_PEER); + NUTS_MATCH(n1, NNG_REQ0_SELF_NAME); + NUTS_MATCH(n2, NNG_REQ0_PEER_NAME); nng_strfree(n1); nng_strfree(n2); } @@ -44,10 +37,10 @@ test_xreq_raw(void) nng_socket s; bool b; - TEST_NNG_PASS(nng_req0_open_raw(&s)); - TEST_NNG_PASS(nng_getopt_bool(s, NNG_OPT_RAW, &b)); - TEST_CHECK(b); - TEST_NNG_PASS(nng_close(s)); + NUTS_PASS(nng_req0_open_raw(&s)); + NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_RAW, &b)); + NUTS_TRUE(b); + NUTS_CLOSE(s); } static void @@ -56,9 +49,9 @@ test_xreq_no_context(void) nng_socket s; nng_ctx ctx; - TEST_NNG_PASS(nng_req0_open_raw(&s)); - TEST_NNG_FAIL(nng_ctx_open(&ctx, s), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(s)); + NUTS_PASS(nng_req0_open_raw(&s)); + NUTS_FAIL(nng_ctx_open(&ctx, s), NNG_ENOTSUP); + NUTS_CLOSE(s); } static void @@ -68,21 +61,21 @@ test_xreq_poll_writeable(void) nng_socket req; nng_socket rep; - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_getopt_int(req, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_get_int(req, NNG_OPT_SENDFD, &fd)); + NUTS_TRUE(fd >= 0); // We can't write until we have a connection. - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); // Now it's writable. - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -93,44 +86,44 @@ test_xreq_poll_readable(void) nng_socket rep; nng_msg * msg; - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_rep0_open(&rep)); - TEST_NNG_PASS(nng_getopt_int(req, NNG_OPT_RECVFD, &fd)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_rep0_open(&rep)); + NUTS_PASS(nng_socket_get_int(req, NNG_OPT_RECVFD, &fd)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_CHECK(fd >= 0); + NUTS_TRUE(fd >= 0); // Not readable if not connected! - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); // Even after connect (no message yet) - TEST_NNG_PASS(testutil_marry(req, rep)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(req, rep); + NUTS_TRUE(nuts_poll_fd(fd) == false); // But once we send messages, it is. // We have to send a request, in order to send a reply. - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); // Request ID - TEST_NNG_PASS(nng_msg_append_u32(msg, 0x80000000)); - TEST_NNG_PASS(nng_sendmsg(req, msg, 0)); + NUTS_PASS(nng_msg_append_u32(msg, 0x80000000)); + NUTS_PASS(nng_sendmsg(req, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(rep, &msg, 0)); - TEST_NNG_PASS(nng_sendmsg(rep, msg, 0)); + NUTS_PASS(nng_recvmsg(rep, &msg, 0)); + NUTS_PASS(nng_sendmsg(rep, msg, 0)); - testutil_sleep(100); + NUTS_SLEEP(100); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); // and receiving makes it no longer ready - TEST_NNG_PASS(nng_recvmsg(req, &msg, 0)); + NUTS_PASS(nng_recvmsg(req, &msg, 0)); nng_msg_free(msg); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -139,28 +132,28 @@ test_xreq_validate_peer(void) nng_socket s1, s2; nng_stat * stats; nng_stat * reject; - char addr[64]; + char * addr; - testutil_scratch_addr("inproc", sizeof(addr), addr); + NUTS_ADDR(addr, "inproc"); - TEST_NNG_PASS(nng_req0_open_raw(&s1)); - TEST_NNG_PASS(nng_req0_open(&s2)); + NUTS_PASS(nng_req0_open_raw(&s1)); + NUTS_PASS(nng_req0_open(&s2)); - TEST_NNG_PASS(nng_listen(s1, addr, NULL, 0)); - TEST_NNG_PASS(nng_dial(s2, addr, NULL, NNG_FLAG_NONBLOCK)); + NUTS_PASS(nng_listen(s1, addr, NULL, 0)); + NUTS_PASS(nng_dial(s2, addr, NULL, NNG_FLAG_NONBLOCK)); - testutil_sleep(100); - TEST_NNG_PASS(nng_stats_get(&stats)); + NUTS_SLEEP(100); + NUTS_PASS(nng_stats_get(&stats)); - TEST_CHECK(stats != NULL); - TEST_CHECK((reject = nng_stat_find_socket(stats, s1)) != NULL); - TEST_CHECK((reject = nng_stat_find(reject, "reject")) != NULL); + NUTS_TRUE(stats != NULL); + NUTS_TRUE((reject = nng_stat_find_socket(stats, s1)) != NULL); + NUTS_TRUE((reject = nng_stat_find(reject, "reject")) != NULL); - TEST_CHECK(nng_stat_type(reject) == NNG_STAT_COUNTER); - TEST_CHECK(nng_stat_value(reject) > 0); + NUTS_TRUE(nng_stat_type(reject) == NNG_STAT_COUNTER); + NUTS_TRUE(nng_stat_value(reject) > 0); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(s2)); + NUTS_CLOSE(s1); + NUTS_CLOSE(s2); nng_stats_free(stats); } @@ -170,14 +163,14 @@ test_xreq_recv_aio_stopped(void) nng_socket req; nng_aio * aio; - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_stop(aio); nng_recv_aio(req, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECANCELED); - TEST_NNG_PASS(nng_close(req)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_CLOSE(req); nng_aio_free(aio); } @@ -189,33 +182,33 @@ test_xreq_recv_garbage(void) nng_msg * m; uint32_t req_id; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(req, rep)); + NUTS_MARRY(req, rep); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_append_u32(m, 0x80000000)); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_append_u32(m, 0x80000000)); + NUTS_PASS(nng_sendmsg(req, m, 0)); - TEST_NNG_PASS(nng_recvmsg(rep, &m, 0)); + NUTS_PASS(nng_recvmsg(rep, &m, 0)); // The message will have a header that contains the 32-bit pipe ID, // followed by the 32-bit request ID. We will discard the request // ID before sending it out. - TEST_CHECK(nng_msg_header_len(m) == 8); - TEST_NNG_PASS(nng_msg_header_chop_u32(m, &req_id)); - TEST_CHECK(req_id == 0x80000000); + NUTS_TRUE(nng_msg_header_len(m) == 8); + NUTS_PASS(nng_msg_header_chop_u32(m, &req_id)); + NUTS_TRUE(req_id == 0x80000000); - TEST_NNG_PASS(nng_sendmsg(rep, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(req, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_sendmsg(rep, m, 0)); + NUTS_FAIL(nng_recvmsg(req, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -227,37 +220,37 @@ test_xreq_recv_header(void) nng_pipe p1, p2; uint32_t id; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry_ex(req, rep, NULL, &p1, &p2)); + NUTS_MARRY_EX(req, rep, NULL, &p1, &p2); // Simulate a few hops. - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_header_append_u32(m, nng_pipe_id(p2))); - TEST_NNG_PASS(nng_msg_header_append_u32(m, 0x2)); - TEST_NNG_PASS(nng_msg_header_append_u32(m, 0x1)); - TEST_NNG_PASS(nng_msg_header_append_u32(m, 0x80000123u)); - - TEST_NNG_PASS(nng_sendmsg(rep, m, 0)); - - TEST_NNG_PASS(nng_recvmsg(req, &m, 0)); - TEST_CHECK(nng_msg_header_len(m) == 12); - TEST_NNG_PASS(nng_msg_header_trim_u32(m, &id)); - TEST_CHECK(id == 0x2); - TEST_NNG_PASS(nng_msg_header_trim_u32(m, &id)); - TEST_CHECK(id == 0x1); - TEST_NNG_PASS(nng_msg_header_trim_u32(m, &id)); - TEST_CHECK(id == 0x80000123u); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_header_append_u32(m, nng_pipe_id(p2))); + NUTS_PASS(nng_msg_header_append_u32(m, 0x2)); + NUTS_PASS(nng_msg_header_append_u32(m, 0x1)); + NUTS_PASS(nng_msg_header_append_u32(m, 0x80000123u)); + + NUTS_PASS(nng_sendmsg(rep, m, 0)); + + NUTS_PASS(nng_recvmsg(req, &m, 0)); + NUTS_TRUE(nng_msg_header_len(m) == 12); + NUTS_PASS(nng_msg_header_trim_u32(m, &id)); + NUTS_TRUE(id == 0x2); + NUTS_PASS(nng_msg_header_trim_u32(m, &id)); + NUTS_TRUE(id == 0x1); + NUTS_PASS(nng_msg_header_trim_u32(m, &id)); + NUTS_TRUE(id == 0x80000123u); nng_msg_free(m); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -269,26 +262,26 @@ test_xreq_close_during_recv(void) nng_pipe p1; nng_pipe p2; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_RECVBUF, 5)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_RECVBUF, 5)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(testutil_marry_ex(req, rep, NULL, &p1, &p2)); - TEST_CHECK(nng_pipe_id(p1) > 0); - TEST_CHECK(nng_pipe_id(p2) > 0); + NUTS_MARRY_EX(req, rep, NULL, &p1, &p2); + NUTS_TRUE(nng_pipe_id(p1) > 0); + NUTS_TRUE(nng_pipe_id(p2) > 0); for (unsigned i = 0; i < 20; i++) { - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_header_append_u32(m, nng_pipe_id(p2))); - TEST_NNG_PASS(nng_msg_header_append_u32(m, i | 0x80000000u)); - testutil_sleep(10); - TEST_NNG_PASS(nng_sendmsg(rep, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_header_append_u32(m, nng_pipe_id(p2))); + NUTS_PASS(nng_msg_header_append_u32(m, i | 0x80000000u)); + NUTS_SLEEP(10); + NUTS_PASS(nng_sendmsg(rep, m, 0)); } - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -300,27 +293,27 @@ test_xreq_close_pipe_during_send(void) nng_pipe p1; nng_pipe p2; - TEST_NNG_PASS(nng_rep0_open_raw(&rep)); - TEST_NNG_PASS(nng_req0_open_raw(&req)); - TEST_NNG_PASS(nng_setopt_ms(rep, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(req, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_int(rep, NNG_OPT_RECVBUF, 5)); - TEST_NNG_PASS(nng_setopt_int(req, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_rep0_open_raw(&rep)); + NUTS_PASS(nng_req0_open_raw(&req)); + NUTS_PASS(nng_socket_set_ms(rep, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(req, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_int(rep, NNG_OPT_RECVBUF, 5)); + NUTS_PASS(nng_socket_set_int(req, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(testutil_marry_ex(req, rep, NULL, &p1, &p2)); - TEST_CHECK(nng_pipe_id(p1) > 0); - TEST_CHECK(nng_pipe_id(p2) > 0); + NUTS_MARRY_EX(req, rep, NULL, &p1, &p2); + NUTS_TRUE(nng_pipe_id(p1) > 0); + NUTS_TRUE(nng_pipe_id(p2) > 0); for (unsigned i = 0; i < 20; i++) { - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_header_append_u32(m, i | 0x80000000u)); - testutil_sleep(10); - TEST_NNG_PASS(nng_sendmsg(req, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_header_append_u32(m, i | 0x80000000u)); + NUTS_SLEEP(10); + NUTS_PASS(nng_sendmsg(req, m, 0)); } - TEST_NNG_PASS(nng_pipe_close(p1)); - TEST_NNG_PASS(nng_close(req)); - TEST_NNG_PASS(nng_close(rep)); + NUTS_PASS(nng_pipe_close(p1)); + NUTS_CLOSE(req); + NUTS_CLOSE(rep); } static void @@ -332,32 +325,32 @@ test_xreq_ttl_option(void) size_t sz; const char *opt = NNG_OPT_MAXTTL; - TEST_NNG_PASS(nng_req0_open_raw(&rep)); + NUTS_PASS(nng_req0_open_raw(&rep)); - TEST_NNG_PASS(nng_setopt_int(rep, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(rep, opt, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(rep, opt, -1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(rep, opt, 16), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(rep, opt, 256), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(rep, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(rep, opt, &v)); - TEST_CHECK(v == 3); + NUTS_PASS(nng_socket_set_int(rep, opt, 1)); + NUTS_FAIL(nng_socket_set_int(rep, opt, 0), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(rep, opt, -1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(rep, opt, 16), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(rep, opt, 256), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(rep, opt, 3)); + NUTS_PASS(nng_socket_get_int(rep, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(rep, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(rep, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_CHECK(nng_setopt(rep, opt, "", 1) == NNG_EINVAL); + NUTS_TRUE(nng_socket_set(rep, opt, "", 1) == NNG_EINVAL); sz = 1; - TEST_CHECK(nng_getopt(rep, opt, &v, &sz) == NNG_EINVAL); - TEST_CHECK(nng_setopt_bool(rep, opt, true) == NNG_EBADTYPE); - TEST_CHECK(nng_getopt_bool(rep, opt, &b) == NNG_EBADTYPE); + NUTS_TRUE(nng_socket_get(rep, opt, &v, &sz) == NNG_EINVAL); + NUTS_TRUE(nng_socket_set_bool(rep, opt, true) == NNG_EBADTYPE); + NUTS_TRUE(nng_socket_get_bool(rep, opt, &b) == NNG_EBADTYPE); - TEST_CHECK(nng_close(rep) == 0); + NUTS_TRUE(nng_close(rep) == 0); } -TEST_LIST = { +NUTS_TESTS = { { "xreq identity", test_xreq_identity }, { "xreq raw", test_xreq_raw }, { "xreq no context", test_xreq_no_context }, |