node/test/parallel/test-stream-writable-clear-buffer.js
Rich Trott 19e44efdfb test: add common.mustCall() to stream test
Refs: https://github.com/nodejs/node/pull/30561/files#r348667256

PR-URL: https://github.com/nodejs/node/pull/30561
Reviewed-By: Denys Otrishko <shishugi@gmail.com>
Reviewed-By: Anna Henningsen <anna@addaleax.net>
2019-11-22 11:41:32 -08:00

36 lines
876 B
JavaScript

'use strict';
// This test ensures that the _writeableState.bufferedRequestCount and
// the actual buffered request count are the same.
const common = require('../common');
const Stream = require('stream');
const assert = require('assert');
class StreamWritable extends Stream.Writable {
constructor() {
super({ objectMode: true });
}
// Refs: https://github.com/nodejs/node/issues/6758
// We need a timer like on the original issue thread.
// Otherwise the code will never reach our test case.
_write(chunk, encoding, cb) {
setImmediate(cb);
}
}
const testStream = new StreamWritable();
testStream.cork();
for (let i = 1; i <= 5; i++) {
testStream.write(i, common.mustCall(() => {
assert.strictEqual(
testStream._writableState.bufferedRequestCount,
testStream._writableState.getBuffer().length
);
}));
}
testStream.end();