"use strict";
const server = createHttpServer({ hosts: ["example.com"] });
server.registerDirectory("/data/", do_get_file("data"));
AddonTestUtils.init(this);
AddonTestUtils.overrideCertDB();
AddonTestUtils.createAppInfo(
"xpcshell@tests.mozilla.org",
"XPCShell",
"1",
"43"
);
let { promiseRestartManager, promiseShutdownManager, promiseStartupManager } =
AddonTestUtils;
const PAGE_HTML = ``;
function trackEvents(wrapper) {
let events = new Map();
for (let event of ["background-script-event", "start-background-script"]) {
events.set(event, false);
wrapper.extension.once(event, () => events.set(event, true));
}
return events;
}
async function test(what, background, script) {
await promiseStartupManager();
let extension = ExtensionTestUtils.loadExtension({
useAddonManager: "permanent",
manifest: {
content_scripts: [
{
matches: ["http://example.com/*"],
js: ["script.js"],
},
],
},
files: {
"page.html": PAGE_HTML,
"script.js": script,
},
background,
});
info(`Set up ${what} listener`);
await extension.startup();
await extension.awaitMessage("bg-ran");
info(`Test wakeup for ${what} from an extension page`);
await promiseRestartManager({ earlyStartup: false });
await extension.awaitStartup();
function awaitBgEvent() {
return new Promise(resolve =>
extension.extension.once("background-script-event", resolve)
);
}
let events = trackEvents(extension);
let url = extension.extension.baseURI.resolve("page.html");
let [, page] = await Promise.all([
awaitBgEvent(),
ExtensionTestUtils.loadContentPage(url, { extension }),
]);
equal(
events.get("background-script-event"),
true,
"Should have gotten a background page event"
);
equal(
events.get("start-background-script"),
false,
"Background page should not be started"
);
equal(extension.messageQueue.size, 0, "Have not yet received bg-ran message");
let promise = extension.awaitMessage("bg-ran");
AddonTestUtils.notifyEarlyStartup();
await promise;
equal(
events.get("start-background-script"),
true,
"Should have gotten start-background-script event"
);
await extension.awaitFinish("messaging-test");
ok(true, "Background page loaded and received message from extension page");
await page.close();
info(`Test wakeup for ${what} from a content script`);
await promiseRestartManager({ earlyStartup: false });
await extension.awaitStartup();
events = trackEvents(extension);
[, page] = await Promise.all([
awaitBgEvent(),
ExtensionTestUtils.loadContentPage(
"http://example.com/data/file_sample.html"
),
]);
equal(
events.get("background-script-event"),
true,
"Should have gotten a background script event"
);
equal(
events.get("start-background-script"),
false,
"Background script should not be started"
);
equal(extension.messageQueue.size, 0, "Have not yet received bg-ran message");
promise = extension.awaitMessage("bg-ran");
AddonTestUtils.notifyEarlyStartup();
await promise;
equal(
events.get("start-background-script"),
true,
"Should have gotten start-background-script event"
);
await extension.awaitFinish("messaging-test");
ok(true, "Background page loaded and received message from content script");
await page.close();
await extension.unload();
await promiseShutdownManager();
}
add_task(function test_onMessage() {
function script() {
browser.runtime.sendMessage("ping").then(reply => {
browser.test.assertEq(
reply,
"pong",
"Extension page received pong reply"
);
browser.test.notifyPass("messaging-test");
});
}
async function background() {
browser.runtime.onMessage.addListener((msg, sender) => {
browser.test.assertEq(
msg,
"ping",
"Background page received ping message"
);
return Promise.resolve("pong");
});
// addListener() returns right away but make a round trip to the
// main process to ensure the persistent onMessage listener is recorded.
await browser.runtime.getBrowserInfo();
browser.test.sendMessage("bg-ran");
}
return test("onMessage", background, script);
});
add_task(function test_onConnect() {
function script() {
let port = browser.runtime.connect();
port.onMessage.addListener(msg => {
browser.test.assertEq(msg, "pong", "Extension page received pong reply");
browser.test.notifyPass("messaging-test");
});
port.postMessage("ping");
}
async function background() {
browser.runtime.onConnect.addListener(port => {
port.onMessage.addListener(msg => {
browser.test.assertEq(
msg,
"ping",
"Background page received ping message"
);
port.postMessage("pong");
});
});
// addListener() returns right away but make a round trip to the
// main process to ensure the persistent onMessage listener is recorded.
await browser.runtime.getBrowserInfo();
browser.test.sendMessage("bg-ran");
}
return test("onConnect", background, script);
});
// Test that messaging works if the background page is started before
// any messages are exchanged. (See bug 1467136 for an example of how
// this broke at one point).
add_task(async function test_other_startup() {
await promiseStartupManager();
let extension = ExtensionTestUtils.loadExtension({
useAddonManager: "permanent",
async background() {
browser.runtime.onMessage.addListener(msg => {
browser.test.notifyPass("startup");
});
// addListener() returns right away but make a round trip to the
// main process to ensure the persistent onMessage listener is recorded.
await browser.runtime.getBrowserInfo();
browser.test.sendMessage("bg-ran");
},
files: {
"page.html": PAGE_HTML,
"script.js"() {
browser.runtime.sendMessage("ping");
},
},
});
await extension.startup();
await extension.awaitMessage("bg-ran");
await promiseRestartManager({ lateStartup: false });
await extension.awaitStartup();
let events = trackEvents(extension);
equal(
events.get("background-script-event"),
false,
"Should not have gotten a background page event"
);
equal(
events.get("start-background-script"),
false,
"Background page should not be started"
);
// Start the background page. No message have been sent at this point.
await AddonTestUtils.notifyLateStartup();
equal(
events.get("start-background-script"),
true,
"Background page should be started"
);
await extension.awaitMessage("bg-ran");
// Now that the background page is fully started, load a new page that
// sends a message to the background page.
let url = extension.extension.baseURI.resolve("page.html");
let page = await ExtensionTestUtils.loadContentPage(url, { extension });
await extension.awaitFinish("startup");
await page.close();
await extension.unload();
await promiseShutdownManager();
});