Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
- Manifest: dom/tests/browser/browser.toml
const { PromptTestUtils } = ChromeUtils.importESModule(
);
function pageScript() {
window.addEventListener(
"beforeunload",
function (event) {
var str = "Leaving?";
event.returnValue = str;
return str;
},
true
);
}
function injectBeforeUnload(browser) {
return ContentTask.spawn(browser, null, async function () {
content.window.addEventListener(
"beforeunload",
function (event) {
sendAsyncMessage("Test:OnBeforeUnloadReceived");
var str = "Leaving?";
event.returnValue = str;
return str;
},
true
);
});
}
// Wait for onbeforeunload dialog, and dismiss it immediately.
function awaitAndCloseBeforeUnloadDialog(browser, doStayOnPage) {
return PromptTestUtils.handleNextPrompt(
browser,
{ modalType: Services.prompt.MODAL_TYPE_CONTENT, promptType: "confirmEx" },
{ buttonNumClick: doStayOnPage ? 1 : 0 }
);
}
SpecialPowers.pushPrefEnv({
set: [["dom.require_user_interaction_for_beforeunload", false]],
});
/**
* Test navigation from a content page to a chrome page. Also check that only
* one beforeunload event is fired.
*/
/* global messageManager */
add_task(async function () {
let beforeUnloadCount = 0;
messageManager.addMessageListener("Test:OnBeforeUnloadReceived", function () {
beforeUnloadCount++;
});
// Open a content page.
let tab = await BrowserTestUtils.openNewForegroundTab(gBrowser, TEST_URL);
let browser = tab.linkedBrowser;
ok(browser.isRemoteBrowser, "Browser should be remote.");
await injectBeforeUnload(browser);
// Navigate to a chrome page.
let dialogShown1 = awaitAndCloseBeforeUnloadDialog(browser, false);
BrowserTestUtils.startLoadingURIString(browser, "about:support");
await Promise.all([dialogShown1, BrowserTestUtils.browserLoaded(browser)]);
is(beforeUnloadCount, 1, "Should have received one beforeunload event.");
ok(!browser.isRemoteBrowser, "Browser should not be remote.");
// Go back to content page.
ok(gBrowser.webNavigation.canGoBack, "Should be able to go back.");
gBrowser.goBack();
await BrowserTestUtils.browserLoaded(browser);
await injectBeforeUnload(browser);
// Test that going forward triggers beforeunload prompt as well.
ok(gBrowser.webNavigation.canGoForward, "Should be able to go forward.");
let dialogShown2 = awaitAndCloseBeforeUnloadDialog(false);
gBrowser.goForward();
await Promise.all([dialogShown2, BrowserTestUtils.browserLoaded(browser)]);
is(beforeUnloadCount, 2, "Should have received two beforeunload events.");
BrowserTestUtils.removeTab(tab);
});
/**
* Test navigation from a chrome page to a content page. Also check that only
* one beforeunload event is fired.
*/
add_task(async function () {
let beforeUnloadCount = 0;
messageManager.addMessageListener("Test:OnBeforeUnloadReceived", function () {
beforeUnloadCount++;
});
// Open a chrome page.
let tab = await BrowserTestUtils.openNewForegroundTab(
gBrowser,
"about:support"
);
let browser = tab.linkedBrowser;
ok(!browser.isRemoteBrowser, "Browser should not be remote.");
await ContentTask.spawn(browser, null, async function () {
content.window.addEventListener(
"beforeunload",
function (event) {
sendAsyncMessage("Test:OnBeforeUnloadReceived");
var str = "Leaving?";
event.returnValue = str;
return str;
},
true
);
});
// Navigate to a content page.
let dialogShown1 = awaitAndCloseBeforeUnloadDialog(false);
BrowserTestUtils.startLoadingURIString(browser, TEST_URL);
await Promise.all([dialogShown1, BrowserTestUtils.browserLoaded(browser)]);
is(beforeUnloadCount, 1, "Should have received one beforeunload event.");
ok(browser.isRemoteBrowser, "Browser should be remote.");
// Go back to chrome page.
ok(gBrowser.webNavigation.canGoBack, "Should be able to go back.");
gBrowser.goBack();
await BrowserTestUtils.browserLoaded(browser);
await ContentTask.spawn(browser, null, async function () {
content.window.addEventListener(
"beforeunload",
function (event) {
sendAsyncMessage("Test:OnBeforeUnloadReceived");
var str = "Leaving?";
event.returnValue = str;
return str;
},
true
);
});
// Test that going forward triggers beforeunload prompt as well.
ok(gBrowser.webNavigation.canGoForward, "Should be able to go forward.");
let dialogShown2 = awaitAndCloseBeforeUnloadDialog(false);
gBrowser.goForward();
await Promise.all([dialogShown2, BrowserTestUtils.browserLoaded(browser)]);
is(beforeUnloadCount, 2, "Should have received two beforeunload events.");
BrowserTestUtils.removeTab(tab);
});