gecko-dev/dom/tests/browser/browser_beforeunload_between_chrome_content.js

const TEST_URL = "http://www.example.com/browser/dom/tests/browser/dummy.html";

const { PromptTestUtils } = ChromeUtils.import(
  "resource://testing-common/PromptTestUtils.jsm"
);

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.loadURI(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.loadURI(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);
});