mirror of
https://github.com/nodejs/node.git
synced 2024-11-29 23:16:30 +01:00
3d2aef3979
Use assert.strictEqual instead of assert.equal in tests, manually convert types where necessary. PR-URL: https://github.com/nodejs/node/pull/10698 Reviewed-By: James M Snell <jasnell@gmail.com> Reviewed-By: Colin Ihrig <cjihrig@gmail.com> Reviewed-By: Michael Dawson <michael_dawson@ca.ibm.com> Reviewed-By: Sakthipriyan Vairamani <thechargingvolcano@gmail.com> Reviewed-By: Michaël Zasso <targos@protonmail.com> Reviewed-By: Teddy Katz <teddy.katz@gmail.com>
40 lines
799 B
JavaScript
40 lines
799 B
JavaScript
'use strict';
|
|
require('../common');
|
|
const assert = require('assert');
|
|
const net = require('net');
|
|
const TCPWrap = process.binding('tcp_wrap').TCP;
|
|
|
|
const echoServer = net.createServer(function(conn) {
|
|
conn.end();
|
|
});
|
|
|
|
const ref = TCPWrap.prototype.ref;
|
|
const unref = TCPWrap.prototype.unref;
|
|
|
|
let refCount = 0;
|
|
|
|
TCPWrap.prototype.ref = function() {
|
|
ref.call(this);
|
|
refCount++;
|
|
assert.strictEqual(refCount, 0);
|
|
};
|
|
|
|
TCPWrap.prototype.unref = function() {
|
|
unref.call(this);
|
|
refCount--;
|
|
assert.strictEqual(refCount, -1);
|
|
};
|
|
|
|
echoServer.listen(0);
|
|
|
|
echoServer.on('listening', function() {
|
|
const sock = new net.Socket();
|
|
sock.unref();
|
|
sock.ref();
|
|
sock.connect(this.address().port);
|
|
sock.on('end', function() {
|
|
assert.strictEqual(refCount, 0);
|
|
echoServer.close();
|
|
});
|
|
});
|