From a2801adffebb6a3679e41789b38ba925ed32832b Mon Sep 17 00:00:00 2001 From: Garrett D'Amore Date: Fri, 6 Jan 2017 08:18:50 -0800 Subject: Message API was awkward. The use of a single function to get both size and length actually turned out to be awkward to use; better to have separate functions to get each. While here, disable some of the initialization/fork checks, because it turns out they aren't needed. --- tests/sock.c | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) (limited to 'tests/sock.c') diff --git a/tests/sock.c b/tests/sock.c index e0743936..78d97a9b 100644 --- a/tests/sock.c +++ b/tests/sock.c @@ -148,8 +148,6 @@ TestMain("Socket Operations", { nng_socket *sock2 = NULL; int len = 1; nng_msg *msg; - size_t sz; - char *ptr; uint64_t second = 1000000; rv = nng_open(&sock2, NNG_PROTO_PAIR); @@ -181,11 +179,9 @@ TestMain("Socket Operations", { rv = nng_msg_alloc(&msg, 3); So(rv == 0); - ptr = nng_msg_body(msg, &sz); - So(ptr != NULL); - So(sz == 3); - - memcpy(ptr, "abc", 3); + So(nng_msg_len(msg) == 3); + So(nng_msg_body(msg) != NULL); + memcpy(nng_msg_body(msg), "abc", 3); rv = nng_sendmsg(sock, msg, 0); So(rv == 0); @@ -194,10 +190,8 @@ TestMain("Socket Operations", { rv = nng_recvmsg(sock2, &msg, 0); So(rv == 0); So(msg != NULL); - ptr = nng_msg_body(msg, &sz); - So(ptr != NULL); - So(sz == 3); - So(memcmp(ptr, "abc", 3) == 0); + So(nng_msg_len(msg) == 3); + So(memcmp(nng_msg_body(msg), "abc", 3) == 0); nng_msg_free(msg); nng_close(sock2); }) -- cgit v1.2.3-70-g09d2