node-red/test/red/runtime/nodes/router/index_spec.js

/**
 * Copyright 2016 IBM Corp.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 **/

var should = require("should");
var sinon = require('sinon');
var RedNode = require("../../../../../red/runtime/nodes/Node");
var router = require("../../../../../red/runtime/nodes/router");
var flows = require("../../../../../red/runtime/nodes/flows");


describe('Router', function() {
    var flowGet;
    afterEach(function() {
        if (flowGet && flowGet.restore) {
            flowGet.restore();
            flowGet = null;
        }
    })

    describe('#add',function() {
        it('adds a route for a node', function(done) {
            var senderNode = {id:'123',metric:function(){}};
            var receiver = sinon.stub();
            flowGet = sinon.stub(flows,"get",function(id) {
                if (id === '456') {
                    return {receive:receiver};
                }
                return null;
            });
            router.send(senderNode,{});
            flowGet.called.should.be.false();

            router.add(senderNode,[['456']]);

            router.send(senderNode,{});
            flowGet.called.should.be.true();
            receiver.called.should.be.true();

            done();
        });
    })

    describe('#remove',function() {
        it('removes a route for a node', function(done) {
            var senderNode = {id:'123',metric:function(){}};
            var receiver = sinon.stub();
            flowGet = sinon.stub(flows,"get",function(id) {
                if (id === '456') {
                    return {receive:receiver};
                }
                return null;
            });
            router.add(senderNode,[['456']]);
            router.send(senderNode,{});
            flowGet.called.should.be.true();
            receiver.called.should.be.true();
            flowGet.reset();
            receiver.reset();

            router.remove(senderNode);
            router.send(senderNode,{});
            flowGet.called.should.be.false();
            done();
        });
    })
})