mirror of
https://github.com/nodejs/node.git
synced 2024-12-01 16:10:02 +01:00
14f5a9becc
In non-buffer tests, change usage of the Buffer constructor to one of the recommended alternatives. PR-URL: https://github.com/nodejs/node/pull/13649 Reviewed-By: Refael Ackermann <refack@gmail.com> Reviewed-By: Colin Ihrig <cjihrig@gmail.com> Reviewed-By: Luigi Pinca <luigipinca@gmail.com> Reviewed-By: James M Snell <jasnell@gmail.com> Reviewed-By: Tobias Nießen <tniessen@tnie.de> Reviewed-By: Brian White <mscdex@mscdex.net>
92 lines
2.1 KiB
JavaScript
92 lines
2.1 KiB
JavaScript
'use strict';
|
|
require('../common');
|
|
const assert = require('assert');
|
|
const stream = require('stream');
|
|
const Writable = stream.Writable;
|
|
|
|
// Test the buffering behaviour of Writable streams.
|
|
//
|
|
// The call to cork() triggers storing chunks which are flushed
|
|
// on calling end() and the stream subsequently ended.
|
|
//
|
|
// node version target: 0.12
|
|
|
|
const expectedChunks = ['please', 'buffer', 'me', 'kindly'];
|
|
const inputChunks = expectedChunks.slice(0);
|
|
let seenChunks = [];
|
|
let seenEnd = false;
|
|
|
|
const w = new Writable();
|
|
// lets arrange to store the chunks
|
|
w._write = function(chunk, encoding, cb) {
|
|
// stream end event is not seen before the last write
|
|
assert.ok(!seenEnd);
|
|
// default encoding given none was specified
|
|
assert.strictEqual(encoding, 'buffer');
|
|
|
|
seenChunks.push(chunk);
|
|
cb();
|
|
};
|
|
// lets record the stream end event
|
|
w.on('finish', () => {
|
|
seenEnd = true;
|
|
});
|
|
|
|
function writeChunks(remainingChunks, callback) {
|
|
const writeChunk = remainingChunks.shift();
|
|
let writeState;
|
|
|
|
if (writeChunk) {
|
|
setImmediate(() => {
|
|
writeState = w.write(writeChunk);
|
|
// we were not told to stop writing
|
|
assert.ok(writeState);
|
|
|
|
writeChunks(remainingChunks, callback);
|
|
});
|
|
} else {
|
|
callback();
|
|
}
|
|
}
|
|
|
|
// do an initial write
|
|
w.write('stuff');
|
|
// the write was immediate
|
|
assert.strictEqual(seenChunks.length, 1);
|
|
// reset the seen chunks
|
|
seenChunks = [];
|
|
|
|
// trigger stream buffering
|
|
w.cork();
|
|
|
|
// write the bufferedChunks
|
|
writeChunks(inputChunks, () => {
|
|
// should not have seen anything yet
|
|
assert.strictEqual(seenChunks.length, 0);
|
|
|
|
// trigger flush and ending the stream
|
|
w.end();
|
|
|
|
// stream should not ended in current tick
|
|
assert.ok(!seenEnd);
|
|
|
|
// buffered bytes should be seen in current tick
|
|
assert.strictEqual(seenChunks.length, 4);
|
|
|
|
// did the chunks match
|
|
for (let i = 0, l = expectedChunks.length; i < l; i++) {
|
|
const seen = seenChunks[i];
|
|
// there was a chunk
|
|
assert.ok(seen);
|
|
|
|
const expected = Buffer.from(expectedChunks[i]);
|
|
// it was what we expected
|
|
assert.ok(seen.equals(expected));
|
|
}
|
|
|
|
setImmediate(() => {
|
|
// stream should have ended in next tick
|
|
assert.ok(seenEnd);
|
|
});
|
|
});
|