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
|
const { HttpServer } = ChromeUtils.importESModule(
"resource://testing-common/httpd.sys.mjs"
);
const { NetUtil } = ChromeUtils.importESModule(
"resource://gre/modules/NetUtil.sys.mjs"
);
// Since this test creates a TYPE_DOCUMENT channel via javascript, it will
// end up using the wrong LoadInfo constructor. Setting this pref will disable
// the ContentPolicyType assertion in the constructor.
Services.prefs.setBoolPref("network.loadinfo.skip_type_assertion", true);
ChromeUtils.defineLazyGetter(this, "URL", function () {
return "http://localhost:" + httpserver.identity.primaryPort;
});
var httpserver = null;
var channel = null;
var curTest = null;
var testpath = "/footpath";
var tests = [
{
description: "should not set request header for TYPE_OTHER",
expectingHeader: false,
contentType: Ci.nsIContentPolicy.TYPE_OTHER,
},
{
description: "should set request header for TYPE_DOCUMENT",
expectingHeader: true,
contentType: Ci.nsIContentPolicy.TYPE_DOCUMENT,
},
{
description: "should set request header for TYPE_SUBDOCUMENT",
expectingHeader: true,
contentType: Ci.nsIContentPolicy.TYPE_SUBDOCUMENT,
},
{
description: "should not set request header for TYPE_IMAGE",
expectingHeader: false,
contentType: Ci.nsIContentPolicy.TYPE_IMAGE,
},
];
function ChannelListener() {}
ChannelListener.prototype = {
onStartRequest(request) {},
onDataAvailable(request, stream, offset, count) {
do_throw("Should not get any data!");
},
onStopRequest(request, status) {
var upgrade_insecure_header = false;
try {
if (request.getRequestHeader("Upgrade-Insecure-Requests")) {
upgrade_insecure_header = true;
}
} catch (e) {
// exception is thrown if header is not available on the request
}
// debug
// dump("executing test: " + curTest.description);
Assert.equal(upgrade_insecure_header, curTest.expectingHeader);
run_next_test();
},
};
function setupChannel(aContentType) {
var chan = NetUtil.newChannel({
uri: URL + testpath,
loadUsingSystemPrincipal: true,
contentPolicyType: aContentType,
});
chan.QueryInterface(Ci.nsIHttpChannel);
chan.requestMethod = "GET";
return chan;
}
function serverHandler(metadata, response) {
// no need to perform anything here
}
function run_next_test() {
curTest = tests.shift();
if (!curTest) {
httpserver.stop(do_test_finished);
return;
}
channel = setupChannel(curTest.contentType);
channel.asyncOpen(new ChannelListener());
}
function run_test() {
do_get_profile();
// set up the test environment
httpserver = new HttpServer();
httpserver.registerPathHandler(testpath, serverHandler);
httpserver.start(-1);
run_next_test();
do_test_pending();
}
|
