/
test-dns-channel-cancel.js
51 lines (39 loc) · 1.2 KB
/
test-dns-channel-cancel.js
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
40
41
42
43
44
45
46
47
48
49
50
51
'use strict';
const common = require('../common');
const { Resolver } = require('dns');
const assert = require('assert');
const dgram = require('dgram');
const server = dgram.createSocket('udp4');
const resolver = new Resolver();
const desiredQueries = 11;
let finishedQueries = 0;
const addMessageListener = () => {
server.removeAllListeners('message');
server.once('message', () => {
server.once('message', common.mustNotCall);
resolver.cancel();
});
};
server.bind(0, common.mustCall(async () => {
resolver.setServers([`127.0.0.1:${server.address().port}`]);
const callback = common.mustCall((err, res) => {
assert.strictEqual(err.code, 'ECANCELLED');
assert.strictEqual(err.syscall, 'queryA');
assert.strictEqual(err.hostname, `example${finishedQueries}.org`);
finishedQueries++;
if (finishedQueries === desiredQueries) {
server.close();
}
}, desiredQueries);
const next = (...args) => {
callback(...args);
addMessageListener();
// Multiple queries
for (let i = 1; i < desiredQueries; i++) {
resolver.resolve4(`example${i}.org`, callback);
}
};
// Single query
addMessageListener();
resolver.resolve4('example0.org', next);
}));