Revision control
Copy as Markdown
Other Tools
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
/* globals messenger */
const { MessageGenerator } = ChromeUtils.importESModule(
);
const tabmail = document.getElementById("tabmail");
const about3Pane = tabmail.currentAbout3Pane;
const { threadTree } = about3Pane;
// The number of messages to generate in the (only) thread.
const messagesInThread = 5;
add_setup(async function () {
Services.prefs.setBoolPref("mailnews.scroll_to_new_message", false);
// Create an account for the test.
const account = MailServices.accounts.createLocalMailAccount();
account.addIdentity(MailServices.accounts.createIdentity());
// Remove test account on cleanup.
registerCleanupFunction(() => {
// Tests following this one may attempt to create a folder at the same URI
// and will fail because our folder lookup code is a mess. Renaming should
// prevent that.
const archiveFolder = rootFolder.getFolderWithFlags(
Ci.nsMsgFolderFlags.Archive
);
archiveFolder?.subFolders[0]?.rename("archive2000", null);
archiveFolder?.rename("archiveArchives", null);
MailServices.accounts.removeAccount(account, false);
// Clear the undo and redo stacks to avoid side-effects on
// tests expecting them to start in a cleared state.
messenger.transactionManager.clear();
Services.prefs.setBoolPref("mailnews.scroll_to_new_message", true);
});
// Create a folder for the account to store test messages.
const rootFolder = account.incomingServer.rootFolder.QueryInterface(
Ci.nsIMsgLocalMailFolder
);
const testFolder = rootFolder
.createLocalSubfolder("test-archive")
.QueryInterface(Ci.nsIMsgLocalMailFolder);
// Generate test messages.
const generator = new MessageGenerator();
testFolder.addMessageBatch(
generator
.makeMessages({
count: messagesInThread,
msgsPerThread: messagesInThread,
})
.map(message => message.toMessageString())
);
// Use the test folder.
about3Pane.displayFolder(testFolder.URI);
});
/**
* Tests undoing after archiving a thread.
*/
add_task(async function testArchiveUndo() {
const row = threadTree.getRowAtIndex(0);
// Simulate a click on the row's subject line to select the row.
const selectPromise = BrowserTestUtils.waitForEvent(threadTree, "select");
EventUtils.synthesizeMouseAtCenter(
row.querySelector(".thread-card-subject-container"),
{ clickCount: 1 },
about3Pane
);
await selectPromise;
// Make sure the thread is selected
Assert.ok(
row.classList.contains("selected"),
"The thread row should be selected"
);
// Archive the message.
EventUtils.synthesizeKey("a");
// Make sure the thread was removed from the thread tree.
await TestUtils.waitForCondition(
() => threadTree.getRowAtIndex(0) === null,
"The thread tree should not have any row"
);
// Undo the operation.
EventUtils.synthesizeKey("z", { accelKey: true });
// Make sure the thread makes it back to the thread tree.
// We want to make sure we've finished moving every message before
// finishing the test.
await TestUtils.waitForCondition(() => {
// The thread will be collapsed by default, in which case
// nothing will exist beyond index 0. So we need to expand
// the thread before we check.
goDoCommand("cmd_expandAllThreads");
// Make sure every message has successfully made it back from the
// archive folder.
for (let i = 0; i < messagesInThread; i++) {
if (threadTree.getRowAtIndex(i) === null) {
return false;
}
}
return true;
}, "The thread should have returned back from the archive");
});