toolkit/components/startup/tests/browser/head.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

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

"use strict";

const { PromptTestUtils } = ChromeUtils.importESModule(
  "resource://testing-common/PromptTestUtils.sys.mjs"
);

function whenBrowserLoaded(browser, callback) {
  return BrowserTestUtils.browserLoaded(browser).then(callback);
}

function waitForOnBeforeUnloadDialog(browser, callback) {
  PromptTestUtils.waitForPrompt(browser, {
    modalType: Services.prompt.MODAL_TYPE_CONTENT,
    promptType: "confirmEx",
  }).then(dialog => {
    SimpleTest.waitForCondition(
      () => Services.focus.activeWindow == browser.ownerGlobal,
      function () {
        let { button0, button1 } = dialog.ui;
        callback(button0, button1);
      },
      "Waited too long for window with dialog to focus"
    );
  });
}