summaryrefslogtreecommitdiffstats
path: root/dom/base/test/browser_messagemanager_loadprocessscript.js
diff options
context:
space:
mode:
Diffstat (limited to 'dom/base/test/browser_messagemanager_loadprocessscript.js')
-rw-r--r--dom/base/test/browser_messagemanager_loadprocessscript.js193
1 files changed, 193 insertions, 0 deletions
diff --git a/dom/base/test/browser_messagemanager_loadprocessscript.js b/dom/base/test/browser_messagemanager_loadprocessscript.js
new file mode 100644
index 0000000000..16686dd566
--- /dev/null
+++ b/dom/base/test/browser_messagemanager_loadprocessscript.js
@@ -0,0 +1,193 @@
+function getBaseNumberOfProcesses() {
+ // We should have three processes for this test, the parent process and two
+ // content processes for the tabs craeted by this test.
+ let processCount = 3;
+
+ // If we run WebExtensions out-of-process (see bug 1190679), there might be
+ // additional processes for those, so let's add these to the base count to
+ // not have system WebExtensions cause test failures.
+ for (let i = 0; i < Services.ppmm.childCount; i++) {
+ if (
+ Services.ppmm.getChildAt(i).remoteType === E10SUtils.EXTENSION_REMOTE_TYPE
+ ) {
+ processCount += 1;
+ }
+ }
+
+ return processCount;
+}
+
+function checkBaseProcessCount(description) {
+ const baseProcessCount = getBaseNumberOfProcesses();
+ const { childCount } = Services.ppmm;
+ // With preloaded activity-stream, process count is a bit undeterministic, so
+ // allow for some variation
+ const extraCount = baseProcessCount + 1;
+ ok(
+ childCount === baseProcessCount || childCount === extraCount,
+ `${description} (${baseProcessCount} or ${extraCount})`
+ );
+}
+
+function processScript() {
+ /* eslint-env mozilla/process-script */
+ if (Services.cpmm !== this) {
+ dump("Test failed: wrong global object\n");
+ return;
+ }
+
+ this.cpmm = Services.cpmm;
+
+ addMessageListener("ProcessTest:Reply", function listener(msg) {
+ removeMessageListener("ProcessTest:Reply", listener);
+ sendAsyncMessage("ProcessTest:Finished");
+ });
+ sendSyncMessage("ProcessTest:Loaded");
+}
+var processScriptURL = "data:,(" + processScript.toString() + ").call(this)";
+
+function initTestScript() {
+ /* eslint-env mozilla/process-script */
+ let init = initialProcessData;
+ if (init.test123 != "hello") {
+ dump("Initial data incorrect\n");
+ return;
+ }
+
+ sendAsyncMessage("ProcessTest:InitGood", init.test456.get("hi"));
+}
+var initTestScriptURL = "data:,(" + initTestScript.toString() + ")()";
+
+var checkProcess = async function (mm) {
+ let { target } = await promiseMessage(mm, "ProcessTest:Loaded");
+ target.sendAsyncMessage("ProcessTest:Reply");
+ await promiseMessage(target, "ProcessTest:Finished");
+ ok(true, "Saw process finished");
+};
+
+function promiseMessage(messageManager, message) {
+ return new Promise(resolve => {
+ let listener = msg => {
+ messageManager.removeMessageListener(message, listener);
+ resolve(msg);
+ };
+
+ messageManager.addMessageListener(message, listener);
+ });
+}
+
+add_task(async function () {
+ // We want to count processes in this test, so let's disable the pre-allocated process manager.
+ await SpecialPowers.pushPrefEnv({
+ set: [["dom.ipc.processPrelaunch.enabled", false]],
+ });
+});
+
+add_task(async function () {
+ // This test is only relevant in e10s.
+ if (!gMultiProcessBrowser) {
+ return;
+ }
+
+ Services.ppmm.releaseCachedProcesses();
+
+ await SpecialPowers.pushPrefEnv({ set: [["dom.ipc.processCount", 5]] });
+ await SpecialPowers.pushPrefEnv({
+ set: [["dom.ipc.keepProcessesAlive.web", 5]],
+ });
+
+ let tabs = [];
+ for (let i = 0; i < 3; i++) {
+ tabs[i] = await BrowserTestUtils.openNewForegroundTab(
+ gBrowser,
+ "about:blank"
+ );
+ }
+
+ for (let i = 0; i < 3; i++) {
+ // FIXME: This should wait for the tab removal gets reflected to the
+ // process count (bug 1446726).
+ let sessionStorePromise = BrowserTestUtils.waitForSessionStoreUpdate(
+ tabs[i]
+ );
+ BrowserTestUtils.removeTab(tabs[i]);
+ await sessionStorePromise;
+ }
+
+ Services.ppmm.releaseCachedProcesses();
+ checkBaseProcessCount(
+ "Should get back to the base number of processes at this point"
+ );
+});
+
+// Test that loading a process script loads in all existing processes
+add_task(async function () {
+ let checks = [];
+ for (let i = 0; i < Services.ppmm.childCount; i++) {
+ checks.push(checkProcess(Services.ppmm.getChildAt(i)));
+ }
+
+ Services.ppmm.loadProcessScript(processScriptURL, false);
+ await Promise.all(checks);
+});
+
+// Test that loading a process script loads in new processes
+add_task(async function () {
+ // This test is only relevant in e10s
+ if (!gMultiProcessBrowser) {
+ return;
+ }
+
+ checkBaseProcessCount(
+ "Should still be at the base number of processes at this point"
+ );
+
+ // Load something in the main process
+ BrowserTestUtils.loadURIString(gBrowser.selectedBrowser, "about:mozilla");
+ await BrowserTestUtils.browserLoaded(gBrowser.selectedBrowser);
+
+ let init = Services.ppmm.initialProcessData;
+ init.test123 = "hello";
+ init.test456 = new Map();
+ init.test456.set("hi", "bye");
+
+ // With no remote frames left we should be down to one process.
+ // However, stuff like remote thumbnails can cause a content
+ // process to exist nonetheless. This should be rare, though,
+ // so the test is useful most of the time.
+ if (Services.ppmm.childCount == 2) {
+ let mainMM = Services.ppmm.getChildAt(0);
+
+ let check = checkProcess(Services.ppmm);
+ Services.ppmm.loadProcessScript(processScriptURL, true);
+
+ // The main process should respond
+ await check;
+
+ check = checkProcess(Services.ppmm);
+ // Reset the default browser to start a new child process
+ gBrowser.updateBrowserRemoteness(gBrowser.selectedBrowser, {
+ remoteType: E10SUtils.DEFAULT_REMOTE_TYPE,
+ });
+ BrowserTestUtils.loadURIString(gBrowser.selectedBrowser, "about:blank");
+ await BrowserTestUtils.browserLoaded(gBrowser.selectedBrowser);
+
+ checkBaseProcessCount(
+ "Should be back to the base number of processes at this point"
+ );
+
+ // The new process should have responded
+ await check;
+
+ Services.ppmm.removeDelayedProcessScript(processScriptURL);
+
+ let childMM;
+ childMM = Services.ppmm.getChildAt(2);
+
+ childMM.loadProcessScript(initTestScriptURL, false);
+ let msg = await promiseMessage(childMM, "ProcessTest:InitGood");
+ is(msg.data, "bye", "initial process data was correct");
+ } else {
+ info("Unable to finish test entirely");
+ }
+});