fune/browser/base/content/test/tabPrompts/browser_beforeunload_urlbar.js

/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

const TEST_ROOT = getRootDirectory(gTestPath).replace(
  "chrome://mochitests/content",
  "http://example.com"
);

add_task(async function test_beforeunload_stay_clears_urlbar() {
  await SpecialPowers.pushPrefEnv({
    set: [["dom.require_user_interaction_for_beforeunload", false]],
  });
  await SpecialPowers.pushPrefEnv({
    set: [["prompts.contentPromptSubDialog", false]],
  });

  const TEST_URL = TEST_ROOT + "file_beforeunload_stop.html";
  await BrowserTestUtils.withNewTab(TEST_URL, async function(browser) {
    gURLBar.focus();
    const inputValue = "http://example.org/?q=typed";
    gURLBar.inputField.value = inputValue.slice(0, -1);
    EventUtils.sendString(inputValue.slice(-1));

    let promptOpenedPromise = TestUtils.topicObserved("tabmodal-dialog-loaded");
    EventUtils.synthesizeKey("VK_RETURN");
    await promptOpenedPromise;
    let promptElement = browser.parentNode.querySelector("tabmodalprompt");

    // Click the cancel button
    promptElement.querySelector(".tabmodalprompt-button1").click();

    await TestUtils.waitForCondition(
      () => promptElement.parentNode == null,
      "tabprompt should be removed"
    );
    // Can't just compare directly with TEST_URL because the URL may be trimmed.
    // Just need it to not be the example.org thing we typed in.
    ok(
      gURLBar.value.endsWith("_stop.html"),
      "Url bar should be reset to point to the stop html file"
    );
    ok(
      gURLBar.value.includes("example.com"),
      "Url bar should be reset to example.com"
    );
    // Check the lock/identity icons are back:
    is(
      gURLBar.textbox.getAttribute("pageproxystate"),
      "valid",
      "Should be in valid pageproxy state."
    );

    // Now we need to get rid of the handler to avoid the prompt coming up when trying to close the
    // tab when we exit `withNewTab`. :-)
    await SpecialPowers.spawn(browser, [], function() {
      content.window.onbeforeunload = null;
    });
  });
});