diff options
Diffstat (limited to 'src/protocol/survey0')
| -rw-r--r-- | src/protocol/survey0/respond_test.c | 503 | ||||
| -rw-r--r-- | src/protocol/survey0/survey_test.c | 540 | ||||
| -rw-r--r-- | src/protocol/survey0/xrespond_test.c | 397 | ||||
| -rw-r--r-- | src/protocol/survey0/xsurvey_test.c | 365 |
4 files changed, 856 insertions, 949 deletions
diff --git a/src/protocol/survey0/respond_test.c b/src/protocol/survey0/respond_test.c index efda181b..51844c76 100644 --- a/src/protocol/survey0/respond_test.c +++ b/src/protocol/survey0/respond_test.c @@ -7,18 +7,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/survey0/respond.h> -#include <nng/protocol/survey0/survey.h> - -#include <acutest.h> -#include <testutil.h> - -#ifndef NNI_PROTO -#define NNI_PROTO(x, y) (((x) << 4u) | (y)) -#endif +#include <nuts.h> void test_resp_identity(void) @@ -27,18 +16,18 @@ test_resp_identity(void) int p; char * n; - TEST_CHECK(nng_respondent0_open(&s) == 0); - TEST_CHECK(nng_getopt_int(s, NNG_OPT_PROTO, &p) == 0); - TEST_CHECK(p == NNI_PROTO(6u, 3u)); - TEST_CHECK(nng_getopt_int(s, NNG_OPT_PEER, &p) == 0); - TEST_CHECK(p == NNI_PROTO(6u, 2u)); - TEST_CHECK(nng_getopt_string(s, NNG_OPT_PROTONAME, &n) == 0); - TEST_CHECK(strcmp(n, "respondent") == 0); + NUTS_PASS(nng_respondent0_open(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p)); + NUTS_TRUE(p == NNG_RESPONDENT0_SELF); + NUTS_TRUE(nng_socket_get_int(s, NNG_OPT_PEER, &p) == 0); + NUTS_TRUE(p == NNG_RESPONDENT0_PEER); + NUTS_TRUE(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n) == 0); + NUTS_MATCH(n, NNG_RESPONDENT0_SELF_NAME); nng_strfree(n); - TEST_CHECK(nng_getopt_string(s, NNG_OPT_PEERNAME, &n) == 0); - TEST_CHECK(strcmp(n, "surveyor") == 0); + NUTS_TRUE(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n) == 0); + NUTS_MATCH(n, NNG_RESPONDENT0_PEER_NAME); nng_strfree(n); - TEST_CHECK(nng_close(s) == 0); + NUTS_CLOSE(s); } void @@ -47,11 +36,11 @@ test_resp_send_bad_state(void) nng_socket resp; nng_msg * msg = NULL; - TEST_CHECK(nng_respondent0_open(&resp) == 0); - TEST_CHECK(nng_msg_alloc(&msg, 0) == 0); - TEST_CHECK(nng_sendmsg(resp, msg, 0) == NNG_ESTATE); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_FAIL(nng_sendmsg(resp, msg, 0), NNG_ESTATE); nng_msg_free(msg); - TEST_CHECK(nng_close(resp) == 0); + NUTS_CLOSE(resp); } void @@ -61,33 +50,33 @@ test_resp_poll_writeable(void) nng_socket surv; nng_socket resp; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_getopt_int(resp, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_socket_get_int(resp, 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(surv, resp)); + NUTS_MARRY(surv, resp); // 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(surv, "abc"); - TEST_NNG_RECV_STR(resp, "abc"); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_SEND(surv, "abc"); + NUTS_RECV(resp, "abc"); + NUTS_TRUE(nuts_poll_fd(fd) == true); // And is no longer writable once we send a message - TEST_NNG_SEND_STR(resp, "def"); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_SEND(resp, "def"); + NUTS_TRUE(nuts_poll_fd(fd) == false); // Even after receiving it - TEST_NNG_RECV_STR(surv, "def"); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_RECV(surv, "def"); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } void @@ -98,34 +87,34 @@ test_resp_poll_readable(void) nng_socket resp; nng_msg * msg; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_getopt_int(resp, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_socket_get_int(resp, 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(surv, resp)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(surv, resp); + 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(surv, "abc"); - testutil_sleep(100); + NUTS_SEND(surv, "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(resp, &msg, 0)); + NUTS_PASS(nng_recvmsg(resp, &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(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } void @@ -135,14 +124,12 @@ test_resp_context_no_poll(void) nng_socket resp; nng_ctx ctx; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_ctx_open(&ctx, resp)); - 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(resp)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_ctx_open(&ctx, resp)); + 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(resp); } void @@ -151,28 +138,28 @@ test_resp_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_respondent0_open(&s1)); - TEST_NNG_PASS(nng_respondent0_open(&s2)); + NUTS_PASS(nng_respondent0_open(&s1)); + NUTS_PASS(nng_respondent0_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); } @@ -183,17 +170,17 @@ test_resp_double_recv(void) nng_aio * aio1; nng_aio * aio2; - TEST_NNG_PASS(nng_respondent0_open(&s1)); - TEST_NNG_PASS(nng_aio_alloc(&aio1, NULL, NULL)); - TEST_NNG_PASS(nng_aio_alloc(&aio2, NULL, NULL)); + NUTS_PASS(nng_respondent0_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); } @@ -207,26 +194,26 @@ test_resp_close_pipe_before_send(void) nng_aio * aio1; nng_msg * m; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_aio_alloc(&aio1, NULL, NULL)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_aio_alloc(&aio1, NULL, NULL)); - TEST_NNG_PASS(testutil_marry(surv, resp)); - TEST_NNG_SEND_STR(surv, "test"); + NUTS_MARRY(surv, resp); + NUTS_SEND(surv, "test"); nng_recv_aio(resp, 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(resp, m, 0)); + NUTS_PASS(nng_pipe_close(p)); + NUTS_PASS(nng_sendmsg(resp, m, 0)); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); nng_aio_free(aio1); } @@ -238,25 +225,24 @@ test_resp_close_pipe_during_send(void) nng_pipe p = NNG_PIPE_INITIALIZER; nng_msg * m; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_RECVBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_setopt_int(resp, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_setopt_int(resp, NNG_OPT_RECVBUF, 20)); + NUTS_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_setopt_int(surv, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); 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(surv, m, 0)); - TEST_NNG_PASS(nng_recvmsg(resp, &m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, (unsigned) i | 0x80000000u)); + NUTS_PASS(nng_sendmsg(surv, m, 0)); + NUTS_PASS(nng_recvmsg(resp, &m, 0)); p = nng_msg_get_pipe(m); rv = nng_sendmsg(resp, m, 0); if (rv == NNG_ETIMEDOUT) { @@ -264,12 +250,12 @@ test_resp_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(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } void @@ -279,16 +265,16 @@ test_resp_ctx_recv_aio_stopped(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_ctx_open(&ctx, resp)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_ctx_open(&ctx, resp)); 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(resp)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_PASS(nng_ctx_close(ctx)); + NUTS_CLOSE(resp); nng_aio_free(aio); } @@ -303,54 +289,53 @@ test_resp_close_pipe_context_send(void) nng_aio * aio[10]; int i; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_setopt_int(resp, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_setopt_int(resp, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_setopt_int(surv, NNG_OPT_RECVBUF, 1)); for (i = 0; i < 10; i++) { - TEST_NNG_PASS(nng_ctx_open(&ctx[i], resp)); - TEST_NNG_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); + NUTS_PASS(nng_ctx_open(&ctx[i], resp)); + NUTS_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); } - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); for (i = 0; i < 10; 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(surv, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, (unsigned) i | 0x80000000u)); + NUTS_PASS(nng_sendmsg(surv, m, 0)); nng_ctx_recv(ctx[i], aio[i]); } for (i = 0; i < 10; 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 SURVEYOR socket is not reading the results. - TEST_NNG_PASS(nng_pipe_close(p)); + NUTS_PASS(nng_pipe_close(p)); for (i = 0; i < 10; 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(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } void @@ -363,33 +348,32 @@ test_resp_close_context_send(void) nng_aio * aio[10]; int i; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_setopt_int(resp, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_setopt_int(resp, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_setopt_int(surv, NNG_OPT_RECVBUF, 1)); for (i = 0; i < 10; i++) { - TEST_NNG_PASS(nng_ctx_open(&ctx[i], resp)); - TEST_NNG_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); + NUTS_PASS(nng_ctx_open(&ctx[i], resp)); + NUTS_PASS(nng_aio_alloc(&aio[i], NULL, NULL)); } - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); for (i = 0; i < 10; 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(surv, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, (unsigned) i | 0x80000000u)); + NUTS_PASS(nng_sendmsg(surv, m, 0)); nng_ctx_recv(ctx[i], aio[i]); } for (i = 0; i < 10; 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]); } @@ -397,17 +381,17 @@ test_resp_close_context_send(void) // Note that REQ socket is not reading the results. for (i = 0; i < 10; 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(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -417,16 +401,16 @@ test_resp_ctx_recv_nonblock(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_ctx_open(&ctx, resp)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_ctx_open(&ctx, resp)); + 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(resp)); + NUTS_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); + NUTS_CLOSE(resp); nng_aio_free(aio); } @@ -439,19 +423,19 @@ test_resp_ctx_send_nonblock(void) nng_aio * aio; nng_msg * msg; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_ctx_open(&ctx, resp)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_ctx_open(&ctx, resp)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_MARRY(surv, resp); - TEST_NNG_SEND_STR(surv, "SEND"); + NUTS_SEND(surv, "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); @@ -459,9 +443,9 @@ test_resp_ctx_send_nonblock(void) nng_ctx_send(ctx, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); nng_aio_free(aio); nng_msg_free(msg); } @@ -473,21 +457,21 @@ test_resp_recv_garbage(void) nng_socket surv; nng_msg * m; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 200)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_append_u32(m, 1u)); - TEST_NNG_PASS(nng_sendmsg(surv, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, 1u)); + NUTS_PASS(nng_sendmsg(surv, m, 0)); + NUTS_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -499,29 +483,29 @@ test_resp_ttl_option(void) size_t sz; const char *opt = NNG_OPT_MAXTTL; - TEST_NNG_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_setopt_int(resp, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(resp, opt, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(resp, opt, -1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(resp, opt, 16), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(resp, opt, 256), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(resp, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(resp, opt, &v)); - TEST_CHECK(v == 3); + NUTS_PASS(nng_setopt_int(resp, opt, 1)); + NUTS_FAIL(nng_setopt_int(resp, opt, 0), NNG_EINVAL); + NUTS_FAIL(nng_setopt_int(resp, opt, -1), NNG_EINVAL); + NUTS_FAIL(nng_setopt_int(resp, opt, 16), NNG_EINVAL); + NUTS_FAIL(nng_setopt_int(resp, opt, 256), NNG_EINVAL); + NUTS_PASS(nng_setopt_int(resp, opt, 3)); + NUTS_PASS(nng_socket_get_int(resp, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(resp, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(resp, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_NNG_FAIL(nng_setopt(resp, opt, "", 1), NNG_EINVAL); + NUTS_FAIL(nng_setopt(resp, opt, "", 1), NNG_EINVAL); sz = 1; - TEST_NNG_FAIL(nng_getopt(resp, opt, &v, &sz), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(resp, opt, true), NNG_EBADTYPE); - TEST_NNG_FAIL(nng_getopt_bool(resp, opt, &b), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get(resp, opt, &v, &sz), NNG_EINVAL); + NUTS_FAIL(nng_setopt_bool(resp, opt, true), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(resp, opt, &b), NNG_EBADTYPE); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(resp); } static void @@ -531,52 +515,52 @@ test_resp_ttl_drop(void) nng_socket surv; nng_msg * m; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_MAXTTL, 3)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_setopt_int(resp, NNG_OPT_MAXTTL, 3)); + NUTS_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 200)); + NUTS_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); // 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(surv, 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(surv, 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(surv, 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(surv, m, 0)); - - TEST_NNG_RECV_STR(resp, "PASS1"); - TEST_NNG_RECV_STR(resp, "PASS3"); - - TEST_NNG_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); - - TEST_NNG_PASS(nng_close(resp)); - TEST_NNG_PASS(nng_close(surv)); + 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(surv, 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(surv, 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(surv, 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(surv, m, 0)); + + NUTS_RECV(resp, "PASS1"); + NUTS_RECV(resp, "PASS3"); + + NUTS_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); + + NUTS_CLOSE(resp); + NUTS_CLOSE(surv); } TEST_LIST = { @@ -590,7 +574,8 @@ TEST_LIST = { { "respond close pipe before send", test_resp_close_pipe_before_send }, { "respond close pipe during send", test_resp_close_pipe_during_send }, { "respond recv aio ctx stopped", test_resp_ctx_recv_aio_stopped }, - { "respond close pipe context send", test_resp_close_pipe_context_send }, + { "respond close pipe context send", + test_resp_close_pipe_context_send }, { "respond close context send", test_resp_close_context_send }, { "respond context send nonblock", test_resp_ctx_send_nonblock }, { "respond context recv nonblock", test_resp_ctx_recv_nonblock }, diff --git a/src/protocol/survey0/survey_test.c b/src/protocol/survey0/survey_test.c index 65395eee..95d27adf 100644 --- a/src/protocol/survey0/survey_test.c +++ b/src/protocol/survey0/survey_test.c @@ -8,14 +8,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/survey0/respond.h> -#include <nng/protocol/survey0/survey.h> - -#include <acutest.h> -#include <testutil.h> +#include <nuts.h> static void test_surv_identity(void) @@ -24,18 +17,18 @@ test_surv_identity(void) int p; char * n; - TEST_NNG_PASS(nng_surveyor0_open(&s)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PROTO, &p)); - TEST_CHECK(p == NNG_SURVEYOR0_SELF); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PEER, &p)); - TEST_CHECK(p == NNG_SURVEYOR0_PEER); // 49 - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PROTONAME, &n)); - TEST_CHECK(strcmp(n, NNG_SURVEYOR0_SELF_NAME) == 0); + NUTS_PASS(nng_surveyor0_open(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p)); + NUTS_TRUE(p == NNG_SURVEYOR0_SELF); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p)); + NUTS_TRUE(p == NNG_SURVEYOR0_PEER); // 49 + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n)); + NUTS_MATCH(n, NNG_SURVEYOR0_SELF_NAME); nng_strfree(n); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PEERNAME, &n)); - TEST_CHECK(strcmp(n, NNG_SURVEYOR0_PEER_NAME) == 0); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n)); + NUTS_MATCH(n, NNG_SURVEYOR0_PEER_NAME); nng_strfree(n); - TEST_NNG_PASS(nng_close(s)); + NUTS_CLOSE(s); } static void @@ -47,31 +40,31 @@ test_surv_ttl_option(void) size_t sz; const char *opt = NNG_OPT_MAXTTL; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_setopt_int(surv, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(surv, opt, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(surv, opt, -1), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(surv, opt, 1)); + NUTS_FAIL(nng_socket_set_int(surv, opt, 0), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(surv, 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(surv, opt, 16), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(surv, opt, 256), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(surv, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(surv, opt, &v)); - TEST_CHECK(v == 3); + NUTS_FAIL(nng_socket_set_int(surv, opt, 16), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(surv, opt, 256), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(surv, opt, 3)); + NUTS_PASS(nng_socket_get_int(surv, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(surv, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(surv, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_NNG_FAIL(nng_setopt(surv, opt, "", 1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set(surv, opt, "", 1), NNG_EINVAL); sz = 1; - TEST_NNG_FAIL(nng_getopt(surv, opt, &v, &sz), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(surv, opt, true), NNG_EBADTYPE); - TEST_NNG_FAIL(nng_getopt_bool(surv, opt, &b), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get(surv, opt, &v, &sz), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_bool(surv, opt, true), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(surv, opt, &b), NNG_EBADTYPE); - TEST_NNG_PASS(nng_close(surv)); + NUTS_CLOSE(surv); } static void @@ -83,17 +76,17 @@ test_surv_survey_time_option(void) size_t sz = sizeof(b); const char * opt = NNG_OPT_SURVEYOR_SURVEYTIME; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_setopt_ms(surv, opt, 10)); - TEST_NNG_FAIL(nng_setopt(surv, opt, "", 1), NNG_EINVAL); - TEST_NNG_FAIL(nng_getopt(surv, opt, &b, &sz), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(surv, opt, true), NNG_EBADTYPE); - TEST_NNG_FAIL(nng_getopt_bool(surv, opt, &b), NNG_EBADTYPE); + NUTS_PASS(nng_socket_set_ms(surv, opt, 10)); + NUTS_FAIL(nng_socket_set(surv, opt, "", 1), NNG_EINVAL); + NUTS_FAIL(nng_socket_get(surv, opt, &b, &sz), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_bool(surv, opt, true), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(surv, opt, &b), NNG_EBADTYPE); - TEST_NNG_PASS(nng_getopt_ms(surv, opt, &d)); - TEST_CHECK(d == 10); - TEST_NNG_PASS(nng_close(surv)); + NUTS_PASS(nng_socket_get_ms(surv, opt, &d)); + NUTS_TRUE(d == 10); + NUTS_CLOSE(surv); } void @@ -102,10 +95,10 @@ test_surv_recv_bad_state(void) nng_socket surv; nng_msg * msg = NULL; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_FAIL(nng_recvmsg(surv, &msg, 0), NNG_ESTATE); - TEST_CHECK(msg == NULL); - TEST_NNG_PASS(nng_close(surv)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_FAIL(nng_recvmsg(surv, &msg, 0), NNG_ESTATE); + NUTS_TRUE(msg == NULL); + NUTS_CLOSE(surv); } static void @@ -116,31 +109,31 @@ test_surv_recv_garbage(void) nng_msg * m; uint32_t surv_id; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_sendmsg(surv, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_sendmsg(surv, m, 0)); - TEST_NNG_PASS(nng_recvmsg(resp, &m, 0)); + NUTS_PASS(nng_recvmsg(resp, &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, &surv_id)); + NUTS_TRUE(nng_msg_header_len(m) == 8); + NUTS_PASS(nng_msg_header_chop_u32(m, &surv_id)); - TEST_NNG_PASS(nng_sendmsg(resp, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(surv, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_sendmsg(resp, m, 0)); + NUTS_FAIL(nng_recvmsg(surv, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } #define SECOND 1000 @@ -150,188 +143,138 @@ test_surv_resp_exchange(void) { nng_socket surv; nng_socket resp; - nng_msg * msg = NULL; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, SECOND)); - - TEST_NNG_PASS(testutil_marry(resp, surv)); - - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append(msg, "ping", 5)); - TEST_CHECK(nng_msg_len(msg) == 5); - TEST_CHECK(strcmp(nng_msg_body(msg), "ping") == 0); - TEST_NNG_PASS(nng_sendmsg(surv, msg, 0)); - msg = NULL; - TEST_NNG_PASS(nng_recvmsg(resp, &msg, 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_NNG_PASS(nng_msg_append(msg, "pong", 5)); - TEST_NNG_PASS(nng_sendmsg(resp, msg, 0)); - msg = NULL; - TEST_NNG_PASS(nng_recvmsg(surv, &msg, 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_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, SECOND)); + + NUTS_MARRY(resp, surv); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_SEND(surv, "ping"); + NUTS_RECV(resp, "ping"); + NUTS_SEND(resp, "pong"); + NUTS_RECV(surv, "pong"); + + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } void test_surv_cancel(void) { - nng_msg * abc; - nng_msg * def; - nng_msg * cmd; nng_socket surv; nng_socket resp; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 5 * SECOND)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 5 * SECOND)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 16)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_surveyor0_open(&surv)); - 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(surv, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_int(surv, NNG_OPT_SENDBUF, 16)); - TEST_NNG_PASS(testutil_marry(resp, surv)); + NUTS_MARRY(resp, surv); // Send req #1 (abc). - TEST_CHECK(nng_sendmsg(surv, abc, 0) == 0); + NUTS_SEND(surv, "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 RESP 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(surv, def, 0)); + NUTS_SEND(surv, "def"); // Receive the first request (should be abc) on the REP server. - TEST_NNG_PASS(nng_recvmsg(resp, &cmd, 0)); - TEST_ASSERT(cmd != NULL); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "abc") == 0); + NUTS_RECV(resp, "abc"); // RESP sends the reply to first command. This will be discarded // by the SURV socket. - TEST_NNG_PASS(nng_sendmsg(resp, cmd, 0)); + NUTS_SEND(resp, "abc"); // Now get the next command from the REP; should be "def". - TEST_NNG_PASS(nng_recvmsg(resp, &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(resp, "def"); // And send it back to REQ. - TEST_NNG_PASS(nng_sendmsg(resp, cmd, 0)); + NUTS_SEND(resp, "def"); // Try a req command. This should give back "def" - TEST_NNG_PASS(nng_recvmsg(surv, &cmd, 0)); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "def") == 0); - nng_msg_free(cmd); + NUTS_RECV(surv, "def"); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } void test_surv_cancel_abort_recv(void) { - - nng_msg * abc; - nng_msg * def; - nng_msg * cmd; nng_aio * aio; nng_duration time = SECOND * 10; // 10s (kind of never) nng_socket surv; nng_socket resp; - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SURVEYOR_SURVEYTIME, time)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 16)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, 5 * SECOND)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 5 * SECOND)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 5 * SECOND)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SURVEYOR_SURVEYTIME, time)); + NUTS_PASS(nng_socket_set_int(surv, NNG_OPT_SENDBUF, 16)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 5 * SECOND)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 5 * SECOND)); - 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)); - - TEST_NNG_PASS(testutil_marry(resp, surv)); + NUTS_MARRY(resp, surv); // Send survey #1 (abc). - TEST_NNG_PASS(nng_sendmsg(surv, abc, 0)); + NUTS_SEND(surv, "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(surv, aio); // Give time for this recv to post properly. - testutil_sleep(100); + NUTS_SLEEP(100); // Send the next next request ("def"). Note that // the respondent 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(surv, def, 0)); + NUTS_SEND(surv, "def"); // Our pending I/O should have been canceled. nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); // Receive the first request (should be abc) on the respondent. - TEST_NNG_PASS(nng_recvmsg(resp, &cmd, 0)); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "abc") == 0); + NUTS_RECV(resp, "abc"); // Respondent sends the reply to first survey. This will be // discarded by the SURV socket. - TEST_CHECK(nng_sendmsg(resp, cmd, 0) == 0); + NUTS_SEND(resp, "abc"); // Now get the next survey from the RESP; should be "def". - TEST_NNG_PASS(nng_recvmsg(resp, &cmd, 0)); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "def") == 0); + NUTS_RECV(resp, "def"); // And send it back to REQ. - TEST_NNG_PASS(nng_sendmsg(resp, cmd, 0)); + NUTS_SEND(resp, "def"); // Try a req command. This should give back "def" - TEST_NNG_PASS(nng_recvmsg(surv, &cmd, 0)); - TEST_CHECK(nng_msg_len(cmd) == 4); - TEST_CHECK(strcmp(nng_msg_body(cmd), "def") == 0); - nng_msg_free(cmd); + NUTS_RECV(surv, "def"); nng_aio_free(aio); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -340,25 +283,25 @@ test_surv_cancel_post_recv(void) nng_socket surv; nng_socket resp; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(surv, resp)); - - TEST_NNG_SEND_STR(surv, "ONE"); - TEST_NNG_RECV_STR(resp, "ONE"); - TEST_NNG_SEND_STR(resp, "one"); - testutil_sleep(100); // Make sure reply arrives! - TEST_NNG_SEND_STR(surv, "TWO"); - TEST_NNG_RECV_STR(resp, "TWO"); - TEST_NNG_SEND_STR(resp, "two"); - TEST_NNG_RECV_STR(surv, "two"); - - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_MARRY(surv, resp); + + NUTS_SEND(surv, "ONE"); + NUTS_RECV(resp, "ONE"); + NUTS_SEND(resp, "one"); + NUTS_SLEEP(100); // Make sure reply arrives! + NUTS_SEND(surv, "TWO"); + NUTS_RECV(resp, "TWO"); + NUTS_SEND(resp, "two"); + NUTS_RECV(surv, "two"); + + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -368,21 +311,21 @@ test_surv_poll_writeable(void) nng_socket surv; nng_socket resp; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_getopt_int(surv, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_socket_get_int(surv, NNG_OPT_SENDFD, &fd)); + NUTS_TRUE(fd >= 0); // Survey is broadcast, so we can always write. - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); // Now it's writable. - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd)); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } void @@ -393,39 +336,39 @@ test_surv_poll_readable(void) nng_socket resp; nng_msg * msg; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_getopt_int(surv, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_socket_get_int(surv, 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(surv, resp)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(surv, resp); + 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(surv, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(resp, &msg, 0)); // recv on rep - TEST_NNG_PASS(nng_sendmsg(resp, 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(surv, msg, 0)); + NUTS_PASS(nng_recvmsg(resp, &msg, 0)); // recv on rep + NUTS_PASS(nng_sendmsg(resp, 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(surv, &msg, 0)); + NUTS_PASS(nng_recvmsg(surv, &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(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -435,14 +378,12 @@ test_surv_ctx_no_poll(void) nng_socket surv; nng_ctx ctx; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_ctx_open(&ctx, surv)); - 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(surv)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_ctx_open(&ctx, surv)); + 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(surv); } static void @@ -454,25 +395,25 @@ test_surv_ctx_recv_nonblock(void) nng_aio * aio; nng_msg * msg; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_ctx_open(&ctx, surv)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_ctx_open(&ctx, surv)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); 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(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); nng_aio_free(aio); } @@ -484,17 +425,17 @@ test_surv_ctx_send_nonblock(void) nng_aio * aio; nng_msg * msg; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_ctx_open(&ctx, surv)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_ctx_open(&ctx, surv)); + 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_PASS(nng_aio_result(aio)); // We never block - TEST_NNG_PASS(nng_close(surv)); + NUTS_PASS(nng_aio_result(aio)); // We never block + NUTS_CLOSE(surv); nng_aio_free(aio); } @@ -504,16 +445,16 @@ test_surv_send_best_effort(void) nng_socket surv; nng_socket resp; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_MARRY(surv, resp); for (int i = 0; i < 200; i++) { - TEST_NNG_SEND_STR(surv, "junk"); + NUTS_SEND(surv, "junk"); } - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -524,23 +465,23 @@ test_surv_survey_timeout(void) char buf[16]; size_t sz; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SURVEYOR_SURVEYTIME, 50)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SURVEYOR_SURVEYTIME, 50)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); - TEST_NNG_SEND_STR(surv, "hello"); - TEST_NNG_RECV_STR(resp, "hello"); + NUTS_SEND(surv, "hello"); + NUTS_RECV(resp, "hello"); sz = sizeof(buf); - TEST_NNG_FAIL(nng_recv(surv, buf, &sz, 0), NNG_ETIMEDOUT); - TEST_NNG_SEND_STR(resp, "world"); - TEST_NNG_FAIL(nng_recv(surv, buf, &sz, 0), NNG_ESTATE); + NUTS_FAIL(nng_recv(surv, buf, &sz, 0), NNG_ETIMEDOUT); + NUTS_SEND(resp, "world"); + NUTS_FAIL(nng_recv(surv, buf, &sz, 0), NNG_ESTATE); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -552,23 +493,23 @@ test_surv_ctx_recv_close_socket(void) nng_aio * aio; nng_msg * m; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_ctx_open(&ctx, surv)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(testutil_marry(surv, resp)); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_ctx_open(&ctx, surv)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_MARRY(surv, resp); + 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(surv); - 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(resp)); + NUTS_CLOSE(resp); } static void @@ -581,52 +522,52 @@ test_surv_context_multi(void) nng_msg * m; int cnt = sizeof(c) / sizeof(c[0]); - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(testutil_marry(surv, resp)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SURVEYOR_SURVEYTIME, 200)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_MARRY(surv, resp); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SURVEYOR_SURVEYTIME, 200)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); for (int i = 0; i < cnt; i++) { - TEST_NNG_PASS(nng_ctx_open(&c[i], surv)); + NUTS_PASS(nng_ctx_open(&c[i], surv)); } for (int i = 0; i < cnt; i++) { - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_append_u32(m, i)); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_append_u32(m, i)); nng_aio_set_msg(aio, m); nng_ctx_send(c[i], aio); nng_aio_wait(aio); - TEST_NNG_PASS(nng_aio_result(aio)); + NUTS_PASS(nng_aio_result(aio)); } for (int i = 0; i < cnt; i++) { - TEST_NNG_PASS(nng_recvmsg(resp, &m, 0)); - TEST_NNG_PASS(nng_sendmsg(resp, m, 0)); + NUTS_PASS(nng_recvmsg(resp, &m, 0)); + NUTS_PASS(nng_sendmsg(resp, m, 0)); } for (int i = cnt - 1; i >= 0; i--) { uint32_t x; nng_ctx_recv(c[i], aio); nng_aio_wait(aio); - TEST_NNG_PASS(nng_aio_result(aio)); + NUTS_PASS(nng_aio_result(aio)); m = nng_aio_get_msg(aio); TEST_ASSERT(m != NULL); - TEST_NNG_PASS(nng_msg_trim_u32(m, &x)); - TEST_CHECK(x == (uint32_t)i); + NUTS_PASS(nng_msg_trim_u32(m, &x)); + NUTS_TRUE(x == (uint32_t) i); nng_msg_free(m); } for (int i = 0; i < cnt; i++) { nng_ctx_recv(c[i], aio); nng_aio_wait(aio); - TEST_CHECK(nng_aio_result(aio) != 0); + NUTS_TRUE(nng_aio_result(aio) != 0); } for (int i = 0; i < cnt; i++) { nng_ctx_close(c[i]); } - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); nng_aio_free(aio); } @@ -636,28 +577,27 @@ test_surv_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_surveyor0_open(&s1)); - TEST_NNG_PASS(nng_surveyor0_open(&s2)); + NUTS_ADDR(addr, "inproc"); + NUTS_PASS(nng_surveyor0_open(&s1)); + NUTS_PASS(nng_surveyor0_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); } diff --git a/src/protocol/survey0/xrespond_test.c b/src/protocol/survey0/xrespond_test.c index 342c8a94..ec5e99a3 100644 --- a/src/protocol/survey0/xrespond_test.c +++ b/src/protocol/survey0/xrespond_test.c @@ -7,14 +7,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/survey0/respond.h> -#include <nng/protocol/survey0/survey.h> - -#include <acutest.h> -#include <testutil.h> +#include <nuts.h> static void test_xresp_identity(void) @@ -24,16 +17,16 @@ test_xresp_identity(void) char * n1; char * n2; - TEST_NNG_PASS(nng_respondent0_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_RESPONDENT0_SELF); - TEST_CHECK(p2 == NNG_RESPONDENT0_PEER); - TEST_CHECK(strcmp(n1, NNG_RESPONDENT0_SELF_NAME) == 0); - TEST_CHECK(strcmp(n2, NNG_RESPONDENT0_PEER_NAME) == 0); + NUTS_PASS(nng_respondent0_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_RESPONDENT0_SELF); + NUTS_TRUE(p2 == NNG_RESPONDENT0_PEER); + NUTS_MATCH(n1, NNG_RESPONDENT0_SELF_NAME); + NUTS_MATCH(n2, NNG_RESPONDENT0_PEER_NAME); nng_strfree(n1); nng_strfree(n2); } @@ -44,10 +37,10 @@ test_xresp_raw(void) nng_socket s; bool b; - TEST_NNG_PASS(nng_respondent0_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_respondent0_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_xresp_no_context(void) nng_socket s; nng_ctx ctx; - TEST_NNG_PASS(nng_respondent0_open_raw(&s)); - TEST_NNG_FAIL(nng_ctx_open(&ctx, s), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(s)); + NUTS_PASS(nng_respondent0_open_raw(&s)); + NUTS_FAIL(nng_ctx_open(&ctx, s), NNG_ENOTSUP); + NUTS_CLOSE(s); } static void @@ -68,24 +61,24 @@ test_xresp_poll_writeable(void) nng_socket surv; nng_socket resp; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_getopt_int(resp, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_socket_get_int(resp, 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(surv, resp)); + NUTS_MARRY(surv, resp); // Now it's writable. - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) == true); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -96,32 +89,32 @@ test_xresp_poll_readable(void) nng_socket resp; nng_msg * msg; - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_getopt_int(resp, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_socket_get_int(resp, 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(surv, resp)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(surv, resp); + 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(surv, "abc"); - testutil_sleep(100); + NUTS_SEND(surv, "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(resp, &msg, 0)); + NUTS_PASS(nng_recvmsg(resp, &msg, 0)); nng_msg_free(msg); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -130,28 +123,28 @@ test_xresp_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_respondent0_open_raw(&s1)); - TEST_NNG_PASS(nng_respondent0_open(&s2)); + NUTS_PASS(nng_respondent0_open_raw(&s1)); + NUTS_PASS(nng_respondent0_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); } @@ -164,26 +157,26 @@ test_xresp_close_pipe_before_send(void) nng_aio * aio1; nng_msg * m; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_aio_alloc(&aio1, NULL, NULL)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open(&surv)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_aio_alloc(&aio1, NULL, NULL)); - TEST_NNG_PASS(testutil_marry(surv, resp)); - TEST_NNG_SEND_STR(surv, "test"); + NUTS_MARRY(surv, resp); + NUTS_SEND(surv, "test"); nng_recv_aio(resp, 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(resp, m, 0)); + NUTS_PASS(nng_pipe_close(p)); + NUTS_PASS(nng_sendmsg(resp, m, 0)); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); nng_aio_free(aio1); } @@ -195,37 +188,37 @@ test_xresp_close_pipe_during_send(void) nng_pipe p; nng_msg * m; - TEST_NNG_PASS(nng_respondent_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_RECVBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_RECVBUF, 1)); - - TEST_NNG_PASS(testutil_marry(surv, resp)); - - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_append_u32(m, (unsigned) 0x81000000u)); - TEST_NNG_PASS(nng_sendmsg(surv, m, 0)); - TEST_NNG_PASS(nng_recvmsg(resp, &m, 0)); + NUTS_PASS(nng_respondent_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_int(resp, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_socket_set_int(resp, NNG_OPT_RECVBUF, 20)); + NUTS_PASS(nng_socket_set_int(surv, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_socket_set_int(surv, NNG_OPT_RECVBUF, 1)); + + NUTS_MARRY(surv, resp); + + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, (unsigned) 0x81000000u)); + NUTS_PASS(nng_sendmsg(surv, m, 0)); + NUTS_PASS(nng_recvmsg(resp, &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)); // protocol does not exert back-pressure - TEST_NNG_PASS(nng_sendmsg(resp, m, 0)); + NUTS_PASS(nng_sendmsg(resp, m, 0)); } - TEST_NNG_PASS(nng_pipe_close(p)); + NUTS_PASS(nng_pipe_close(p)); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -235,27 +228,27 @@ test_xresp_close_during_recv(void) nng_socket surv; nng_msg * m; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_RECVBUF, 5)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_int(resp, NNG_OPT_RECVBUF, 5)); + NUTS_PASS(nng_socket_set_int(surv, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); 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(surv, m, 0); if (rv == NNG_ETIMEDOUT) { nng_msg_free(m); break; } } - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -264,14 +257,14 @@ test_xresp_recv_aio_stopped(void) nng_socket resp; nng_aio * aio; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_stop(aio); nng_recv_aio(resp, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECANCELED); - TEST_NNG_PASS(nng_close(resp)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_CLOSE(resp); nng_aio_free(aio); } @@ -282,21 +275,21 @@ test_xresp_send_no_header(void) nng_socket surv; nng_msg * m; - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_sendmsg(resp, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_sendmsg(resp, m, 0)); + NUTS_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -306,21 +299,21 @@ test_xresp_recv_garbage(void) nng_socket surv; nng_msg * m; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); - TEST_NNG_PASS(nng_msg_alloc(&m, 4)); - TEST_NNG_PASS(nng_msg_append_u32(m, 1u)); - TEST_NNG_PASS(nng_sendmsg(surv, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 4)); + NUTS_PASS(nng_msg_append_u32(m, 1u)); + NUTS_PASS(nng_sendmsg(surv, m, 0)); + NUTS_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -332,29 +325,29 @@ test_xresp_ttl_option(void) size_t sz; const char *opt = NNG_OPT_MAXTTL; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_setopt_int(resp, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(resp, opt, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(resp, opt, -1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(resp, opt, 16), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(resp, opt, 256), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(resp, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(resp, opt, &v)); - TEST_CHECK(v == 3); + NUTS_PASS(nng_socket_set_int(resp, opt, 1)); + NUTS_FAIL(nng_socket_set_int(resp, opt, 0), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(resp, opt, -1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(resp, opt, 16), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(resp, opt, 256), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(resp, opt, 3)); + NUTS_PASS(nng_socket_get_int(resp, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(resp, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(resp, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_CHECK(nng_setopt(resp, opt, "", 1) == NNG_EINVAL); + NUTS_FAIL(nng_socket_set(resp, opt, "", 1), NNG_EINVAL); sz = 1; - TEST_CHECK(nng_getopt(resp, opt, &v, &sz) == NNG_EINVAL); - TEST_CHECK(nng_setopt_bool(resp, opt, true) == NNG_EBADTYPE); - TEST_CHECK(nng_getopt_bool(resp, opt, &b) == NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get(resp, opt, &v, &sz), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_bool(resp, opt, true), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(resp, opt, &b), NNG_EBADTYPE); - TEST_CHECK(nng_close(resp) == 0); + NUTS_CLOSE(resp); } static void @@ -364,65 +357,65 @@ test_xresp_ttl_drop(void) nng_socket surv; nng_msg * m; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_MAXTTL, 3)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_socket_set_int(resp, NNG_OPT_MAXTTL, 3)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 200)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); // Send messages. Note that xresp 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(surv, 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(surv, 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(surv, 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(surv, 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(surv, 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(surv, 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(surv, 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(surv, m, 0)); // So on receive we should see 80000001 and 80000003. - TEST_NNG_PASS(nng_recvmsg(resp, &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(resp, &m, 0)); + NUTS_TRUE(nng_msg_header_len(m) == 12); + NUTS_TRUE(nng_msg_len(m) == 6); + NUTS_MATCH(nng_msg_body(m), "PASS1"); nng_msg_free(m); - TEST_NNG_PASS(nng_recvmsg(resp, &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(resp, &m, 0)); + NUTS_TRUE(nng_msg_header_len(m) == 16); // 3 hops + ID + NUTS_TRUE(nng_msg_len(m) == 6); + NUTS_MATCH(nng_msg_body(m), "PASS3"); nng_msg_free(m); - TEST_NNG_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); + NUTS_FAIL(nng_recvmsg(resp, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } -TEST_LIST = { +NUTS_TESTS = { { "xrespond identity", test_xresp_identity }, { "xrespond raw", test_xresp_raw }, { "xrespond no context", test_xresp_no_context }, diff --git a/src/protocol/survey0/xsurvey_test.c b/src/protocol/survey0/xsurvey_test.c index ca7a2dd6..f8e9d401 100644 --- a/src/protocol/survey0/xsurvey_test.c +++ b/src/protocol/survey0/xsurvey_test.c @@ -7,18 +7,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/survey0/respond.h> -#include <nng/protocol/survey0/survey.h> - -#include <acutest.h> -#include <testutil.h> - -#ifndef NNI_PROTO -#define NNI_PROTO(x, y) (((x) << 4u) | (y)) -#endif +#include <nuts.h> static void test_xsurveyor_identity(void) @@ -27,18 +16,18 @@ test_xsurveyor_identity(void) int p; char * n; - TEST_CHECK(nng_surveyor0_open_raw(&s) == 0); - TEST_CHECK(nng_getopt_int(s, NNG_OPT_PROTO, &p) == 0); - TEST_CHECK(p == NNI_PROTO(6u, 2u)); // 0x62 - TEST_CHECK(nng_getopt_int(s, NNG_OPT_PEER, &p) == 0); - TEST_CHECK(p == NNI_PROTO(6u, 3u)); // 0x63 - TEST_CHECK(nng_getopt_string(s, NNG_OPT_PROTONAME, &n) == 0); - TEST_CHECK(strcmp(n, "surveyor") == 0); + NUTS_PASS(nng_surveyor0_open_raw(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p)); + NUTS_TRUE(p == NNG_SURVEYOR0_SELF); // 0x62 + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p)); + NUTS_TRUE(p == NNG_SURVEYOR0_PEER); // 0x62 + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n)); + NUTS_MATCH(n, NNG_SURVEYOR0_SELF_NAME); nng_strfree(n); - TEST_CHECK(nng_getopt_string(s, NNG_OPT_PEERNAME, &n) == 0); - TEST_CHECK(strcmp(n, "respondent") == 0); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n)); + NUTS_MATCH(n, NNG_SURVEYOR0_PEER_NAME); nng_strfree(n); - TEST_CHECK(nng_close(s) == 0); + NUTS_CLOSE(s); } static void @@ -47,10 +36,10 @@ test_xsurveyor_raw(void) nng_socket s; bool b; - TEST_NNG_PASS(nng_surveyor0_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_surveyor0_open_raw(&s)); + NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_RAW, &b)); + NUTS_TRUE(b); + NUTS_CLOSE(s); } static void @@ -59,9 +48,9 @@ test_xsurvey_no_context(void) nng_socket s; nng_ctx ctx; - TEST_NNG_PASS(nng_surveyor0_open_raw(&s)); - TEST_NNG_FAIL(nng_ctx_open(&ctx, s), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(s)); + NUTS_PASS(nng_surveyor0_open_raw(&s)); + NUTS_FAIL(nng_ctx_open(&ctx, s), NNG_ENOTSUP); + NUTS_CLOSE(s); } static void @@ -71,21 +60,21 @@ test_xsurvey_poll_writeable(void) nng_socket surv; nng_socket resp; - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_getopt_int(surv, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_socket_get_int(surv, NNG_OPT_SENDFD, &fd)); + NUTS_TRUE(fd >= 0); // Survey is broadcast, so we can always write. - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); // Now it's writable. - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd)); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -96,44 +85,44 @@ test_xsurvey_poll_readable(void) nng_socket resp; nng_msg * msg; - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_respondent0_open(&resp)); - TEST_NNG_PASS(nng_getopt_int(surv, NNG_OPT_RECVFD, &fd)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_respondent0_open(&resp)); + NUTS_PASS(nng_socket_get_int(surv, NNG_OPT_RECVFD, &fd)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, 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(surv, resp)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(surv, resp); + 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(surv, msg, 0)); + NUTS_PASS(nng_msg_append_u32(msg, 0x80000000)); + NUTS_PASS(nng_sendmsg(surv, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(resp, &msg, 0)); - TEST_NNG_PASS(nng_sendmsg(resp, msg, 0)); + NUTS_PASS(nng_recvmsg(resp, &msg, 0)); + NUTS_PASS(nng_sendmsg(resp, msg, 0)); - testutil_sleep(100); + NUTS_SLEEP(100); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd) ); // and receiving makes it no longer ready - TEST_NNG_PASS(nng_recvmsg(surv, &msg, 0)); + NUTS_PASS(nng_recvmsg(surv, &msg, 0)); nng_msg_free(msg); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -142,28 +131,28 @@ test_xsurvey_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_surveyor0_open_raw(&s1)); - TEST_NNG_PASS(nng_surveyor0_open(&s2)); + NUTS_PASS(nng_surveyor0_open_raw(&s1)); + NUTS_PASS(nng_surveyor0_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); } @@ -173,14 +162,14 @@ test_xsurvey_recv_aio_stopped(void) nng_socket surv; nng_aio * aio; - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_stop(aio); nng_recv_aio(surv, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECANCELED); - TEST_NNG_PASS(nng_close(surv)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_CLOSE(surv); nng_aio_free(aio); } @@ -192,33 +181,33 @@ test_xsurvey_recv_garbage(void) nng_msg * m; uint32_t req_id; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(surv, resp)); + NUTS_MARRY(surv, resp); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_append_u32(m, 0x80000000)); - TEST_NNG_PASS(nng_sendmsg(surv, m, 0)); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_append_u32(m, 0x80000000)); + NUTS_PASS(nng_sendmsg(surv, m, 0)); - TEST_NNG_PASS(nng_recvmsg(resp, &m, 0)); + NUTS_PASS(nng_recvmsg(resp, &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(resp, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(surv, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_sendmsg(resp, m, 0)); + NUTS_FAIL(nng_recvmsg(surv, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -230,37 +219,37 @@ test_xsurvey_recv_header(void) nng_pipe p; uint32_t id; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry_ex(surv, resp, NULL, NULL, &p)); + NUTS_MARRY_EX(surv, resp, NULL, NULL, &p); // Simulate a few hops. - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_header_append_u32(m, nng_pipe_id(p))); - 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(resp, m, 0)); - - TEST_NNG_PASS(nng_recvmsg(surv, &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(p))); + 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(resp, m, 0)); + + NUTS_PASS(nng_recvmsg(surv, &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(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -272,26 +261,26 @@ test_xsurvey_close_during_recv(void) nng_pipe p1; nng_pipe p2; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_int(surv, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_socket_set_int(resp, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(testutil_marry_ex(surv, resp, NULL, &p1, &p2)); - TEST_CHECK(nng_pipe_id(p1) > 0); - TEST_CHECK(nng_pipe_id(p2) > 0); + NUTS_MARRY_EX(surv, resp, 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(resp, 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(resp, m, 0)); } - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -303,27 +292,27 @@ test_xsurvey_close_pipe_during_send(void) nng_pipe p1; nng_pipe p2; - TEST_NNG_PASS(nng_respondent0_open_raw(&resp)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_int(resp, NNG_OPT_RECVBUF, 5)); - TEST_NNG_PASS(nng_setopt_int(surv, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_respondent0_open_raw(&resp)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_socket_set_ms(resp, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_int(resp, NNG_OPT_RECVBUF, 5)); + NUTS_PASS(nng_socket_set_int(surv, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(testutil_marry_ex(surv, resp, NULL, &p1, &p2)); - TEST_CHECK(nng_pipe_id(p1) > 0); - TEST_CHECK(nng_pipe_id(p2) > 0); + NUTS_MARRY_EX(surv, resp, 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(surv, 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(surv, m, 0)); } - TEST_NNG_PASS(nng_pipe_close(p1)); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp)); + NUTS_PASS(nng_pipe_close(p1)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp); } static void @@ -335,29 +324,29 @@ test_xsurvey_ttl_option(void) size_t sz; const char *opt = NNG_OPT_MAXTTL; - TEST_NNG_PASS(nng_surveyor0_open_raw(&s)); + NUTS_PASS(nng_surveyor0_open_raw(&s)); - TEST_NNG_PASS(nng_setopt_int(s, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(s, opt, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(s, opt, -1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(s, opt, 16), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(s, opt, 256), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(s, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(s, opt, &v)); - TEST_CHECK(v == 3); + NUTS_PASS(nng_socket_set_int(s, opt, 1)); + NUTS_FAIL(nng_socket_set_int(s, opt, 0), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(s, opt, -1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(s, opt, 16), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(s, opt, 256), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(s, opt, 3)); + NUTS_PASS(nng_socket_get_int(s, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(s, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(s, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_CHECK(nng_setopt(s, opt, "", 1) == NNG_EINVAL); + NUTS_FAIL(nng_socket_set(s, opt, "", 1) , NNG_EINVAL); sz = 1; - TEST_CHECK(nng_getopt(s, opt, &v, &sz) == NNG_EINVAL); - TEST_CHECK(nng_setopt_bool(s, opt, true) == NNG_EBADTYPE); - TEST_CHECK(nng_getopt_bool(s, opt, &b) == NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get(s, opt, &v, &sz) , NNG_EINVAL); + NUTS_FAIL(nng_socket_set_bool(s, opt, true) , NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(s, opt, &b) , NNG_EBADTYPE); - TEST_CHECK(nng_close(s) == 0); + NUTS_CLOSE(s); } static void @@ -368,27 +357,27 @@ test_xsurvey_broadcast(void) nng_socket surv; nng_msg * m; - TEST_NNG_PASS(nng_respondent0_open(&resp1)); - TEST_NNG_PASS(nng_respondent0_open(&resp2)); - TEST_NNG_PASS(nng_surveyor0_open_raw(&surv)); - TEST_NNG_PASS(nng_setopt_ms(resp1, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(resp2, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(surv, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_respondent0_open(&resp1)); + NUTS_PASS(nng_respondent0_open(&resp2)); + NUTS_PASS(nng_surveyor0_open_raw(&surv)); + NUTS_PASS(nng_socket_set_ms(resp1, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(resp2, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(surv, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(testutil_marry(surv, resp1)); - TEST_NNG_PASS(testutil_marry(surv, resp2)); + NUTS_MARRY(surv, resp1); + NUTS_MARRY(surv, resp2); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_header_append_u32(m, 0x80000002u)); - TEST_NNG_PASS(nng_msg_append(m, "hello", 6)); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_header_append_u32(m, 0x80000002u)); + NUTS_PASS(nng_msg_append(m, "hello", 6)); - TEST_NNG_PASS(nng_sendmsg(surv, m, 0)); - TEST_NNG_RECV_STR(resp1, "hello"); - TEST_NNG_RECV_STR(resp2, "hello"); + NUTS_PASS(nng_sendmsg(surv, m, 0)); + NUTS_RECV(resp1, "hello"); + NUTS_RECV(resp2, "hello"); - TEST_NNG_PASS(nng_close(surv)); - TEST_NNG_PASS(nng_close(resp1)); - TEST_NNG_PASS(nng_close(resp2)); + NUTS_CLOSE(surv); + NUTS_CLOSE(resp1); + NUTS_CLOSE(resp2); } TEST_LIST = { |