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
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
|
/* 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/. */
/**
* Tests that we do the right thing wrt. message encoding when editing or
* replying to messages.
*/
"use strict";
var utils = ChromeUtils.import("resource://testing-common/mozmill/utils.jsm");
var {
close_compose_window,
open_compose_with_reply,
save_compose_message,
wait_for_compose_window,
} = ChromeUtils.import("resource://testing-common/mozmill/ComposeHelpers.jsm");
var {
add_message_to_folder,
assert_selected_and_displayed,
be_in_folder,
create_message,
get_special_folder,
get_about_message,
make_display_unthreaded,
mc,
press_delete,
select_click_row,
} = ChromeUtils.import(
"resource://testing-common/mozmill/FolderDisplayHelpers.jsm"
);
var { SyntheticPartLeaf } = ChromeUtils.import(
"resource://testing-common/mailnews/MessageGenerator.jsm"
);
var { wait_for_notification_to_show, get_notification } = ChromeUtils.import(
"resource://testing-common/mozmill/NotificationBoxHelpers.jsm"
);
var { plan_for_new_window } = ChromeUtils.import(
"resource://testing-common/mozmill/WindowHelpers.jsm"
);
var { MailServices } = ChromeUtils.import(
"resource:///modules/MailServices.jsm"
);
var { MimeParser } = ChromeUtils.import("resource:///modules/mimeParser.jsm");
let aboutMessage = get_about_message();
var gDrafts;
add_setup(async function () {
gDrafts = await get_special_folder(Ci.nsMsgFolderFlags.Drafts, true);
});
/**
* Helper to get the full message content.
*
* @param aMsgHdr: nsIMsgDBHdr object whose text body will be read
* @param aGetText: if true, return header objects. if false, return body data.
* @returns Map(partnum -> message headers)
*/
function getMsgHeaders(aMsgHdr, aGetText = false) {
let msgFolder = aMsgHdr.folder;
let msgUri = msgFolder.getUriForMsg(aMsgHdr);
let handler = {
_done: false,
_data: new Map(),
_text: new Map(),
endMessage() {
this._done = true;
},
deliverPartData(num, text) {
this._text.set(num, this._text.get(num) + text);
},
startPart(num, headers) {
this._data.set(num, headers);
this._text.set(num, "");
},
};
let streamListener = MimeParser.makeStreamListenerParser(handler, {
strformat: "unicode",
});
MailServices.messageServiceFromURI(msgUri).streamMessage(
msgUri,
streamListener,
null,
null,
false,
"",
false
);
utils.waitFor(() => handler._done);
return aGetText ? handler._text : handler._data;
}
/**
* Test that if we reply to a message in an invalid charset, we don't try to compose
* in that charset. Instead, we should be using UTF-8.
*/
add_task(async function test_wrong_reply_charset() {
let folder = gDrafts;
let msg0 = create_message({
bodyPart: new SyntheticPartLeaf("Some text", {
charset: "invalid-charset",
}),
});
await add_message_to_folder([folder], msg0);
await be_in_folder(folder);
// Make the folder unthreaded for easier message selection.
make_display_unthreaded();
let msg = select_click_row(0);
assert_selected_and_displayed(mc, msg);
Assert.equal(getMsgHeaders(msg).get("").charset, "invalid-charset");
let rwc = open_compose_with_reply();
await save_compose_message(rwc.window);
await TestUtils.waitForCondition(
() => folder.getTotalMessages(false) == 2,
"message saved to drafts folder"
);
close_compose_window(rwc);
let draftMsg = select_click_row(1);
Assert.equal(getMsgHeaders(draftMsg).get("").charset, "UTF-8");
press_delete(mc); // Delete message
// Edit the original message. Charset should be UTF-8 now.
msg = select_click_row(0);
// Wait for the notification with the Edit button.
wait_for_notification_to_show(
aboutMessage,
"mail-notification-top",
"draftMsgContent"
);
plan_for_new_window("msgcompose");
let box = get_notification(
aboutMessage,
"mail-notification-top",
"draftMsgContent"
);
// Click on the "Edit" button in the draft notification.
EventUtils.synthesizeMouseAtCenter(
box.buttonContainer.firstElementChild,
{},
aboutMessage
);
rwc = wait_for_compose_window();
await save_compose_message(rwc.window);
close_compose_window(rwc);
msg = select_click_row(0);
await TestUtils.waitForCondition(
() => getMsgHeaders(msg).get("").charset == "UTF-8",
"The charset matches"
);
press_delete(mc); // Delete message
});
/**
* Test that replying to bad charsets don't screw up the existing text.
*/
add_task(async function test_no_mojibake() {
let folder = gDrafts;
let nonASCII = "ケツァルコアトル";
let UTF7 = "+MLEwxDChMOswszCiMMgw6w-";
let msg0 = create_message({
bodyPart: new SyntheticPartLeaf(UTF7, { charset: "utf-7" }),
});
await add_message_to_folder([folder], msg0);
await be_in_folder(folder);
let msg = select_click_row(0);
assert_selected_and_displayed(mc, msg);
await TestUtils.waitForCondition(
() => getMsgHeaders(msg).get("").charset == "utf-7",
"message charset correctly set"
);
Assert.equal(getMsgHeaders(msg, true).get("").trim(), nonASCII);
let rwc = open_compose_with_reply();
await save_compose_message(rwc.window);
await TestUtils.waitForCondition(
() => folder.getTotalMessages(false) == 2,
"message saved to drafts folder"
);
close_compose_window(rwc);
let draftMsg = select_click_row(1);
Assert.equal(getMsgHeaders(draftMsg).get("").charset.toUpperCase(), "UTF-8");
let text = getMsgHeaders(draftMsg, true).get("");
// Delete message first before throwing so subsequent tests are not affected.
press_delete(mc);
if (!text.includes(nonASCII)) {
throw new Error("Expected to find " + nonASCII + " in " + text);
}
// Edit the original message. Charset should be UTF-8 now.
msg = select_click_row(0);
// Wait for the notification with the Edit button.
wait_for_notification_to_show(
aboutMessage,
"mail-notification-top",
"draftMsgContent"
);
plan_for_new_window("msgcompose");
let box = get_notification(
aboutMessage,
"mail-notification-top",
"draftMsgContent"
);
// Click on the "Edit" button in the draft notification.
EventUtils.synthesizeMouseAtCenter(
box.buttonContainer.firstElementChild,
{},
aboutMessage
);
rwc = wait_for_compose_window();
await save_compose_message(rwc.window);
close_compose_window(rwc);
msg = select_click_row(0);
Assert.equal(getMsgHeaders(msg).get("").charset.toUpperCase(), "UTF-8");
Assert.equal(getMsgHeaders(msg, true).get("").trim(), nonASCII);
press_delete(mc); // Delete message
});
|
