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
|
/* 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 ts = Cc["@mozilla.org/msg-trait-service;1"].getService(
Ci.nsIMsgTraitService
);
// junk-related traits set by default
var kJunkId = "mailnews@mozilla.org#junk";
var kGoodId = "mailnews@mozilla.org#good";
var kGoodIndex = Ci.nsIJunkMailPlugin.GOOD_TRAIT;
var kJunkIndex = Ci.nsIJunkMailPlugin.JUNK_TRAIT;
// a dummy set of traits
var proId = "TheProTrait";
var proName = "ProName";
var antiId = "TheAntiTrait";
function run_test() {
// Check lastIndex prior to adding, 3 - 1000 are reserved for mailnews
Assert.equal(ts.lastIndex, 1000);
// basic junk as traits should be setup automatically
Assert.equal(
kGoodId,
Services.prefs.getCharPref("mailnews.traits.id." + kGoodIndex)
);
Assert.equal(
kJunkId,
Services.prefs.getCharPref("mailnews.traits.id." + kJunkIndex)
);
Assert.equal(
kGoodId,
Services.prefs.getCharPref("mailnews.traits.antiId." + kJunkIndex)
);
Assert.ok(
Services.prefs.getBoolPref("mailnews.traits.enabled." + kJunkIndex)
);
// add the pro and anti test traits
Assert.ok(!ts.isRegistered(proId));
var proIndex = ts.registerTrait(proId);
Assert.ok(ts.isRegistered(proId));
Assert.equal(proIndex, 1001);
Assert.equal(proIndex, ts.getIndex(proId));
Assert.equal(proId, ts.getId(proIndex));
var antiIndex = ts.registerTrait(antiId);
Assert.equal(proIndex, 1001);
Assert.equal(antiIndex, 1002);
// check setting and getting things through the service
ts.setName(proId, proName);
Assert.equal(proName, ts.getName(proId));
Assert.ok(!ts.getEnabled(proId));
ts.setEnabled(proId, true);
Assert.ok(ts.getEnabled(proId));
ts.setAntiId(proId, antiId);
Assert.equal(antiId, ts.getAntiId(proId));
let proArray = ts.getEnabledProIndices();
let antiArray = ts.getEnabledAntiIndices();
Assert.equal(proArray.length, 2);
Assert.equal(antiArray.length, 2);
Assert.equal(proArray[1], proIndex);
Assert.equal(antiArray[1], antiIndex);
// check of aliases
// add three random aliases
ts.addAlias(1, 501);
ts.addAlias(1, 502);
ts.addAlias(1, 601);
let aliases = ts.getAliases(1);
Assert.equal(aliases[0], 501);
Assert.equal(aliases[1], 502);
Assert.equal(aliases[2], 601);
// remove the middle one
ts.removeAlias(1, 502);
aliases = ts.getAliases(1);
Assert.equal(aliases.length, 2);
Assert.equal(aliases[0], 501);
Assert.equal(aliases[1], 601);
// try to add an existing value
ts.addAlias(1, 501);
aliases = ts.getAliases(1);
Assert.equal(aliases.length, 2);
Assert.equal(aliases[0], 501);
Assert.equal(aliases[1], 601);
// now let's make sure this got saved in preferences
Assert.equal(
proId,
Services.prefs.getCharPref("mailnews.traits.id." + proIndex)
);
Assert.equal(
proName,
Services.prefs.getCharPref("mailnews.traits.name." + proIndex)
);
Assert.ok(Services.prefs.getBoolPref("mailnews.traits.enabled." + proIndex));
Assert.equal(
antiId,
Services.prefs.getCharPref("mailnews.traits.antiId." + proIndex)
);
// remove the pro trait
ts.unRegisterTrait(proId);
Assert.ok(!ts.isRegistered(proId));
// check that this is also removed from prefs. The get calls should fail
try {
Services.prefs.getCharPref("mailnews.traits.id." + proIndex);
Assert.ok(false);
} catch (e) {}
try {
Services.prefs.getCharPref("mailnews.traits.name." + proIndex);
Assert.ok(false);
} catch (e) {}
try {
Services.prefs.getBoolPref("mailnews.traits.enabled." + proIndex);
Assert.ok(false);
} catch (e) {}
try {
Services.prefs.getCharPref("mailnews.traits.antiId." + proIndex);
Assert.ok(false);
} catch (e) {}
}
|
