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
|
/* 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/. */
"use strict";
var { mc } = ChromeUtils.import(
"resource://testing-common/mozmill/FolderDisplayHelpers.jsm"
);
var { click_account_tree_row, get_account_tree_row, open_advanced_settings } =
ChromeUtils.import(
"resource://testing-common/mozmill/AccountManagerHelpers.jsm"
);
var { FAKE_SERVER_HOSTNAME } = ChromeUtils.import(
"resource://testing-common/mozmill/FolderDisplayHelpers.jsm"
);
var { MailServices } = ChromeUtils.import(
"resource:///modules/MailServices.jsm"
);
var gOldWhiteList = null;
var gKeyString = null;
var gAccount = null;
add_setup(function () {
let server = MailServices.accounts.findServer(
"tinderbox",
FAKE_SERVER_HOSTNAME,
"pop3"
);
gAccount = MailServices.accounts.FindAccountForServer(server);
let serverKey = server.key;
gKeyString = "mail.server." + serverKey + ".whiteListAbURI";
gOldWhiteList = Services.prefs.getCharPref(gKeyString);
Services.prefs.setCharPref(gKeyString, "");
});
registerCleanupFunction(function () {
Services.prefs.setCharPref(gKeyString, gOldWhiteList);
});
/**
* First, test that when we initially load the account manager, that
* we're not whitelisting any address books. Then, we'll check all
* address books and save.
*
* @param {object} tab - The account manager tab.
*/
function subtest_check_whitelist_init_and_save(tab) {
// Ok, the advanced settings window is open. Let's choose
// the junkmail settings.
let accountRow = get_account_tree_row(gAccount.key, "am-junk.xhtml", tab);
click_account_tree_row(tab, accountRow);
let doc =
tab.browser.contentWindow.document.getElementById(
"contentFrame"
).contentDocument;
// At this point, we shouldn't have anything checked, but we should have
// the two default address books (Personal and Collected) displayed
let list = doc.getElementById("whiteListAbURI");
Assert.equal(
2,
list.getRowCount(),
"There was an unexpected number of address books"
);
// Now we'll check both address books
for (let i = 0; i < list.getRowCount(); i++) {
let abNode = list.getItemAtIndex(i);
EventUtils.synthesizeMouseAtCenter(
abNode.firstElementChild,
{ clickCount: 1 },
abNode.firstElementChild.ownerGlobal
);
}
}
/**
* Next, we'll make sure that the address books we checked in
* subtest_check_whitelist_init_and_save were properly saved.
* Then, we'll clear the address books and save.
*
* @param {object} tab - The account manager tab.
*/
function subtest_check_whitelist_load_and_clear(tab) {
let accountRow = get_account_tree_row(gAccount.key, "am-junk.xhtml", tab);
click_account_tree_row(tab, accountRow);
let doc =
tab.browser.contentWindow.document.getElementById(
"contentFrame"
).contentDocument;
let list = doc.getElementById("whiteListAbURI");
let whiteListURIs = Services.prefs.getCharPref(gKeyString).split(" ");
for (let i = 0; i < list.getRowCount(); i++) {
let abNode = list.getItemAtIndex(i);
Assert.equal(
true,
abNode.firstElementChild.checked,
"Should have been checked"
);
// Also ensure that the address book URI was properly saved in the
// prefs
Assert.ok(whiteListURIs.includes(abNode.getAttribute("value")));
// Now un-check that address book
EventUtils.synthesizeMouseAtCenter(
abNode.firstElementChild,
{ clickCount: 1 },
abNode.firstElementChild.ownerGlobal
);
}
}
/**
* Finally, we'll make sure that the address books we cleared
* were actually cleared.
*
* @param {object} tab - The account manager tab.
*/
function subtest_check_whitelist_load_cleared(tab) {
let accountRow = get_account_tree_row(gAccount.key, "am-junk.xhtml", tab);
click_account_tree_row(tab, accountRow);
let doc =
tab.browser.contentWindow.document.getElementById(
"contentFrame"
).contentDocument;
let list = doc.getElementById("whiteListAbURI");
let whiteListURIs = "";
try {
whiteListURIs = Services.prefs.getCharPref(gKeyString);
// We should have failed here, because the pref should have been cleared
// out.
throw Error(
"The whitelist preference for this server wasn't properly cleared."
);
} catch (e) {}
for (let i = 0; i < list.getRowCount(); i++) {
let abNode = list.getItemAtIndex(i);
Assert.equal(
false,
abNode.firstElementChild.checked,
"Should not have been checked"
);
// Also ensure that the address book URI was properly cleared in the
// prefs
Assert.ok(!whiteListURIs.includes(abNode.getAttribute("value")));
}
}
add_task(async function test_address_book_whitelist() {
await open_advanced_settings(subtest_check_whitelist_init_and_save);
await open_advanced_settings(subtest_check_whitelist_load_and_clear);
await open_advanced_settings(subtest_check_whitelist_load_cleared);
});
|
