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
|
/* eslint-disable mozilla/no-arbitrary-setTimeout */
/**
* This test ensures that http-on-modify-request is dispatched for channels that
* are blocked by tracking protection. It sets up a page with a third-party script
* resource on it that is blocked by TP, and sets up an http-on-modify-request
* observer which waits to be notified about that resource. The test would time out
* if the http-on-modify-request notification isn't dispatched before the channel is
* canceled.
*/
let gExpectedResourcesSeen = 0;
async function onModifyRequest() {
return new Promise((resolve, reject) => {
Services.obs.addObserver(function observer(subject, topic, data) {
let httpChannel = subject.QueryInterface(Ci.nsIHttpChannel);
let spec = httpChannel.URI.spec;
info("Observed channel for " + spec);
if (httpChannel.URI.prePath + "/" != TEST_3RD_PARTY_DOMAIN_TP) {
return;
}
if (spec.endsWith("empty.js")) {
ok(true, "Correct resource observed");
++gExpectedResourcesSeen;
} else if (spec.endsWith("empty.js?redirect")) {
httpChannel.redirectTo(
Services.io.newURI(spec.replace("empty.js?redirect", "head.js"))
);
} else if (spec.endsWith("empty.js?redirect2")) {
httpChannel.suspend();
setTimeout(() => {
httpChannel.redirectTo(
Services.io.newURI(spec.replace("empty.js?redirect2", "head.js"))
);
httpChannel.resume();
}, 100);
} else if (spec.endsWith("head.js")) {
++gExpectedResourcesSeen;
}
if (gExpectedResourcesSeen == 3) {
Services.obs.removeObserver(observer, "http-on-modify-request");
resolve();
}
}, "http-on-modify-request");
});
}
add_task(async function () {
info("Starting subResources test");
await SpecialPowers.flushPrefEnv();
await SpecialPowers.pushPrefEnv({
set: [
["privacy.trackingprotection.enabled", true],
// the test doesn't open a private window, so we don't care about this pref's value
["privacy.trackingprotection.pbmode.enabled", false],
// tracking annotations aren't needed in this test, only TP is needed
["privacy.trackingprotection.annotate_channels", false],
[
"privacy.restrict3rdpartystorage.userInteractionRequiredForHosts",
"tracking.example.com,tracking.example.org",
],
["privacy.trackingprotection.testing.report_blocked_node", true],
],
});
await UrlClassifierTestUtils.addTestTrackers();
let promise = onModifyRequest();
info("Creating a new tab");
let tab = BrowserTestUtils.addTab(gBrowser, TEST_EMBEDDER_PAGE);
gBrowser.selectedTab = tab;
let browser = gBrowser.getBrowserForTab(tab);
await BrowserTestUtils.browserLoaded(browser);
await promise;
info("Verify the number of tracking nodes found");
await SpecialPowers.spawn(
browser,
[{ expected: gExpectedResourcesSeen }],
async function (obj) {
is(
content.document.blockedNodeByClassifierCount,
obj.expected,
"Expected tracking nodes found"
);
}
);
info("Removing the tab");
BrowserTestUtils.removeTab(tab);
UrlClassifierTestUtils.cleanupTestTrackers();
});
|
