gecko-dev/toolkit/components/startup/tests/browser/browser_bug511456.js

/* 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_URL = "http://example.com/browser/toolkit/components/startup/tests/browser/beforeunload.html";

SpecialPowers.pushPrefEnv({"set": [["dom.require_user_interaction_for_beforeunload", false]]});

function test() {
  waitForExplicitFinish();
  ignoreAllUncaughtExceptions();

  let win2 = window.openDialog(location, "", "chrome,all,dialog=no", "about:blank");
  win2.addEventListener("load", function onLoad() {
    win2.removeEventListener("load", onLoad);

    gBrowser.selectedTab = gBrowser.addTab(TEST_URL);
    let browser = gBrowser.selectedBrowser;

    whenBrowserLoaded(browser, function () {
      let seenDialog = false;

      // Cancel the prompt the first time.
      waitForOnBeforeUnloadDialog(browser, (btnLeave, btnStay) => {
        seenDialog = true;
        btnStay.click();
      });

      let appStartup = Cc['@mozilla.org/toolkit/app-startup;1'].
                         getService(Ci.nsIAppStartup);
      appStartup.quit(Ci.nsIAppStartup.eAttemptQuit);
      ok(seenDialog, "Should have seen a prompt dialog");
      ok(!win2.closed, "Shouldn't have closed the additional window");
      win2.close();

      // Leave the page the second time.
      waitForOnBeforeUnloadDialog(browser, (btnLeave, btnStay) => {
        btnLeave.click();
      });

      gBrowser.removeTab(gBrowser.selectedTab);
      executeSoon(finish);
    });
  });
}