summaryrefslogtreecommitdiff
path: root/test/parallel/test-net-persistent-ref-unref.js
blob: b3ea0969f39961e5231d59288824d6c5003193cf (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';
var common = require('../common');
var assert = require('assert');
var net = require('net');
var TCPWrap = process.binding('tcp_wrap').TCP;

var echoServer = net.createServer(function(conn) {
  conn.end();
});

var ref = TCPWrap.prototype.ref;
var unref = TCPWrap.prototype.unref;

var refCount = 0;

TCPWrap.prototype.ref = function() {
  ref.call(this);
  refCount++;
  assert.equal(refCount, 0);
};

TCPWrap.prototype.unref = function() {
  unref.call(this);
  refCount--;
  assert.equal(refCount, -1);
};

echoServer.listen(common.PORT);

echoServer.on('listening', function() {
  var sock = new net.Socket();
  sock.unref();
  sock.ref();
  sock.connect(common.PORT);
  sock.on('end', function() {
    assert.equal(refCount, 0);
    echoServer.close();
  });
});