From 93529d8c5db599a45171942c4510f1b84ed09e6a Mon Sep 17 00:00:00 2001 From: Chuck Remes Date: Sun, 6 Nov 2011 14:03:51 +0100 Subject: Add zmq_getmsgopt to the API The new function allows to retrieve options (flags) from zmq_msg_t. Signed-off-by: Chuck Remes Renamed from zmq_msg_flags to zmq_getmsgopt Signed-off-by: Martin Sustrik --- tests/Makefile.am | 4 ++- tests/test_msg_flags.cpp | 82 ++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 85 insertions(+), 1 deletion(-) create mode 100644 tests/test_msg_flags.cpp (limited to 'tests') diff --git a/tests/Makefile.am b/tests/Makefile.am index 5f0cfc1..bbae270 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -8,7 +8,8 @@ noinst_PROGRAMS = test_pair_inproc \ test_hwm \ test_reqrep_device \ test_sub_forward \ - test_invalid_rep + test_invalid_rep \ + test_msg_flags if !ON_MINGW noinst_PROGRAMS += test_shutdown_stress \ @@ -25,6 +26,7 @@ test_hwm_SOURCES = test_hwm.cpp test_reqrep_device_SOURCES = test_reqrep_device.cpp test_sub_forward_SOURCES = test_sub_forward.cpp test_invalid_rep_SOURCES = test_invalid_rep.cpp +test_msg_flags_SOURCES = test_msg_flags.cpp if !ON_MINGW test_shutdown_stress_SOURCES = test_shutdown_stress.cpp diff --git a/tests/test_msg_flags.cpp b/tests/test_msg_flags.cpp new file mode 100644 index 0000000..10fd526 --- /dev/null +++ b/tests/test_msg_flags.cpp @@ -0,0 +1,82 @@ +/* + Copyright (c) 2011 250bpm s.r.o. + + This file is part of 0MQ. + + 0MQ is free software; you can redistribute it and/or modify it under + the terms of the GNU Lesser General Public License as published by + the Free Software Foundation; either version 3 of the License, or + (at your option) any later version. + + 0MQ is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU Lesser General Public License for more details. + + You should have received a copy of the GNU Lesser General Public License + along with this program. If not, see . +*/ + +#include +#include + +#include "../include/zmq.h" + +int main (int argc, char *argv []) +{ + // Create the infrastructure + void *ctx = zmq_init (0); + assert (ctx); + void *sb = zmq_socket (ctx, ZMQ_XREP); + assert (sb); + int rc = zmq_bind (sb, "inproc://a"); + assert (rc == 0); + void *sc = zmq_socket (ctx, ZMQ_XREQ); + assert (sc); + rc = zmq_connect (sc, "inproc://a"); + assert (rc == 0); + + // Send 2-part message. + rc = zmq_send (sc, "A", 1, ZMQ_SNDMORE); + assert (rc == 1); + rc = zmq_send (sc, "B", 1, 0); + assert (rc == 1); + + // Identity comes first. + zmq_msg_t msg; + rc = zmq_msg_init (&msg); + assert (rc == 0); + rc = zmq_recvmsg (sb, &msg, 0); + assert (rc >= 0); + int more; + size_t more_size = sizeof (more); + rc = zmq_getmsgopt (&msg, ZMQ_MORE, &more, &more_size); + assert (rc == 0); + assert (more == 1); + + // Then the first part of the message body. + rc = zmq_recvmsg (sb, &msg, 0); + assert (rc == 1); + more_size = sizeof (more); + rc = zmq_getmsgopt (&msg, ZMQ_MORE, &more, &more_size); + assert (rc == 0); + assert (more == 1); + + // And finally, the second part of the message body. + rc = zmq_recvmsg (sb, &msg, 0); + assert (rc == 1); + more_size = sizeof (more); + rc = zmq_getmsgopt (&msg, ZMQ_MORE, &more, &more_size); + assert (rc == 0); + assert (more == 0); + + // Deallocate the infrastructure. + rc = zmq_close (sc); + assert (rc == 0); + rc = zmq_close (sb); + assert (rc == 0); + rc = zmq_term (ctx); + assert (rc == 0); + return 0 ; +} + -- cgit v1.2.3