Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
"use strict";
requestLongerTimeout(2);
add_setup(async function () {
await SpecialPowers.pushPrefEnv({
set: [["dom.ipc.processCount", 1]],
});
});
function promiseTabsRestored(win, nExpected) {
return new Promise(resolve => {
let nReceived = 0;
function handler() {
if (++nReceived === nExpected) {
win.gBrowser.tabContainer.removeEventListener(
"SSTabRestored",
handler,
true
);
resolve();
}
}
win.gBrowser.tabContainer.addEventListener("SSTabRestored", handler, true);
});
}
add_task(async function () {
let win = await BrowserTestUtils.openNewBrowserWindow();
// Create 4 tabs with different userContextId.
for (let userContextId = 1; userContextId < 5; userContextId++) {
userContextId,
});
await promiseBrowserLoaded(tab.linkedBrowser);
await TabStateFlusher.flush(tab.linkedBrowser);
}
// Move the default tab of window to the end.
// We want the 1st tab to have non-default userContextId, so later when we
// restore into win2 we can test restore into an existing tab with different
// userContextId.
win.gBrowser.moveTabTo(win.gBrowser.tabs[0], win.gBrowser.tabs.length - 1);
let winState = ss.getWindowState(win);
for (let i = 0; i < 4; i++) {
Assert.equal(
winState.windows[0].tabs[i].userContextId,
i + 1,
"1st Window: tabs[" + i + "].userContextId should exist."
);
}
let win2 = await BrowserTestUtils.openNewBrowserWindow();
// Create tabs with different userContextId, but this time we create them with
// fewer tabs and with different order with win.
for (let userContextId = 3; userContextId > 0; userContextId--) {
userContextId,
});
await promiseBrowserLoaded(tab.linkedBrowser);
await TabStateFlusher.flush(tab.linkedBrowser);
}
let tabsRestored = promiseTabsRestored(win2, 5);
await setWindowState(win2, winState, true);
await tabsRestored;
for (let i = 0; i < 4; i++) {
let browser = win2.gBrowser.tabs[i].linkedBrowser;
await ContentTask.spawn(
browser,
{ expectedId: i + 1 },
async function (args) {
Assert.equal(
docShell.getOriginAttributes().userContextId,
args.expectedId,
"The docShell has the correct userContextId"
);
Assert.equal(
content.document.nodePrincipal.originAttributes.userContextId,
args.expectedId,
"The document has the correct userContextId"
);
}
);
}
// Test the last tab, which doesn't have userContextId.
let browser = win2.gBrowser.tabs[4].linkedBrowser;
await SpecialPowers.spawn(
browser,
[{ expectedId: 0 }],
async function (args) {
Assert.equal(
docShell.getOriginAttributes().userContextId,
args.expectedId,
"The docShell has the correct userContextId"
);
Assert.equal(
content.document.nodePrincipal.originAttributes.userContextId,
args.expectedId,
"The document has the correct userContextId"
);
}
);
await BrowserTestUtils.closeWindow(win);
await BrowserTestUtils.closeWindow(win2);
});
add_task(async function () {
let win = await BrowserTestUtils.openNewBrowserWindow();
await TabStateFlusher.flush(win.gBrowser.selectedBrowser);
userContextId: 1,
});
await promiseBrowserLoaded(tab.linkedBrowser);
await TabStateFlusher.flush(tab.linkedBrowser);
// win should have 1 default tab, and 1 container tab.
Assert.equal(win.gBrowser.tabs.length, 2, "win should have 2 tabs");
let winState = ss.getWindowState(win);
for (let i = 0; i < 2; i++) {
Assert.equal(
winState.windows[0].tabs[i].userContextId,
i,
"1st Window: tabs[" + i + "].userContextId should be " + i
);
}
let win2 = await BrowserTestUtils.openNewBrowserWindow();
userContextId: 1,
});
await promiseBrowserLoaded(tab2.linkedBrowser);
await TabStateFlusher.flush(tab2.linkedBrowser);
// Move the first normal tab to end, so the first tab of win2 will be a
// container tab.
win2.gBrowser.moveTabTo(win2.gBrowser.tabs[0], win2.gBrowser.tabs.length - 1);
await TabStateFlusher.flush(win2.gBrowser.tabs[0].linkedBrowser);
let tabsRestored = promiseTabsRestored(win2, 2);
await setWindowState(win2, winState, true);
await tabsRestored;
for (let i = 0; i < 2; i++) {
let browser = win2.gBrowser.tabs[i].linkedBrowser;
await ContentTask.spawn(browser, { expectedId: i }, async function (args) {
Assert.equal(
docShell.getOriginAttributes().userContextId,
args.expectedId,
"The docShell has the correct userContextId"
);
Assert.equal(
content.document.nodePrincipal.originAttributes.userContextId,
args.expectedId,
"The document has the correct userContextId"
);
});
}
await BrowserTestUtils.closeWindow(win);
await BrowserTestUtils.closeWindow(win2);
});