test: refactor test-stream-unshift-read-race
* add RegExp as second argument to assert.throws() * replace process.on('exit', ...) boolean checks with common.mustCall() * assert.equal() -> assert.strictEqual() * add 1 ms duration as second argument to setTimeout() * var -> const PR-URL: https://github.com/nodejs/node/pull/10532 Reviewed-By: Luigi Pinca <luigipinca@gmail.com> Reviewed-By: James M Snell <jasnell@gmail.com>
This commit is contained in:
parent
e5674e1c04
commit
55f530159f
@ -1,6 +1,6 @@
|
|||||||
'use strict';
|
'use strict';
|
||||||
require('../common');
|
const common = require('../common');
|
||||||
var assert = require('assert');
|
const assert = require('assert');
|
||||||
|
|
||||||
// This test verifies that:
|
// This test verifies that:
|
||||||
// 1. unshift() does not cause colliding _read() calls.
|
// 1. unshift() does not cause colliding _read() calls.
|
||||||
@ -9,19 +9,19 @@ var assert = require('assert');
|
|||||||
// 3. push() after the EOF signaling null is an error.
|
// 3. push() after the EOF signaling null is an error.
|
||||||
// 4. _read() is not called after pushing the EOF null chunk.
|
// 4. _read() is not called after pushing the EOF null chunk.
|
||||||
|
|
||||||
var stream = require('stream');
|
const stream = require('stream');
|
||||||
var hwm = 10;
|
const hwm = 10;
|
||||||
var r = stream.Readable({ highWaterMark: hwm });
|
const r = stream.Readable({ highWaterMark: hwm });
|
||||||
var chunks = 10;
|
const chunks = 10;
|
||||||
|
|
||||||
var data = Buffer.allocUnsafe(chunks * hwm + Math.ceil(hwm / 2));
|
const data = Buffer.allocUnsafe(chunks * hwm + Math.ceil(hwm / 2));
|
||||||
for (var i = 0; i < data.length; i++) {
|
for (let i = 0; i < data.length; i++) {
|
||||||
var c = 'asdf'.charCodeAt(i % 4);
|
const c = 'asdf'.charCodeAt(i % 4);
|
||||||
data[i] = c;
|
data[i] = c;
|
||||||
}
|
}
|
||||||
|
|
||||||
var pos = 0;
|
let pos = 0;
|
||||||
var pushedNull = false;
|
let pushedNull = false;
|
||||||
r._read = function(n) {
|
r._read = function(n) {
|
||||||
assert(!pushedNull, '_read after null push');
|
assert(!pushedNull, '_read after null push');
|
||||||
|
|
||||||
@ -30,7 +30,7 @@ r._read = function(n) {
|
|||||||
|
|
||||||
function push(fast) {
|
function push(fast) {
|
||||||
assert(!pushedNull, 'push() after null push');
|
assert(!pushedNull, 'push() after null push');
|
||||||
var c = pos >= data.length ? null : data.slice(pos, pos + n);
|
const c = pos >= data.length ? null : data.slice(pos, pos + n);
|
||||||
pushedNull = c === null;
|
pushedNull = c === null;
|
||||||
if (fast) {
|
if (fast) {
|
||||||
pos += n;
|
pos += n;
|
||||||
@ -41,7 +41,7 @@ r._read = function(n) {
|
|||||||
pos += n;
|
pos += n;
|
||||||
r.push(c);
|
r.push(c);
|
||||||
if (c === null) pushError();
|
if (c === null) pushError();
|
||||||
});
|
}, 1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@ -49,29 +49,26 @@ r._read = function(n) {
|
|||||||
function pushError() {
|
function pushError() {
|
||||||
assert.throws(function() {
|
assert.throws(function() {
|
||||||
r.push(Buffer.allocUnsafe(1));
|
r.push(Buffer.allocUnsafe(1));
|
||||||
});
|
}, /^Error: stream.push\(\) after EOF$/);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
var w = stream.Writable();
|
const w = stream.Writable();
|
||||||
var written = [];
|
const written = [];
|
||||||
w._write = function(chunk, encoding, cb) {
|
w._write = function(chunk, encoding, cb) {
|
||||||
written.push(chunk.toString());
|
written.push(chunk.toString());
|
||||||
cb();
|
cb();
|
||||||
};
|
};
|
||||||
|
|
||||||
var ended = false;
|
r.on('end', common.mustCall(function() {
|
||||||
r.on('end', function() {
|
|
||||||
assert(!ended, 'end emitted more than once');
|
|
||||||
assert.throws(function() {
|
assert.throws(function() {
|
||||||
r.unshift(Buffer.allocUnsafe(1));
|
r.unshift(Buffer.allocUnsafe(1));
|
||||||
});
|
}, /^Error: stream.unshift\(\) after end event$/);
|
||||||
ended = true;
|
|
||||||
w.end();
|
w.end();
|
||||||
});
|
}));
|
||||||
|
|
||||||
r.on('readable', function() {
|
r.on('readable', function() {
|
||||||
var chunk;
|
let chunk;
|
||||||
while (null !== (chunk = r.read(10))) {
|
while (null !== (chunk = r.read(10))) {
|
||||||
w.write(chunk);
|
w.write(chunk);
|
||||||
if (chunk.length > 4)
|
if (chunk.length > 4)
|
||||||
@ -79,21 +76,19 @@ r.on('readable', function() {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
var finished = false;
|
w.on('finish', common.mustCall(function() {
|
||||||
w.on('finish', function() {
|
|
||||||
finished = true;
|
|
||||||
// each chunk should start with 1234, and then be asfdasdfasdf...
|
// each chunk should start with 1234, and then be asfdasdfasdf...
|
||||||
// The first got pulled out before the first unshift('1234'), so it's
|
// The first got pulled out before the first unshift('1234'), so it's
|
||||||
// lacking that piece.
|
// lacking that piece.
|
||||||
assert.equal(written[0], 'asdfasdfas');
|
assert.strictEqual(written[0], 'asdfasdfas');
|
||||||
var asdf = 'd';
|
let asdf = 'd';
|
||||||
console.error('0: %s', written[0]);
|
console.error('0: %s', written[0]);
|
||||||
for (var i = 1; i < written.length; i++) {
|
for (let i = 1; i < written.length; i++) {
|
||||||
console.error('%s: %s', i.toString(32), written[i]);
|
console.error('%s: %s', i.toString(32), written[i]);
|
||||||
assert.equal(written[i].slice(0, 4), '1234');
|
assert.strictEqual(written[i].slice(0, 4), '1234');
|
||||||
for (var j = 4; j < written[i].length; j++) {
|
for (let j = 4; j < written[i].length; j++) {
|
||||||
var c = written[i].charAt(j);
|
const c = written[i].charAt(j);
|
||||||
assert.equal(c, asdf);
|
assert.strictEqual(c, asdf);
|
||||||
switch (asdf) {
|
switch (asdf) {
|
||||||
case 'a': asdf = 's'; break;
|
case 'a': asdf = 's'; break;
|
||||||
case 's': asdf = 'd'; break;
|
case 's': asdf = 'd'; break;
|
||||||
@ -102,11 +97,9 @@ w.on('finish', function() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
});
|
}));
|
||||||
|
|
||||||
process.on('exit', function() {
|
process.on('exit', function() {
|
||||||
assert.equal(written.length, 18);
|
assert.strictEqual(written.length, 18);
|
||||||
assert(ended, 'stream ended');
|
|
||||||
assert(finished, 'stream finished');
|
|
||||||
console.log('ok');
|
console.log('ok');
|
||||||
});
|
});
|
||||||
|
Loading…
x
Reference in New Issue
Block a user