diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/Makefile.am | 19 | ||||
| -rw-r--r-- | tests/test_pair_inproc.cpp | 30 | ||||
| -rw-r--r-- | tests/test_pair_ipc.cpp | 30 | ||||
| -rw-r--r-- | tests/test_pair_tcp.cpp | 30 | ||||
| -rw-r--r-- | tests/test_reqrep_inproc.cpp | 30 | ||||
| -rw-r--r-- | tests/test_reqrep_ipc.cpp | 30 | ||||
| -rw-r--r-- | tests/test_reqrep_tcp.cpp | 30 | ||||
| -rw-r--r-- | tests/testutil.hpp | 129 | 
8 files changed, 328 insertions, 0 deletions
| diff --git a/tests/Makefile.am b/tests/Makefile.am new file mode 100644 index 0000000..0825ae8 --- /dev/null +++ b/tests/Makefile.am @@ -0,0 +1,19 @@ +INCLUDES = -I$(top_builddir)/include +LDADD = $(top_builddir)/src/libzmq.la + +noinst_PROGRAMS = test_pair_inproc \ +	          test_pair_ipc \ +                  test_pair_tcp \ +                  test_reqrep_inproc \ +                  test_reqrep_ipc \ +                  test_reqrep_tcp + +test_pair_inproc_SOURCES = test_pair_inproc.cpp testutil.hpp +test_pair_ipc_SOURCES = test_pair_ipc.cpp testutil.hpp +test_pair_tcp_SOURCES = test_pair_tcp.cpp testutil.hpp + +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 + +TESTS = $(noinst_PROGRAMS) diff --git a/tests/test_pair_inproc.cpp b/tests/test_pair_inproc.cpp new file mode 100644 index 0000000..d5da401 --- /dev/null +++ b/tests/test_pair_inproc.cpp @@ -0,0 +1,30 @@ +/* +    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 "testutil.hpp" + +using namespace std; +using namespace zmqtestutil; + +int main (int argc, char *argv []) +{ +    const char *transport = "inproc://tester" ; +    basic_tests (transport, ZMQ_PAIR, ZMQ_PAIR); +    return 0 ; +} diff --git a/tests/test_pair_ipc.cpp b/tests/test_pair_ipc.cpp new file mode 100644 index 0000000..5bf2621 --- /dev/null +++ b/tests/test_pair_ipc.cpp @@ -0,0 +1,30 @@ +/* +    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 "testutil.hpp" + +using namespace std; +using namespace zmqtestutil; + +int main (int argc, char *argv []) +{ +    const char *transport = "ipc:///tmp/tester" ; +    basic_tests (transport, ZMQ_PAIR, ZMQ_PAIR); +    return 0 ; +} diff --git a/tests/test_pair_tcp.cpp b/tests/test_pair_tcp.cpp new file mode 100644 index 0000000..60b73ac --- /dev/null +++ b/tests/test_pair_tcp.cpp @@ -0,0 +1,30 @@ +/* +    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 "testutil.hpp" + +using namespace std; +using namespace zmqtestutil; + +int main (int argc, char *argv []) +{ +    const char *transport = "tcp://127.0.0.1:5555" ; +    basic_tests (transport, ZMQ_PAIR, ZMQ_PAIR); +    return 0 ; +} diff --git a/tests/test_reqrep_inproc.cpp b/tests/test_reqrep_inproc.cpp new file mode 100644 index 0000000..847d925 --- /dev/null +++ b/tests/test_reqrep_inproc.cpp @@ -0,0 +1,30 @@ +/* +    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 "testutil.hpp" + +using namespace std; +using namespace zmqtestutil; + +int main (int argc, char *argv []) +{ +    const char *transport = "inproc://tester" ; +    basic_tests (transport, ZMQ_REQ, ZMQ_REP); +    return 0 ; +} diff --git a/tests/test_reqrep_ipc.cpp b/tests/test_reqrep_ipc.cpp new file mode 100644 index 0000000..990834e --- /dev/null +++ b/tests/test_reqrep_ipc.cpp @@ -0,0 +1,30 @@ +/* +    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 "testutil.hpp" + +using namespace std; +using namespace zmqtestutil; + +int main (int argc, char *argv []) +{ +    const char *transport = "ipc:///tmp/tester" ; +    basic_tests (transport, ZMQ_REQ, ZMQ_REP); +    return 0 ; +} diff --git a/tests/test_reqrep_tcp.cpp b/tests/test_reqrep_tcp.cpp new file mode 100644 index 0000000..e841d0e --- /dev/null +++ b/tests/test_reqrep_tcp.cpp @@ -0,0 +1,30 @@ +/* +    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 "testutil.hpp" + +using namespace std; +using namespace zmqtestutil; + +int main (int argc, char *argv []) +{ +    const char *transport = "tcp://127.0.0.1:5555" ; +    basic_tests (transport, ZMQ_REQ, ZMQ_REP); +    return 0 ; +} diff --git a/tests/testutil.hpp b/tests/testutil.hpp new file mode 100644 index 0000000..313fad4 --- /dev/null +++ b/tests/testutil.hpp @@ -0,0 +1,129 @@ +/* +    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/>. +*/ + +#ifndef __ZMQ_TEST_TESTUTIL_HPP_INCLUDED__ +#define __ZMQ_TEST_TESTUTIL_HPP_INCLUDED__ + +#include <assert.h> +#include <iostream> +#include <string> +#include <utility> + +#include "../include/zmq.hpp" + +namespace zmqtestutil +{ + +    using namespace std ; + +    typedef std::pair <zmq::socket_t*, zmq::socket_t*> socket_pair; + +    //  Create a pair of sockets connected to each other. +    socket_pair create_bound_pair (zmq::context_t *context_, +        int t1_, int t2_, const char *transport_) +    { +        zmq::socket_t *s1 = new zmq::socket_t (*context_, t1_); +        zmq::socket_t *s2 = new zmq::socket_t (*context_, t2_); +        s1->bind (transport_); +        s2->connect (transport_); +        return socket_pair (s1, s2); +    } + +    //  Send a message from one socket in the pair to the other and back. +    std::string ping_pong (const socket_pair &sp_, const std::string &orig_msg_) +    { +        zmq::socket_t &s1 = *sp_.first; +        zmq::socket_t &s2 = *sp_.second; + +        //  Construct message to send. +        zmq::message_t ping (orig_msg_.size ()); +        memcpy (ping.data (), orig_msg_.c_str (), orig_msg_.size ()); + +        //  Send ping out. +        s1.send (ping, 0); + +        //  Get pong from connected socket. +        zmq::message_t pong; +        s2.recv (&pong, 0); + +        //  Send message via s2, so state is clean in case of req/rep. +        std::string ret ((char*) pong.data(), pong.size ()); +        s2.send (pong, 0); + +        //  Return received data as std::string. +        return ret ; +    } + +    /*  Run basic tests for the given transport. + +        Basic tests are: +        * ping pong as defined above. +        * send receive where the receive is signalled by zmq::poll +    */ +    void basic_tests (const char *transport_, int t1_, int t2_) +    { +        zmq::context_t context (1); + +        zmq::pollitem_t items [2]; +        socket_pair p = create_bound_pair (&context, t1_, t2_, transport_); + +        //  First test simple ping pong. +        const string expect ("XXX"); + +        { +            const string returned = zmqtestutil::ping_pong (p, expect); +            assert (expect == returned); + +            //  Adjust socket state so that poll shows only 1 pending message. +            zmq::message_t mx ; +            p.first->recv (&mx, 0); +        } + +        { +            //  Now poll is used to singal that a message is ready to read. +            zmq::message_t m1 (expect.size ()); +            memcpy (m1.data (), expect.c_str (), expect.size ()); +            items [0].socket = *p.first; +            items [0].fd = 0; +            items [0].events = ZMQ_POLLIN; +            items [0].revents = 0; +            items [1].socket = *p.second; +            items [1].fd = 0; +            items [1].events = ZMQ_POLLIN; +            items [1].revents = 0; + +            p.first->send (m1, 0); + +            int rc = zmq::poll (&items [0], 2, -1); +            assert (rc == 1); +            assert ((items [1].revents & ZMQ_POLLIN) != 0); + +            zmq::message_t m2; +            p.second->recv (&m2, 0); +            const string ret ((char*) m2.data (), m2.size ()); +            assert (expect == ret); +        } + +        //  Delete sockets. +        delete (p.first); +        delete (p.second); +    } +} + +#endif | 
