node/test/parallel/test-net-dns-custom-lookup.js

'use strict';
var common = require('../common');
var assert = require('assert');
var net = require('net');
var ok = false;

function check(addressType, cb) {
  var server = net.createServer(function(client) {
    client.end();
    server.close();
    cb && cb();
  });

  var address = addressType === 4 ? common.localhostIPv4 : '::1';
  server.listen(common.PORT, address, function() {
    net.connect({
      port: common.PORT,
      host: 'localhost',
      family: addressType,
      lookup: lookup
    }).on('lookup', function(err, ip, type) {
      assert.equal(err, null);
      assert.equal(address, ip);
      assert.equal(type, addressType);
      ok = true;
    });
  });

  function lookup(host, dnsopts, cb) {
    dnsopts.family = addressType;
    if (addressType === 4) {
      process.nextTick(function() {
        cb(null, common.localhostIPv4, 4);
      });
    } else {
      process.nextTick(function() {
        cb(null, '::1', 6);
      });
    }
  }
}

check(4, function() {
  common.hasIPv6 && check(6);
});

process.on('exit', function() {
  assert.ok(ok);
});