test_probe_router.cpp 2.12 KB
Newer Older
Pieter Hintjens's avatar
Pieter Hintjens committed
1
/*
2
    Copyright (c) 2007-2014 Contributors as noted in the AUTHORS file
Pieter Hintjens's avatar
Pieter Hintjens committed
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

    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 <http://www.gnu.org/licenses/>.
*/

20
#include "testutil.hpp"
Pieter Hintjens's avatar
Pieter Hintjens committed
21 22 23

int main (void)
{
24
    setup_test_environment();
Pieter Hintjens's avatar
Pieter Hintjens committed
25 26 27 28 29 30
    void *ctx = zmq_ctx_new ();
    assert (ctx);
    
    //  Create server and bind to endpoint
    void *server = zmq_socket (ctx, ZMQ_ROUTER);
    assert (server);
31
    int rc = zmq_bind (server, "tcp://127.0.0.1:5560");
Pieter Hintjens's avatar
Pieter Hintjens committed
32 33 34
    assert (rc == 0);

    //  Create client and connect to server, doing a probe
35
    void *client = zmq_socket (ctx, ZMQ_ROUTER);
Pieter Hintjens's avatar
Pieter Hintjens committed
36 37 38 39
    assert (client);
    rc = zmq_setsockopt (client, ZMQ_IDENTITY, "X", 1);
    assert (rc == 0);
    int probe = 1;
40
    rc = zmq_setsockopt (client, ZMQ_PROBE_ROUTER, &probe, sizeof (probe));
Pieter Hintjens's avatar
Pieter Hintjens committed
41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
    assert (rc == 0);
    rc = zmq_connect (client, "tcp://localhost:5560");
    assert (rc == 0);

    //  We expect an identity=X + empty message from client
    unsigned char buffer [255];
    rc = zmq_recv (server, buffer, 255, 0);
    assert (rc == 1);
    assert (buffer [0] ==  'X');
    rc = zmq_recv (server, buffer, 255, 0);
    assert (rc == 0);

    //  Send a message to client now
    rc = zmq_send (server, "X", 1, ZMQ_SNDMORE);
    assert (rc == 1);
    rc = zmq_send (server, "Hello", 5, 0);
    assert (rc == 5);
    
    rc = zmq_recv (client, buffer, 255, 0);
    assert (rc == 5);
    
    rc = zmq_close (server);
    assert (rc == 0);

    rc = zmq_close (client);
    assert (rc == 0);

    rc = zmq_ctx_term (ctx);
    assert (rc == 0);

    return 0 ;
}