1 files changed, 47 insertions, 0 deletions
diff --git a/toolkit/components/startup/tests/browser/browser_bug511456.js b/toolkit/components/startup/tests/browser/browser_bug511456.js
new file mode 100644
index 000000000..652a34ea2
--- /dev/null
+++ b/toolkit/components/startup/tests/browser/browser_bug511456.js
@@ -0,0 +1,47 @@
+/* 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);
+    });
+  });
+}