diff options
| -rw-r--r-- | tests/Makefile.am | 7 | ||||
| -rw-r--r-- | tests/test_shutdown_stress.cpp | 83 |
2 files changed, 88 insertions, 2 deletions
diff --git a/tests/Makefile.am b/tests/Makefile.am index 0825ae8..e580802 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -2,11 +2,12 @@ INCLUDES = -I$(top_builddir)/include LDADD = $(top_builddir)/src/libzmq.la noinst_PROGRAMS = test_pair_inproc \ - test_pair_ipc \ + test_pair_ipc \ test_pair_tcp \ test_reqrep_inproc \ test_reqrep_ipc \ - test_reqrep_tcp + test_reqrep_tcp \ + test_shutdown_stress test_pair_inproc_SOURCES = test_pair_inproc.cpp testutil.hpp test_pair_ipc_SOURCES = test_pair_ipc.cpp testutil.hpp @@ -16,4 +17,6 @@ test_reqrep_inproc_SOURCES = test_reqrep_inproc.cpp testutil.hpp test_reqrep_ipc_SOURCES = test_reqrep_ipc.cpp testutil.hpp test_reqrep_tcp_SOURCES = test_reqrep_tcp.cpp testutil.hpp +test_shutdown_stress_SOURCES = test_shutdown_stress.cpp + TESTS = $(noinst_PROGRAMS) diff --git a/tests/test_shutdown_stress.cpp b/tests/test_shutdown_stress.cpp new file mode 100644 index 0000000..9f24138 --- /dev/null +++ b/tests/test_shutdown_stress.cpp @@ -0,0 +1,83 @@ +/* + 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 Lesser GNU 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 + Lesser GNU General Public License for more details. + + You should have received a copy of the Lesser GNU General Public License + along with this program. If not, see <http://www.gnu.org/licenses/>. +*/ + +#include "../include/zmq.h" +#include <assert.h> +#include <pthread.h> +#include <stddef.h> + +#define THREAD_COUNT 100 + +void *worker (void *s) +{ + int rc; + + rc = zmq_connect (s, "tcp://127.0.0.1:5555"); + assert (rc == 0); + + // Start closing the socket while the connecting process is underway. + rc = zmq_close (s); + assert (rc == 0); + + return NULL; +} + +int main (int argc, char *argv []) +{ + void *ctx; + void *s1; + void *s2; + int i; + int j; + int rc; + pthread_t threads [THREAD_COUNT]; + + for (j = 0; j != 10; j++) { + + // Check the shutdown with many parallel I/O threads. + ctx = zmq_init (7); + assert (ctx); + + s1 = zmq_socket (ctx, ZMQ_REP); + assert (s1); + + rc = zmq_bind (s1, "tcp://127.0.0.1:5555"); + assert (rc == 0); + + for (i = 0; i != THREAD_COUNT; i++) { + s2 = zmq_socket (ctx, ZMQ_SUB); + assert (s2); + rc = pthread_create (&threads [i], NULL, worker, s2); + assert (rc == 0); + } + + for (i = 0; i != THREAD_COUNT; i++) { + rc = pthread_join (threads [i], NULL); + assert (rc == 0); + } + + rc = zmq_close (s1); + assert (rc == 0); + + rc = zmq_term (ctx); + assert (rc == 0); + } + + return 0; +} |