From 72d320366f2a1076ced81f4e871c7affbb9ffb9d Mon Sep 17 00:00:00 2001 From: Mikko Koppanen Date: Sun, 30 Jan 2011 11:51:30 +0100 Subject: Added test for HWM Signed-off-by: Mikko Koppanen --- tests/Makefile.am | 5 +++- tests/test_hwm.cpp | 67 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 71 insertions(+), 1 deletion(-) create mode 100644 tests/test_hwm.cpp (limited to 'tests') diff --git a/tests/Makefile.am b/tests/Makefile.am index 30ca62c..ebbc46c 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -4,7 +4,8 @@ LDADD = $(top_builddir)/src/libzmq.la noinst_PROGRAMS = test_pair_inproc \ test_pair_tcp \ test_reqrep_inproc \ - test_reqrep_tcp + test_reqrep_tcp \ + test_hwm if !ON_MINGW noinst_PROGRAMS += test_shutdown_stress \ @@ -18,6 +19,8 @@ test_pair_tcp_SOURCES = test_pair_tcp.cpp testutil.hpp test_reqrep_inproc_SOURCES = test_reqrep_inproc.cpp testutil.hpp test_reqrep_tcp_SOURCES = test_reqrep_tcp.cpp testutil.hpp +test_hwm_SOURCES = test_hwm.cpp + if !ON_MINGW test_shutdown_stress_SOURCES = test_shutdown_stress.cpp test_pair_ipc_SOURCES = test_pair_ipc.cpp testutil.hpp diff --git a/tests/test_hwm.cpp b/tests/test_hwm.cpp new file mode 100644 index 0000000..a3b007b --- /dev/null +++ b/tests/test_hwm.cpp @@ -0,0 +1,67 @@ +/* + Copyright (c) 2007-2010 iMatix Corporation + + 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 "testutil.hpp" +#include "../src/stdint.hpp" + +using namespace std; +using namespace zmqtestutil; + +int main (int argc, char *argv []) +{ + uint64_t hwm = 5; + int linger = 0; + + zmq::context_t context (1); + zmq::socket_t s1 (context, ZMQ_PULL); + zmq::socket_t s2 (context, ZMQ_PUSH); + + s2.setsockopt (ZMQ_LINGER, &linger, sizeof (int)); + s2.setsockopt (ZMQ_HWM, &hwm, sizeof (uint64_t)); + + s1.bind ("tcp://127.0.0.1:5858"); + s2.connect ("tcp://127.0.0.1:5858"); + + for (int i = 0; i < 10; i++) + { + zmq::message_t msg (sizeof ("test") - 1); + memcpy (msg.data (), "test", sizeof ("test") - 1); + + bool sent = s2.send (msg, ZMQ_NOBLOCK); + + // Anything below HWM should be sent + if (i < 5) { + assert (sent); + } else { + assert (!sent && errno == EAGAIN); + } + } + + // There should be now 5 messages pending, consume one + zmq::message_t msg; + + bool received = s1.recv (&msg, 0); + assert (received); + + // Now it should be possible to send one more + bool sent = s2.send (msg, 0); + assert (sent); + + return 0; +} -- cgit v1.2.3