// Copyright Joyent, Inc. and other Node contributors. // // Permission is hereby granted, free of charge, to any person obtaining a // copy of this software and associated documentation files (the // "Software"), to deal in the Software without restriction, including // without limitation the rights to use, copy, modify, merge, publish, // distribute, sublicense, and/or sell copies of the Software, and to permit // persons to whom the Software is furnished to do so, subject to the // following conditions: // // The above copyright notice and this permission notice shall be included // in all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN // NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR // OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE // USE OR OTHER DEALINGS IN THE SOFTWARE. var common = require('../common.js'); var Readable = require('_stream_readable'); var Writable = require('_stream_writable'); var assert = require('assert'); // tiny node-tap lookalike. var tests = []; var count = 0; function test(name, fn) { count++; tests.push([name, fn]); } function run() { var next = tests.shift(); if (!next) return console.error('ok'); var name = next[0]; var fn = next[1]; console.log('# %s', name); fn({ same: assert.deepEqual, equal: assert.equal, end: function() { count--; run(); } }); } // ensure all tests have run process.on('exit', function() { assert.equal(count, 0); }); process.nextTick(run); function toArray(callback) { var stream = new Writable(); var list = []; stream.write = function(chunk) { list.push(chunk); }; stream.end = function() { callback(list); }; return stream; } function fromArray(list) { var r = new Readable(); r._read = noop; list.forEach(function(chunk) { r.push(chunk); }); r.push(null); return r; } function noop() {} test('can read objects from stream', function(t) { var r = fromArray([{ one: '1'}, { two: '2' }]); var v1 = r.read(); var v2 = r.read(); var v3 = r.read(); assert.deepEqual(v1, { one: '1' }); assert.deepEqual(v2, { two: '2' }); assert.deepEqual(v3, null); t.end(); }); test('can pipe objects into stream', function(t) { var r = fromArray([{ one: '1'}, { two: '2' }]); r.pipe(toArray(function(list) { assert.deepEqual(list, [ { one: '1' }, { two: '2' } ]); t.end(); })); }); test('read(n) is ignored', function(t) { var r = fromArray([{ one: '1'}, { two: '2' }]); var value = r.read(2); assert.deepEqual(value, { one: '1' }); t.end(); }); test('can read objects from _read (sync)', function(t) { var r = new Readable(); var list = [{ one: '1'}, { two: '2' }]; r._read = function(n, cb) { var item = list.shift(); cb(null, item || null); }; r.pipe(toArray(function(list) { assert.deepEqual(list, [ { one: '1' }, { two: '2' } ]); t.end(); })); }); test('can read objects from _read (async)', function(t) { var r = new Readable(); var list = [{ one: '1'}, { two: '2' }]; r._read = function(n, cb) { var item = list.shift(); process.nextTick(function() { cb(null, item || null); }); }; r.pipe(toArray(function(list) { assert.deepEqual(list, [ { one: '1' }, { two: '2' } ]); t.end(); })); }); test('can read strings as objects', function(t) { var r = new Readable({ objectMode: true }); r._read = noop; var list = ['one', 'two', 'three']; list.forEach(function(str) { r.push(str); }); r.push(null); r.pipe(toArray(function(array) { assert.deepEqual(array, list); t.end(); })); }); test('read(0) for object streams', function(t) { var r = new Readable({ objectMode: true }); r._read = noop; r.push('foobar'); r.push(null); var v = r.read(0); r.pipe(toArray(function(array) { assert.deepEqual(array, ['foobar']); t.end(); })); }); test('falsey values', function(t) { var r = new Readable({ objectMode: true }); r._read = noop; r.push(false); r.push(0); r.push(''); r.push(null); r.pipe(toArray(function(array) { assert.deepEqual(array, [false, 0, '']); t.end(); })); }); test('high watermark _read', function(t) { var r = new Readable({ highWaterMark: 6, objectMode: true }); var calls = 0; var list = ['1', '2', '3', '4', '5', '6', '7', '8']; r._read = function() { calls++; }; list.forEach(function(c) { r.push(c); }); var v = r.read(); assert.equal(calls, 0); assert.equal(v, '1'); var v2 = r.read(); assert.equal(calls, 1); assert.equal(v2, '2'); t.end(); }); test('high watermark push', function(t) { var r = new Readable({ highWaterMark: 6, objectMode: true }); r._read = function() {}; for (var i = 0; i < 6; i++) { var bool = r.push(i); assert.equal(bool, i === 5 ? false : true); } t.end(); }); test('stream of buffers converted to object halfway through', function(t) { var r = new Readable(); r._read = noop; r.push(new Buffer('fus')); r.push(new Buffer('do')); r.push(new Buffer('rah')); var str = r.read(4); assert.equal(str, 'fusd'); r.push({ foo: 'bar' }); r.push(null); r.pipe(toArray(function(list) { assert.deepEqual(list, [ new Buffer('o'), new Buffer('rah'), { foo: 'bar'} ]); t.end(); })); }); test('stream of strings converted to objects halfway through', function(t) { var r = new Readable({ encoding: 'utf8' }); r._read = noop; r.push('fus'); r.push('do'); r.push('rah'); var str = r.read(4); assert.equal(str, 'fusd'); r.push({ foo: 'bar' }); r.push(null); r.pipe(toArray(function(list) { assert.deepEqual(list, [ 'o', 'rah', { foo: 'bar'} ]); t.end(); })); }); test('can write objects to stream', function(t) { var w = new Writable(); w._write = function(chunk, cb) { assert.deepEqual(chunk, { foo: 'bar' }); cb(); }; w.on('finish', function() { t.end(); }); w.write({ foo: 'bar' }); w.end(); }); test('can write multiple objects to stream', function(t) { var w = new Writable(); var list = []; w._write = function(chunk, cb) { list.push(chunk); cb(); }; w.on('finish', function() { assert.deepEqual(list, [0, 1, 2, 3, 4]); t.end(); }); w.write(0); w.write(1); w.write(2); w.write(3); w.write(4); w.end(); }); test('can write strings as objects', function(t) { var w = new Writable({ objectMode: true }); var list = []; w._write = function(chunk, cb) { list.push(chunk); process.nextTick(cb); }; w.on('finish', function() { assert.deepEqual(list, ['0', '1', '2', '3', '4']); t.end(); }); w.write('0'); w.write('1'); w.write('2'); w.write('3'); w.write('4'); w.end(); }); test('buffers finish until cb is called', function(t) { var w = new Writable({ objectMode: true }); var called = false; w._write = function(chunk, cb) { assert.equal(chunk, 'foo'); process.nextTick(function() { called = true; cb(); }); }; w.on('finish', function() { assert.equal(called, true); t.end(); }); w.write('foo'); w.end(); });