1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
'use strict';
require('../common');
const assert = require('assert');
const zlib = require('zlib');
const bigData = Buffer.alloc(10240, 'x');
const opts = {
level: 0,
highWaterMark: 16
};
const deflater = zlib.createDeflate(opts);
// shim deflater.flush so we can count times executed
let flushCount = 0;
let drainCount = 0;
const flush = deflater.flush;
deflater.flush = function(kind, callback) {
flushCount++;
flush.call(this, kind, callback);
};
deflater.write(bigData);
const ws = deflater._writableState;
const beforeFlush = ws.needDrain;
let afterFlush = ws.needDrain;
deflater.flush(function(err) {
afterFlush = ws.needDrain;
});
deflater.on('drain', function() {
drainCount++;
});
process.once('exit', function() {
assert.equal(beforeFlush, true,
'before calling flush, writable stream should need to drain');
assert.equal(afterFlush, false,
'after calling flush, writable stream should not need to drain');
assert.equal(drainCount, 1,
'the deflater should have emitted a single drain event');
assert.equal(flushCount, 2,
'flush should be called twice');
});
|
