Revision control
Copy as Markdown
Other Tools
/* Any copyright is dedicated to the Public Domain.
"use strict";
const { ForcedQuitHandler } = ChromeUtils.importESModule(
);
const { close_compose_window, open_compose_new_mail } =
ChromeUtils.importESModule(
);
const { get_special_folder } = ChromeUtils.importESModule(
);
let gDraftFolder;
add_setup(async function () {
gDraftFolder = await get_special_folder(Ci.nsMsgFolderFlags.Drafts, true);
});
add_task(async function test_forced_shutdown_flushes_compose_windows() {
const initialDraftKeys = new Set(
Array.from(gDraftFolder.messages, message => message.messageKey)
);
const subject = "Enterprise forced shutdown draft";
const modifiedCompose = await open_compose_new_mail(window);
const unmodifiedCompose = await open_compose_new_mail(window);
registerCleanupFunction(async () => {
for (const composeWindow of [modifiedCompose, unmodifiedCompose]) {
if (!composeWindow.closed) {
await close_compose_window(composeWindow);
}
}
});
modifiedCompose.document.getElementById("msgSubject").focus();
EventUtils.sendString(subject, modifiedCompose);
modifiedCompose.document.getElementById("messageEditor").focus();
EventUtils.sendString(
"Content preserved during forced shutdown.",
modifiedCompose
);
const hook = ForcedQuitHandler._appForcedQuitHook();
Assert.equal(
typeof hook,
"function",
"The Thunderbird forced-quit hook should be registered"
);
await hook(Ci.nsIAppStartup.eForceQuit);
Assert.ok(
!modifiedCompose.closed,
"Preparing for forced shutdown does not close the modified compose window"
);
Assert.ok(
!unmodifiedCompose.closed,
"Preparing for forced shutdown does not close the unmodified compose window"
);
for (const [kind, composeWindow] of [
["modified", modifiedCompose],
["unmodified", unmodifiedCompose],
]) {
Assert.equal(
composeWindow.document.documentElement.dataset.enterpriseForcedShutdown,
"true",
`The ${kind} compose window is prepared for the forced quit`
);
Assert.ok(
composeWindow.ComposeCanClose(),
`The ${kind} compose window does not block the forced quit`
);
Assert.ok(
!(
"enterpriseForcedShutdown" in
composeWindow.document.documentElement.dataset
),
`The ${kind} compose window consumed its forced-close marker`
);
}
await TestUtils.waitForCondition(
() => gDraftFolder.getTotalMessages(false) == initialDraftKeys.size + 1,
"Waiting for the forced-shutdown draft to be created"
);
const newDrafts = Array.from(gDraftFolder.messages).filter(
message => !initialDraftKeys.has(message.messageKey)
);
Assert.equal(
newDrafts.length,
1,
"Only the modified compose window created a draft"
);
Assert.equal(
newDrafts[0]?.subject,
subject,
"The forced-shutdown draft has the expected subject"
);
await close_compose_window(modifiedCompose);
await close_compose_window(unmodifiedCompose);
});