344 lines
6.8 KiB
JavaScript
344 lines
6.8 KiB
JavaScript
|
'use strict';
|
||
|
|
||
|
var es = require('../')
|
||
|
, it = require('it-is')
|
||
|
, u = require('ubelt')
|
||
|
, spec = require('stream-spec')
|
||
|
, Stream = require('stream')
|
||
|
, from = require('from')
|
||
|
, through = require('through')
|
||
|
|
||
|
//REFACTOR THIS TEST TO USE es.readArray and es.writeArray
|
||
|
|
||
|
function writeArray(array, stream) {
|
||
|
|
||
|
array.forEach( function (j) {
|
||
|
stream.write(j)
|
||
|
})
|
||
|
stream.end()
|
||
|
|
||
|
}
|
||
|
|
||
|
function readStream(stream, done) {
|
||
|
|
||
|
var array = []
|
||
|
stream.on('data', function (data) {
|
||
|
array.push(data)
|
||
|
})
|
||
|
stream.on('error', done)
|
||
|
stream.on('end', function (data) {
|
||
|
done(null, array)
|
||
|
})
|
||
|
|
||
|
}
|
||
|
|
||
|
//call sink on each write,
|
||
|
//and complete when finished.
|
||
|
|
||
|
function pauseStream (prob, delay) {
|
||
|
var pauseIf = (
|
||
|
'number' == typeof prob
|
||
|
? function () {
|
||
|
return Math.random() < prob
|
||
|
}
|
||
|
: 'function' == typeof prob
|
||
|
? prob
|
||
|
: 0.1
|
||
|
)
|
||
|
var delayer = (
|
||
|
!delay
|
||
|
? process.nextTick
|
||
|
: 'number' == typeof delay
|
||
|
? function (next) { setTimeout(next, delay) }
|
||
|
: delay
|
||
|
)
|
||
|
|
||
|
return es.through(function (data) {
|
||
|
if(!this.paused && pauseIf()) {
|
||
|
console.log('PAUSE STREAM PAUSING')
|
||
|
this.pause()
|
||
|
var self = this
|
||
|
delayer(function () {
|
||
|
console.log('PAUSE STREAM RESUMING')
|
||
|
self.resume()
|
||
|
})
|
||
|
}
|
||
|
console.log("emit ('data', " + data + ')')
|
||
|
this.emit('data', data)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
exports ['simple map'] = function (test) {
|
||
|
|
||
|
var input = u.map(1, 1000, function () {
|
||
|
return Math.random()
|
||
|
})
|
||
|
var expected = input.map(function (v) {
|
||
|
return v * 2
|
||
|
})
|
||
|
|
||
|
var pause = pauseStream(0.1)
|
||
|
var fs = from(input)
|
||
|
var ts = es.writeArray(function (err, ar) {
|
||
|
it(ar).deepEqual(expected)
|
||
|
test.done()
|
||
|
})
|
||
|
var map = es.through(function (data) {
|
||
|
this.emit('data', data * 2)
|
||
|
})
|
||
|
|
||
|
spec(map).through().validateOnExit()
|
||
|
spec(pause).through().validateOnExit()
|
||
|
|
||
|
fs.pipe(map).pipe(pause).pipe(ts)
|
||
|
}
|
||
|
|
||
|
exports ['simple map applied to a stream'] = function (test) {
|
||
|
|
||
|
var input = [1,2,3,7,5,3,1,9,0,2,4,6]
|
||
|
//create event stream from
|
||
|
|
||
|
var doubler = es.map(function (data, cb) {
|
||
|
cb(null, data * 2)
|
||
|
})
|
||
|
|
||
|
spec(doubler).through().validateOnExit()
|
||
|
|
||
|
//a map is only a middle man, so it is both readable and writable
|
||
|
|
||
|
it(doubler).has({
|
||
|
readable: true,
|
||
|
writable: true,
|
||
|
})
|
||
|
|
||
|
readStream(doubler, function (err, output) {
|
||
|
it(output).deepEqual(input.map(function (j) {
|
||
|
return j * 2
|
||
|
}))
|
||
|
// process.nextTick(x.validate)
|
||
|
test.done()
|
||
|
})
|
||
|
|
||
|
writeArray(input, doubler)
|
||
|
|
||
|
}
|
||
|
|
||
|
exports['pipe two maps together'] = function (test) {
|
||
|
|
||
|
var input = [1,2,3,7,5,3,1,9,0,2,4,6]
|
||
|
//create event stream from
|
||
|
function dd (data, cb) {
|
||
|
cb(null, data * 2)
|
||
|
}
|
||
|
var doubler1 = es.map(dd), doubler2 = es.map(dd)
|
||
|
|
||
|
doubler1.pipe(doubler2)
|
||
|
|
||
|
spec(doubler1).through().validateOnExit()
|
||
|
spec(doubler2).through().validateOnExit()
|
||
|
|
||
|
readStream(doubler2, function (err, output) {
|
||
|
it(output).deepEqual(input.map(function (j) {
|
||
|
return j * 4
|
||
|
}))
|
||
|
test.done()
|
||
|
})
|
||
|
|
||
|
writeArray(input, doubler1)
|
||
|
|
||
|
}
|
||
|
|
||
|
//next:
|
||
|
//
|
||
|
// test pause, resume and drian.
|
||
|
//
|
||
|
|
||
|
// then make a pipe joiner:
|
||
|
//
|
||
|
// plumber (evStr1, evStr2, evStr3, evStr4, evStr5)
|
||
|
//
|
||
|
// will return a single stream that write goes to the first
|
||
|
|
||
|
exports ['map will not call end until the callback'] = function (test) {
|
||
|
|
||
|
var ticker = es.map(function (data, cb) {
|
||
|
process.nextTick(function () {
|
||
|
cb(null, data * 2)
|
||
|
})
|
||
|
})
|
||
|
|
||
|
spec(ticker).through().validateOnExit()
|
||
|
|
||
|
ticker.write('x')
|
||
|
ticker.end()
|
||
|
|
||
|
ticker.on('end', function () {
|
||
|
test.done()
|
||
|
})
|
||
|
}
|
||
|
|
||
|
|
||
|
exports ['emit error thrown'] = function (test) {
|
||
|
|
||
|
var err = new Error('INTENSIONAL ERROR')
|
||
|
, mapper =
|
||
|
es.map(function () {
|
||
|
throw err
|
||
|
})
|
||
|
|
||
|
mapper.on('error', function (_err) {
|
||
|
it(_err).equal(err)
|
||
|
test.done()
|
||
|
})
|
||
|
|
||
|
// onExit(spec(mapper).basic().validate)
|
||
|
//need spec that says stream may error.
|
||
|
|
||
|
mapper.write('hello')
|
||
|
|
||
|
}
|
||
|
|
||
|
exports ['emit error calledback'] = function (test) {
|
||
|
|
||
|
var err = new Error('INTENSIONAL ERROR')
|
||
|
, mapper =
|
||
|
es.map(function (data, callback) {
|
||
|
callback(err)
|
||
|
})
|
||
|
|
||
|
mapper.on('error', function (_err) {
|
||
|
it(_err).equal(err)
|
||
|
test.done()
|
||
|
})
|
||
|
|
||
|
mapper.write('hello')
|
||
|
|
||
|
}
|
||
|
|
||
|
exports ['do not emit drain if not paused'] = function (test) {
|
||
|
|
||
|
var map = es.map(function (data, callback) {
|
||
|
u.delay(callback)(null, 1)
|
||
|
return true
|
||
|
})
|
||
|
|
||
|
spec(map).through().pausable().validateOnExit()
|
||
|
|
||
|
map.on('drain', function () {
|
||
|
it(false).ok('should not emit drain unless the stream is paused')
|
||
|
})
|
||
|
|
||
|
it(map.write('hello')).equal(true)
|
||
|
it(map.write('hello')).equal(true)
|
||
|
it(map.write('hello')).equal(true)
|
||
|
setTimeout(function () {map.end()},10)
|
||
|
map.on('end', test.done)
|
||
|
}
|
||
|
|
||
|
exports ['emits drain if paused, when all '] = function (test) {
|
||
|
var active = 0
|
||
|
var drained = false
|
||
|
var map = es.map(function (data, callback) {
|
||
|
active ++
|
||
|
u.delay(function () {
|
||
|
active --
|
||
|
callback(null, 1)
|
||
|
})()
|
||
|
console.log('WRITE', false)
|
||
|
return false
|
||
|
})
|
||
|
|
||
|
spec(map).through().validateOnExit()
|
||
|
|
||
|
map.on('drain', function () {
|
||
|
drained = true
|
||
|
it(active).equal(0, 'should emit drain when all maps are done')
|
||
|
})
|
||
|
|
||
|
it(map.write('hello')).equal(false)
|
||
|
it(map.write('hello')).equal(false)
|
||
|
it(map.write('hello')).equal(false)
|
||
|
|
||
|
process.nextTick(function () {map.end()},10)
|
||
|
|
||
|
map.on('end', function () {
|
||
|
console.log('end')
|
||
|
it(drained).ok('shoud have emitted drain before end')
|
||
|
test.done()
|
||
|
})
|
||
|
|
||
|
}
|
||
|
|
||
|
exports ['map applied to a stream with filtering'] = function (test) {
|
||
|
|
||
|
var input = [1,2,3,7,5,3,1,9,0,2,4,6]
|
||
|
|
||
|
var doubler = es.map(function (data, callback) {
|
||
|
if (data % 2)
|
||
|
callback(null, data * 2)
|
||
|
else
|
||
|
callback()
|
||
|
})
|
||
|
|
||
|
readStream(doubler, function (err, output) {
|
||
|
it(output).deepEqual(input.filter(function (j) {
|
||
|
return j % 2
|
||
|
}).map(function (j) {
|
||
|
return j * 2
|
||
|
}))
|
||
|
test.done()
|
||
|
})
|
||
|
|
||
|
spec(doubler).through().validateOnExit()
|
||
|
|
||
|
writeArray(input, doubler)
|
||
|
|
||
|
}
|
||
|
|
||
|
exports ['simple mapSync applied to a stream'] = function (test) {
|
||
|
|
||
|
var input = [1,2,3,7,5,3,1,9,0,2,4,6]
|
||
|
|
||
|
var doubler = es.mapSync(function (data) {
|
||
|
return data * 2
|
||
|
})
|
||
|
|
||
|
readStream(doubler, function (err, output) {
|
||
|
it(output).deepEqual(input.map(function (j) {
|
||
|
return j * 2
|
||
|
}))
|
||
|
test.done()
|
||
|
})
|
||
|
|
||
|
spec(doubler).through().validateOnExit()
|
||
|
|
||
|
writeArray(input, doubler)
|
||
|
|
||
|
}
|
||
|
|
||
|
exports ['mapSync applied to a stream with filtering'] = function (test) {
|
||
|
|
||
|
var input = [1,2,3,7,5,3,1,9,0,2,4,6]
|
||
|
|
||
|
var doubler = es.mapSync(function (data) {
|
||
|
if (data % 2)
|
||
|
return data * 2
|
||
|
})
|
||
|
|
||
|
readStream(doubler, function (err, output) {
|
||
|
it(output).deepEqual(input.filter(function (j) {
|
||
|
return j % 2
|
||
|
}).map(function (j) {
|
||
|
return j * 2
|
||
|
}))
|
||
|
test.done()
|
||
|
})
|
||
|
|
||
|
spec(doubler).through().validateOnExit()
|
||
|
|
||
|
writeArray(input, doubler)
|
||
|
|
||
|
}
|
||
|
|
||
|
require('./helper')(module)
|