test: writable stream needDrain state

Add a test for _writableState.needDrain. PR-URL: #8799 Reviewed-By: Colin Ihrig <[email protected]> Reviewed-By: Matteo Collina <[email protected]> Reviewed-By: Anna Henningsen <[email protected]> Reviewed-By: James M Snell <[email protected]> Related: #8686
nodejs · Nov 19, 2016 · 1fee6c1 · 1fee6c1
1 parent b3f8f89
commit 1fee6c1
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/test/parallel/test-stream-writable-needdrain-state.js b/test/parallel/test-stream-writable-needdrain-state.js
@@ -0,0 +1,23 @@
+'use strict';
+
+const common = require('../common');
+const stream = require('stream');
+const assert = require('assert');
+
+const transform = new stream.Transform({
+  transform: _transform,
+  highWaterMark: 1
+});
+
+function _transform(chunk, encoding, cb) {
+  assert.strictEqual(transform._writableState.needDrain, true);
+  cb();
+}
+
+assert.strictEqual(transform._writableState.needDrain, false);
+
+transform.write('asdasd', common.mustCall(() => {
+  assert.strictEqual(transform._writableState.needDrain, false);
+}));
+
+assert.strictEqual(transform._writableState.needDrain, true);