1 files changed, 152 insertions, 0 deletions
diff --git a/dom/tests/browser/browser_beforeunload_between_chrome_content.js b/dom/tests/browser/browser_beforeunload_between_chrome_content.js
new file mode 100644
index 0000000000..67579592ed
--- /dev/null
+++ b/dom/tests/browser/browser_beforeunload_between_chrome_content.js
@@ -0,0 +1,152 @@
+const TEST_URL = "http://www.example.com/browser/dom/tests/browser/dummy.html";
+
+const { PromptTestUtils } = ChromeUtils.importESModule(
+  "resource://testing-common/PromptTestUtils.sys.mjs"
+);
+
+function pageScript() {
+  window.addEventListener(
+    "beforeunload",
+    function (event) {
+      var str = "Leaving?";
+      event.returnValue = str;
+      return str;
+    },
+    true
+  );
+}
+
+function injectBeforeUnload(browser) {
+  return ContentTask.spawn(browser, null, async function () {
+    content.window.addEventListener(
+      "beforeunload",
+      function (event) {
+        sendAsyncMessage("Test:OnBeforeUnloadReceived");
+        var str = "Leaving?";
+        event.returnValue = str;
+        return str;
+      },
+      true
+    );
+  });
+}
+
+// Wait for onbeforeunload dialog, and dismiss it immediately.
+function awaitAndCloseBeforeUnloadDialog(browser, doStayOnPage) {
+  return PromptTestUtils.handleNextPrompt(
+    browser,
+    { modalType: Services.prompt.MODAL_TYPE_CONTENT, promptType: "confirmEx" },
+    { buttonNumClick: doStayOnPage ? 1 : 0 }
+  );
+}
+
+SpecialPowers.pushPrefEnv({
+  set: [["dom.require_user_interaction_for_beforeunload", false]],
+});
+
+/**
+ * Test navigation from a content page to a chrome page. Also check that only
+ * one beforeunload event is fired.
+ */
+/* global messageManager */
+add_task(async function () {
+  let beforeUnloadCount = 0;
+  messageManager.addMessageListener("Test:OnBeforeUnloadReceived", function () {
+    beforeUnloadCount++;
+  });
+
+  // Open a content page.
+  let tab = await BrowserTestUtils.openNewForegroundTab(gBrowser, TEST_URL);
+  let browser = tab.linkedBrowser;
+
+  ok(browser.isRemoteBrowser, "Browser should be remote.");
+
+  await injectBeforeUnload(browser);
+  // Navigate to a chrome page.
+  let dialogShown1 = awaitAndCloseBeforeUnloadDialog(browser, false);
+  BrowserTestUtils.loadURIString(browser, "about:support");
+  await Promise.all([dialogShown1, BrowserTestUtils.browserLoaded(browser)]);
+
+  is(beforeUnloadCount, 1, "Should have received one beforeunload event.");
+  ok(!browser.isRemoteBrowser, "Browser should not be remote.");
+
+  // Go back to content page.
+  ok(gBrowser.webNavigation.canGoBack, "Should be able to go back.");
+  gBrowser.goBack();
+  await BrowserTestUtils.browserLoaded(browser);
+  await injectBeforeUnload(browser);
+
+  // Test that going forward triggers beforeunload prompt as well.
+  ok(gBrowser.webNavigation.canGoForward, "Should be able to go forward.");
+  let dialogShown2 = awaitAndCloseBeforeUnloadDialog(false);
+  gBrowser.goForward();
+  await Promise.all([dialogShown2, BrowserTestUtils.browserLoaded(browser)]);
+  is(beforeUnloadCount, 2, "Should have received two beforeunload events.");
+
+  BrowserTestUtils.removeTab(tab);
+});
+
+/**
+ * Test navigation from a chrome page to a content page. Also check that only
+ * one beforeunload event is fired.
+ */
+add_task(async function () {
+  let beforeUnloadCount = 0;
+  messageManager.addMessageListener("Test:OnBeforeUnloadReceived", function () {
+    beforeUnloadCount++;
+  });
+
+  // Open a chrome page.
+  let tab = await BrowserTestUtils.openNewForegroundTab(
+    gBrowser,
+    "about:support"
+  );
+  let browser = tab.linkedBrowser;
+
+  ok(!browser.isRemoteBrowser, "Browser should not be remote.");
+  await ContentTask.spawn(browser, null, async function () {
+    content.window.addEventListener(
+      "beforeunload",
+      function (event) {
+        sendAsyncMessage("Test:OnBeforeUnloadReceived");
+        var str = "Leaving?";
+        event.returnValue = str;
+        return str;
+      },
+      true
+    );
+  });
+
+  // Navigate to a content page.
+  let dialogShown1 = awaitAndCloseBeforeUnloadDialog(false);
+  BrowserTestUtils.loadURIString(browser, TEST_URL);
+  await Promise.all([dialogShown1, BrowserTestUtils.browserLoaded(browser)]);
+  is(beforeUnloadCount, 1, "Should have received one beforeunload event.");
+  ok(browser.isRemoteBrowser, "Browser should be remote.");
+
+  // Go back to chrome page.
+  ok(gBrowser.webNavigation.canGoBack, "Should be able to go back.");
+  gBrowser.goBack();
+  await BrowserTestUtils.browserLoaded(browser);
+  await ContentTask.spawn(browser, null, async function () {
+    content.window.addEventListener(
+      "beforeunload",
+      function (event) {
+        sendAsyncMessage("Test:OnBeforeUnloadReceived");
+        var str = "Leaving?";
+        event.returnValue = str;
+        return str;
+      },
+      true
+    );
+  });
+
+  // Test that going forward triggers beforeunload prompt as well.
+  ok(gBrowser.webNavigation.canGoForward, "Should be able to go forward.");
+  let dialogShown2 = awaitAndCloseBeforeUnloadDialog(false);
+  gBrowser.goForward();
+  await Promise.all([dialogShown2, BrowserTestUtils.browserLoaded(browser)]);
+  is(beforeUnloadCount, 2, "Should have received two beforeunload events.");
+
+  BrowserTestUtils.removeTab(tab);
+});