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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
/* Any copyright is dedicated to the Public Domain.
http://creativecommons.org/publicdomain/zero/1.0/ */
"use strict";
let { HPKEConfigManager } = ChromeUtils.importESModule(
"resource://gre/modules/HPKEConfigManager.sys.mjs"
);
const { HttpServer } = ChromeUtils.importESModule(
"resource://testing-common/httpd.sys.mjs"
);
let gHttpServer;
let gValidRequestCount = 0;
let gFickleIsWorking = true;
add_setup(async function () {
gHttpServer = new HttpServer();
let invalidHandler = (req, res) => {
res.setStatusLine(req.httpVersion, 500, "Oh no, it broke");
res.write("Uh oh, it broke.");
};
let validHandler = (req, res) => {
res.setHeader("Content-Type", "application/ohttp-keys");
res.write("1234");
gValidRequestCount++;
};
gHttpServer.registerPathHandler("/.wellknown/invalid", invalidHandler);
gHttpServer.registerPathHandler("/.wellknown/valid", validHandler);
gHttpServer.registerPathHandler("/.wellknown/fickle", (req, res) => {
if (gFickleIsWorking) {
return validHandler(req, res);
}
return invalidHandler(req, res);
});
gHttpServer.start(-1);
});
function getLocalURL(path) {
return `http://localhost:${gHttpServer.identity.primaryPort}/.wellknown/${path}`;
}
add_task(async function test_broken_url_returns_null() {
Assert.equal(await HPKEConfigManager.get(getLocalURL("invalid")), null);
});
add_task(async function test_working_url_returns_data() {
Assert.deepEqual(
await HPKEConfigManager.get(getLocalURL("valid")),
new TextEncoder().encode("1234")
);
});
add_task(async function test_we_only_request_once() {
Assert.deepEqual(
await HPKEConfigManager.get(getLocalURL("valid")),
new TextEncoder().encode("1234")
);
let oldRequestCount = gValidRequestCount;
Assert.deepEqual(
await HPKEConfigManager.get(getLocalURL("valid")),
new TextEncoder().encode("1234")
);
Assert.equal(
oldRequestCount,
gValidRequestCount,
"Shouldn't have made another request."
);
});
add_task(async function test_maxAge_forces_refresh() {
Assert.deepEqual(
await HPKEConfigManager.get(getLocalURL("valid")),
new TextEncoder().encode("1234")
);
let oldRequestCount = gValidRequestCount;
Assert.deepEqual(
await HPKEConfigManager.get(getLocalURL("valid"), { maxAge: 0 }),
new TextEncoder().encode("1234")
);
Assert.equal(
oldRequestCount + 1,
gValidRequestCount,
"Should have made another request due to maxAge."
);
});
add_task(async function test_maxAge_handling_of_invalid_requests() {
Assert.deepEqual(
await HPKEConfigManager.get(getLocalURL("fickle")),
new TextEncoder().encode("1234")
);
gFickleIsWorking = false;
Assert.deepEqual(
await HPKEConfigManager.get(getLocalURL("fickle"), { maxAge: 0 }),
null
);
Assert.deepEqual(
await HPKEConfigManager.get(getLocalURL("fickle")),
new TextEncoder().encode("1234"),
"Should still have the cached config if no max age is passed."
);
});
|
