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
/**
* Tests the live view adapter for displaying conversations. It only
* displays one row per conversation, and the rows can't be expanded.
* Ordering is only allowed by date (ascending or descending).
*/
const { LiveViewConversationsDataAdapter } = ChromeUtils.importESModule(
);
const LiveView = Components.Constructor(
"@mozilla.org/mailnews/live-view;1",
"nsILiveView"
);
const tree = new ListenerTree();
add_setup(async function () {
await installDBFromFile("db/conversations.sql");
});
/**
* Tests LiveViewConversationsDataAdapter, when sorted by date, descending.
*/
add_task(async function testDateDescending() {
const liveView = new LiveView();
const adapter = new LiveViewConversationsDataAdapter(liveView);
adapter.sortBy("date", "descending");
adapter.setTree(tree);
await tree.promiseInvalidated(0, 5);
try {
Assert.equal(adapter.rowCount, 6);
Assert.ok(!adapter.isContainer(0));
Assert.equal(adapter.rowAt(0).messageCount, 1);
Assert.equal(adapter.getCellText(0, "threadId"), "10");
Assert.ok(!adapter.isContainer(1));
Assert.equal(adapter.rowAt(1).messageCount, 1);
Assert.equal(adapter.getCellText(1, "threadId"), "9");
Assert.ok(!adapter.isContainer(2));
Assert.equal(adapter.rowAt(2).messageCount, 1);
Assert.equal(adapter.getCellText(2, "threadId"), "8");
Assert.ok(!adapter.isContainer(3));
Assert.equal(adapter.rowAt(3).messageCount, 3);
Assert.equal(adapter.getCellText(3, "threadId"), "3");
Assert.ok(!adapter.isContainer(4));
Assert.equal(adapter.rowAt(4).messageCount, 3);
Assert.equal(adapter.getCellText(4, "threadId"), "7");
Assert.ok(!adapter.isContainer(5));
Assert.equal(adapter.rowAt(5).messageCount, 1);
Assert.equal(adapter.getCellText(5, "threadId"), "1");
} finally {
adapter.setTree(null);
}
});
/**
* Test what happens when messages are added or removed.
*/
add_task(async function testAddRemoveDateDescending() {
const liveView = new LiveView();
const adapter = new LiveViewConversationsDataAdapter(liveView);
adapter.sortBy("date", "descending");
adapter.setTree(tree);
await tree.promiseInvalidated(0, 5);
try {
Assert.equal(adapter.rowCount, 6);
Assert.deepEqual(listMessages(adapter), [10, 9, 8, 6, 4, 1]);
// Add a message to a thread.
Assert.equal(adapter.rowAt(3).messageCount, 3);
Assert.equal(adapter.getCellText(3, "threadId"), 3);
const added1 = addMessage({
messageId: "added1@invalid",
references: ["message3@invalid"],
});
Assert.equal(added1, 11);
tree.assertInvalidated(3, 3);
Assert.equal(adapter.rowCount, 6);
Assert.ok(!adapter.isContainer(3));
Assert.equal(adapter.rowAt(3).messageCount, 4);
Assert.deepEqual(listMessages(adapter), [10, 9, 8, 6, 4, 1]);
Assert.equal(adapter.getCellText(3, "threadId"), 3);
// Add a message to a non-thread.
Assert.equal(adapter.rowAt(1).messageCount, 1);
Assert.equal(adapter.getCellText(1, "threadId"), 9);
const added4 = addMessage({
messageId: "added4@invalid",
references: ["message9@invalid"],
});
Assert.equal(added4, 12);
tree.assertInvalidated(1, 1);
Assert.equal(adapter.rowCount, 6);
Assert.ok(!adapter.isContainer(1));
Assert.equal(adapter.rowAt(1).messageCount, 2);
Assert.deepEqual(listMessages(adapter), [10, 9, 8, 6, 4, 1]);
Assert.equal(adapter.getCellText(1, "threadId"), 9);
// Add a message that isn't part of a thread, at the start.
const added5 = addMessage({
messageId: "added5@invalid",
date: "2026-03-03",
});
Assert.equal(added5, 13);
tree.assertRowCountChanged(0, 1);
Assert.equal(adapter.rowCount, 7);
Assert.ok(!adapter.isContainer(0));
Assert.equal(adapter.rowAt(0).messageCount, 1);
Assert.deepEqual(listMessages(adapter), [13, 10, 9, 8, 6, 4, 1]);
Assert.equal(adapter.getCellText(0, "threadId"), 13);
// Add another message that isn't part of a thread, in the middle.
const added6 = addMessage({
messageId: "added6@invalid",
date: "2025-09-07",
});
Assert.equal(added6, 14);
tree.assertRowCountChanged(4, 1);
Assert.equal(adapter.rowCount, 8);
Assert.ok(!adapter.isContainer(4));
Assert.equal(adapter.rowAt(4).messageCount, 1);
Assert.deepEqual(listMessages(adapter), [13, 10, 9, 8, 14, 6, 4, 1]);
Assert.equal(adapter.getCellText(4, "threadId"), 14);
// Add another message that isn't part of a thread, at the end.
const added7 = addMessage({
messageId: "added7@invalid",
date: "1998-03-03",
});
Assert.equal(added7, 15);
tree.assertRowCountChanged(8, 1);
Assert.equal(adapter.rowCount, 9);
Assert.ok(!adapter.isContainer(8));
Assert.equal(adapter.rowAt(8).messageCount, 1);
Assert.deepEqual(listMessages(adapter), [13, 10, 9, 8, 14, 6, 4, 1, 15]);
Assert.equal(adapter.getCellText(8, "threadId"), 15);
// Remove the messages in the order they were added.
messageDB.removeMessage(added1);
tree.assertInvalidated(5, 5);
Assert.ok(!adapter.isContainer(5));
Assert.equal(adapter.rowAt(5).messageCount, 3);
Assert.deepEqual(listMessages(adapter), [13, 10, 9, 8, 14, 6, 4, 1, 15]);
messageDB.removeMessage(added4);
tree.assertInvalidated(2, 2);
Assert.ok(!adapter.isContainer(2));
Assert.equal(adapter.rowAt(2).messageCount, 1);
Assert.deepEqual(listMessages(adapter), [13, 10, 9, 8, 14, 6, 4, 1, 15]);
messageDB.removeMessage(added5);
tree.assertRowCountChanged(0, -1);
Assert.deepEqual(listMessages(adapter), [10, 9, 8, 14, 6, 4, 1, 15]);
messageDB.removeMessage(added6);
tree.assertRowCountChanged(3, -1);
Assert.deepEqual(listMessages(adapter), [10, 9, 8, 6, 4, 1, 15]);
messageDB.removeMessage(added7);
tree.assertRowCountChanged(6, -1);
Assert.deepEqual(listMessages(adapter), [10, 9, 8, 6, 4, 1]);
} finally {
adapter.setTree(null);
}
});
/**
* Tests LiveViewConversationsDataAdapter, when sorted by date, ascending.
*/
add_task(async function testDateAscending() {
const liveView = new LiveView();
const adapter = new LiveViewConversationsDataAdapter(liveView);
adapter.sortBy("date", "ascending");
adapter.setTree(tree);
await tree.promiseInvalidated(0, 5);
try {
Assert.equal(adapter.rowCount, 6);
Assert.ok(!adapter.isContainer(0));
Assert.equal(adapter.rowAt(0).messageCount, 1);
Assert.equal(adapter.getCellText(0, "threadId"), "1");
Assert.ok(!adapter.isContainer(1));
Assert.equal(adapter.rowAt(1).messageCount, 3);
Assert.equal(adapter.getCellText(1, "threadId"), "7");
Assert.ok(!adapter.isContainer(2));
Assert.equal(adapter.rowAt(2).messageCount, 3);
Assert.equal(adapter.getCellText(2, "threadId"), "3");
Assert.ok(!adapter.isContainer(3));
Assert.equal(adapter.rowAt(3).messageCount, 1);
Assert.equal(adapter.getCellText(3, "threadId"), "8");
Assert.ok(!adapter.isContainer(4));
Assert.equal(adapter.rowAt(4).messageCount, 1);
Assert.equal(adapter.getCellText(4, "threadId"), "9");
Assert.ok(!adapter.isContainer(5));
Assert.equal(adapter.rowAt(5).messageCount, 1);
Assert.equal(adapter.getCellText(5, "threadId"), "10");
} finally {
adapter.setTree(null);
}
});
/**
* Test what happens when messages are added or removed.
*/
add_task(async function testAddRemoveDateAscending() {
const liveView = new LiveView();
const adapter = new LiveViewConversationsDataAdapter(liveView);
adapter.sortBy("date", "ascending");
adapter.setTree(tree);
await tree.promiseInvalidated(0, 5);
try {
Assert.equal(adapter.rowCount, 6);
Assert.deepEqual(listMessages(adapter), [1, 4, 6, 8, 9, 10]);
// Add a message to a thread.
Assert.equal(adapter.rowAt(2).messageCount, 3);
Assert.equal(adapter.getCellText(2, "threadId"), 3);
const added1 = addMessage({
messageId: "added1@invalid",
references: ["message3@invalid"],
});
Assert.equal(added1, 11);
tree.assertInvalidated(2, 2);
Assert.equal(adapter.rowCount, 6);
Assert.ok(!adapter.isContainer(2));
Assert.equal(adapter.rowAt(2).messageCount, 4);
Assert.deepEqual(listMessages(adapter), [1, 4, 6, 8, 9, 10]);
Assert.equal(adapter.getCellText(2, "threadId"), 3);
// Add a message to a non-thread.
Assert.equal(adapter.rowAt(4).messageCount, 1);
Assert.equal(adapter.getCellText(4, "threadId"), 9);
const added4 = addMessage({
messageId: "added4@invalid",
references: ["message9@invalid"],
});
Assert.equal(added4, 12);
tree.assertInvalidated(4, 4);
Assert.equal(adapter.rowCount, 6);
Assert.ok(!adapter.isContainer(4));
Assert.equal(adapter.rowAt(4).messageCount, 2);
Assert.deepEqual(listMessages(adapter), [1, 4, 6, 8, 9, 10]);
Assert.equal(adapter.getCellText(4, "threadId"), 9);
// Add a message that isn't part of a thread, at the end.
const added5 = addMessage({
messageId: "added5@invalid",
date: "2026-03-03",
});
Assert.equal(added5, 13);
tree.assertRowCountChanged(6, 1);
Assert.equal(adapter.rowCount, 7);
Assert.ok(!adapter.isContainer(6));
Assert.equal(adapter.rowAt(6).messageCount, 1);
Assert.deepEqual(listMessages(adapter), [1, 4, 6, 8, 9, 10, 13]);
Assert.equal(adapter.getCellText(6, "threadId"), 13);
// Add another message that isn't part of a thread, in the middle.
const added6 = addMessage({
messageId: "added6@invalid",
date: "2025-09-07",
});
Assert.equal(added6, 14);
tree.assertRowCountChanged(3, 1);
Assert.equal(adapter.rowCount, 8);
Assert.ok(!adapter.isContainer(3));
Assert.equal(adapter.rowAt(3).messageCount, 1);
Assert.deepEqual(listMessages(adapter), [1, 4, 6, 14, 8, 9, 10, 13]);
Assert.equal(adapter.getCellText(3, "threadId"), 14);
// Add another message that isn't part of a thread, at the start.
const added7 = addMessage({
messageId: "added7@invalid",
date: "1998-03-03",
});
Assert.equal(added7, 15);
tree.assertRowCountChanged(0, 1);
Assert.equal(adapter.rowCount, 9);
Assert.ok(!adapter.isContainer(0));
Assert.equal(adapter.rowAt(0).messageCount, 1);
Assert.deepEqual(listMessages(adapter), [15, 1, 4, 6, 14, 8, 9, 10, 13]);
Assert.equal(adapter.getCellText(0, "threadId"), 15);
// Remove the messages in the order they were added.
messageDB.removeMessage(added1);
tree.assertInvalidated(3, 3);
Assert.ok(!adapter.isContainer(3));
Assert.equal(adapter.rowAt(3).messageCount, 3);
Assert.deepEqual(listMessages(adapter), [15, 1, 4, 6, 14, 8, 9, 10, 13]);
messageDB.removeMessage(added4);
tree.assertInvalidated(6, 6);
Assert.ok(!adapter.isContainer(6));
Assert.equal(adapter.rowAt(6).messageCount, 1);
Assert.deepEqual(listMessages(adapter), [15, 1, 4, 6, 14, 8, 9, 10, 13]);
messageDB.removeMessage(added5);
tree.assertRowCountChanged(8, -1);
Assert.deepEqual(listMessages(adapter), [15, 1, 4, 6, 14, 8, 9, 10]);
messageDB.removeMessage(added6);
tree.assertRowCountChanged(4, -1);
Assert.deepEqual(listMessages(adapter), [15, 1, 4, 6, 8, 9, 10]);
messageDB.removeMessage(added7);
tree.assertRowCountChanged(0, -1);
Assert.deepEqual(listMessages(adapter), [1, 4, 6, 8, 9, 10]);
} finally {
adapter.setTree(null);
}
});
/**
* Test that only sort by date is allowed.
*/
add_task(async function testSortColumn() {
const liveView = new LiveView();
const adapter = new LiveViewConversationsDataAdapter(liveView);
try {
await adapter.sortBy("subject", "descending");
Assert.equal(adapter.sortColumn, "date");
Assert.equal(adapter.sortDirection, "descending");
await adapter.sortBy("subject", "ascending");
Assert.equal(adapter.sortColumn, "date");
Assert.equal(adapter.sortDirection, "ascending");
await adapter.sortBy("sender", "ascending");
Assert.equal(adapter.sortColumn, "date");
Assert.equal(adapter.sortDirection, "ascending");
await adapter.sortBy("flagged", "descending");
Assert.equal(adapter.sortColumn, "date");
Assert.equal(adapter.sortDirection, "descending");
await adapter.sortBy();
Assert.equal(adapter.sortColumn, "date");
Assert.equal(adapter.sortDirection, "descending");
} finally {
adapter.setTree(null);
}
});