nanomsg/test/inproc.js

Node bindings for nanomsg

// http://tim.dysinger.net/posts/2013-09-16-getting-started-with-nanomsg.html

var nano = require('../');
var test = require('tape');

test('inproc socket pub sub', function (t) {
    t.plan(1);

    var pub = nano.socket('pub');
    var sub = nano.socket('sub');

    var addr = 'inproc://pubsub';
    var msg = 'hello world';

    pub.bind(addr);
    sub.connect(addr);

    sub.on('data', function (buf) {
        t.equal(buf.toString(), msg);

        pub.close();
        sub.close();
    });

    setTimeout(function () {
        pub.send(msg);
    }, 100);
});


test('inproc socket pairs', function (t) {
    t.plan(1);

    var s1 = nano.socket('pair');
    var s2 = nano.socket('pair');

    var addr = 'inproc://pairs';
    var msg = 'hello world';

    s1.bind(addr);
    s2.connect(addr);

    s1.on('data', function (buf) {
        t.equal(buf.toString(), msg);

        s1.close();
        s2.close();
    });

    setTimeout(function () {
        s2.send(msg);
    }, 100);
});


test('inproc socket req rep', function (t) {
    t.plan(2);

    var req = nano.socket('req');
    var rep = nano.socket('rep');

    var addr = 'inproc://reqrep';
    var msg1 = 'knock knock';
    var msg2 = "who's there?";

    rep.bind(addr);
    req.connect(addr);

    rep.on('data', function (buf) {
        t.equal(buf.toString(), msg1, 'request received');
        rep.send(msg2);
    });

    req.on('data', function (buf) {
        t.equal(buf.toString(), msg2, 'reply received');
        req.close();
        rep.close();
    });

    setTimeout(function () {
        req.send(msg1);
    }, 100);
});

test('inproc socket survey', function (t) {
    t.plan(3);

    var sur = nano.socket('surveyor');
    var rep1 = nano.socket('respondent');
    var rep2 = nano.socket('respondent');
    var rep3 = nano.socket('respondent');

    var addr = 'inproc://survey';
    var msg1 = 'knock knock';
    var msg2 = "who's there?";

    sur.bind(addr);
    rep1.connect(addr);
    rep2.connect(addr);
    rep3.connect(addr);

    function answer (buf) {
        this.send(msg2);
    }

    rep1.on('data', answer);
    rep2.on('data', answer);
    rep3.on('data', answer);

    var count = 0;
    sur.on('data', function (buf) {
        t.ok(buf.toString() == msg2, buf.toString() + ' == ' + msg2);

        if (++count == 3) {
            sur.close();
            rep1.close();
            rep2.close();
            rep3.close();
        }
    });

    setTimeout(function () {
        sur.send(msg1);
    }, 100);
});

test('inproc socket bus', function (t) {
    // http://250bpm.com/blog:17

    // Number of buses to create.
    var count = 3;
    var total = count * (count-1), current = 0;
    t.plan(count);

    // Create buses.
    var buses = {};
    for (var i = 0; i < count; i++) {
        (function (i) {
            var bus = nano.socket('bus');
            var addr = 'inproc://bus' + i;
            bus.bind(addr);
            buses[addr] = bus;

            // Add a "response count" for each bus.
            // We want this to equal the number of other buses.
            bus.responseCount = 0;

            // Tally messages from other buses.
            bus.on('data', function (msg) {
                //console.error('#', 'received data from', msg.toString(), 'on', addr)
                this.responseCount++;
                current++;

                if (this.responseCount == count - 1) {
                    // All set! bus received all messages.
                    t.ok(true, 'all messages received on ' + addr);
                }

                if (current == total) {
                    // close all buses.
                    Object.keys(buses).forEach(function (addr) {
                        buses[addr].close();
                    })
                }
            });
        })(i);
    }

    // Connect all possible pairs of buses.
    setTimeout(function () {
        var keys = Object.keys(buses);

        for (var i = 0; i < keys.length; i++) {
            for (var j = i+1; j < keys.length; j++) {
                //console.error('#', 'connecting', keys[i], 'to', keys[j]);
                buses[keys[i]].connect(keys[j]);
            }
        }
    }, 500);

    // Send messages on every bus.
    setTimeout(function () {
        Object.keys(buses).forEach(function (addr) {
            //console.error('#', 'writing on', addr, addr);
            buses[addr].send(addr);
        });
    }, 1000);
});


test('inproc multiple binds on same address', function (t) {
    t.plan(1);

    var s1 = nano.socket('pair');
    var s2 = nano.socket('pair');

    var addr = 'inproc://multiplebinds';

    s2.on('error', function (err) {
        t.ok(err, 'error was thrown on multiple binds.');
        s1.close();
        s2.close();
    });

    s1.bind(addr);
    s2.bind(addr);
});

test('inproc multiple socket pub sub', function (t) {
    t.plan(3);

    var pub = nano.socket('pub');
    var sub1 = nano.socket('sub');
    var sub2 = nano.socket('sub');
    var sub3 = nano.socket('sub');

    var addr = 'inproc://pubsub';
    var msg = 'hello world';

    pub.bind(addr);
    sub1.connect(addr);
    sub2.connect(addr);
    sub3.connect(addr);

    var responses = 0;

    var resp_handler = function (buf) {
        t.equal(buf.toString(), msg);

        responses++;

        if(responses == 3) {
            pub.close();
            sub1.close();
            sub2.close();
            sub3.close();
        }
    };

    sub1.on('data', resp_handler);
    sub2.on('data', resp_handler);
    sub3.on('data', resp_handler);

    setTimeout(function () {
        pub.send(msg);
    }, 100);
});