1 files changed, 199 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..af33da721d
--- /dev/null
+++ b/dom/base/test/browser_messagemanager_loadprocessscript.js
@@ -0,0 +1,199 @@
+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.startLoadingURIString(
+    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.startLoadingURIString(
+      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");
+  }
+});