mirror of
https://github.com/mozilla/gecko-dev.git
synced 2025-11-11 05:39:41 +02:00
5f24a04aa2
Differential Revision: https://phabricator.services.mozilla.com/D29446 --HG-- extra : moz-landing-system : lando
151 lines
5.2 KiB
JavaScript
151 lines
5.2 KiB
JavaScript
"use strict";
|
|
|
|
/* eslint no-unused-vars: ["error", {"args": "none", "varsIgnorePattern": "^(FindProxyForURL)$"}] */
|
|
|
|
const server = createHttpServer({hosts: ["example.com"]});
|
|
|
|
server.registerPathHandler("/dummy", (request, response) => {
|
|
response.setStatusLine(request.httpVersion, 200, "OK");
|
|
response.setHeader("Content-Type", "text/html", false);
|
|
response.write("<!DOCTYPE html><html></html>");
|
|
});
|
|
|
|
|
|
add_task(async function test_incognito_proxy_onRequest_access() {
|
|
// No specific support exists in the proxy api for this test,
|
|
// rather it depends on functionality existing in ChannelWrapper
|
|
// that prevents notification of private channels if the
|
|
// extension does not have permission.
|
|
Services.prefs.setBoolPref("extensions.allowPrivateBrowsingByDefault", false);
|
|
|
|
// This extension will fail if it gets a private request.
|
|
let extension = ExtensionTestUtils.loadExtension({
|
|
manifest: {
|
|
permissions: ["proxy", "<all_urls>"],
|
|
},
|
|
async background() {
|
|
browser.proxy.onRequest.addListener(async (details) => {
|
|
browser.test.assertFalse(details.incognito, "incognito flag is not set");
|
|
browser.test.notifyPass("proxy.onRequest");
|
|
}, {urls: ["<all_urls>"], types: ["main_frame"]});
|
|
|
|
// Actual call arguments do not matter here.
|
|
await browser.test.assertRejects(
|
|
browser.proxy.settings.set({value: {
|
|
proxyType: "none",
|
|
}}),
|
|
/proxy.settings requires private browsing permission/,
|
|
"proxy.settings requires private browsing permission.");
|
|
|
|
browser.test.sendMessage("ready");
|
|
},
|
|
});
|
|
await extension.startup();
|
|
await extension.awaitMessage("ready");
|
|
|
|
let pextension = ExtensionTestUtils.loadExtension({
|
|
incognitoOverride: "spanning",
|
|
manifest: {
|
|
permissions: ["proxy", "<all_urls>"],
|
|
},
|
|
background() {
|
|
browser.proxy.onRequest.addListener(async (details) => {
|
|
browser.test.assertTrue(details.incognito, "incognito flag is set with filter");
|
|
browser.test.sendMessage("proxy.onRequest.private");
|
|
}, {urls: ["<all_urls>"], types: ["main_frame"], incognito: true});
|
|
|
|
browser.proxy.onRequest.addListener(async (details) => {
|
|
browser.test.assertFalse(details.incognito, "incognito flag is not set with filter");
|
|
browser.test.notifyPass("proxy.onRequest.spanning");
|
|
}, {urls: ["<all_urls>"], types: ["main_frame"], incognito: false});
|
|
},
|
|
});
|
|
await pextension.startup();
|
|
|
|
let contentPage = await ExtensionTestUtils.loadContentPage("http://example.com/dummy", {privateBrowsing: true});
|
|
await pextension.awaitMessage("proxy.onRequest.private");
|
|
await contentPage.close();
|
|
|
|
contentPage = await ExtensionTestUtils.loadContentPage("http://example.com/dummy");
|
|
await extension.awaitFinish("proxy.onRequest");
|
|
await pextension.awaitFinish("proxy.onRequest.spanning");
|
|
await contentPage.close();
|
|
|
|
await pextension.unload();
|
|
await extension.unload();
|
|
|
|
Services.prefs.clearUserPref("extensions.allowPrivateBrowsingByDefault");
|
|
});
|
|
|
|
function scriptData(script) {
|
|
return String(script).replace(/^.*?\{([^]*)\}$/, "$1");
|
|
}
|
|
|
|
add_task(async function test_incognito_proxy_register_access() {
|
|
Services.prefs.setBoolPref("extensions.allowPrivateBrowsingByDefault", false);
|
|
|
|
// This extension will fail if it gets a request
|
|
let extension = ExtensionTestUtils.loadExtension({
|
|
manifest: {
|
|
permissions: ["proxy"],
|
|
},
|
|
async background() {
|
|
browser.runtime.onMessage.addListener((message, sender) => {
|
|
if (sender.url === browser.extension.getURL("proxy.js")) {
|
|
browser.test.fail(message);
|
|
}
|
|
});
|
|
await browser.proxy.register("proxy.js");
|
|
browser.test.sendMessage("ready");
|
|
},
|
|
files: {
|
|
"proxy.js": scriptData(() => {
|
|
function FindProxyForURL() {
|
|
// Shoot a message off to the background.
|
|
browser.runtime.sendMessage("incognito fail");
|
|
return null;
|
|
}
|
|
}),
|
|
},
|
|
});
|
|
await extension.startup();
|
|
await extension.awaitMessage("ready");
|
|
|
|
// This extension will succeed if it gets a request
|
|
let pb_extension = ExtensionTestUtils.loadExtension({
|
|
incognitoOverride: "spanning",
|
|
manifest: {
|
|
permissions: ["proxy"],
|
|
},
|
|
async background() {
|
|
browser.runtime.onMessage.addListener((message, sender) => {
|
|
if (sender.url === browser.extension.getURL("proxy.js")) {
|
|
browser.test.notifyPass(message);
|
|
}
|
|
});
|
|
await browser.proxy.register("proxy.js");
|
|
browser.test.sendMessage("ready");
|
|
},
|
|
files: {
|
|
"proxy.js": scriptData(() => {
|
|
function FindProxyForURL() {
|
|
// Shoot a message off to the background.
|
|
browser.runtime.sendMessage("success");
|
|
return null;
|
|
}
|
|
}),
|
|
},
|
|
});
|
|
await pb_extension.startup();
|
|
await pb_extension.awaitMessage("ready");
|
|
|
|
let finished = pb_extension.awaitFinish("success");
|
|
let contentPage = await ExtensionTestUtils.loadContentPage("http://example.com/dummy", {privateBrowsing: true});
|
|
await finished;
|
|
|
|
await extension.unload();
|
|
await pb_extension.unload();
|
|
await contentPage.close();
|
|
|
|
Services.prefs.clearUserPref("extensions.allowPrivateBrowsingByDefault");
|
|
});
|