From 2afb791ec2f9853cc642cf8a042c2acc7fbb7acd Mon Sep 17 00:00:00 2001 From: Martin Sustrik Date: Mon, 11 Jul 2011 15:54:13 +0200 Subject: [PATCH] Subscription forwarding test added Signed-off-by: Martin Sustrik --- tests/Makefile.am | 4 +- tests/test_sub_forward.cpp | 94 ++++++++++++++++++++++++++++++++++++++ 2 files changed, 97 insertions(+), 1 deletion(-) create mode 100644 tests/test_sub_forward.cpp diff --git a/tests/Makefile.am b/tests/Makefile.am index 8ec3b0cb..6389e21b 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -7,7 +7,8 @@ noinst_PROGRAMS = test_pair_inproc \ test_reqrep_tcp \ test_hwm \ test_reqrep_device \ - test_reqrep_drop + test_reqrep_drop \ + test_sub_forward if !ON_MINGW noinst_PROGRAMS += test_shutdown_stress \ @@ -23,6 +24,7 @@ test_reqrep_tcp_SOURCES = test_reqrep_tcp.cpp testutil.hpp test_hwm_SOURCES = test_hwm.cpp test_reqrep_device_SOURCES = test_reqrep_device.cpp test_reqrep_drop_SOURCES = test_reqrep_drop.cpp +test_sub_forward_SOURCES = test_sub_forward.cpp if !ON_MINGW test_shutdown_stress_SOURCES = test_shutdown_stress.cpp diff --git a/tests/test_sub_forward.cpp b/tests/test_sub_forward.cpp new file mode 100644 index 00000000..d69f923d --- /dev/null +++ b/tests/test_sub_forward.cpp @@ -0,0 +1,94 @@ +/* + Copyright (c) 2007-2011 iMatix Corporation + Copyright (c) 2007-2011 Other contributors as noted in the AUTHORS file + + 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/zmq.h" +#include "../include/zmq_utils.h" + +int main (int argc, char *argv []) +{ + void *ctx = zmq_init (1); + assert (ctx); + + // First, create an intermediate device. + void *xpub = zmq_socket (ctx, ZMQ_XPUB); + assert (xpub); + int rc = zmq_bind (xpub, "tcp://127.0.0.1:5560"); + assert (rc == 0); + void *xsub = zmq_socket (ctx, ZMQ_XSUB); + assert (xsub); + rc = zmq_bind (xsub, "tcp://127.0.0.1:5561"); + assert (rc == 0); + + // Create a publisher. + void *pub = zmq_socket (ctx, ZMQ_PUB); + assert (pub); + rc = zmq_connect (pub, "tcp://127.0.0.1:5561"); + assert (rc == 0); + + // Create a subscriber. + void *sub = zmq_socket (ctx, ZMQ_SUB); + assert (sub); + rc = zmq_connect (sub, "tcp://127.0.0.1:5560"); + assert (rc == 0); + + // Subscribe for all messages. + rc = zmq_setsockopt (sub, ZMQ_SUBSCRIBE, "", 0); + assert (rc == 0); + + // Pass the subscription upstream through the device. + char buff [32]; + rc = zmq_recv (xpub, buff, sizeof (buff), 0); + assert (rc >= 0); + rc = zmq_send (xsub, buff, rc, 0); + assert (rc >= 0); + + // Wait a bit till the subscription gets to the publisher. + zmq_sleep (1); + + // Send an empty message. + rc = zmq_send (pub, NULL, 0, 0); + assert (rc == 0); + + // Pass the message downstream through the device. + rc = zmq_recv (xsub, buff, sizeof (buff), 0); + assert (rc >= 0); + rc = zmq_send (xpub, buff, rc, 0); + assert (rc >= 0); + + // Receive the message in the subscriber. + rc = zmq_recv (sub, buff, sizeof (buff), 0); + assert (rc == 0); + + // Clean up. + rc = zmq_close (xpub); + assert (rc == 0); + rc = zmq_close (xsub); + assert (rc == 0); + rc = zmq_close (pub); + assert (rc == 0); + rc = zmq_close (sub); + assert (rc == 0); + rc = zmq_term (ctx); + assert (rc == 0); + + return 0 ; +}