зеркало из https://github.com/mozilla/gecko-dev.git
Bug 1536094 - Add test for importing from WebExtension content scripts r=smaug,robwu
Differential Revision: https://phabricator.services.mozilla.com/D108236
This commit is contained in:
Tom Schuster
Родитель
68c7cd7283
Коммит
0583e7e496
|
|
@ -0,0 +1,274 @@
|
|||
"use strict";
|
||||
|
||||
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>");
|
||||
});
|
||||
|
||||
server.registerPathHandler("/script.js", (request, response) => {
|
||||
ok(false, "Unexpected request to /script.js");
|
||||
});
|
||||
|
||||
/* eslint-disable no-unsanitized/method, no-eval, no-implied-eval */
|
||||
|
||||
const MODULE1 = `
|
||||
import {foo} from "./module2.js";
|
||||
export let bar = foo;
|
||||
|
||||
let count = 0;
|
||||
|
||||
export function counter () { return count++; }
|
||||
`;
|
||||
|
||||
const MODULE2 = `export let foo = 2;`;
|
||||
|
||||
add_task(async function test_disallowed_import() {
|
||||
let extension = ExtensionTestUtils.loadExtension({
|
||||
manifest: {
|
||||
content_scripts: [
|
||||
{
|
||||
matches: ["http://example.com/dummy"],
|
||||
js: ["main.js"],
|
||||
},
|
||||
],
|
||||
},
|
||||
|
||||
files: {
|
||||
"main.js": async function() {
|
||||
let disallowedURLs = [
|
||||
"data:text/javascript,void 0",
|
||||
"javascript:void 0",
|
||||
"http://example.com/script.js",
|
||||
URL.createObjectURL(
|
||||
new Blob(["void 0", { type: "text/javascript" }])
|
||||
),
|
||||
];
|
||||
|
||||
for (let url of disallowedURLs) {
|
||||
await browser.test.assertRejects(
|
||||
import(url),
|
||||
/error loading dynamically imported module/,
|
||||
`should reject import("${url}")`
|
||||
);
|
||||
}
|
||||
|
||||
browser.test.sendMessage("done");
|
||||
},
|
||||
},
|
||||
});
|
||||
|
||||
await extension.startup();
|
||||
let contentPage = await ExtensionTestUtils.loadContentPage(
|
||||
"http://example.com/dummy"
|
||||
);
|
||||
await extension.awaitMessage("done");
|
||||
await extension.unload();
|
||||
await contentPage.close();
|
||||
});
|
||||
|
||||
add_task(async function test_normal_import() {
|
||||
let extension = ExtensionTestUtils.loadExtension({
|
||||
manifest: {
|
||||
content_scripts: [
|
||||
{
|
||||
matches: ["http://example.com/dummy"],
|
||||
js: ["main.js"],
|
||||
},
|
||||
],
|
||||
},
|
||||
|
||||
files: {
|
||||
"main.js": async function() {
|
||||
/* global exportFunction */
|
||||
const url = browser.runtime.getURL("module1.js");
|
||||
let mod = await import(url);
|
||||
browser.test.assertEq(mod.bar, 2);
|
||||
|
||||
browser.test.assertEq(mod.counter(), 0);
|
||||
browser.test.assertEq(mod.counter(), 1);
|
||||
|
||||
await browser.test.assertRejects(
|
||||
window.eval(`import("${url}")`),
|
||||
/error loading dynamically imported module/,
|
||||
"Cannot import script that is not web-accessible from page context"
|
||||
);
|
||||
|
||||
let promise = new Promise((resolve, reject) => {
|
||||
exportFunction(resolve, window, { defineAs: "resolve" });
|
||||
exportFunction(reject, window, { defineAs: "reject" });
|
||||
});
|
||||
|
||||
window.setTimeout(`import("${url}").then(resolve, reject)`, 0);
|
||||
|
||||
await browser.test.assertRejects(
|
||||
promise,
|
||||
/error loading dynamically imported module/,
|
||||
"Cannot import script that is not web-accessible from page context"
|
||||
);
|
||||
|
||||
browser.test.sendMessage("done");
|
||||
},
|
||||
"module1.js": MODULE1,
|
||||
"module2.js": MODULE2,
|
||||
},
|
||||
});
|
||||
|
||||
await extension.startup();
|
||||
let contentPage = await ExtensionTestUtils.loadContentPage(
|
||||
"http://example.com/dummy"
|
||||
);
|
||||
|
||||
await extension.awaitMessage("done");
|
||||
|
||||
// Web page can not import non-web-accessible files.
|
||||
await contentPage.spawn(extension.uuid, async uuid => {
|
||||
let files = ["main.js", "module1.js", "module2.js"];
|
||||
|
||||
for (let file of files) {
|
||||
let url = `moz-extension://${uuid}/${file}`;
|
||||
await Assert.rejects(
|
||||
content.eval(`import("${url}")`),
|
||||
/error loading dynamically imported module/,
|
||||
"Cannot import script that is not web-accessible"
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
await extension.unload();
|
||||
await contentPage.close();
|
||||
});
|
||||
|
||||
add_task(async function test_import_web_accessible() {
|
||||
let extension = ExtensionTestUtils.loadExtension({
|
||||
manifest: {
|
||||
content_scripts: [
|
||||
{
|
||||
matches: ["http://example.com/dummy"],
|
||||
js: ["main.js"],
|
||||
},
|
||||
],
|
||||
web_accessible_resources: ["module1.js", "module2.js"],
|
||||
},
|
||||
|
||||
files: {
|
||||
"main.js": async function() {
|
||||
let mod = await import(browser.runtime.getURL("module1.js"));
|
||||
browser.test.assertEq(mod.bar, 2);
|
||||
browser.test.assertEq(mod.counter(), 0);
|
||||
browser.test.sendMessage("done");
|
||||
},
|
||||
"module1.js": MODULE1,
|
||||
"module2.js": MODULE2,
|
||||
},
|
||||
});
|
||||
|
||||
await extension.startup();
|
||||
let contentPage = await ExtensionTestUtils.loadContentPage(
|
||||
"http://example.com/dummy"
|
||||
);
|
||||
await extension.awaitMessage("done");
|
||||
|
||||
// Web page can import web-accessible files,
|
||||
// even after WebExtension imported the same files.
|
||||
await contentPage.spawn(extension.uuid, async uuid => {
|
||||
let base = `moz-extension://${uuid}`;
|
||||
|
||||
await Assert.rejects(
|
||||
content.eval(`import("${base}/main.js")`),
|
||||
/error loading dynamically imported module/,
|
||||
"Cannot import script that is not web-accessible"
|
||||
);
|
||||
|
||||
let promise = content.eval(`import("${base}/module1.js")`);
|
||||
let mod = (await promise.wrappedJSObject).wrappedJSObject;
|
||||
Assert.equal(mod.bar, 2, "exported value should match");
|
||||
Assert.equal(mod.counter(), 0, "Counter should be fresh");
|
||||
Assert.equal(mod.counter(), 1, "Counter should be fresh");
|
||||
|
||||
promise = content.eval(`import("${base}/module2.js")`);
|
||||
mod = (await promise.wrappedJSObject).wrappedJSObject;
|
||||
Assert.equal(mod.foo, 2, "exported value should match");
|
||||
});
|
||||
|
||||
await extension.unload();
|
||||
await contentPage.close();
|
||||
});
|
||||
|
||||
add_task(async function test_import_web_accessible_after_page() {
|
||||
let extension = ExtensionTestUtils.loadExtension({
|
||||
manifest: {
|
||||
content_scripts: [
|
||||
{
|
||||
matches: ["http://example.com/dummy"],
|
||||
js: ["main.js"],
|
||||
},
|
||||
],
|
||||
web_accessible_resources: ["module1.js", "module2.js"],
|
||||
},
|
||||
|
||||
files: {
|
||||
"main.js": async function() {
|
||||
browser.test.onMessage.addListener(async msg => {
|
||||
browser.test.assertEq(msg, "import");
|
||||
|
||||
const url = browser.runtime.getURL("module1.js");
|
||||
let mod = await import(url);
|
||||
browser.test.assertEq(mod.bar, 2);
|
||||
browser.test.assertEq(mod.counter(), 0, "Counter should be fresh");
|
||||
|
||||
let promise = window.eval(`import("${url}")`);
|
||||
let mod2 = (await promise.wrappedJSObject).wrappedJSObject;
|
||||
browser.test.assertEq(
|
||||
mod2.counter(),
|
||||
2,
|
||||
"Counter should have been incremented by page"
|
||||
);
|
||||
|
||||
browser.test.sendMessage("done");
|
||||
});
|
||||
browser.test.sendMessage("ready");
|
||||
},
|
||||
"module1.js": MODULE1,
|
||||
"module2.js": MODULE2,
|
||||
},
|
||||
});
|
||||
|
||||
await extension.startup();
|
||||
let contentPage = await ExtensionTestUtils.loadContentPage(
|
||||
"http://example.com/dummy"
|
||||
);
|
||||
await extension.awaitMessage("ready");
|
||||
|
||||
// The web page imports the web-accessible files first,
|
||||
// when the WebExtension imports the same file, they should
|
||||
// not be shared.
|
||||
await contentPage.spawn(extension.uuid, async uuid => {
|
||||
let base = `moz-extension://${uuid}`;
|
||||
|
||||
await Assert.rejects(
|
||||
content.eval(`import("${base}/main.js")`),
|
||||
/error loading dynamically imported module/,
|
||||
"Cannot import script that is not web-accessible"
|
||||
);
|
||||
|
||||
let promise = content.eval(`import("${base}/module1.js")`);
|
||||
let mod = (await promise.wrappedJSObject).wrappedJSObject;
|
||||
Assert.equal(mod.bar, 2, "exported value should match");
|
||||
Assert.equal(mod.counter(), 0);
|
||||
Assert.equal(mod.counter(), 1);
|
||||
|
||||
promise = content.eval(`import("${base}/module2.js")`);
|
||||
mod = (await promise.wrappedJSObject).wrappedJSObject;
|
||||
Assert.equal(mod.foo, 2, "exported value should match");
|
||||
});
|
||||
|
||||
extension.sendMessage("import");
|
||||
|
||||
await extension.awaitMessage("done");
|
||||
|
||||
await extension.unload();
|
||||
await contentPage.close();
|
||||
});
|
||||
|
|
@ -61,6 +61,7 @@ skip-if = tsan # Bug 1683730
|
|||
[test_ext_contentscript_css.js]
|
||||
[test_ext_contentscript_exporthelpers.js]
|
||||
[test_ext_contentscript_in_background.js]
|
||||
[test_ext_contentscript_module_import.js]
|
||||
[test_ext_contentscript_restrictSchemes.js]
|
||||
[test_ext_contentscript_teardown.js]
|
||||
skip-if = tsan # Bug 1683730
|
||||
|
|
|
|||
Загрузка…
Ссылка в новой задаче