Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
- Manifest: netwerk/test/unit/xpcshell.toml
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
"use strict";
trr_test_setup();
registerCleanupFunction(async () => {
trr_clear_prefs();
});
let trrServer = null;
add_task(async function start_trr_server() {
trrServer = new TRRServer();
registerCleanupFunction(async () => {
await trrServer.stop();
});
await trrServer.start();
dump(`port = ${trrServer.port()}\n`);
Services.prefs.setCharPref(
"network.trr.uri",
);
Services.prefs.setIntPref("network.trr.mode", Ci.nsIDNSService.MODE_TRRFIRST);
});
add_task(async function sanity_check() {
await trrServer.registerDoHAnswers("example.com", "A", {
answers: [
{
name: "example.com",
ttl: 55,
type: "A",
flush: false,
data: "1.2.3.4",
},
],
});
// Simple check to see that TRR works.
await new TRRDNSListener("example.com", { expectedAnswer: "1.2.3.4" });
});
// Cancelling the request is not sync when using the socket process, so
// we skip this test when it's enabled.
add_task(
{ skip_if: () => mozinfo.socketprocess_networking },
async function cancel_immediately() {
await trrServer.registerDoHAnswers("example.org", "A", {
answers: [
{
name: "example.org",
ttl: 55,
type: "A",
flush: false,
data: "2.3.4.5",
},
],
});
let r1 = new TRRDNSListener("example.org", { expectedSuccess: false });
let r2 = new TRRDNSListener("example.org", { expectedAnswer: "2.3.4.5" });
r1.cancel();
let { inStatus } = await r1;
equal(inStatus, Cr.NS_ERROR_ABORT);
await r2;
equal(await trrServer.requestCount("example.org", "A"), 1);
// Now we cancel both of them
Services.dns.clearCache(true);
r1 = new TRRDNSListener("example.org", { expectedSuccess: false });
r2 = new TRRDNSListener("example.org", { expectedSuccess: false });
r1.cancel();
r2.cancel();
({ inStatus } = await r1);
equal(inStatus, Cr.NS_ERROR_ABORT);
({ inStatus } = await r2);
equal(inStatus, Cr.NS_ERROR_ABORT);
await new Promise(resolve => do_timeout(50, resolve));
equal(await trrServer.requestCount("example.org", "A"), 2);
}
);
add_task(async function cancel_delayed() {
Services.dns.clearCache(true);
await trrServer.registerDoHAnswers("example.com", "A", {
answers: [
{
name: "example.com",
ttl: 55,
type: "A",
flush: false,
data: "1.1.1.1",
},
],
delay: 500,
});
let r1 = new TRRDNSListener("example.com", { expectedSuccess: false });
let r2 = new TRRDNSListener("example.com", { expectedAnswer: "1.1.1.1" });
await new Promise(resolve => do_timeout(50, resolve));
r1.cancel();
let { inStatus } = await r1;
equal(inStatus, Cr.NS_ERROR_ABORT);
await r2;
});
add_task(async function cancel_after_completed() {
Services.dns.clearCache(true);
await trrServer.registerDoHAnswers("example.com", "A", {
answers: [
{
name: "example.com",
ttl: 55,
type: "A",
flush: false,
data: "2.2.2.2",
},
],
});
let r1 = new TRRDNSListener("example.com", { expectedAnswer: "2.2.2.2" });
await r1;
let r2 = new TRRDNSListener("example.com", { expectedAnswer: "2.2.2.2" });
// Check that cancelling r1 after it's complete does not affect r2 in any way.
r1.cancel();
await r2;
});
add_task(async function clearCacheWhileResolving() {
Services.dns.clearCache(true);
await trrServer.registerDoHAnswers("example.com", "A", {
answers: [
{
name: "example.com",
ttl: 55,
type: "A",
flush: false,
data: "3.3.3.3",
},
],
delay: 500,
});
// Check that calling clearCache does not leave the request hanging.
let r1 = new TRRDNSListener("example.com", { expectedAnswer: "3.3.3.3" });
let r2 = new TRRDNSListener("example.com", { expectedAnswer: "3.3.3.3" });
Services.dns.clearCache(true);
await r1;
await r2;
// Also check the same for HTTPS records
await trrServer.registerDoHAnswers("httpsvc.com", "HTTPS", {
answers: [
{
name: "httpsvc.com",
ttl: 55,
type: "HTTPS",
flush: false,
data: {
priority: 1,
name: "test.p1.com",
values: [{ key: "alpn", value: ["h2", "h3"] }],
},
},
],
delay: 500,
});
let r3 = new TRRDNSListener("httpsvc.com", {
type: Ci.nsIDNSService.RESOLVE_TYPE_HTTPSSVC,
});
let r4 = new TRRDNSListener("httpsvc.com", {
type: Ci.nsIDNSService.RESOLVE_TYPE_HTTPSSVC,
});
Services.dns.clearCache(true);
await r3;
await r4;
equal(await trrServer.requestCount("httpsvc.com", "HTTPS"), 1);
Services.dns.clearCache(true);
await new TRRDNSListener("httpsvc.com", {
type: Ci.nsIDNSService.RESOLVE_TYPE_HTTPSSVC,
});
equal(await trrServer.requestCount("httpsvc.com", "HTTPS"), 2);
});