firefox-main/toolkit/components/extensions/test/xpcshell/test_ext_urlclassifier.js (file symbol)

Source code

Revision control

Copy as Markdown

Other Tools

Test Info:

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
"use strict";
add_setup(async () => {
do_get_profile();
Services.fog.initializeFOG();
Services.prefs.setCharPref(
"urlclassifier.features.harmfuladdon.blocklistHosts",
"example.org"
);
Services.prefs.setCharPref(
"urlclassifier.features.harmfuladdon.entitylistHosts",
""
);
Services.prefs.setCharPref(
"urlclassifier.features.harmfuladdon.skipURLs",
""
);
registerCleanupFunction(() => {
Services.prefs.clearUserPref(
"urlclassifier.features.harmfuladdon.blocklistHosts"
);
Services.prefs.clearUserPref(
"urlclassifier.features.harmfuladdon.entitylistHosts"
);
Services.prefs.clearUserPref(
"urlclassifier.features.harmfuladdon.skipURLs"
);
});
});
const server = AddonTestUtils.createHttpServer({
hosts: ["example.com", "example.org"],
});
server.registerPathHandler("/dummy", (request, response) => {
response.setStatusLine(request.httpVersion, 200, "OK");
response.setHeader("Content-Type", "text/html", false);
response.write("<!DOCTYPE html><html></html>");
});
server.registerPathHandler("/contentScript", (request, response) => {
response.setStatusLine(request.httpVersion, 200, "OK");
response.setHeader("Content-Type", "text/html", false);
response.write("<h1>Content Script</h1>");
});
server.registerPathHandler("/backgroundScript", (request, response) => {
response.setStatusLine(request.httpVersion, 200, "OK");
response.setHeader("Content-Type", "text/html", false);
response.write("<h1>Background Script</h1>");
});
add_task(
{ pref_set: [["privacy.trackingprotection.harmfuladdon.enabled", true]] },
async function test_addon_blocked_by_url_classifier() {
await runTest("backgroundScript_failed", "contentScript_failed", true);
}
);
add_task(
{ pref_set: [["privacy.trackingprotection.harmfuladdon.enabled", false]] },
async function test_addon_blocked_by_url_classifier() {
await runTest("backgroundScript_loaded", "contentScript_loaded", false);
}
);
async function runTest(message1, message2, expectGleanEvent) {
Services.fog.testResetFOG();
const extension = ExtensionTestUtils.loadExtension({
manifest: {
host_permissions: ["http://example.org/"],
content_scripts: [
{
matches: ["http://example.com/*"],
run_at: "document_end",
js: ["contentscript.js"],
},
],
},
background: async () => {
try {
await fetch("http://example.org/backgroundScript").then(r => r.text());
browser.test.sendMessage("backgroundScript_loaded");
} catch (e) {
browser.test.sendMessage("backgroundScript_failed");
}
},
files: {
"contentscript.js": async () => {
try {
await fetch("http://example.org/contentScript").then(r => r.text());
browser.test.sendMessage("contentScript_loaded");
} catch (e) {
browser.test.sendMessage("contentScript_failed");
}
},
},
});
await extension.startup();
const contentPage = await ExtensionTestUtils.loadContentPage(
);
await extension.awaitMessage(message1);
await extension.awaitMessage(message2);
if (expectGleanEvent) {
const events = Glean.network.urlclassifierAddonBlock.testGetValue();
Assert.greater(events.length, 1, "We have received glean events");
let glean = events[0];
Assert.greater(glean.extra.addon_id.length, 0);
Assert.equal(glean.extra.table, "harmfuladdon-blocklist-pref");
Assert.equal(glean.extra.etld, "example.org");
glean = events[1];
Assert.greater(glean.extra.addon_id.length, 0);
Assert.equal(glean.extra.table, "harmfuladdon-blocklist-pref");
Assert.equal(glean.extra.etld, "example.org");
} else {
const events = Glean.network.urlclassifierAddonBlock.testGetValue();
Assert.equal(events, undefined, "We haven't received glean events");
}
await contentPage.close();
await extension.unload();
}