summaryrefslogtreecommitdiffstats
path: root/browser/components/doh/test/unit/test_DNSLookup.js
blob: 5951445f13caf322be431fc6fbb659eda10b0623 (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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
/* Any copyright is dedicated to the Public Domain.
 * http://creativecommons.org/publicdomain/zero/1.0/
 */

"use strict";

add_task(setup);

add_task(async function test_SuccessfulRandomDNSLookup() {
  let deferred = PromiseUtils.defer();
  let lookup = new DNSLookup(
    null,
    trrServer1,
    (request, record, status, usedDomain, retryCount) => {
      deferred.resolve({ request, record, status, usedDomain, retryCount });
    }
  );
  lookup.doLookup();
  let result = await deferred.promise;
  Assert.ok(result.usedDomain.endsWith(".firefox-dns-perf-test.net."));
  Assert.equal(result.status, Cr.NS_OK);
  Assert.ok(result.record.QueryInterface(Ci.nsIDNSAddrRecord));
  Assert.ok(result.record.IsTRR());
  Assert.greater(result.record.trrFetchDuration, 0);
  Assert.equal(result.retryCount, 1);
});

add_task(async function test_SuccessfulSpecifiedDNSLookup() {
  let deferred = PromiseUtils.defer();
  let lookup = new DNSLookup(
    "foo.example.com",
    trrServer1,
    (request, record, status, usedDomain, retryCount) => {
      deferred.resolve({ request, record, status, usedDomain, retryCount });
    }
  );
  lookup.doLookup();
  let result = await deferred.promise;
  Assert.equal(result.usedDomain, "foo.example.com");
  Assert.equal(result.status, Cr.NS_OK);
  Assert.ok(result.record.QueryInterface(Ci.nsIDNSAddrRecord));
  Assert.ok(result.record.IsTRR());
  Assert.greater(result.record.trrFetchDuration, 0);
  Assert.equal(result.retryCount, 1);
});

add_task(async function test_FailedDNSLookup() {
  let deferred = PromiseUtils.defer();
  let lookup = new DNSLookup(
    null,
    `https://foo.example.com:${h2Port}/doh?responseIP=none`,
    (request, record, status, usedDomain, retryCount) => {
      deferred.resolve({ request, record, status, usedDomain, retryCount });
    }
  );
  lookup.doLookup();
  let result = await deferred.promise;
  Assert.ok(result.usedDomain.endsWith(".firefox-dns-perf-test.net."));
  Assert.notEqual(result.status, Cr.NS_OK);
  Assert.equal(result.record, null);
  Assert.equal(result.retryCount, 3);
});