nsq/nsqadmin/node_modules/stream-splicer/test/unshift.js

var pipeline = require('../');
var through = require('through2');
var test = require('tape');

test('unshift', function (t) {
    var expected = {};
    expected.a = [ 5, 6 ];
    expected.b = [ 3, 4, 500, 600 ];
    expected.c = [ 13, 14, 510, 610 ];
    expected.output = [ 13/2, 7, 255, 305 ];
    
    t.plan(2 + 4 + 4 + 4);
    
    var a = through.obj(function (x, enc, next) {
        var ex = expected.a.shift();
        t.equal(x, ex, 'a');
        this.push(x * 100);
        next();
    });
    var b = through.obj(function (x, enc, next) {
        var ex = expected.b.shift();
        t.equal(x, ex, 'b');
        if (expected.b.length === 2) p.unshift(a)
        this.push(x + 10);
        next();
    });
    var c = through.obj(function (x, enc, next) {
        var ex = expected.c.shift();
        t.equal(x, ex, 'c');
        this.push(x / 2);
        next();
    });
    
    var p = pipeline.obj([ b, c ]);
    p.pipe(through.obj(function (x, enc, next) {
        var ex = expected.output.shift();
        t.equal(x, ex);
        next();
    }));
    
    p.write(3);
    p.write(4);
    p.write(5);
    p.write(6);
    p.end();
});