diff options
| -rw-r--r-- | tests/testutil.hpp | 24 |
1 files changed, 23 insertions, 1 deletions
diff --git a/tests/testutil.hpp b/tests/testutil.hpp index 1fbf1f3..57db3c4 100644 --- a/tests/testutil.hpp +++ b/tests/testutil.hpp @@ -31,13 +31,27 @@ inline void bounce (void *sb, void *sc) const char *content = "12345678ABCDEFGH12345678abcdefgh"; // Send the message. - int rc = zmq_send (sc, content, 32, 0); + int rc = zmq_send (sc, content, 32, ZMQ_SNDMORE); + assert (rc == 32); + rc = zmq_send (sc, content, 32, 0); assert (rc == 32); // Bounce the message back. char buf1 [32]; rc = zmq_recv (sb, buf1, 32, 0); assert (rc == 32); + int rcvmore; + size_t sz = sizeof (rcvmore); + rc = zmq_getsockopt (sb, ZMQ_RCVMORE, &rcvmore, &sz); + assert (rc == 0); + assert (rcvmore); + rc = zmq_recv (sb, buf1, 32, 0); + assert (rc == 32); + rc = zmq_getsockopt (sb, ZMQ_RCVMORE, &rcvmore, &sz); + assert (rc == 0); + assert (!rcvmore); + rc = zmq_send (sb, buf1, 32, ZMQ_SNDMORE); + assert (rc == 32); rc = zmq_send (sb, buf1, 32, 0); assert (rc == 32); @@ -45,6 +59,14 @@ inline void bounce (void *sb, void *sc) char buf2 [32]; rc = zmq_recv (sc, buf2, 32, 0); assert (rc == 32); + rc = zmq_getsockopt (sc, ZMQ_RCVMORE, &rcvmore, &sz); + assert (rc == 0); + assert (rcvmore); + rc = zmq_recv (sc, buf2, 32, 0); + assert (rc == 32); + rc = zmq_getsockopt (sc, ZMQ_RCVMORE, &rcvmore, &sz); + assert (rc == 0); + assert (!rcvmore); // Check whether the message is still the same. assert (memcmp (buf2, content, 32) == 0); |