diff options
Diffstat (limited to 'src/protocol')
| -rw-r--r-- | src/protocol/bus0/bug1247_test.c | 22 | ||||
| -rw-r--r-- | src/protocol/pair1/pair1_poly_test.c | 367 | ||||
| -rw-r--r-- | src/protocol/pair1/pair1_test.c | 447 | ||||
| -rw-r--r-- | src/protocol/pubsub0/pub_test.c | 260 | ||||
| -rw-r--r-- | src/protocol/pubsub0/sub_test.c | 515 | ||||
| -rw-r--r-- | src/protocol/pubsub0/xsub_test.c | 267 | ||||
| -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 | ||||
| -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 |
14 files changed, 2759 insertions, 2954 deletions
diff --git a/src/protocol/bus0/bug1247_test.c b/src/protocol/bus0/bug1247_test.c index 6f418f53..bbc6958b 100644 --- a/src/protocol/bus0/bug1247_test.c +++ b/src/protocol/bus0/bug1247_test.c @@ -7,30 +7,26 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> +#include <nuts.h> -#include <nng/nng.h> #include <nng/protocol/bus0/bus.h> -#include "acutest.h" -#include "testutil.h" - void test_bug1247(void) { nng_socket bus1, bus2; - char addr[64]; + char * addr; - testutil_scratch_addr("tcp", sizeof(addr), addr); + NUTS_ADDR(addr, "tcp"); - TEST_NNG_PASS(nng_bus0_open(&bus1)); - TEST_NNG_PASS(nng_bus0_open(&bus2)); + NUTS_PASS(nng_bus0_open(&bus1)); + NUTS_PASS(nng_bus0_open(&bus2)); - TEST_NNG_PASS(nng_listen(bus1, addr, NULL, 0)); - TEST_NNG_FAIL(nng_listen(bus2, addr, NULL, 0), NNG_EADDRINUSE); + NUTS_PASS(nng_listen(bus1, addr, NULL, 0)); + NUTS_FAIL(nng_listen(bus2, addr, NULL, 0), NNG_EADDRINUSE); - TEST_NNG_PASS(nng_close(bus2)); - TEST_NNG_PASS(nng_close(bus1)); + NUTS_PASS(nng_close(bus2)); + NUTS_PASS(nng_close(bus1)); } TEST_LIST = { diff --git a/src/protocol/pair1/pair1_poly_test.c b/src/protocol/pair1/pair1_poly_test.c index c4acbd86..f26f7809 100644 --- a/src/protocol/pair1/pair1_poly_test.c +++ b/src/protocol/pair1/pair1_poly_test.c @@ -8,22 +8,35 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/pair0/pair.h> -#include <nng/protocol/pair1/pair.h> - -#include <testutil.h> - -#include <acutest.h> +#include <nuts.h> #define SECOND 1000 -#define APPEND_STR(m, s) TEST_NNG_PASS(nng_msg_append(m, s, strlen(s))) -#define CHECK_STR(m, s) \ - TEST_CHECK(nng_msg_len(m) == strlen(s)); \ - TEST_CHECK(memcmp(nng_msg_body(m), s, strlen(s)) == 0) +#define APPEND_STR(m, s) NUTS_PASS(nng_msg_append(m, s, strlen(s))) +#define CHECK_STR(m, s) \ + NUTS_TRUE(nng_msg_len(m) == strlen(s)); \ + NUTS_TRUE(memcmp(nng_msg_body(m), s, strlen(s)) == 0) + +static void +test_poly_identity(void) +{ + nng_socket s; + int p; + char * n; + + NUTS_PASS(nng_pair1_open_poly(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p)); + NUTS_TRUE(p == NUTS_PROTO(1u, 1u)); // 32 + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p)); + NUTS_TRUE(p == NUTS_PROTO(1u, 1u)); // 33 + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n)); + NUTS_MATCH(n, "pair1"); + nng_strfree(n); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n)); + NUTS_MATCH(n, "pair1"); + nng_strfree(n); + NUTS_CLOSE(s); +} void test_poly_best_effort(void) @@ -32,23 +45,23 @@ test_poly_best_effort(void) nng_socket c1; nng_msg * msg; - TEST_NNG_PASS(nng_pair1_open_poly(&s1)); - TEST_NNG_PASS(nng_pair1_open(&c1)); + NUTS_PASS(nng_pair1_open_poly(&s1)); + NUTS_PASS(nng_pair1_open(&c1)); - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(c1, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_setopt_int(c1, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(testutil_marry(s1, c1)); + NUTS_MARRY(s1, c1); for (int i = 0; i < 10; i++) { - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_sendmsg(s1, msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_sendmsg(s1, msg, 0)); } - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(c1)); + NUTS_CLOSE(s1); + NUTS_CLOSE(c1); } void @@ -62,69 +75,69 @@ test_poly_cooked(void) nng_pipe p1; nng_pipe p2; - TEST_NNG_PASS(nng_pair1_open_poly(&s1)); - TEST_NNG_PASS(nng_pair1_open(&c1)); - TEST_NNG_PASS(nng_pair1_open(&c2)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(c1, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(c2, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 10)); - TEST_NNG_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND / 10)); - TEST_NNG_PASS(nng_setopt_ms(c2, NNG_OPT_RECVTIMEO, SECOND / 10)); + NUTS_PASS(nng_pair1_open_poly(&s1)); + NUTS_PASS(nng_pair1_open(&c1)); + NUTS_PASS(nng_pair1_open(&c2)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_setopt_ms(c1, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_setopt_ms(c2, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 10)); + NUTS_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND / 10)); + NUTS_PASS(nng_setopt_ms(c2, NNG_OPT_RECVTIMEO, SECOND / 10)); - TEST_NNG_PASS(nng_getopt_bool(s1, NNG_OPT_PAIR1_POLY, &v)); - TEST_CHECK(v == true); + NUTS_PASS(nng_getopt_bool(s1, NNG_OPT_PAIR1_POLY, &v)); + NUTS_TRUE(v); - TEST_NNG_PASS(testutil_marry(s1, c1)); - TEST_NNG_PASS(testutil_marry(s1, c2)); + NUTS_MARRY(s1, c1); + NUTS_MARRY(s1, c2); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); APPEND_STR(msg, "ONE"); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); CHECK_STR(msg, "ONE"); p1 = nng_msg_get_pipe(msg); - TEST_CHECK(nng_pipe_id(p1) > 0); + NUTS_TRUE(nng_pipe_id(p1) > 0); nng_msg_free(msg); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); APPEND_STR(msg, "TWO"); - TEST_NNG_PASS(nng_sendmsg(c2, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_sendmsg(c2, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); CHECK_STR(msg, "TWO"); p2 = nng_msg_get_pipe(msg); - TEST_CHECK(nng_pipe_id(p2) > 0); + NUTS_TRUE(nng_pipe_id(p2) > 0); nng_msg_free(msg); - TEST_CHECK(nng_pipe_id(p1) != nng_pipe_id(p2)); + NUTS_TRUE(nng_pipe_id(p1) != nng_pipe_id(p2)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); nng_msg_set_pipe(msg, p1); APPEND_STR(msg, "UNO"); - TEST_NNG_PASS(nng_sendmsg(s1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(c1, &msg, 0)); + NUTS_PASS(nng_sendmsg(s1, msg, 0)); + NUTS_PASS(nng_recvmsg(c1, &msg, 0)); CHECK_STR(msg, "UNO"); nng_msg_free(msg); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); nng_msg_set_pipe(msg, p2); APPEND_STR(msg, "DOS"); - TEST_NNG_PASS(nng_sendmsg(s1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(c2, &msg, 0)); + NUTS_PASS(nng_sendmsg(s1, msg, 0)); + NUTS_PASS(nng_recvmsg(c2, &msg, 0)); CHECK_STR(msg, "DOS"); nng_msg_free(msg); - TEST_NNG_PASS(nng_close(c1)); + NUTS_PASS(nng_close(c1)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); nng_msg_set_pipe(msg, p1); APPEND_STR(msg, "EIN"); - TEST_NNG_PASS(nng_sendmsg(s1, msg, 0)); - TEST_NNG_FAIL(nng_recvmsg(c2, &msg, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_sendmsg(s1, msg, 0)); + NUTS_FAIL(nng_recvmsg(c2, &msg, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(c2)); + NUTS_CLOSE(s1); + NUTS_CLOSE(c2); } void @@ -135,38 +148,38 @@ test_poly_default(void) nng_socket c2; nng_msg * msg; - TEST_NNG_PASS(nng_pair1_open_poly(&s1)); - TEST_NNG_PASS(nng_pair1_open(&c1)); - TEST_NNG_PASS(nng_pair1_open(&c2)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(c1, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(c2, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_pair1_open_poly(&s1)); + NUTS_PASS(nng_pair1_open(&c1)); + NUTS_PASS(nng_pair1_open(&c2)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_setopt_ms(c1, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_setopt_ms(c2, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(testutil_marry(s1, c1)); - TEST_NNG_PASS(testutil_marry(s1, c2)); + NUTS_MARRY(s1, c1); + NUTS_MARRY(s1, c2); // This assumes poly picks the first suitor. Applications // should not make the same assumption. - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); APPEND_STR(msg, "YES"); - TEST_NNG_PASS(nng_sendmsg(s1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(c1, &msg, 0)); + NUTS_PASS(nng_sendmsg(s1, msg, 0)); + NUTS_PASS(nng_recvmsg(c1, &msg, 0)); CHECK_STR(msg, "YES"); nng_msg_free(msg); - TEST_NNG_PASS(nng_close(c1)); - testutil_sleep(10); + NUTS_CLOSE(c1); + NUTS_SLEEP(10); // Verify that the other pipe is chosen as the next suitor. - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); APPEND_STR(msg, "AGAIN"); - TEST_NNG_PASS(nng_sendmsg(s1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(c2, &msg, 0)); + NUTS_PASS(nng_sendmsg(s1, msg, 0)); + NUTS_PASS(nng_recvmsg(c2, &msg, 0)); CHECK_STR(msg, "AGAIN"); nng_msg_free(msg); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(c2)); + NUTS_CLOSE(s1); + NUTS_CLOSE(c2); } void @@ -175,25 +188,24 @@ test_poly_close_abort(void) nng_socket s; nng_socket c; - TEST_NNG_PASS(nng_pair1_open_poly(&s)); - TEST_NNG_PASS(nng_pair1_open(&c)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_int(s, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(c, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_pair1_open_poly(&s)); + NUTS_PASS(nng_pair1_open(&c)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_setopt_int(s, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_setopt_int(c, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(testutil_marry(c, s)); + NUTS_MARRY(c, s); for (int i = 0; i < 20; i++) { - TEST_NNG_SEND_STR(c, "TEST"); + NUTS_SEND(c, "TEST"); } - testutil_sleep(50); + NUTS_SLEEP(50); - TEST_NNG_PASS(nng_close(s)); - TEST_NNG_PASS(nng_close(c)); + NUTS_CLOSE(s); + NUTS_CLOSE(c); } - void test_poly_recv_no_header(void) { @@ -201,20 +213,20 @@ test_poly_recv_no_header(void) nng_socket c; nng_msg * m; - TEST_NNG_PASS(nng_pair1_open_poly(&s)); - TEST_NNG_PASS(nng_pair1_open(&c)); - TEST_NNG_PASS(nng_setopt_bool(c, "pair1_test_inject_header", true)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_pair1_open_poly(&s)); + NUTS_PASS(nng_pair1_open(&c)); + NUTS_PASS(nng_setopt_bool(c, "pair1_test_inject_header", true)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(testutil_marry(c, s)); + NUTS_MARRY(c, s); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_sendmsg(c, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(s, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_sendmsg(c, m, 0)); + NUTS_FAIL(nng_recvmsg(s, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(c)); - TEST_NNG_PASS(nng_close(s)); + NUTS_CLOSE(c); + NUTS_CLOSE(s); } void @@ -224,22 +236,22 @@ test_poly_recv_garbage(void) nng_socket c; nng_msg * m; - TEST_NNG_PASS(nng_pair1_open_poly(&s)); - TEST_NNG_PASS(nng_pair1_open(&c)); - TEST_NNG_PASS(nng_setopt_bool(c, "pair1_test_inject_header", true)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_pair1_open_poly(&s)); + NUTS_PASS(nng_pair1_open(&c)); + NUTS_PASS(nng_setopt_bool(c, "pair1_test_inject_header", true)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(testutil_marry(c, s)); + NUTS_MARRY(c, s); // ridiculous hop count - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_append_u32(m, 0x1000)); - TEST_NNG_PASS(nng_sendmsg(c, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(s, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_append_u32(m, 0x1000)); + NUTS_PASS(nng_sendmsg(c, m, 0)); + NUTS_FAIL(nng_recvmsg(s, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(c)); - TEST_NNG_PASS(nng_close(s)); + NUTS_CLOSE(c); + NUTS_CLOSE(s); } void @@ -251,65 +263,65 @@ test_poly_ttl(void) uint32_t val; int ttl; - TEST_NNG_PASS(nng_pair1_open_poly(&s1)); - TEST_NNG_PASS(nng_pair1_open_raw(&c1)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 5)); - TEST_NNG_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND / 5)); + NUTS_PASS(nng_pair1_open_poly(&s1)); + NUTS_PASS(nng_pair1_open_raw(&c1)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 5)); + NUTS_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND / 5)); // cannot set insane TTLs - TEST_NNG_FAIL(nng_setopt_int(s1, NNG_OPT_MAXTTL, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(s1, NNG_OPT_MAXTTL, 1000), NNG_EINVAL); + NUTS_FAIL(nng_setopt_int(s1, NNG_OPT_MAXTTL, 0), NNG_EINVAL); + NUTS_FAIL(nng_setopt_int(s1, NNG_OPT_MAXTTL, 1000), NNG_EINVAL); ttl = 8; - TEST_NNG_FAIL(nng_setopt(s1, NNG_OPT_MAXTTL, &ttl, 1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(s1, NNG_OPT_MAXTTL, true), NNG_EBADTYPE); + NUTS_FAIL(nng_setopt(s1, NNG_OPT_MAXTTL, &ttl, 1), NNG_EINVAL); + NUTS_FAIL(nng_setopt_bool(s1, NNG_OPT_MAXTTL, true), NNG_EBADTYPE); - TEST_NNG_PASS(testutil_marry(s1, c1)); + NUTS_MARRY(s1, c1); // Let's check enforcement of TTL - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 4)); - TEST_NNG_PASS(nng_getopt_int(s1, NNG_OPT_MAXTTL, &ttl)); - TEST_CHECK(ttl == 4); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 4)); + NUTS_PASS(nng_getopt_int(s1, NNG_OPT_MAXTTL, &ttl)); + NUTS_TRUE(ttl == 4); // Bad TTL bounces - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 4)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_header_append_u32(msg, 4)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); // Good TTL passes - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append_u32(msg, 0xFEEDFACE)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 3)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); - TEST_NNG_PASS(nng_msg_trim_u32(msg, &val)); - TEST_CHECK(val == 0xFEEDFACE); - TEST_NNG_PASS(nng_msg_header_trim_u32(msg, &val)); - TEST_CHECK(val == 4); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_append_u32(msg, 0xFEEDFACE)); + NUTS_PASS(nng_msg_header_append_u32(msg, 3)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_msg_trim_u32(msg, &val)); + NUTS_TRUE(val == 0xFEEDFACE); + NUTS_PASS(nng_msg_header_trim_u32(msg, &val)); + NUTS_TRUE(val == 4); nng_msg_free(msg); // Large TTL passes - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 15)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append_u32(msg, 1234)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 14)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); - TEST_NNG_PASS(nng_msg_trim_u32(msg, &val)); - TEST_CHECK(val == 1234); - TEST_NNG_PASS(nng_msg_header_trim_u32(msg, &val)); - TEST_CHECK(val == 15); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 15)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_append_u32(msg, 1234)); + NUTS_PASS(nng_msg_header_append_u32(msg, 14)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_msg_trim_u32(msg, &val)); + NUTS_TRUE(val == 1234); + NUTS_PASS(nng_msg_header_trim_u32(msg, &val)); + NUTS_TRUE(val == 15); nng_msg_free(msg); // Max TTL fails - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 15)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 15)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); - - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(c1)); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 15)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_header_append_u32(msg, 15)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); + + NUTS_CLOSE(s1); + NUTS_CLOSE(c1); } void @@ -318,40 +330,41 @@ test_poly_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_pair1_open_poly(&s1)); - TEST_NNG_PASS(nng_pair0_open(&s2)); + NUTS_PASS(nng_pair1_open_poly(&s1)); + NUTS_PASS(nng_pair0_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 = { - { "pair1 polyamorous best effort", test_poly_best_effort }, - { "pair1 polyamorous cooked", test_poly_cooked }, - { "pair1 polyamorous default", test_poly_default }, - { "pair1 polyamorous recv no header", test_poly_recv_no_header }, - { "pair1 polyamorous recv garbage", test_poly_recv_garbage }, - { "pair1 polyamorous ttl", test_poly_ttl }, - { "pair1 polyamorous close abort", test_poly_close_abort }, - { "pair1 polyamorous validate peer", test_poly_validate_peer }, + { "pair1 poly identity", test_poly_identity }, + { "pair1 poly best effort", test_poly_best_effort }, + { "pair1 poly cooked", test_poly_cooked }, + { "pair1 poly default", test_poly_default }, + { "pair1 poly recv no header", test_poly_recv_no_header }, + { "pair1 poly recv garbage", test_poly_recv_garbage }, + { "pair1 poly ttl", test_poly_ttl }, + { "pair1 poly close abort", test_poly_close_abort }, + { "pair1 poly validate peer", test_poly_validate_peer }, { NULL, NULL }, }; diff --git a/src/protocol/pair1/pair1_test.c b/src/protocol/pair1/pair1_test.c index 0f0131b5..881c4ac8 100644 --- a/src/protocol/pair1/pair1_test.c +++ b/src/protocol/pair1/pair1_test.c @@ -8,22 +8,35 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/pair0/pair.h> -#include <nng/protocol/pair1/pair.h> - -#include <testutil.h> - -#include <acutest.h> +#include <nuts.h> #define SECOND 1000 -#define APPEND_STR(m, s) TEST_CHECK(nng_msg_append(m, s, strlen(s)) == 0) -#define CHECK_STR(m, s) \ - TEST_CHECK(nng_msg_len(m) == strlen(s)); \ - TEST_CHECK(memcmp(nng_msg_body(m), s, strlen(s)) == 0) +#define APPEND_STR(m, s) NUTS_TRUE(nng_msg_append(m, s, strlen(s)) == 0) +#define CHECK_STR(m, s) \ + NUTS_TRUE(nng_msg_len(m) == strlen(s)); \ + NUTS_TRUE(memcmp(nng_msg_body(m), s, strlen(s)) == 0) + +static void +test_mono_identity(void) +{ + nng_socket s; + int p; + char * n; + + NUTS_PASS(nng_pair1_open(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p)); + NUTS_TRUE(p == NUTS_PROTO(1u, 1u)); // 32 + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p)); + NUTS_TRUE(p == NUTS_PROTO(1u, 1u)); // 33 + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n)); + NUTS_MATCH(n, "pair1"); + nng_strfree(n); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n)); + NUTS_MATCH(n, "pair1"); + nng_strfree(n); + NUTS_CLOSE(s); +} void test_mono_cooked(void) @@ -32,28 +45,28 @@ test_mono_cooked(void) nng_socket c1; nng_msg * msg; - TEST_NNG_PASS(nng_pair1_open(&s1)); - TEST_NNG_PASS(nng_pair1_open(&c1)); - TEST_NNG_PASS(testutil_marry(s1, c1)); + NUTS_PASS(nng_pair1_open(&s1)); + NUTS_PASS(nng_pair1_open(&c1)); + NUTS_PASS(nuts_marry(s1, c1)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append(msg, "ALPHA", strlen("ALPHA") + 1)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); - TEST_CHECK(nng_msg_len(msg) == strlen("ALPHA") + 1); - TEST_CHECK(strcmp(nng_msg_body(msg), "ALPHA") == 0); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_append(msg, "ALPHA", strlen("ALPHA") + 1)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_TRUE(nng_msg_len(msg) == strlen("ALPHA") + 1); + NUTS_MATCH(nng_msg_body(msg), "ALPHA"); nng_msg_free(msg); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append(msg, "BETA", strlen("BETA") + 1)); - TEST_NNG_PASS(nng_sendmsg(s1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(c1, &msg, 0)); - TEST_CHECK(nng_msg_len(msg) == strlen("BETA") + 1); - TEST_CHECK(strcmp(nng_msg_body(msg), "BETA") == 0); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_append(msg, "BETA", strlen("BETA") + 1)); + NUTS_PASS(nng_sendmsg(s1, msg, 0)); + NUTS_PASS(nng_recvmsg(c1, &msg, 0)); + NUTS_TRUE(nng_msg_len(msg) == strlen("BETA") + 1); + NUTS_MATCH(nng_msg_body(msg), "BETA"); nng_msg_free(msg); - TEST_NNG_PASS(nng_close(c1)); - TEST_NNG_PASS(nng_close(s1)); + NUTS_CLOSE(c1); + NUTS_CLOSE(s1); } void @@ -65,35 +78,35 @@ test_mono_faithful(void) nng_msg * msg; const char *addr = "inproc://pair1_mono_faithful"; - TEST_NNG_PASS(nng_pair1_open(&s1)); - TEST_NNG_PASS(nng_pair1_open(&c1)); - TEST_NNG_PASS(nng_pair1_open(&c2)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 4)); - TEST_NNG_PASS(nng_setopt_ms(c1, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(c2, NNG_OPT_SENDTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_int(c2, NNG_OPT_SENDBUF, 2)); + NUTS_PASS(nng_pair1_open(&s1)); + NUTS_PASS(nng_pair1_open(&c1)); + NUTS_PASS(nng_pair1_open(&c2)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 4)); + NUTS_PASS(nng_setopt_ms(c1, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_setopt_ms(c2, NNG_OPT_SENDTIMEO, SECOND)); + NUTS_PASS(nng_setopt_int(c2, NNG_OPT_SENDBUF, 2)); - TEST_NNG_PASS(nng_listen(s1, addr, NULL, 0)); - TEST_NNG_PASS(testutil_marry(s1, c1)); - TEST_NNG_PASS(nng_dial(c2, addr, NULL, 0)); + NUTS_PASS(nng_listen(s1, addr, NULL, 0)); + NUTS_MARRY(s1, c1); + NUTS_PASS(nng_dial(c2, addr, NULL, 0)); - testutil_sleep(100); + NUTS_SLEEP(100); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); APPEND_STR(msg, "ONE"); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); CHECK_STR(msg, "ONE"); nng_msg_free(msg); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); APPEND_STR(msg, "TWO"); - TEST_NNG_PASS(nng_sendmsg(c2, msg, 0)); - TEST_NNG_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_sendmsg(c2, msg, 0)); + NUTS_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(c1)); - TEST_NNG_PASS(nng_close(c2)); + NUTS_CLOSE(s1); + NUTS_CLOSE(c1); + NUTS_CLOSE(c2); } void @@ -106,28 +119,28 @@ test_mono_back_pressure(void) nng_msg * msg; nng_duration to = 100; - TEST_NNG_PASS(nng_pair1_open(&s1)); - TEST_NNG_PASS(nng_pair1_open(&c1)); - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_SENDBUF, 1)); - TEST_NNG_PASS(nng_setopt_int(c1, NNG_OPT_RECVBUF, 1)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_SENDTIMEO, to)); + NUTS_PASS(nng_pair1_open(&s1)); + NUTS_PASS(nng_pair1_open(&c1)); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_SENDBUF, 1)); + NUTS_PASS(nng_setopt_int(c1, NNG_OPT_RECVBUF, 1)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_SENDTIMEO, to)); - TEST_NNG_PASS(testutil_marry(s1, c1)); + NUTS_MARRY(s1, c1); // We choose to allow some buffering. In reality the // buffer size is just 1, and we will fail after 2. for (i = 0, rv = 0; i < 10; i++) { - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); if ((rv = nng_sendmsg(s1, msg, 0)) != 0) { nng_msg_free(msg); break; } } - TEST_NNG_FAIL(rv, NNG_ETIMEDOUT); - TEST_CHECK(i < 10); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(c1)); + NUTS_FAIL(rv, NNG_ETIMEDOUT); + NUTS_TRUE(i < 10); + NUTS_CLOSE(s1); + NUTS_CLOSE(c1); } void @@ -139,45 +152,45 @@ test_mono_raw_exchange(void) nng_msg *msg; uint32_t hops; - TEST_NNG_PASS(nng_pair1_open_raw(&s1)); - TEST_NNG_PASS(nng_pair1_open_raw(&c1)); + NUTS_PASS(nng_pair1_open_raw(&s1)); + NUTS_PASS(nng_pair1_open_raw(&c1)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND)); - TEST_NNG_PASS(testutil_marry(s1, c1)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND)); + NUTS_MARRY(s1, c1); nng_pipe p = NNG_PIPE_INITIALIZER; - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); APPEND_STR(msg, "GAMMA"); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 1)); - TEST_CHECK(nng_msg_header_len(msg) == sizeof(uint32_t)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_msg_header_append_u32(msg, 1)); + NUTS_TRUE(nng_msg_header_len(msg) == sizeof(uint32_t)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); p = nng_msg_get_pipe(msg); - TEST_CHECK(nng_pipe_id(p) > 0); + NUTS_TRUE(nng_pipe_id(p) > 0); CHECK_STR(msg, "GAMMA"); - TEST_CHECK(nng_msg_header_len(msg) == sizeof(uint32_t)); - TEST_NNG_PASS(nng_msg_header_trim_u32(msg, &hops)); - TEST_CHECK(hops == 2); + NUTS_TRUE(nng_msg_header_len(msg) == sizeof(uint32_t)); + NUTS_PASS(nng_msg_header_trim_u32(msg, &hops)); + NUTS_TRUE(hops == 2); nng_msg_free(msg); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); APPEND_STR(msg, "EPSILON"); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 1)); - TEST_NNG_PASS(nng_sendmsg(s1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(c1, &msg, 0)); + NUTS_PASS(nng_msg_header_append_u32(msg, 1)); + NUTS_PASS(nng_sendmsg(s1, msg, 0)); + NUTS_PASS(nng_recvmsg(c1, &msg, 0)); CHECK_STR(msg, "EPSILON"); - TEST_CHECK(nng_msg_header_len(msg) == sizeof(uint32_t)); - TEST_NNG_PASS(nng_msg_header_trim_u32(msg, &hops)); + NUTS_TRUE(nng_msg_header_len(msg) == sizeof(uint32_t)); + NUTS_PASS(nng_msg_header_trim_u32(msg, &hops)); p = nng_msg_get_pipe(msg); - TEST_CHECK(nng_pipe_id(p) > 0); + NUTS_TRUE(nng_pipe_id(p) > 0); - TEST_CHECK(hops == 2); + NUTS_TRUE(hops == 2); nng_msg_free(msg); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(c1)); + NUTS_CLOSE(s1); + NUTS_CLOSE(c1); } void @@ -188,52 +201,52 @@ test_mono_raw_header(void) nng_msg * msg; uint32_t v; - TEST_NNG_PASS(nng_pair1_open_raw(&s1)); - TEST_NNG_PASS(nng_pair1_open_raw(&c1)); + NUTS_PASS(nng_pair1_open_raw(&s1)); + NUTS_PASS(nng_pair1_open_raw(&c1)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 5)); - TEST_NNG_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND / 5)); - TEST_NNG_PASS(testutil_marry(s1, c1)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 5)); + NUTS_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND / 5)); + NUTS_MARRY(s1, c1); // Missing bits in the header - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); // Valid header works - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append_u32(msg, 0xFEEDFACE)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 1)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); - TEST_NNG_PASS(nng_msg_trim_u32(msg, &v)); - TEST_CHECK(v == 0xFEEDFACE); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_append_u32(msg, 0xFEEDFACE)); + NUTS_PASS(nng_msg_header_append_u32(msg, 1)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_msg_trim_u32(msg, &v)); + NUTS_TRUE(v == 0xFEEDFACE); nng_msg_free(msg); // Header with reserved bits set dropped - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 0xDEAD0000)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_header_append_u32(msg, 0xDEAD0000)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); // Header with no chance to add another hop gets dropped - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 0xff)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_header_append_u32(msg, 0xff)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); // With the same bits clear it works - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append_u32(msg, 0xFEEDFACE)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 1)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); - TEST_NNG_PASS(nng_msg_trim_u32(msg, &v)); - TEST_CHECK(v == 0xFEEDFACE); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_append_u32(msg, 0xFEEDFACE)); + NUTS_PASS(nng_msg_header_append_u32(msg, 1)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_msg_trim_u32(msg, &v)); + NUTS_TRUE(v == 0xFEEDFACE); nng_msg_free(msg); - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(c1)); + NUTS_CLOSE(s1); + NUTS_CLOSE(c1); } void @@ -242,17 +255,17 @@ test_pair1_raw(void) nng_socket s1; bool raw; - TEST_NNG_PASS(nng_pair1_open(&s1)); - TEST_NNG_PASS(nng_getopt_bool(s1, NNG_OPT_RAW, &raw)); - TEST_CHECK(raw == false); - TEST_NNG_FAIL(nng_setopt_bool(s1, NNG_OPT_RAW, true), NNG_EREADONLY); - TEST_NNG_PASS(nng_close(s1)); - - TEST_NNG_PASS(nng_pair1_open_raw(&s1)); - TEST_NNG_PASS(nng_getopt_bool(s1, NNG_OPT_RAW, &raw)); - TEST_CHECK(raw == true); - TEST_NNG_FAIL(nng_setopt_bool(s1, NNG_OPT_RAW, false), NNG_EREADONLY); - TEST_NNG_PASS(nng_close(s1)); + NUTS_PASS(nng_pair1_open(&s1)); + NUTS_PASS(nng_getopt_bool(s1, NNG_OPT_RAW, &raw)); + NUTS_TRUE(raw == false); + NUTS_FAIL(nng_setopt_bool(s1, NNG_OPT_RAW, true), NNG_EREADONLY); + NUTS_PASS(nng_close(s1)); + + NUTS_PASS(nng_pair1_open_raw(&s1)); + NUTS_PASS(nng_getopt_bool(s1, NNG_OPT_RAW, &raw)); + NUTS_TRUE(raw == true); + NUTS_FAIL(nng_setopt_bool(s1, NNG_OPT_RAW, false), NNG_EREADONLY); + NUTS_PASS(nng_close(s1)); } void @@ -264,65 +277,65 @@ test_pair1_ttl(void) uint32_t val; int ttl; - TEST_NNG_PASS(nng_pair1_open_raw(&s1)); - TEST_NNG_PASS(nng_pair1_open_raw(&c1)); - TEST_NNG_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 5)); - TEST_NNG_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND / 5)); + NUTS_PASS(nng_pair1_open_raw(&s1)); + NUTS_PASS(nng_pair1_open_raw(&c1)); + NUTS_PASS(nng_setopt_ms(s1, NNG_OPT_RECVTIMEO, SECOND / 5)); + NUTS_PASS(nng_setopt_ms(c1, NNG_OPT_RECVTIMEO, SECOND / 5)); // cannot set insane TTLs - TEST_NNG_FAIL(nng_setopt_int(s1, NNG_OPT_MAXTTL, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(s1, NNG_OPT_MAXTTL, 1000), NNG_EINVAL); + NUTS_FAIL(nng_setopt_int(s1, NNG_OPT_MAXTTL, 0), NNG_EINVAL); + NUTS_FAIL(nng_setopt_int(s1, NNG_OPT_MAXTTL, 1000), NNG_EINVAL); ttl = 8; - TEST_NNG_FAIL(nng_setopt(s1, NNG_OPT_MAXTTL, &ttl, 1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(s1, NNG_OPT_MAXTTL, true), NNG_EBADTYPE); + NUTS_FAIL(nng_setopt(s1, NNG_OPT_MAXTTL, &ttl, 1), NNG_EINVAL); + NUTS_FAIL(nng_setopt_bool(s1, NNG_OPT_MAXTTL, true), NNG_EBADTYPE); - TEST_NNG_PASS(testutil_marry(s1, c1)); + NUTS_MARRY(s1, c1); // Let's check enforcement of TTL - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 4)); - TEST_NNG_PASS(nng_getopt_int(s1, NNG_OPT_MAXTTL, &ttl)); - TEST_CHECK(ttl == 4); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 4)); + NUTS_PASS(nng_getopt_int(s1, NNG_OPT_MAXTTL, &ttl)); + NUTS_TRUE(ttl == 4); // Bad TTL bounces - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 4)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_header_append_u32(msg, 4)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); // Good TTL passes - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append_u32(msg, 0xFEEDFACE)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 3)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); - TEST_NNG_PASS(nng_msg_trim_u32(msg, &val)); - TEST_CHECK(val == 0xFEEDFACE); - TEST_NNG_PASS(nng_msg_header_trim_u32(msg, &val)); - TEST_CHECK(val == 4); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_append_u32(msg, 0xFEEDFACE)); + NUTS_PASS(nng_msg_header_append_u32(msg, 3)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_msg_trim_u32(msg, &val)); + NUTS_TRUE(val == 0xFEEDFACE); + NUTS_PASS(nng_msg_header_trim_u32(msg, &val)); + NUTS_TRUE(val == 4); nng_msg_free(msg); // Large TTL passes - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 15)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_append_u32(msg, 1234)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 14)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_PASS(nng_recvmsg(s1, &msg, 0)); - TEST_NNG_PASS(nng_msg_trim_u32(msg, &val)); - TEST_CHECK(val == 1234); - TEST_NNG_PASS(nng_msg_header_trim_u32(msg, &val)); - TEST_CHECK(val == 15); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 15)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_append_u32(msg, 1234)); + NUTS_PASS(nng_msg_header_append_u32(msg, 14)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_PASS(nng_recvmsg(s1, &msg, 0)); + NUTS_PASS(nng_msg_trim_u32(msg, &val)); + NUTS_TRUE(val == 1234); + NUTS_PASS(nng_msg_header_trim_u32(msg, &val)); + NUTS_TRUE(val == 15); nng_msg_free(msg); // Max TTL fails - TEST_NNG_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 15)); - TEST_NNG_PASS(nng_msg_alloc(&msg, 0)); - TEST_NNG_PASS(nng_msg_header_append_u32(msg, 15)); - TEST_NNG_PASS(nng_sendmsg(c1, msg, 0)); - TEST_NNG_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); - - TEST_NNG_PASS(nng_close(s1)); - TEST_NNG_PASS(nng_close(c1)); + NUTS_PASS(nng_setopt_int(s1, NNG_OPT_MAXTTL, 15)); + NUTS_PASS(nng_msg_alloc(&msg, 0)); + NUTS_PASS(nng_msg_header_append_u32(msg, 15)); + NUTS_PASS(nng_sendmsg(c1, msg, 0)); + NUTS_FAIL(nng_recvmsg(s1, &msg, 0), NNG_ETIMEDOUT); + + NUTS_CLOSE(s1); + NUTS_CLOSE(c1); } void @@ -331,28 +344,27 @@ test_pair1_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_pair1_open(&s1)); - TEST_NNG_PASS(nng_pair0_open(&s2)); + NUTS_ADDR(addr, "inproc"); + NUTS_PASS(nng_pair1_open(&s1)); + NUTS_PASS(nng_pair0_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); } @@ -363,20 +375,20 @@ test_pair1_recv_no_header(void) nng_socket c; nng_msg * m; - TEST_NNG_PASS(nng_pair1_open(&s)); - TEST_NNG_PASS(nng_pair1_open(&c)); - TEST_NNG_PASS(nng_setopt_bool(c, "pair1_test_inject_header", true)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_pair1_open(&s)); + NUTS_PASS(nng_pair1_open(&c)); + NUTS_PASS(nng_setopt_bool(c, "pair1_test_inject_header", true)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(testutil_marry(c, s)); + NUTS_MARRY(c, s); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_sendmsg(c, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(s, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_sendmsg(c, m, 0)); + NUTS_FAIL(nng_recvmsg(s, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(c)); - TEST_NNG_PASS(nng_close(s)); + NUTS_CLOSE(c); + NUTS_CLOSE(s); } void @@ -386,30 +398,31 @@ test_pair1_recv_garbage(void) nng_socket c; nng_msg * m; - TEST_NNG_PASS(nng_pair1_open(&s)); - TEST_NNG_PASS(nng_pair1_open(&c)); - TEST_NNG_PASS(nng_setopt_bool(c, "pair1_test_inject_header", true)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); + NUTS_PASS(nng_pair1_open(&s)); + NUTS_PASS(nng_pair1_open(&c)); + NUTS_PASS(nng_setopt_bool(c, "pair1_test_inject_header", true)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_RECVTIMEO, 100)); + NUTS_PASS(nng_setopt_ms(s, NNG_OPT_SENDTIMEO, 200)); - TEST_NNG_PASS(testutil_marry(c, s)); + NUTS_MARRY(c, s); // ridiculous hop count - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_msg_append_u32(m, 0x1000)); - TEST_NNG_PASS(nng_sendmsg(c, m, 0)); - TEST_NNG_FAIL(nng_recvmsg(s, &m, 0), NNG_ETIMEDOUT); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_msg_append_u32(m, 0x1000)); + NUTS_PASS(nng_sendmsg(c, m, 0)); + NUTS_FAIL(nng_recvmsg(s, &m, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(c)); - TEST_NNG_PASS(nng_close(s)); + NUTS_CLOSE(c); + NUTS_CLOSE(s); } -TEST_LIST = { - { "pair1 monogamous cooked", test_mono_cooked }, - { "pair1 monogamous faithful", test_mono_faithful }, - { "pair1 monogamous back pressure", test_mono_back_pressure }, - { "pair1 monogamous raw exchange", test_mono_raw_exchange }, - { "pair1 monogamous raw header", test_mono_raw_header }, +NUTS_TESTS = { + { "pair1 mono identity", test_mono_identity }, + { "pair1 mono cooked", test_mono_cooked }, + { "pair1 mono faithful", test_mono_faithful }, + { "pair1 mono back pressure", test_mono_back_pressure }, + { "pair1 mono raw exchange", test_mono_raw_exchange }, + { "pair1 mono raw header", test_mono_raw_header }, { "pair1 raw", test_pair1_raw }, { "pair1 ttl", test_pair1_ttl }, { "pair1 validate peer", test_pair1_validate_peer }, diff --git a/src/protocol/pubsub0/pub_test.c b/src/protocol/pubsub0/pub_test.c index 94d5c333..a430b610 100644 --- a/src/protocol/pubsub0/pub_test.c +++ b/src/protocol/pubsub0/pub_test.c @@ -7,18 +7,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/pubsub0/pub.h> -#include <nng/protocol/pubsub0/sub.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_pub_identity(void) @@ -27,18 +16,18 @@ test_pub_identity(void) int p; char * n; - TEST_NNG_PASS(nng_pub0_open(&s)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PROTO, &p)); - TEST_CHECK(p == NNI_PROTO(2u, 0u)); // 32 - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PEER, &p)); - TEST_CHECK(p == NNI_PROTO(2u, 1u)); // 33 - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PROTONAME, &n)); - TEST_CHECK(strcmp(n, "pub") == 0); + NUTS_PASS(nng_pub0_open(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p)); + NUTS_TRUE(p == NUTS_PROTO(2u, 0u)); // 32 + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p)); + NUTS_TRUE(p == NUTS_PROTO(2u, 1u)); // 33 + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n)); + NUTS_MATCH(n, "pub"); nng_strfree(n); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PEERNAME, &n)); - TEST_CHECK(strcmp(n, "sub") == 0); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n)); + NUTS_MATCH(n, "sub"); nng_strfree(n); - TEST_NNG_PASS(nng_close(s)); + NUTS_CLOSE(s); } static void @@ -46,9 +35,9 @@ test_pub_cannot_recv(void) { nng_socket pub; - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_FAIL(nng_recv(pub, "", 0, 0), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(pub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_FAIL(nng_recv(pub, "", 0, 0), NNG_ENOTSUP); + NUTS_CLOSE(pub); } static void @@ -57,9 +46,9 @@ test_pub_no_context(void) nng_socket pub; nng_ctx ctx; - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_FAIL(nng_ctx_open(&ctx, pub), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(pub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_FAIL(nng_ctx_open(&ctx, pub), NNG_ENOTSUP); + NUTS_CLOSE(pub); } static void @@ -68,9 +57,9 @@ test_pub_not_readable(void) int fd; nng_socket pub; - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_FAIL(nng_getopt_int(pub, NNG_OPT_RECVFD, &fd), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(pub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_FAIL(nng_socket_get_int(pub, NNG_OPT_RECVFD, &fd), NNG_ENOTSUP); + NUTS_CLOSE(pub); } static void @@ -80,25 +69,24 @@ test_pub_poll_writeable(void) nng_socket pub; nng_socket sub; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_getopt_int(pub, NNG_OPT_SENDFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_socket_get_int(pub, NNG_OPT_SENDFD, &fd)); + NUTS_TRUE(fd >= 0); // Pub is *always* writeable - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd)); // Even after connect (no message yet) - TEST_NNG_PASS(testutil_marry(pub, sub)); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_MARRY(pub, sub); + NUTS_TRUE(nuts_poll_fd(fd)); - // But once we send messages, it is. - // We have to send a request, in order to send a reply. - TEST_NNG_SEND_STR(pub, "abc"); - TEST_CHECK(testutil_pollfd(fd) == true); + // Even if we send messages. + NUTS_SEND(pub, "abc"); + NUTS_TRUE(nuts_poll_fd(fd)); - TEST_NNG_PASS(nng_close(pub)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(pub); + NUTS_CLOSE(sub); } static void @@ -106,10 +94,10 @@ test_pub_send_no_pipes(void) { nng_socket pub; - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_SEND_STR(pub, "DROP1"); - TEST_NNG_SEND_STR(pub, "DROP2"); - TEST_NNG_PASS(nng_close(pub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_SEND(pub, "DROP1"); + NUTS_SEND(pub, "DROP2"); + NUTS_CLOSE(pub); } void @@ -118,28 +106,28 @@ test_pub_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_pub0_open(&s1)); - TEST_NNG_PASS(nng_pub0_open(&s2)); + NUTS_PASS(nng_pub0_open(&s1)); + NUTS_PASS(nng_pub0_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); } @@ -152,26 +140,26 @@ test_pub_send_queued(void) // MB: What we really need is a mock so that we can send harder // than we receive -- we need a way to apply back-pressure for this // test to be really meaningful. - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_SUBSCRIBE, "", 0)); - TEST_NNG_PASS(nng_setopt_int(pub, NNG_OPT_SENDBUF, 10)); - TEST_NNG_PASS(nng_setopt_int(sub, NNG_OPT_RECVBUF, 10)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(pub, sub)); - TEST_NNG_SEND_STR(pub, "first"); - TEST_NNG_SEND_STR(pub, "second"); - TEST_NNG_SEND_STR(pub, "three musketeers"); - TEST_NNG_SEND_STR(pub, "four"); - testutil_sleep(50); - TEST_NNG_RECV_STR(sub, "first"); - TEST_NNG_RECV_STR(sub, "second"); - TEST_NNG_RECV_STR(sub, "three musketeers"); - TEST_NNG_RECV_STR(sub, "four"); - - TEST_NNG_PASS(nng_close(pub)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_SUBSCRIBE, "", 0)); + NUTS_PASS(nng_socket_set_int(pub, NNG_OPT_SENDBUF, 10)); + NUTS_PASS(nng_socket_set_int(sub, NNG_OPT_RECVBUF, 10)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 1000)); + NUTS_MARRY(pub, sub); + NUTS_SEND(pub, "first"); + NUTS_SEND(pub, "second"); + NUTS_SEND(pub, "three musketeers"); + NUTS_SEND(pub, "four"); + NUTS_SLEEP(50); + NUTS_RECV(sub, "first"); + NUTS_RECV(sub, "second"); + NUTS_RECV(sub, "three musketeers"); + NUTS_RECV(sub, "four"); + + NUTS_CLOSE(pub); + NUTS_CLOSE(sub); } static void test_sub_recv_ctx_closed(void) @@ -179,15 +167,15 @@ test_sub_recv_ctx_closed(void) nng_socket sub; nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_ctx_close(ctx); nng_ctx_recv(ctx, aio); 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); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); } static void @@ -197,16 +185,16 @@ test_sub_ctx_recv_aio_stopped(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); 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(sub)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_PASS(nng_ctx_close(ctx)); + NUTS_CLOSE(sub); nng_aio_free(aio); } @@ -217,16 +205,16 @@ test_sub_close_context_recv(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_set_timeout(aio, 1000); nng_ctx_recv(ctx, aio); - TEST_NNG_PASS(nng_ctx_close(ctx)); + NUTS_PASS(nng_ctx_close(ctx)); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECLOSED); + NUTS_FAIL(nng_aio_result(aio), NNG_ECLOSED); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); nng_aio_free(aio); } @@ -237,16 +225,16 @@ test_sub_ctx_recv_nonblock(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + 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(sub)); + NUTS_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); + NUTS_CLOSE(sub); nng_aio_free(aio); } @@ -257,17 +245,17 @@ test_sub_ctx_recv_cancel(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_set_timeout(aio, 1000); nng_ctx_recv(ctx, aio); nng_aio_abort(aio, NNG_ECANCELED); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECANCELED); - TEST_NNG_PASS(nng_close(sub)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_CLOSE(sub); nng_aio_free(aio); } @@ -280,28 +268,28 @@ test_pub_send_buf_option(void) size_t sz; const char *opt = NNG_OPT_SENDBUF; - TEST_NNG_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_setopt_int(pub, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(pub, opt, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(pub, opt, -1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(pub, opt, 1000000), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(pub, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(pub, opt, &v)); - TEST_CHECK(v == 3); + NUTS_PASS(nng_socket_set_int(pub, opt, 1)); + NUTS_FAIL(nng_socket_set_int(pub, opt, 0), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(pub, opt, -1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(pub, opt, 1000000), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(pub, opt, 3)); + NUTS_PASS(nng_socket_get_int(pub, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(pub, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(pub, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_NNG_FAIL(nng_setopt(pub, opt, "", 1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set(pub, opt, "", 1), NNG_EINVAL); sz = 1; - TEST_NNG_FAIL(nng_getopt(pub, opt, &v, &sz), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(pub, opt, true), NNG_EBADTYPE); - TEST_NNG_FAIL(nng_getopt_bool(pub, opt, &b), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get(pub, opt, &v, &sz), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_bool(pub, opt, true), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(pub, opt, &b), NNG_EBADTYPE); - TEST_NNG_PASS(nng_close(pub)); + NUTS_CLOSE(pub); } static void @@ -310,20 +298,20 @@ test_pub_cooked(void) nng_socket s; bool b; - TEST_NNG_PASS(nng_pub0_open(&s)); - TEST_NNG_PASS(nng_getopt_bool(s, NNG_OPT_RAW, &b)); - TEST_CHECK(!b); - TEST_NNG_FAIL(nng_setopt_bool(s, NNG_OPT_RAW, true), NNG_EREADONLY); - TEST_NNG_PASS(nng_close(s)); + NUTS_PASS(nng_pub0_open(&s)); + NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_RAW, &b)); + NUTS_TRUE(!b); + NUTS_FAIL(nng_socket_set_bool(s, NNG_OPT_RAW, true), NNG_EREADONLY); + NUTS_PASS(nng_close(s)); // raw pub only differs in the option setting - TEST_NNG_PASS(nng_pub0_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_pub0_open_raw(&s)); + NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_RAW, &b)); + NUTS_TRUE(b); + NUTS_CLOSE(s); } -TEST_LIST = { +NUTS_TESTS = { { "pub identity", test_pub_identity }, { "pub cannot recv", test_pub_cannot_recv }, { "put no context", test_pub_no_context }, diff --git a/src/protocol/pubsub0/sub_test.c b/src/protocol/pubsub0/sub_test.c index 3edf1d16..b830ae80 100644 --- a/src/protocol/pubsub0/sub_test.c +++ b/src/protocol/pubsub0/sub_test.c @@ -7,18 +7,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/pubsub0/pub.h> -#include <nng/protocol/pubsub0/sub.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_sub_identity(void) @@ -27,18 +16,18 @@ test_sub_identity(void) int p; char * n; - TEST_NNG_PASS(nng_sub0_open(&s)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PROTO, &p)); - TEST_CHECK(p == NNI_PROTO(2u, 1u)); // 33 - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PEER, &p)); - TEST_CHECK(p == NNI_PROTO(2u, 0u)); // 32 - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PROTONAME, &n)); - TEST_CHECK(strcmp(n, "sub") == 0); + NUTS_PASS(nng_sub0_open(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p)); + NUTS_TRUE(p == NUTS_PROTO(2u, 1u)); // 33 + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p)); + NUTS_TRUE(p == NUTS_PROTO(2u, 0u)); // 32 + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n)); + NUTS_MATCH(n, "sub"); nng_strfree(n); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PEERNAME, &n)); - TEST_CHECK(strcmp(n, "pub") == 0); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n)); + NUTS_MATCH(n, "pub"); nng_strfree(n); - TEST_NNG_PASS(nng_close(s)); + NUTS_CLOSE(s); } static void @@ -46,9 +35,9 @@ test_sub_cannot_send(void) { nng_socket sub; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_FAIL(nng_send(sub, "", 0, 0), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_FAIL(nng_send(sub, "", 0, 0), NNG_ENOTSUP); + NUTS_CLOSE(sub); } static void @@ -59,17 +48,17 @@ test_sub_context_cannot_send(void) nng_msg * m; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - TEST_NNG_PASS(nng_msg_alloc(&m, 0)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + NUTS_PASS(nng_msg_alloc(&m, 0)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_set_msg(aio, m); nng_aio_set_timeout(aio, 1000); nng_ctx_send(ctx, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ENOTSUP); - TEST_NNG_PASS(nng_ctx_close(ctx)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_FAIL(nng_aio_result(aio), NNG_ENOTSUP); + NUTS_PASS(nng_ctx_close(ctx)); + NUTS_CLOSE(sub); nng_aio_free(aio); nng_msg_free(m); } @@ -80,9 +69,9 @@ test_sub_not_writeable(void) int fd; nng_socket sub; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_FAIL(nng_getopt_int(sub, NNG_OPT_SENDFD, &fd), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_FAIL(nng_socket_get_int(sub, NNG_OPT_SENDFD, &fd), NNG_ENOTSUP); + NUTS_CLOSE(sub); } static void @@ -92,38 +81,38 @@ test_sub_poll_readable(void) nng_socket pub; nng_socket sub; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_SUBSCRIBE, "a", 1)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_getopt_int(sub, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_SUBSCRIBE, "a", 1)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_get_int(sub, 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(pub, sub)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(pub, sub); + NUTS_TRUE(nuts_poll_fd(fd) == false); // If we send a message we didn't subscribe to, that doesn't matter. - TEST_NNG_SEND_STR(pub, "def"); - testutil_sleep(100); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_SEND(pub, "def"); + NUTS_SLEEP(100); + 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(pub, "abc"); - testutil_sleep(100); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_SEND(pub, "abc"); + NUTS_SLEEP(100); + NUTS_TRUE(nuts_poll_fd(fd)); // and receiving makes it no longer ready - TEST_NNG_RECV_STR(sub, "abc"); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_RECV(sub, "abc"); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(nng_close(pub)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(pub); + NUTS_CLOSE(sub); } static void @@ -135,41 +124,41 @@ test_sub_recv_late(void) nng_aio * aio; nng_msg * msg; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_SUBSCRIBE, "", 0)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_getopt_int(sub, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_SUBSCRIBE, "", 0)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_get_int(sub, 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(pub, sub)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(pub, sub); + NUTS_TRUE(nuts_poll_fd(fd) == false); nng_recv_aio(sub, aio); // But once we send messages, it is. // We have to send a request, in order to send a reply. - TEST_NNG_SEND_STR(pub, "abc"); - testutil_sleep(200); + NUTS_SEND(pub, "abc"); + NUTS_SLEEP(200); nng_aio_wait(aio); - TEST_NNG_PASS(nng_aio_result(aio)); + NUTS_PASS(nng_aio_result(aio)); msg = nng_aio_get_msg(aio); nng_aio_set_msg(aio, NULL); - TEST_CHECK(nng_msg_len(msg) == 4); - TEST_CHECK(strcmp(nng_msg_body(msg), "abc") == 0); + NUTS_TRUE(nng_msg_len(msg) == 4); + NUTS_MATCH(nng_msg_body(msg), "abc"); nng_msg_free(msg); nng_aio_free(aio); - TEST_NNG_PASS(nng_close(pub)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(pub); + NUTS_CLOSE(sub); } void @@ -179,14 +168,12 @@ test_sub_context_no_poll(void) nng_socket sub; nng_ctx ctx; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - 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(sub)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + 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(sub); } void @@ -195,28 +182,28 @@ test_sub_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_sub0_open(&s1)); - TEST_NNG_PASS(nng_sub0_open(&s2)); + NUTS_PASS(nng_sub0_open(&s1)); + NUTS_PASS(nng_sub0_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); } @@ -226,15 +213,15 @@ test_sub_recv_ctx_closed(void) nng_socket sub; nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_ctx_close(ctx); nng_ctx_recv(ctx, aio); 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); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); } static void @@ -244,16 +231,16 @@ test_sub_ctx_recv_aio_stopped(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); 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(sub)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_PASS(nng_ctx_close(ctx)); + NUTS_CLOSE(sub); nng_aio_free(aio); } @@ -264,16 +251,16 @@ test_sub_close_context_recv(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_set_timeout(aio, 1000); nng_ctx_recv(ctx, aio); - TEST_NNG_PASS(nng_ctx_close(ctx)); + NUTS_PASS(nng_ctx_close(ctx)); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECLOSED); + NUTS_FAIL(nng_aio_result(aio), NNG_ECLOSED); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); nng_aio_free(aio); } @@ -284,16 +271,16 @@ test_sub_ctx_recv_nonblock(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + 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(sub)); + NUTS_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); + NUTS_CLOSE(sub); nng_aio_free(aio); } @@ -304,17 +291,17 @@ test_sub_ctx_recv_cancel(void) nng_ctx ctx; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_ctx_open(&ctx, sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_ctx_open(&ctx, sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_set_timeout(aio, 1000); nng_ctx_recv(ctx, aio); nng_aio_abort(aio, NNG_ECANCELED); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECANCELED); - TEST_NNG_PASS(nng_close(sub)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_CLOSE(sub); nng_aio_free(aio); } @@ -327,28 +314,28 @@ test_sub_recv_buf_option(void) size_t sz; const char *opt = NNG_OPT_RECVBUF; - TEST_NNG_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_setopt_int(sub, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(sub, opt, 0), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(sub, opt, -1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(sub, opt, 1000000), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(sub, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(sub, opt, &v)); - TEST_CHECK(v == 3); + NUTS_PASS(nng_socket_set_int(sub, opt, 1)); + NUTS_FAIL(nng_socket_set_int(sub, opt, 0), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(sub, opt, -1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(sub, opt, 1000000), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(sub, opt, 3)); + NUTS_PASS(nng_socket_get_int(sub, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(sub, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(sub, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_NNG_FAIL(nng_setopt(sub, opt, "", 1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set(sub, opt, "", 1), NNG_EINVAL); sz = 1; - TEST_NNG_FAIL(nng_getopt(sub, opt, &v, &sz), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(sub, opt, true), NNG_EBADTYPE); - TEST_NNG_FAIL(nng_getopt_bool(sub, opt, &b), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get(sub, opt, &v, &sz), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_bool(sub, opt, true), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(sub, opt, &b), NNG_EBADTYPE); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); } static void @@ -359,16 +346,16 @@ test_sub_subscribe_option(void) int v; const char *opt = NNG_OPT_SUB_SUBSCRIBE; - TEST_NNG_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_setopt(sub, opt, "abc", 3)); - TEST_NNG_PASS(nng_setopt(sub, opt, "abc", 3)); // duplicate - TEST_NNG_PASS(nng_setopt_bool(sub, opt, false)); - TEST_NNG_PASS(nng_setopt_int(sub, opt, 32)); + NUTS_PASS(nng_socket_set(sub, opt, "abc", 3)); + NUTS_PASS(nng_socket_set(sub, opt, "abc", 3)); // duplicate + NUTS_PASS(nng_socket_set_bool(sub, opt, false)); + NUTS_PASS(nng_socket_set_int(sub, opt, 32)); sz = sizeof(v); - TEST_NNG_FAIL(nng_getopt(sub, opt, &v, &sz), NNG_EWRITEONLY); + NUTS_FAIL(nng_socket_get(sub, opt, &v, &sz), NNG_EWRITEONLY); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); } static void @@ -380,19 +367,19 @@ test_sub_unsubscribe_option(void) const char *opt1 = NNG_OPT_SUB_SUBSCRIBE; const char *opt2 = NNG_OPT_SUB_UNSUBSCRIBE; - TEST_NNG_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_setopt(sub, opt1, "abc", 3)); - TEST_NNG_FAIL(nng_setopt(sub, opt2, "abcdef", 6), NNG_ENOENT); - TEST_NNG_PASS(nng_setopt(sub, opt2, "abc", 3)); - TEST_NNG_FAIL(nng_setopt(sub, opt2, "abc", 3), NNG_ENOENT); - TEST_NNG_PASS(nng_setopt_int(sub, opt1, 32)); - TEST_NNG_FAIL(nng_setopt_int(sub, opt2, 23), NNG_ENOENT); - TEST_NNG_PASS(nng_setopt_int(sub, opt2, 32)); + NUTS_PASS(nng_socket_set(sub, opt1, "abc", 3)); + NUTS_FAIL(nng_socket_set(sub, opt2, "abc123", 6), NNG_ENOENT); + NUTS_PASS(nng_socket_set(sub, opt2, "abc", 3)); + NUTS_FAIL(nng_socket_set(sub, opt2, "abc", 3), NNG_ENOENT); + NUTS_PASS(nng_socket_set_int(sub, opt1, 32)); + NUTS_FAIL(nng_socket_set_int(sub, opt2, 23), NNG_ENOENT); + NUTS_PASS(nng_socket_set_int(sub, opt2, 32)); sz = sizeof(v); - TEST_NNG_FAIL(nng_getopt(sub, opt2, &v, &sz), NNG_EWRITEONLY); + NUTS_FAIL(nng_socket_get(sub, opt2, &v, &sz), NNG_EWRITEONLY); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); } static void @@ -403,22 +390,22 @@ test_sub_prefer_new_option(void) size_t sz; const char *opt = NNG_OPT_SUB_PREFNEW; - TEST_NNG_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_setopt_bool(sub, opt, true)); - TEST_NNG_PASS(nng_setopt_bool(sub, opt, false)); - TEST_NNG_PASS(nng_getopt_bool(sub, opt, &b)); - TEST_CHECK(b == false); + NUTS_PASS(nng_socket_set_bool(sub, opt, true)); + NUTS_PASS(nng_socket_set_bool(sub, opt, false)); + NUTS_PASS(nng_socket_get_bool(sub, opt, &b)); + NUTS_TRUE(b == false); sz = sizeof(b); b = true; - TEST_NNG_PASS(nng_getopt(sub, opt, &b, &sz)); - TEST_CHECK(b == false); - TEST_CHECK(sz == sizeof(bool)); + NUTS_PASS(nng_socket_get(sub, opt, &b, &sz)); + NUTS_TRUE(b == false); + NUTS_TRUE(sz == sizeof(bool)); - TEST_NNG_FAIL(nng_setopt(sub, opt, "abc", 3), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(sub, opt, 1), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_set(sub, opt, "abc", 3), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(sub, opt, 1), NNG_EBADTYPE); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); } void @@ -428,23 +415,23 @@ test_sub_drop_new(void) nng_socket pub; nng_msg * msg; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_setopt_int(sub, NNG_OPT_RECVBUF, 2)); - TEST_NNG_PASS(nng_setopt_bool(sub, NNG_OPT_SUB_PREFNEW, false)); - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_SUBSCRIBE, NULL, 0)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(pub, sub)); - TEST_NNG_SEND_STR(pub, "one"); - TEST_NNG_SEND_STR(pub, "two"); - TEST_NNG_SEND_STR(pub, "three"); - testutil_sleep(100); - TEST_NNG_RECV_STR(sub, "one"); - TEST_NNG_RECV_STR(sub, "two"); - TEST_NNG_FAIL(nng_recvmsg(sub, &msg, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(pub)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_socket_set_int(sub, NNG_OPT_RECVBUF, 2)); + NUTS_PASS(nng_socket_set_bool(sub, NNG_OPT_SUB_PREFNEW, false)); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_SUBSCRIBE, NULL, 0)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 200)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 1000)); + NUTS_MARRY(pub, sub); + NUTS_SEND(pub, "one"); + NUTS_SEND(pub, "two"); + NUTS_SEND(pub, "three"); + NUTS_SLEEP(100); + NUTS_RECV(sub, "one"); + NUTS_RECV(sub, "two"); + NUTS_FAIL(nng_recvmsg(sub, &msg, 0), NNG_ETIMEDOUT); + NUTS_CLOSE(pub); + NUTS_CLOSE(sub); } void @@ -454,23 +441,23 @@ test_sub_drop_old(void) nng_socket pub; nng_msg * msg; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_setopt_int(sub, NNG_OPT_RECVBUF, 2)); - TEST_NNG_PASS(nng_setopt_bool(sub, NNG_OPT_SUB_PREFNEW, true)); - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_SUBSCRIBE, NULL, 0)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 200)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(testutil_marry(pub, sub)); - TEST_NNG_SEND_STR(pub, "one"); - TEST_NNG_SEND_STR(pub, "two"); - TEST_NNG_SEND_STR(pub, "three"); - testutil_sleep(100); - TEST_NNG_RECV_STR(sub, "two"); - TEST_NNG_RECV_STR(sub, "three"); - TEST_NNG_FAIL(nng_recvmsg(sub, &msg, 0), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(pub)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_socket_set_int(sub, NNG_OPT_RECVBUF, 2)); + NUTS_PASS(nng_socket_set_bool(sub, NNG_OPT_SUB_PREFNEW, true)); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_SUBSCRIBE, NULL, 0)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 200)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 1000)); + NUTS_MARRY(pub, sub); + NUTS_SEND(pub, "one"); + NUTS_SEND(pub, "two"); + NUTS_SEND(pub, "three"); + NUTS_SLEEP(100); + NUTS_RECV(sub, "two"); + NUTS_RECV(sub, "three"); + NUTS_FAIL(nng_recvmsg(sub, &msg, 0), NNG_ETIMEDOUT); + NUTS_CLOSE(pub); + NUTS_CLOSE(sub); } static void @@ -481,46 +468,46 @@ test_sub_filter(void) char buf[32]; size_t sz; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_int(sub, NNG_OPT_RECVBUF, 10)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_int(sub, NNG_OPT_RECVBUF, 10)); // Set up some default filters - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_SUBSCRIBE, "abc", 3)); - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_SUBSCRIBE, "def", 3)); - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_SUBSCRIBE, "ghi", 3)); - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_SUBSCRIBE, "jkl", 3)); - - TEST_NNG_PASS(testutil_marry(pub, sub)); - - TEST_NNG_PASS(nng_send(pub, "def", 3, 0)); - TEST_NNG_PASS(nng_send(pub, "de", 2, 0)); // will not go through - TEST_NNG_PASS(nng_send(pub, "abc123", 6, 0)); - TEST_NNG_PASS(nng_send(pub, "xzy", 3, 0)); // does not match - TEST_NNG_PASS(nng_send(pub, "ghidrop", 7, 0)); // dropped by unsub - TEST_NNG_PASS(nng_send(pub, "jklmno", 6, 0)); - - testutil_sleep(100); - TEST_NNG_PASS(nng_setopt(sub, NNG_OPT_SUB_UNSUBSCRIBE, "ghi", 3)); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_SUBSCRIBE, "abc", 3)); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_SUBSCRIBE, "def", 3)); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_SUBSCRIBE, "ghi", 3)); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_SUBSCRIBE, "jkl", 3)); + + NUTS_MARRY(pub, sub); + + NUTS_PASS(nng_send(pub, "def", 3, 0)); + NUTS_PASS(nng_send(pub, "de", 2, 0)); // will not go through + NUTS_PASS(nng_send(pub, "abc123", 6, 0)); + NUTS_PASS(nng_send(pub, "xzy", 3, 0)); // does not match + NUTS_PASS(nng_send(pub, "ghi-drop", 7, 0)); // dropped by unsub + NUTS_PASS(nng_send(pub, "jkl-mno", 6, 0)); + + NUTS_SLEEP(100); + NUTS_PASS(nng_socket_set(sub, NNG_OPT_SUB_UNSUBSCRIBE, "ghi", 3)); sz = sizeof(buf); - TEST_NNG_PASS(nng_recv(sub, buf, &sz, 0)); - TEST_CHECK(sz == 3); - TEST_CHECK(memcmp(buf, "def", 3) == 0); + NUTS_PASS(nng_recv(sub, buf, &sz, 0)); + NUTS_TRUE(sz == 3); + NUTS_TRUE(memcmp(buf, "def", 3) == 0); sz = sizeof(buf); - TEST_NNG_PASS(nng_recv(sub, buf, &sz, 0)); - TEST_CHECK(sz == 6); - TEST_CHECK(memcmp(buf, "abc123", 6) == 0); + NUTS_PASS(nng_recv(sub, buf, &sz, 0)); + NUTS_TRUE(sz == 6); + NUTS_TRUE(memcmp(buf, "abc123", 6) == 0); sz = sizeof(buf); - TEST_NNG_PASS(nng_recv(sub, buf, &sz, 0)); - TEST_CHECK(sz == 6); - TEST_CHECK(memcmp(buf, "jklmno", 6) == 0); + NUTS_PASS(nng_recv(sub, buf, &sz, 0)); + NUTS_TRUE(sz == 6); + NUTS_TRUE(memcmp(buf, "jkl-mno", 6) == 0); - TEST_NNG_PASS(nng_close(sub)); - TEST_NNG_PASS(nng_close(pub)); + NUTS_CLOSE(sub); + NUTS_CLOSE(pub); } static void @@ -534,55 +521,55 @@ test_sub_multi_context(void) nng_aio * aio2; nng_msg * m; - TEST_NNG_PASS(nng_sub0_open(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_aio_alloc(&aio1, NULL, NULL)); - TEST_NNG_PASS(nng_aio_alloc(&aio2, NULL, NULL)); - TEST_NNG_PASS(nng_ctx_open(&c1, sub)); - TEST_NNG_PASS(nng_ctx_open(&c2, sub)); + NUTS_PASS(nng_sub0_open(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_aio_alloc(&aio1, NULL, NULL)); + NUTS_PASS(nng_aio_alloc(&aio2, NULL, NULL)); + NUTS_PASS(nng_ctx_open(&c1, sub)); + NUTS_PASS(nng_ctx_open(&c2, sub)); - TEST_NNG_PASS(nng_ctx_setopt(c1, NNG_OPT_SUB_SUBSCRIBE, "one", 3)); - TEST_NNG_PASS(nng_ctx_setopt(c1, NNG_OPT_SUB_SUBSCRIBE, "all", 3)); + NUTS_PASS(nng_ctx_set(c1, NNG_OPT_SUB_SUBSCRIBE, "one", 3)); + NUTS_PASS(nng_ctx_set(c1, NNG_OPT_SUB_SUBSCRIBE, "all", 3)); - TEST_NNG_PASS(nng_ctx_setopt(c2, NNG_OPT_SUB_SUBSCRIBE, "two", 3)); - TEST_NNG_PASS(nng_ctx_setopt(c2, NNG_OPT_SUB_SUBSCRIBE, "all", 3)); + NUTS_PASS(nng_ctx_set(c2, NNG_OPT_SUB_SUBSCRIBE, "two", 3)); + NUTS_PASS(nng_ctx_set(c2, NNG_OPT_SUB_SUBSCRIBE, "all", 3)); nng_aio_set_timeout(aio1, 100); nng_aio_set_timeout(aio2, 100); - TEST_NNG_PASS(testutil_marry(pub, sub)); + NUTS_MARRY(pub, sub); - TEST_NNG_SEND_STR(pub, "one for the money"); - TEST_NNG_SEND_STR(pub, "all dogs go to heaven"); - TEST_NNG_SEND_STR(pub, "nobody likes a snitch"); - TEST_NNG_SEND_STR(pub, "two for the show"); + NUTS_SEND(pub, "one for the money"); + NUTS_SEND(pub, "all dogs go to heaven"); + NUTS_SEND(pub, "nobody likes a snitch"); + NUTS_SEND(pub, "two for the show"); nng_ctx_recv(c1, aio1); nng_aio_wait(aio1); - TEST_NNG_PASS(nng_aio_result(aio1)); + NUTS_PASS(nng_aio_result(aio1)); m = nng_aio_get_msg(aio1); - TEST_CHECK(strcmp(nng_msg_body(m), "one for the money") == 0); + NUTS_MATCH(nng_msg_body(m), "one for the money"); nng_msg_free(m); nng_ctx_recv(c1, aio1); nng_aio_wait(aio1); - TEST_NNG_PASS(nng_aio_result(aio1)); + NUTS_PASS(nng_aio_result(aio1)); m = nng_aio_get_msg(aio1); - TEST_CHECK(strcmp(nng_msg_body(m), "all dogs go to heaven") == 0); + NUTS_MATCH(nng_msg_body(m), "all dogs go to heaven"); nng_msg_free(m); nng_ctx_recv(c2, aio1); nng_aio_wait(aio1); - TEST_NNG_PASS(nng_aio_result(aio1)); + NUTS_PASS(nng_aio_result(aio1)); m = nng_aio_get_msg(aio1); - TEST_CHECK(strcmp(nng_msg_body(m), "all dogs go to heaven") == 0); + NUTS_MATCH(nng_msg_body(m), "all dogs go to heaven"); nng_msg_free(m); nng_ctx_recv(c2, aio1); nng_aio_wait(aio1); - TEST_NNG_PASS(nng_aio_result(aio1)); + NUTS_PASS(nng_aio_result(aio1)); m = nng_aio_get_msg(aio1); - TEST_CHECK(strcmp(nng_msg_body(m), "two for the show") == 0); + NUTS_MATCH(nng_msg_body(m), "two for the show"); nng_msg_free(m); nng_ctx_recv(c1, aio1); @@ -590,10 +577,10 @@ test_sub_multi_context(void) nng_aio_wait(aio1); nng_aio_wait(aio2); - TEST_NNG_FAIL(nng_aio_result(aio1), NNG_ETIMEDOUT); - TEST_NNG_FAIL(nng_aio_result(aio2), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(sub)); - TEST_NNG_PASS(nng_close(pub)); + NUTS_FAIL(nng_aio_result(aio1), NNG_ETIMEDOUT); + NUTS_FAIL(nng_aio_result(aio2), NNG_ETIMEDOUT); + NUTS_CLOSE(sub); + NUTS_CLOSE(pub); nng_aio_free(aio1); nng_aio_free(aio2); } @@ -604,10 +591,10 @@ test_sub_cooked(void) nng_socket s; bool b; - TEST_NNG_PASS(nng_sub0_open(&s)); - TEST_NNG_PASS(nng_getopt_bool(s, NNG_OPT_RAW, &b)); - TEST_CHECK(!b); - TEST_NNG_PASS(nng_close(s)); + NUTS_PASS(nng_sub0_open(&s)); + NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_RAW, &b)); + NUTS_TRUE(!b); + NUTS_CLOSE(s); } TEST_LIST = { diff --git a/src/protocol/pubsub0/xsub_test.c b/src/protocol/pubsub0/xsub_test.c index a05b893c..19815661 100644 --- a/src/protocol/pubsub0/xsub_test.c +++ b/src/protocol/pubsub0/xsub_test.c @@ -7,18 +7,7 @@ // found online at https://opensource.org/licenses/MIT. // -#include <string.h> - -#include <nng/nng.h> -#include <nng/protocol/pubsub0/pub.h> -#include <nng/protocol/pubsub0/sub.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_xsub_identity(void) @@ -27,18 +16,18 @@ test_xsub_identity(void) int p; char * n; - TEST_NNG_PASS(nng_sub0_open_raw(&s)); - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PROTO, &p)); - TEST_CHECK(p == NNI_PROTO(2u, 1u)); // 33 - TEST_NNG_PASS(nng_getopt_int(s, NNG_OPT_PEER, &p)); - TEST_CHECK(p == NNI_PROTO(2u, 0u)); // 32 - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PROTONAME, &n)); - TEST_CHECK(strcmp(n, "sub") == 0); + NUTS_PASS(nng_sub0_open_raw(&s)); + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PROTO, &p)); + NUTS_TRUE(p == NUTS_PROTO(2u, 1u)); // 33 + NUTS_PASS(nng_socket_get_int(s, NNG_OPT_PEER, &p)); + NUTS_TRUE(p == NUTS_PROTO(2u, 0u)); // 32 + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PROTONAME, &n)); + NUTS_MATCH(n, "sub"); nng_strfree(n); - TEST_NNG_PASS(nng_getopt_string(s, NNG_OPT_PEERNAME, &n)); - TEST_CHECK(strcmp(n, "pub") == 0); + NUTS_PASS(nng_socket_get_string(s, NNG_OPT_PEERNAME, &n)); + NUTS_MATCH(n, "pub"); nng_strfree(n); - TEST_NNG_PASS(nng_close(s)); + NUTS_CLOSE(s); } static void @@ -46,9 +35,9 @@ test_xsub_cannot_send(void) { nng_socket sub; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_FAIL(nng_send(sub, "", 0, 0), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_FAIL(nng_send(sub, "", 0, 0), NNG_ENOTSUP); + NUTS_CLOSE(sub); } static void @@ -57,9 +46,9 @@ test_xsub_not_writeable(void) int fd; nng_socket sub; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_FAIL(nng_getopt_int(sub, NNG_OPT_SENDFD, &fd), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_FAIL(nng_socket_get_int(sub, NNG_OPT_SENDFD, &fd), NNG_ENOTSUP); + NUTS_CLOSE(sub); } static void @@ -69,33 +58,33 @@ test_xsub_poll_readable(void) nng_socket pub; nng_socket sub; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_getopt_int(sub, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_get_int(sub, 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(pub, sub)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(pub, sub); + 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(pub, "abc"); - testutil_sleep(200); + NUTS_SEND(pub, "abc"); + NUTS_SLEEP(200); - TEST_CHECK(testutil_pollfd(fd) == true); + NUTS_TRUE(nuts_poll_fd(fd)); // and receiving makes it no longer ready - TEST_NNG_RECV_STR(sub, "abc"); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_RECV(sub, "abc"); + NUTS_TRUE(nuts_poll_fd(fd) == false); - TEST_NNG_PASS(nng_close(pub)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(pub); + NUTS_CLOSE(sub); } static void @@ -107,40 +96,40 @@ test_xsub_recv_late(void) nng_aio * aio; nng_msg * msg; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 1000)); - TEST_NNG_PASS(nng_getopt_int(sub, NNG_OPT_RECVFD, &fd)); - TEST_CHECK(fd >= 0); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 1000)); + NUTS_PASS(nng_socket_get_int(sub, 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(pub, sub)); - TEST_CHECK(testutil_pollfd(fd) == false); + NUTS_MARRY(pub, sub); + NUTS_TRUE(nuts_poll_fd(fd) == false); nng_recv_aio(sub, aio); // But once we send messages, it is. // We have to send a request, in order to send a reply. - TEST_NNG_SEND_STR(pub, "abc"); - testutil_sleep(200); + NUTS_SEND(pub, "abc"); + NUTS_SLEEP(200); nng_aio_wait(aio); - TEST_NNG_PASS(nng_aio_result(aio)); + NUTS_PASS(nng_aio_result(aio)); msg = nng_aio_get_msg(aio); nng_aio_set_msg(aio, NULL); - TEST_CHECK(nng_msg_len(msg) == 4); - TEST_CHECK(strcmp(nng_msg_body(msg), "abc") == 0); + NUTS_TRUE(nng_msg_len(msg) == 4); + NUTS_TRUE(strcmp(nng_msg_body(msg), "abc") == 0); nng_msg_free(msg); nng_aio_free(aio); - TEST_NNG_PASS(nng_close(pub)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(pub); + NUTS_CLOSE(sub); } void @@ -149,9 +138,9 @@ test_xsub_no_context(void) nng_socket sub; nng_ctx ctx; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_FAIL(nng_ctx_open(&ctx, sub), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_FAIL(nng_ctx_open(&ctx, sub), NNG_ENOTSUP); + NUTS_CLOSE(sub); } void @@ -160,28 +149,28 @@ test_xsub_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_sub0_open_raw(&s1)); - TEST_NNG_PASS(nng_sub0_open_raw(&s2)); + NUTS_PASS(nng_sub0_open_raw(&s1)); + NUTS_PASS(nng_sub0_open_raw(&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); } @@ -190,12 +179,12 @@ test_xsub_recv_closed(void) { nng_socket sub; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); - nng_close(sub); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_CLOSE(sub); nng_recv_aio(sub, aio); 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); } @@ -205,13 +194,13 @@ test_xsub_close_recv(void) nng_socket sub; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_set_timeout(aio, 1000); nng_recv_aio(sub, aio); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); 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); } @@ -222,15 +211,15 @@ test_xsub_recv_nonblock(void) nng_socket sub; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_set_timeout(aio, 0); // Instant timeout nng_recv_aio(sub, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); - TEST_NNG_PASS(nng_close(sub)); + NUTS_FAIL(nng_aio_result(aio), NNG_ETIMEDOUT); + NUTS_CLOSE(sub); nng_aio_free(aio); } @@ -243,27 +232,27 @@ test_xsub_recv_buf_option(void) size_t sz; const char *opt = NNG_OPT_RECVBUF; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); + NUTS_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_PASS(nng_setopt_int(sub, opt, 1)); - TEST_NNG_FAIL(nng_setopt_int(sub, opt, -1), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_int(sub, opt, 1000000), NNG_EINVAL); - TEST_NNG_PASS(nng_setopt_int(sub, opt, 3)); - TEST_NNG_PASS(nng_getopt_int(sub, opt, &v)); - TEST_CHECK(v == 3); + NUTS_PASS(nng_socket_set_int(sub, opt, 1)); + NUTS_FAIL(nng_socket_set_int(sub, opt, -1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_int(sub, opt, 1000000), NNG_EINVAL); + NUTS_PASS(nng_socket_set_int(sub, opt, 3)); + NUTS_PASS(nng_socket_get_int(sub, opt, &v)); + NUTS_TRUE(v == 3); v = 0; sz = sizeof(v); - TEST_NNG_PASS(nng_getopt(sub, opt, &v, &sz)); - TEST_CHECK(v == 3); - TEST_CHECK(sz == sizeof(v)); + NUTS_PASS(nng_socket_get(sub, opt, &v, &sz)); + NUTS_TRUE(v == 3); + NUTS_TRUE(sz == sizeof(v)); - TEST_NNG_FAIL(nng_setopt(sub, opt, "", 1), NNG_EINVAL); + NUTS_FAIL(nng_socket_set(sub, opt, "", 1), NNG_EINVAL); sz = 1; - TEST_NNG_FAIL(nng_getopt(sub, opt, &v, &sz), NNG_EINVAL); - TEST_NNG_FAIL(nng_setopt_bool(sub, opt, true), NNG_EBADTYPE); - TEST_NNG_FAIL(nng_getopt_bool(sub, opt, &b), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get(sub, opt, &v, &sz), NNG_EINVAL); + NUTS_FAIL(nng_socket_set_bool(sub, opt, true), NNG_EBADTYPE); + NUTS_FAIL(nng_socket_get_bool(sub, opt, &b), NNG_EBADTYPE); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); } static void @@ -272,9 +261,9 @@ test_xsub_subscribe_option(void) nng_socket sub; const char *opt = NNG_OPT_SUB_SUBSCRIBE; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_FAIL(nng_setopt(sub, opt, "abc", 3), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_FAIL(nng_socket_set(sub, opt, "abc", 3), NNG_ENOTSUP); + NUTS_CLOSE(sub); } static void @@ -283,9 +272,9 @@ test_xsub_unsubscribe_option(void) nng_socket sub; const char *opt = NNG_OPT_SUB_UNSUBSCRIBE; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_FAIL(nng_setopt(sub, opt, "abc", 3), NNG_ENOTSUP); - TEST_NNG_PASS(nng_close(sub)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_FAIL(nng_socket_set(sub, opt, "abc", 3), NNG_ENOTSUP); + NUTS_CLOSE(sub); } static void @@ -294,10 +283,10 @@ test_xsub_raw(void) nng_socket s; bool b; - TEST_NNG_PASS(nng_sub0_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_sub0_open_raw(&s)); + NUTS_PASS(nng_socket_get_bool(s, NNG_OPT_RAW, &b)); + NUTS_TRUE(b); + NUTS_CLOSE(s); } static void @@ -306,20 +295,20 @@ test_xsub_close_during_recv(void) nng_socket sub; nng_socket pub; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_int(sub, NNG_OPT_RECVBUF, 5)); - TEST_NNG_PASS(nng_setopt_int(pub, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_int(sub, NNG_OPT_RECVBUF, 5)); + NUTS_PASS(nng_socket_set_int(pub, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(testutil_marry(pub, sub)); + NUTS_MARRY(pub, sub); for (unsigned i = 0; i < 100; i++) { - TEST_NNG_PASS(nng_send(pub, "abc", 3, 0)); + NUTS_PASS(nng_send(pub, "abc", 3, 0)); } - TEST_NNG_PASS(nng_close(pub)); - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(pub); + NUTS_CLOSE(sub); } static void @@ -328,14 +317,14 @@ test_xsub_close_during_pipe_recv(void) nng_socket sub; nng_socket pub; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_PASS(nng_pub0_open(&pub)); - TEST_NNG_PASS(nng_setopt_ms(sub, NNG_OPT_RECVTIMEO, 1000)); - TEST_NNG_PASS(nng_setopt_ms(pub, NNG_OPT_SENDTIMEO, 100)); - TEST_NNG_PASS(nng_setopt_int(sub, NNG_OPT_RECVBUF, 5)); - TEST_NNG_PASS(nng_setopt_int(pub, NNG_OPT_SENDBUF, 20)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_PASS(nng_pub0_open(&pub)); + NUTS_PASS(nng_socket_set_ms(sub, NNG_OPT_RECVTIMEO, 1000)); + NUTS_PASS(nng_socket_set_ms(pub, NNG_OPT_SENDTIMEO, 100)); + NUTS_PASS(nng_socket_set_int(sub, NNG_OPT_RECVBUF, 5)); + NUTS_PASS(nng_socket_set_int(pub, NNG_OPT_SENDBUF, 20)); - TEST_NNG_PASS(testutil_marry(pub, sub)); + NUTS_MARRY(pub, sub); for (unsigned i = 0; i < 100; i++) { int rv; @@ -343,9 +332,9 @@ test_xsub_close_during_pipe_recv(void) if (rv == NNG_ETIMEDOUT) { break; } - testutil_sleep(1); + NUTS_SLEEP(1); } - TEST_NNG_PASS(nng_close(sub)); + NUTS_CLOSE(sub); } static void @@ -354,14 +343,14 @@ test_xsub_recv_aio_stopped(void) nng_socket sub; nng_aio * aio; - TEST_NNG_PASS(nng_sub0_open_raw(&sub)); - TEST_NNG_PASS(nng_aio_alloc(&aio, NULL, NULL)); + NUTS_PASS(nng_sub0_open_raw(&sub)); + NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL)); nng_aio_stop(aio); nng_recv_aio(sub, aio); nng_aio_wait(aio); - TEST_NNG_FAIL(nng_aio_result(aio), NNG_ECANCELED); - TEST_NNG_PASS(nng_close(sub)); + NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED); + NUTS_CLOSE(sub); nng_aio_free(aio); } 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 }, 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 = { |