1 files changed, 59 insertions, 0 deletions
diff --git a/browser/components/sessionstore/test/browser_restore_private_tab_os.js b/browser/components/sessionstore/test/browser_restore_private_tab_os.js
new file mode 100644
index 0000000000..3041fe6f39
--- /dev/null
+++ b/browser/components/sessionstore/test/browser_restore_private_tab_os.js
@@ -0,0 +1,59 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+"use strict";
+
+const TEST_URI =
+  "https://example.com/" +
+  "browser/browser/components/sessionstore/test/empty.html";
+
+add_setup(async function () {
+  await SpecialPowers.pushPrefEnv({
+    set: [
+      // Set the pref to true so we know exactly how many tabs should be restoring at
+      // any given time. This guarantees that a finishing load won't start another.
+      ["browser.sessionstore.restore_on_demand", true],
+      // Don't restore tabs lazily.
+      ["browser.sessionstore.restore_tabs_lazily", true],
+      // don't preload tabs so we don't have extra XULFrameLoaderCreated events
+      // firing
+      ["browser.newtab.preload", false],
+    ],
+  });
+});
+
+add_task(async function testRestore() {
+  // Clear the list of closed windows.
+  forgetClosedWindows();
+
+  // Create a new private window
+  let win = await promiseNewWindowLoaded({ private: true });
+
+  // Create a new private tab
+  let tab = BrowserTestUtils.addTab(win.gBrowser, TEST_URI);
+  let browser = tab.linkedBrowser;
+  await promiseBrowserLoaded(browser);
+  await TabStateFlusher.flush(browser);
+
+  // Ensure that closed tabs in a private windows can be restored.
+  win.gBrowser.removeTab(tab);
+  is(ss.getClosedTabCountForWindow(win), 1, "there is a single tab to restore");
+
+  tab = SessionStore.undoCloseTab(win, 0);
+  info(`Undo close tab`);
+  browser = tab.linkedBrowser;
+  await promiseTabRestored(tab);
+  info(`Private tab restored`);
+
+  let expectedRemoteType = gFissionBrowser
+    ? "webIsolated=https://example.com^privateBrowsingId=1"
+    : "web";
+  is(browser.remoteType, expectedRemoteType, "correct remote type");
+
+  await BrowserTestUtils.closeWindow(win);
+
+  // Cleanup
+  info("Forgetting closed tabs");
+  forgetClosedTabs(window);
+});