1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
|
/* 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
* file, you can obtain one at http://mozilla.org/MPL/2.0/. */
var { FeedUtils } = ChromeUtils.import("resource:///modules/FeedUtils.jsm");
var { VirtualFolderHelper } = ChromeUtils.import(
"resource:///modules/VirtualFolderWrapper.jsm"
);
let about3Pane = document.getElementById("tabmail").currentAbout3Pane;
let rootFolder;
let folders = {};
const FOLDER_PREFIX = "mailbox://nobody@smart%20mailboxes/tags/";
const DEFAULT_TAGS = new Map([
["$label1", { label: "Important", color: "#FF0000" }],
["$label2", { label: "Work", color: "#FF9900" }],
["$label3", { label: "Personal", color: "#009900" }],
["$label4", { label: "To Do", color: "#3333FF" }],
["$label5", { label: "Later", color: "#993399" }],
]);
add_setup(async function () {
let allTags = MailServices.tags.getAllTags();
Assert.deepEqual(
allTags.map(t => t.key),
[...DEFAULT_TAGS.keys()],
"sanity check tag keys"
);
Assert.deepEqual(
allTags.map(t => ({ label: t.tag, color: t.color })),
[...DEFAULT_TAGS.values()],
"sanity check tag labels"
);
about3Pane.folderPane.activeModes = ["all"];
resetSmartMailboxes();
let account = MailServices.accounts.createAccount();
account.incomingServer = MailServices.accounts.createIncomingServer(
`${account.key}user`,
"localhost",
"pop3"
);
account.addIdentity(MailServices.accounts.createIdentity());
rootFolder = account.incomingServer.rootFolder.QueryInterface(
Ci.nsIMsgLocalMailFolder
);
for (let flag of [
"Inbox",
"Drafts",
"Templates",
"SentMail",
"Archive",
"Junk",
"Trash",
"Queue",
"Virtual",
]) {
let folder = rootFolder.getFolderWithFlags(Ci.nsMsgFolderFlags[flag]);
if (!folder) {
folder = rootFolder.createLocalSubfolder(`tagsMode${flag}`);
folder.setFlag(Ci.nsMsgFolderFlags[flag]);
}
folders[flag] = folder.QueryInterface(Ci.nsIMsgLocalMailFolder);
}
folders.Plain = rootFolder.createLocalSubfolder("tagsModePlain");
let msgDatabase = folders.Virtual.msgDatabase;
let folderInfo = msgDatabase.dBFolderInfo;
folderInfo.setCharProperty("searchStr", "ALL");
folderInfo.setCharProperty("searchFolderUri", folders.Inbox.URI);
registerCleanupFunction(function () {
MailServices.accounts.removeAccount(account, false);
about3Pane.folderPane.activeModes = ["all"];
});
});
async function checkFolderTree(expectedTags) {
let tagsList = about3Pane.folderTree.querySelector(`li[data-mode="tags"] ul`);
await TestUtils.waitForCondition(
() => tagsList.childElementCount == expectedTags.size,
"waiting for folder tree to update"
);
let keys = expectedTags.keys();
let values = expectedTags.values();
for (let row of tagsList.children) {
let key = keys.next().value;
let { label, color } = values.next().value;
Assert.equal(row.uri, FOLDER_PREFIX + encodeURIComponent(key));
Assert.equal(row.name, label);
Assert.equal(row.icon.style.getPropertyValue("--icon-color"), color);
}
Assert.ok(keys.next().done, "all tags should have a row in the tree");
}
add_task(async function testFolderTree() {
// Check the default tags are shown initially.
let expectedTags = new Map(DEFAULT_TAGS);
about3Pane.folderPane.activeModes = ["all", "tags"];
await checkFolderTree(DEFAULT_TAGS);
// Add two custom tags and check they are shown.
MailServices.tags.addTagForKey("testkey", "testLabel", "#000000", "");
await TestUtils.waitForCondition(
() => MailServices.tags.getAllTags().length == 6,
"waiting for tag to be created"
);
expectedTags.set("testkey", { label: "testLabel", color: "#000000" });
await checkFolderTree(expectedTags);
MailServices.tags.addTagForKey("anotherkey", "anotherLabel", "#333333", "");
await TestUtils.waitForCondition(
() => MailServices.tags.getAllTags().length == 7,
"waiting for tag to be created"
);
expectedTags.set("anotherkey", { label: "anotherLabel", color: "#333333" });
await checkFolderTree(expectedTags);
// Delete the first custom tag and check it is removed.
MailServices.tags.deleteKey("testkey");
await TestUtils.waitForCondition(
() => MailServices.tags.getAllTags().length == 6,
"waiting for tag to be removed"
);
expectedTags.delete("testkey");
await checkFolderTree(expectedTags);
// Hide and reinitialise the Tags mode and check the list is the same.
about3Pane.folderPane.activeModes = ["all"];
about3Pane.folderPane.activeModes = ["all", "tags"];
await checkFolderTree(expectedTags);
// Delete the second custom tag.
MailServices.tags.deleteKey("anotherkey");
await TestUtils.waitForCondition(
() => MailServices.tags.getAllTags().length == 5,
"waiting for tag to be removed"
);
expectedTags.delete("anotherkey");
await checkFolderTree(expectedTags);
});
function checkVirtualFolder(tagKey, tagLabel, expectedFolderURIs) {
let folder = MailServices.folderLookup.getFolderForURL(
FOLDER_PREFIX + encodeURIComponent(tagKey)
);
Assert.ok(folder);
let wrappedFolder = VirtualFolderHelper.wrapVirtualFolder(folder);
Assert.equal(folder.prettyName, tagLabel);
Assert.equal(wrappedFolder.searchString, `AND (tag,contains,${tagKey})`);
Assert.equal(wrappedFolder.searchFolderURIs, "*");
about3Pane.displayFolder(folder);
Assert.deepEqual(
about3Pane.gViewWrapper._underlyingFolders.map(f => f.URI).sort(),
expectedFolderURIs.sort()
);
}
add_task(async function testFolderSelection() {
let expectedFolderURIs = [
folders.Inbox.URI,
folders.Drafts.URI,
folders.Templates.URI,
folders.SentMail.URI,
folders.Archive.URI,
folders.Plain.URI,
];
for (let [key, { label }] of DEFAULT_TAGS) {
checkVirtualFolder(key, label, expectedFolderURIs);
}
// Add another plain folder. It should be added to the searched folders.
let newPlainFolder = rootFolder.createLocalSubfolder("tagsModePlain2");
expectedFolderURIs.push(newPlainFolder.URI);
checkVirtualFolder("$label1", "Important", expectedFolderURIs);
// Add a subfolder to the inbox. It should be added to the searched folders.
let newInboxFolder = folders.Inbox.createLocalSubfolder("tagsModeInbox2");
expectedFolderURIs.push(newInboxFolder.URI);
checkVirtualFolder("$label2", "Work", expectedFolderURIs);
// Add a subfolder to the trash. It should NOT be added to the searched folders.
folders.Trash.createLocalSubfolder("tagsModeTrash2");
checkVirtualFolder("$label1", "Important", expectedFolderURIs);
let rssAccount = FeedUtils.createRssAccount("rss");
let rssRootFolder = rssAccount.incomingServer.rootFolder;
FeedUtils.subscribeToFeed(
"https://example.org/browser/comm/mail/base/test/browser/files/rss.xml?tagsMode",
rssRootFolder,
null
);
await TestUtils.waitForCondition(() => rssRootFolder.subFolders.length == 2);
let rssFeedFolder = rssRootFolder.getChildNamed("Test Feed");
expectedFolderURIs.push(rssFeedFolder.URI);
checkVirtualFolder("$label2", "Work", expectedFolderURIs);
// Delete the smart mailboxes server and check it is correctly recreated.
about3Pane.folderPane.activeModes = ["all"];
resetSmartMailboxes();
about3Pane.folderPane.activeModes = ["all", "tags"];
for (let [key, { label }] of DEFAULT_TAGS) {
checkVirtualFolder(key, label, expectedFolderURIs);
}
MailServices.accounts.removeAccount(rssAccount, false);
});
|
