1 files changed, 176 insertions, 0 deletions
diff --git a/browser/components/sessionstore/test/browser_windowStateContainer.js b/browser/components/sessionstore/test/browser_windowStateContainer.js
new file mode 100644
index 0000000000..f0d6f42d39
--- /dev/null
+++ b/browser/components/sessionstore/test/browser_windowStateContainer.js
@@ -0,0 +1,176 @@
+"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(event) {
+      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++) {
+    let tab = BrowserTestUtils.addTab(win.gBrowser, "http://example.com/", {
+      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--) {
+    let tab = BrowserTestUtils.addTab(win2.gBrowser, "http://example.com/", {
+      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);
+
+  let tab = BrowserTestUtils.addTab(win.gBrowser, "http://example.com/", {
+    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();
+
+  let tab2 = BrowserTestUtils.addTab(win2.gBrowser, "http://example.com/", {
+    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);
+});