60 lines
1.8 KiB
JavaScript
60 lines
1.8 KiB
JavaScript
|
'use strict';
|
||
|
|
require('../common');
|
||
|
|
const assert = require('assert');
|
||
|
|
const { Writable } = require('stream');
|
||
|
|
|
||
|
|
// Test interaction between calling .destroy() on a writable and pending
|
||
|
|
// writes.
|
||
|
|
|
||
|
|
for (const withPendingData of [ false, true ]) {
|
||
|
|
for (const useEnd of [ false, true ]) {
|
||
|
|
const callbacks = [];
|
||
|
|
|
||
|
|
const w = new Writable({
|
||
|
|
write(data, enc, cb) {
|
||
|
|
callbacks.push(cb);
|
||
|
|
},
|
||
|
|
// Effectively disable the HWM to observe 'drain' events more easily.
|
||
|
|
highWaterMark: 1
|
||
|
|
});
|
||
|
|
|
||
|
|
let chunksWritten = 0;
|
||
|
|
let drains = 0;
|
||
|
|
let finished = false;
|
||
|
|
w.on('drain', () => drains++);
|
||
|
|
w.on('finish', () => finished = true);
|
||
|
|
|
||
|
|
w.write('abc', () => chunksWritten++);
|
||
|
|
assert.strictEqual(chunksWritten, 0);
|
||
|
|
assert.strictEqual(drains, 0);
|
||
|
|
callbacks.shift()();
|
||
|
|
assert.strictEqual(chunksWritten, 1);
|
||
|
|
assert.strictEqual(drains, 1);
|
||
|
|
|
||
|
|
if (withPendingData) {
|
||
|
|
// Test 2 cases: There either is or is not data still in the write queue.
|
||
|
|
// (The second write will never actually get executed either way.)
|
||
|
|
w.write('def', () => chunksWritten++);
|
||
|
|
}
|
||
|
|
if (useEnd) {
|
||
|
|
// Again, test 2 cases: Either we indicate that we want to end the
|
||
|
|
// writable or not.
|
||
|
|
w.end('ghi', () => chunksWritten++);
|
||
|
|
} else {
|
||
|
|
w.write('ghi', () => chunksWritten++);
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.strictEqual(chunksWritten, 1);
|
||
|
|
w.destroy();
|
||
|
|
assert.strictEqual(chunksWritten, 1);
|
||
|
|
callbacks.shift()();
|
||
|
|
assert.strictEqual(chunksWritten, 2);
|
||
|
|
assert.strictEqual(callbacks.length, 0);
|
||
|
|
assert.strictEqual(drains, 1);
|
||
|
|
|
||
|
|
// When we used `.end()`, we see the 'finished' event if and only if
|
||
|
|
// we actually finished processing the write queue.
|
||
|
|
assert.strictEqual(finished, !withPendingData && useEnd);
|
||
|
|
}
|
||
|
|
}
|