1 files changed, 58 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 0000000000..d8b99417d0
--- /dev/null
+++ b/toolkit/components/startup/tests/browser/browser_bug511456.js
@@ -0,0 +1,58 @@
+/* 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();
+
+  // Create foreground window
+  let win2 = window.openDialog(
+    location,
+    "",
+    "chrome,all,dialog=no",
+    "about:blank"
+  );
+  win2.addEventListener(
+    "load",
+    function() {
+      // Create background test tab
+      let browser = BrowserTestUtils.addTab(gBrowser, TEST_URL).linkedBrowser;
+
+      whenBrowserLoaded(browser, function() {
+        let seenDialog = false;
+
+        // Cancel the prompt the first time.
+        waitForOnBeforeUnloadDialog(browser, (btnLeave, btnStay) => {
+          seenDialog =
+            Services.focus.activeWindow == window &&
+            gBrowser.selectedBrowser == browser;
+          btnStay.click();
+        });
+
+        Services.startup.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);
+      });
+    },
+    { once: true }
+  );
+}