1berry/nodejs
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: add mustcall in test-net-bytes-read.js

Browse Source 
* add mustcall() on createServerListener
* add mustcall() on listenPortListener
* add mustCall() on onConnectListener
* add mustCallAtLeast() on onDataListener

PR-URL: https://github.com/nodejs/node/pull/27471
Reviewed-By: Ouyang Yadong <oyydoibh@gmail.com>
Reviewed-By: Masashi Hirano <shisama07@gmail.com>
Reviewed-By: Rich Trott <rtrott@gmail.com>
This commit is contained in:
imhype 2019-04-29 11:08:53 +08:00 committed by Rich Trott
parent 64284aad38
commit aacc046e6b
1 changed files with 24 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
test/parallel/test-net-bytes-read.js
View File

@ -6,10 +6,12 @@ const net = require('net');
const big = Buffer.alloc(1024 * 1024);
const server = net.createServer((socket) => {
const handler = common.mustCall((socket) => {
socket.end(big);
server.close();
}).listen(0, () => {
});
const onListen = common.mustCall(() => {
let prev = 0;
function checkRaise(value) {
@ -17,22 +19,29 @@ const server = net.createServer((socket) => {
prev = value;
}
const socket = net.connect(server.address().port, () => {
socket.on('data', (chunk) => {
checkRaise(socket.bytesRead);
});
const onData = common.mustCallAtLeast((chunk) => {
checkRaise(socket.bytesRead);
});
socket.on('end', common.mustCall(() => {
assert.strictEqual(socket.bytesRead, prev);
assert.strictEqual(big.length, prev);
}));
const onEnd = common.mustCall(() => {
assert.strictEqual(socket.bytesRead, prev);
assert.strictEqual(big.length, prev);
});
socket.on('close', common.mustCall(() => {
assert(!socket._handle);
assert.strictEqual(socket.bytesRead, prev);
assert.strictEqual(big.length, prev);
}));
const onClose = common.mustCall(() => {
assert(!socket._handle);
assert.strictEqual(socket.bytesRead, prev);
assert.strictEqual(big.length, prev);
});
const onConnect = common.mustCall(() => {
socket.on('data', onData);
socket.on('end', onEnd);
socket.on('close', onClose);
socket.end();
});
const socket = net.connect(server.address().port, onConnect);
});
const server = net.createServer(handler).listen(0, onListen);