forked from nodejs/node
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test-dns-channel-cancel.js
56 lines (45 loc) · 1.48 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
52
53
54
55
56
'use strict';
const common = require('../common');
const dnstools = require('../common/dns');
const { Resolver } = require('dns');
const assert = require('assert');
const dgram = require('dgram');
const server = dgram.createSocket('udp4');
const resolver = new Resolver();
const expectedDomains = [];
const receivedDomains = [];
const desiredQueries = 11;
let finishedQueries = 0;
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) {
assert.deepStrictEqual(expectedDomains.sort(), receivedDomains.sort());
server.close();
}
}, desiredQueries);
const next = (...args) => {
callback(...args);
// Multiple queries
for (let i = 1; i < desiredQueries; i++) {
const domain = `example${i}.org`;
expectedDomains.push(domain);
resolver.resolve4(domain, callback);
}
};
// Single query
expectedDomains.push('example0.org');
resolver.resolve4('example0.org', next);
}));
server.on('message', (msg, { address, port }) => {
const parsed = dnstools.parseDNSPacket(msg);
for (const question of parsed.questions) {
receivedDomains.push(question.domain);
}
// Do not send a reply.
resolver.cancel();
});