1 files changed, 30 insertions, 0 deletions
diff --git a/toolkit/components/startup/tests/browser/head.js b/toolkit/components/startup/tests/browser/head.js
new file mode 100644
index 0000000000..55c4528e13
--- /dev/null
+++ b/toolkit/components/startup/tests/browser/head.js
@@ -0,0 +1,30 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+SpecialPowers.pushPrefEnv({
+  set: [["prompts.contentPromptSubDialog", false]],
+});
+
+function whenBrowserLoaded(browser, callback) {
+  return BrowserTestUtils.browserLoaded(browser).then(callback);
+}
+
+function waitForOnBeforeUnloadDialog(browser, callback) {
+  browser.addEventListener(
+    "DOMWillOpenModalDialog",
+    function onModalDialog(event) {
+      SimpleTest.waitForCondition(
+        () => Services.focus.activeWindow == browser.ownerGlobal,
+        function() {
+          let prompt = browser.tabModalPromptBox.listPrompts()[0];
+          let { button0, button1 } = prompt.ui;
+          callback(button0, button1);
+        },
+        "Waited too long for window with dialog to focus"
+      );
+    },
+    { capture: true, once: true }
+  );
+}