Skip to content

Commit

Permalink
test: add test-zlib-flush-drain
Browse files Browse the repository at this point in the history
This test assures that if flush is called while the zlib object needs
to be drained that it will defer the callback until after the drain.

PR-URL: #3534
Reviewed-By: Ben Noordhuis <[email protected]>
Reviewed-By: James M Snell <[email protected]>
  • Loading branch information
Myles Borins authored and Fishrock123 committed Nov 17, 2015
1 parent 33ffc62 commit 2ec5e17
Showing 1 changed file with 49 additions and 0 deletions.
49 changes: 49 additions & 0 deletions test/parallel/test-zlib-flush-drain.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
'use strict';
const common = require('../common');
const assert = require('assert');
const zlib = require('zlib');
const path = require('path');

const bigData = new Buffer(10240).fill('x');

const opts = {
level: 0,
highWaterMark: 16
};

const deflater = zlib.createDeflate(opts);

// shim deflater.flush so we can count times executed
var flushCount = 0;
var 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;
var 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 the writable stream should need to drain');
assert.equal(afterFlush, false,
'after calling flush the 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');
});

0 comments on commit 2ec5e17

Please sign in to comment.