1 files changed, 115 insertions, 0 deletions
diff --git a/browser/base/content/test/tabs/browser_removeTabs_skipPermitUnload.js b/browser/base/content/test/tabs/browser_removeTabs_skipPermitUnload.js
new file mode 100644
index 0000000000..1db1b22195
--- /dev/null
+++ b/browser/base/content/test/tabs/browser_removeTabs_skipPermitUnload.js
@@ -0,0 +1,115 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+/**
+ * Tests for waiting for beforeunload before replacing a session.
+ */
+
+const { PromptTestUtils } = ChromeUtils.import(
+  "resource://testing-common/PromptTestUtils.jsm"
+);
+
+// The first two urls are intentionally different domains to force pages
+// to load in different tabs.
+const TEST_PATH = getRootDirectory(gTestPath).replace(
+  "chrome://mochitests/content",
+  "https://example.com"
+);
+const TEST_URL = "https://example.com/";
+
+const BUILDER_URL = "https://example.com/document-builder.sjs?html=";
+const PAGE_MARKUP = `
+<html>
+<head>
+  <script>
+    window.onbeforeunload = function() {
+      return true;
+    };
+  </script>
+</head>
+<body>TEST PAGE</body>
+</html>
+`;
+const TEST_URL2 = BUILDER_URL + encodeURI(PAGE_MARKUP);
+
+let win;
+let nonBeforeUnloadTab;
+let beforeUnloadTab;
+
+add_setup(async function() {
+  await SpecialPowers.pushPrefEnv({
+    set: [["dom.require_user_interaction_for_beforeunload", false]],
+  });
+
+  // Run tests in a new window to avoid affecting the main test window.
+  win = await BrowserTestUtils.openNewBrowserWindow();
+
+  BrowserTestUtils.loadURI(win.gBrowser.selectedBrowser, TEST_URL);
+  await BrowserTestUtils.browserLoaded(
+    win.gBrowser.selectedBrowser,
+    false,
+    TEST_URL
+  );
+  nonBeforeUnloadTab = win.gBrowser.selectedTab;
+  beforeUnloadTab = await BrowserTestUtils.openNewForegroundTab(
+    win.gBrowser,
+    TEST_URL2
+  );
+
+  registerCleanupFunction(async () => {
+    await BrowserTestUtils.closeWindow(win);
+  });
+});
+
+add_task(async function test_runBeforeUnloadForTabs() {
+  let unloadDialogPromise = PromptTestUtils.handleNextPrompt(
+    win,
+    {
+      modalType: Ci.nsIPrompt.MODAL_TYPE_CONTENT,
+      promptType: "confirmEx",
+    },
+    // Click the cancel button.
+    { buttonNumClick: 1 }
+  );
+
+  let unloadBlocked = await win.gBrowser.runBeforeUnloadForTabs(
+    win.gBrowser.tabs
+  );
+
+  await unloadDialogPromise;
+
+  Assert.ok(unloadBlocked, "Should have reported the unload was blocked");
+  Assert.equal(win.gBrowser.tabs.length, 2, "Should have left all tabs open");
+
+  unloadDialogPromise = PromptTestUtils.handleNextPrompt(
+    win,
+    {
+      modalType: Ci.nsIPrompt.MODAL_TYPE_CONTENT,
+      promptType: "confirmEx",
+    },
+    // Click the ok button.
+    { buttonNumClick: 0 }
+  );
+
+  unloadBlocked = await win.gBrowser.runBeforeUnloadForTabs(win.gBrowser.tabs);
+
+  await unloadDialogPromise;
+
+  Assert.ok(!unloadBlocked, "Should have reported the unload was not blocked");
+  Assert.equal(win.gBrowser.tabs.length, 2, "Should have left all tabs open");
+});
+
+add_task(async function test_skipPermitUnload() {
+  let closePromise = BrowserTestUtils.waitForTabClosing(beforeUnloadTab);
+
+  await win.gBrowser.removeAllTabsBut(nonBeforeUnloadTab, {
+    animate: false,
+    skipPermitUnload: true,
+  });
+
+  await closePromise;
+
+  Assert.equal(win.gBrowser.tabs.length, 1, "Should have left one tab open");
+});