Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
/* Any copyright is dedicated to the Public Domain.
"use strict";
const { PromptTestUtils } = ChromeUtils.importESModule(
);
const { TabStateFlusher } = ChromeUtils.importESModule(
"resource:///modules/sessionstore/TabStateFlusher.sys.mjs"
);
const TEST_ROOT = getRootDirectory(gTestPath).replace(
);
async function addTab(url) {
let tab = await BrowserTestUtils.openNewForegroundTab({
gBrowser,
url,
animate: false,
});
return tab;
}
async function addBeforeUnloadTab() {
let tab = await addTab(TEST_ROOT + "file_beforeunload_stop.html");
await BrowserTestUtils.synthesizeMouseAtCenter(
"body",
{},
tab.linkedBrowser.browsingContext
);
Assert.ok(
tab.linkedBrowser.hasBeforeUnload,
"Added tab has a beforeUnload prompt"
);
return tab;
}
/**
* When closing a group with a tab prompt and accepting the prompt,
* the group should close.
*/
add_task(async function test_closeGroupAndAcceptPrompt() {
await SimpleTest.requestCompleteLog();
let tab1 = await addTab("about:mozilla");
let tab2 = await addBeforeUnloadTab();
let group = gBrowser.addTabGroup([tab1, tab2]);
let promptHandled = PromptTestUtils.handleNextPrompt(
window,
{
modalType: Services.prompt.MODAL_TYPE_CONTENT,
},
{
buttonNumClick: 0, // 0 = leave page (accept), 1= stay on page (cancel)
}
);
let groupRemoved = BrowserTestUtils.waitForEvent(group, "TabGroupRemoved");
gBrowser.removeTabGroup(group);
await Promise.allSettled([promptHandled, groupRemoved]);
Assert.ok(!gBrowser.getAllTabGroups().length, "Tab group was removed");
});
/**
* When closing a group with a tab prompt and rejecting the prompt,
* all tabs in the group should remain open.
*/
add_task(async function test_closeGroupAndRejectPrompt() {
let tab1 = await addTab("about:mozilla");
let tab2 = await addBeforeUnloadTab();
let groupCreated = BrowserTestUtils.waitForEvent(window, "TabGroupCreate");
let group = gBrowser.addTabGroup([tab1, tab2]);
await groupCreated;
let promptHandled = PromptTestUtils.handleNextPrompt(
tab2.linkedBrowser,
{
modalType: Services.prompt.MODAL_TYPE_CONTENT,
},
{
buttonNumClick: 1, // 0 = leave page (accept), 1= stay on page (cancel)
}
);
gBrowser.removeTabGroup(group);
await promptHandled;
Assert.equal(gBrowser.getAllTabGroups().length, 1, "Tab group is still open");
Assert.equal(group.tabs.length, 2, "Both tabs in group remain open");
let groupRemoved = BrowserTestUtils.waitForEvent(group, "TabGroupRemoved");
gBrowser.removeTabGroup(group, { skipPermitUnload: true });
await groupRemoved;
});
/**
* When closing a group with multiple tab prompts and rejecting ANY of them,
* all tabs in the group should remain open.
*/
add_task(async function test_closeGroupAndRejectAnyPrompt() {
// maybe disable this test and try running a11y tests
let tab1 = await addBeforeUnloadTab();
let tab2 = await addBeforeUnloadTab();
let group = gBrowser.addTabGroup([tab1, tab2]);
let firstPromptHandled = PromptTestUtils.handleNextPrompt(
tab1.linkedBrowser,
{
modalType: Services.prompt.MODAL_TYPE_CONTENT,
},
{
buttonNumClick: 0, // 0 = leave page (accept), 1= stay on page (cancel)
}
);
let secondPromptHandled = PromptTestUtils.handleNextPrompt(
tab2.linkedBrowser,
{
modalType: Services.prompt.MODAL_TYPE_CONTENT,
},
{
buttonNumClick: 1, // 0 = leave page (accept), 1= stay on page (cancel)
}
);
let groupRemoved = gBrowser.removeTabGroup(group);
await Promise.race([firstPromptHandled, secondPromptHandled]);
await groupRemoved;
Assert.equal(gBrowser.getAllTabGroups().length, 1, "Tab group is still open");
Assert.equal(group.tabs.length, 2, "Both tabs in group remain open");
groupRemoved = BrowserTestUtils.waitForEvent(group, "TabGroupRemoved");
gBrowser.removeTabGroup(group, { skipPermitUnload: true });
await groupRemoved;
});
/**
* When a bunch of tabs are closed, e.g. when using "close tabs to the right",
* any whole tabgroups with beforeunload handlers should be kept open if
* cancelled by the user.
*/
add_task(async function test_closeMultipleTabsAndRejectFromGroup() {
let group1Tab1 = await addTab("about:mozilla");
let group1Tab2 = await addBeforeUnloadTab();
let group2Tab1 = await addBeforeUnloadTab();
let group2Tab2 = await addTab("about:mozilla");
let group1 = gBrowser.addTabGroup([group1Tab1, group1Tab2], {
id: "leave-open",
label: "leave open",
});
gBrowser.addTabGroup([group2Tab1, group2Tab2], {
id: "close",
label: "close",
});
let firstPromptHandled = PromptTestUtils.handleNextPrompt(
group1Tab2.linkedBrowser,
{
modalType: Services.prompt.MODAL_TYPE_CONTENT,
},
{
buttonNumClick: 1, // 0 = leave page (accept), 1= stay on page (cancel)
}
);
let secondPromptHandled = PromptTestUtils.handleNextPrompt(
group2Tab1.linkedBrowser,
{
modalType: Services.prompt.MODAL_TYPE_CONTENT,
},
{
buttonNumClick: 0, // 0 = leave page (accept), 1= stay on page (cancel)
}
);
gBrowser.removeTabsToTheEndFrom(gBrowser.tabs[0]);
await Promise.allSettled([firstPromptHandled, secondPromptHandled]);
// initial tab and 1 tab group should remain open
await BrowserTestUtils.waitForCondition(
() => gBrowser.tabs.length == 3,
"3 tabs remain open"
);
Assert.equal(
gBrowser.getAllTabGroups()[0].id,
group1.id,
"Group 1 is still open"
);
await gBrowser.removeTabGroup(group1, { skipPermitUnload: true });
});
/**
* When a tab group is saved & closed, it should not end up saved if the user
* cancels the group close.
*/
add_task(async function test_cancelClosingASavingGroup() {
let tab1 = await addBeforeUnloadTab();
let tab2 = await addTab("about:mozilla");
await TabStateFlusher.flush(tab1.linkedBrowser);
let group = gBrowser.addTabGroup([tab1, tab2]);
group.save();
Assert.equal(
SessionStore.getCurrentState().savedGroups.length,
1,
"Group is saved"
);
let promptHandled = PromptTestUtils.handleNextPrompt(
tab1.linkedBrowser,
{
modalType: Services.prompt.MODAL_TYPE_CONTENT,
},
{
buttonNumClick: 1, // 0 = leave page (accept), 1= stay on page (cancel)
}
);
let groupRemoved = gBrowser.removeTabGroup(group);
await Promise.allSettled([promptHandled, groupRemoved]);
Assert.equal(gBrowser.getAllTabGroups().length, 1, "Group is still open");
Assert.ok(
!SessionStore.getCurrentState().savedGroups.length,
"Group is no longer saved after close was canceled"
);
await gBrowser.removeTabGroup(group, { skipPermitUnload: true });
});