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
"use strict";
const { ensure_table_view } = ChromeUtils.importESModule(
);
let gAccount, gRootFolder, gSubFolders, gDefaultTabmail;
add_setup(async () => {
gAccount = createAccount();
gRootFolder = gAccount.incomingServer.rootFolder;
await createSubfolder(gRootFolder, "test1");
await createSubfolder(gRootFolder, "test2");
await createSubfolder(gRootFolder, "test3");
gSubFolders = {};
for (const folder of gRootFolder.subFolders) {
gSubFolders[folder.name] = folder;
}
await createMessages(gSubFolders.test1, 10);
await createMessages(gSubFolders.test2, 50);
await createMessages(gSubFolders.test3, 5);
await createMessages(gSubFolders.test3, {
count: 3,
msgsPerThread: 3,
});
await createMessages(gSubFolders.test3, 5);
await createMessages(gSubFolders.test3, {
count: 4,
msgsPerThread: 4,
});
await createMessages(gSubFolders.test3, 5);
gDefaultTabmail = document.getElementById("tabmail");
gDefaultTabmail.currentTabInfo.folder = gRootFolder;
await ensure_table_view(document);
});
add_task(async function test_setSelectedMessages() {
async function background() {
const [accountId] = await window.waitForMessage();
const { rootFolder: accRootFolder } = await browser.accounts.get(
accountId,
true
);
const allTabs = await browser.tabs.query({});
const queryTabs = await browser.tabs.query({ type: "mail" });
const allMailTabs = await browser.mailTabs.query({});
// Helper function to make sure the entire list has been awaited, before
// the test ends.
async function pullEntireList(listPromise) {
const msgs = [];
let list = await listPromise;
while (list) {
for (const m of list.messages) {
msgs.push(m);
}
if (!list.id) {
break;
}
list = await browser.messages.continueList(list.id);
}
return msgs;
}
const folder1 = accRootFolder.subFolders.find(f => f.path == "/test1");
const folder2 = accRootFolder.subFolders.find(f => f.path == "/test2");
const messages1 = await pullEntireList(browser.messages.list(folder1.id));
browser.test.assertTrue(
messages1.length > 7,
"There should be more than 7 messages in /test1"
);
const messages2 = await pullEntireList(browser.messages.list(folder2.id));
browser.test.assertTrue(
messages2.length > 4,
"There should be more than 4 messages in /test2"
);
browser.test.assertEq(3, allMailTabs.length);
browser.test.assertEq(5, allTabs.length);
browser.test.assertEq(3, queryTabs.length);
const foregroundTabId = allMailTabs[1].tabId;
browser.test.assertEq(accountId, allMailTabs[1].displayedFolder.accountId);
browser.test.assertEq("/test1", allMailTabs[1].displayedFolder.path);
browser.test.assertTrue(allMailTabs[1].active);
const backgroundTabId = allMailTabs[2].tabId;
browser.test.assertEq(accountId, allMailTabs[2].displayedFolder.accountId);
browser.test.assertEq("/", allMailTabs[2].displayedFolder.path);
// Check the initial real state.
await window.sendMessage("checkRealLayout", {
messagePaneVisible: true,
folderPaneVisible: true,
displayedFolder: "/test1",
});
// Change the selection in the foreground tab.
await browser.mailTabs.setSelectedMessages(foregroundTabId, [
messages1.at(-7).id,
messages1.at(-8).id,
]);
// Check the current real state.
await window.sendMessage("checkRealLayout", {
messagePaneVisible: true,
folderPaneVisible: true,
displayedFolder: "/test1",
});
// Check API return value of the foreground tab.
const { messages: readMessagesA } =
await browser.mailTabs.getSelectedMessages(foregroundTabId);
window.assertDeepEqual(
[messages1.at(-7).id, messages1.at(-8).id],
readMessagesA.map(m => m.id)
);
// Change the selection in the background tab.
await browser.mailTabs.setSelectedMessages(backgroundTabId, [
messages2.at(-1).id,
messages2.at(-4).id,
]);
// Real state should be the same, since we're updating a background tab.
await window.sendMessage("checkRealLayout", {
messagePaneVisible: true,
folderPaneVisible: true,
displayedFolder: "/test1",
});
// Check unchanged API return value of the foreground tab.
const { messages: readMessagesB } =
await browser.mailTabs.getSelectedMessages(foregroundTabId);
window.assertDeepEqual(
[messages1.at(-7).id, messages1.at(-8).id],
readMessagesB.map(m => m.id)
);
// Check API return value of the inactive background tab.
const { messages: readMessagesC } =
await browser.mailTabs.getSelectedMessages(backgroundTabId);
window.assertDeepEqual(
[messages2.at(-1).id, messages2.at(-4).id],
readMessagesC.map(m => m.id)
);
// Switch to the background tab.
await browser.tabs.update(backgroundTabId, { active: true });
// Check API return value of the background tab (now active).
const { messages: readMessagesD } =
await browser.mailTabs.getSelectedMessages(backgroundTabId);
window.assertDeepEqual(
[messages2.at(-1).id, messages2.at(-4).id],
readMessagesD.map(m => m.id)
);
// Check real state, should now match the active background tab.
await window.sendMessage("checkRealLayout", {
messagePaneVisible: true,
folderPaneVisible: true,
displayedFolder: "/test2",
});
// Check unchanged API return value of the foreground tab (now inactive).
const { messages: readMessagesE } =
await browser.mailTabs.getSelectedMessages(foregroundTabId);
window.assertDeepEqual(
[messages1.at(-7).id, messages1.at(-8).id],
readMessagesE.map(m => m.id)
);
// Switch back to the foreground tab.
await browser.tabs.update(foregroundTabId, { active: true });
// Change the selection in the foreground tab.
await browser.mailTabs.setSelectedMessages(foregroundTabId, [
messages2.at(-3).id,
messages2.at(-5).id,
]);
// Check API return value of the foreground tab.
const { messages: readMessagesF } =
await browser.mailTabs.getSelectedMessages(foregroundTabId);
window.assertDeepEqual(
[messages2.at(-3).id, messages2.at(-5).id],
readMessagesF.map(m => m.id)
);
// Check real state.
await window.sendMessage("checkRealLayout", {
messagePaneVisible: true,
folderPaneVisible: true,
displayedFolder: "/test2",
});
// Check API return value of the inactive background tab.
const { messages: readMessagesG } =
await browser.mailTabs.getSelectedMessages(backgroundTabId);
window.assertDeepEqual(
[messages2.at(-1).id, messages2.at(-4).id],
readMessagesG.map(m => m.id)
);
// Clear selection in background tab.
await browser.mailTabs.setSelectedMessages(backgroundTabId, []);
// Check API return value of the inactive background tab.
const { messages: readMessagesH } =
await browser.mailTabs.getSelectedMessages(backgroundTabId);
browser.test.assertEq(0, readMessagesH.length);
// Clear selection in foreground tab.
await browser.mailTabs.setSelectedMessages(foregroundTabId, []);
// Check API return value of the foreground tab.
const { messages: readMessagesI } =
await browser.mailTabs.getSelectedMessages(foregroundTabId);
browser.test.assertEq(0, readMessagesI.length);
// Should throw if messages belong to different folders.
await browser.test.assertRejects(
browser.mailTabs.setSelectedMessages(foregroundTabId, [
messages2.at(-3).id,
messages1.at(-5).id,
]),
/Requested messages are not in the same folder and are also not in the current view/,
"browser.mailTabs.setSelectedMessages() should reject, if the requested message do not belong to the same folder."
);
browser.test.notifyPass("mailTabs");
}
const extension = ExtensionTestUtils.loadExtension({
files: {
"background.js": background,
"utils.js": await getUtilsJS(),
},
manifest: {
manifest_version: 3,
background: { scripts: ["utils.js", "background.js"] },
permissions: ["accountsRead", "messagesRead"],
},
});
extension.onMessage("checkRealLayout", async expected => {
await check3PaneState(
expected.folderPaneVisible,
expected.messagePaneVisible
);
Assert.equal(
"/" + (gDefaultTabmail.currentTabInfo.folder.URI || "").split("/").pop(),
expected.displayedFolder,
"Should display the correct folder"
);
extension.sendMessage();
});
window.openContentTab("about:buildconfig");
window.openContentTab("about:mozilla");
gDefaultTabmail.openTab("mail3PaneTab", { folderURI: gSubFolders.test1.URI });
gDefaultTabmail.openTab("mail3PaneTab", {
folderURI: gRootFolder.URI,
background: true,
});
await BrowserTestUtils.waitForEvent(
gDefaultTabmail.currentTabInfo.chromeBrowser,
"folderURIChanged",
false,
event => event.detail == gSubFolders.test1.URI
);
await extension.startup();
extension.sendMessage(gAccount.key);
await extension.awaitFinish("mailTabs");
await extension.unload();
gDefaultTabmail.closeOtherTabs(0);
gDefaultTabmail.currentTabInfo.folder = gRootFolder;
});
add_task(async function test_setSelectedMessages_with_thread() {
async function background() {
const [testFolder] = await browser.folders.query({ name: "test3" });
await browser.mailTabs.update({ displayedFolderId: testFolder.id });
// Get the listed messages (some messages are collapsed and not visible).
const { messages: listedMessages } =
await browser.mailTabs.getListedMessages();
browser.test.assertEq(
17,
listedMessages.length,
"Should find the correct number of messages in collapsed threads"
);
// Get all messages in the test folder.
const { messages: allMessages } = await browser.messages.list(
testFolder.id
);
browser.test.assertEq(
22,
allMessages.length,
"Should find the correct number of messages in the folder"
);
// Find the collapsed messages.
const listedMessagesIds = listedMessages.map(m => m.id);
const collapsedMessages = allMessages.filter(
m => !listedMessagesIds.includes(m.id)
);
browser.test.assertEq(
5,
collapsedMessages.length,
"Should find the correct number of collapsed messages"
);
// Select the first, the last and the collapsed messages. Expanding the hidden
// threads should not cause the wrong "last" message to be selected (see bug
// 1953713).
const testMessages = [
listedMessages[0],
listedMessages[16],
...collapsedMessages,
];
await browser.mailTabs.setSelectedMessages(testMessages.map(m => m.id));
const { messages: selectedMessages } =
await browser.mailTabs.getSelectedMessages();
window.assertDeepEqual(
testMessages.map(m => m.id).sort(),
selectedMessages.map(m => m.id).sort(),
"The correct messages should be selected"
);
browser.test.notifyPass("mailTabs");
}
const extension = ExtensionTestUtils.loadExtension({
files: {
"background.js": background,
"utils.js": await getUtilsJS(),
},
manifest: {
manifest_version: 3,
background: { scripts: ["utils.js", "background.js"] },
permissions: ["accountsRead", "messagesRead"],
},
});
await extension.startup();
await extension.awaitFinish("mailTabs");
await extension.unload();
});