Source code

Revision control

Copy as Markdown

Other Tools

Test Info:

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");
}
});