fune/toolkit/components/antitracking/test/browser/browser_onBeforeRequestNotificationForTrackingResources.js

91 lines
3.4 KiB
JavaScript

ChromeUtils.import("resource://gre/modules/Services.jsm");
/**
* This test ensures that onBeforeRequest is dispatched for webRequest loads 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 onBeforeRequest listener
* which waits to be notified about that resource. The test would time out if the
* onBeforeRequest listener isn't called dispatched before the load is canceled.
*/
let extension;
add_task(async function() {
extension = ExtensionTestUtils.loadExtension({
manifest: {permissions: ["webRequest", "webRequestBlocking", "*://*/*"]},
async background() {
let gExpectedResourcesSeen = 0;
function onBeforeRequest(details) {
let spec = details.url;
browser.test.log("Observed channel for " + spec);
// We would use TEST_3RD_PARTY_DOMAIN_TP here, but the variable is inaccessible
// since it is defined in head.js!
if (!spec.startsWith("https://tracking.example.com/")) {
return undefined;
}
if (spec.endsWith("empty.js")) {
browser.test.succeed("Correct resource observed");
++gExpectedResourcesSeen;
} else if (spec.endsWith("empty.js?redirect")) {
return {redirectUrl: spec.replace("empty.js?redirect", "head.js")};
} else if (spec.endsWith("head.js")) {
++gExpectedResourcesSeen;
}
if (gExpectedResourcesSeen == 2) {
browser.webRequest.onBeforeRequest.removeListener(onBeforeRequest);
browser.test.sendMessage("finish");
}
return undefined;
}
browser.webRequest.onBeforeRequest.addListener(onBeforeRequest,
{urls: ["*://*/*"]},
["blocking"]);
browser.test.sendMessage("ready");
},
});
await extension.startup();
await extension.awaitMessage("ready");
});
add_task(async function() {
info("Starting subResources test");
await SpecialPowers.flushPrefEnv();
await SpecialPowers.pushPrefEnv({"set": [
["browser.contentblocking.enabled", true],
["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],
// prevent the content blocking on-boarding UI to start mid-way through the test!
["privacy.trackingprotection.introCount", ContentBlocking.MAX_INTROS],
]});
await UrlClassifierTestUtils.addTestTrackers();
let promise = extension.awaitMessage("finish");
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 ContentTask.spawn(browser,
{ expected: 3,
},
async function(obj) {
is(content.document.blockedTrackingNodeCount, obj.expected, "Expected tracking nodes found");
});
info("Removing the tab");
BrowserTestUtils.removeTab(tab);
UrlClassifierTestUtils.cleanupTestTrackers();
await extension.unload();
});