blob: d8de7ba3f2877acb603e21c30e7dc56d2b36e0a7 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
'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();
});
});
|