1 files changed, 99 insertions, 0 deletions
diff --git a/comm/mail/test/browser/message-reader/browser_detectCharset.js b/comm/mail/test/browser/message-reader/browser_detectCharset.js
new file mode 100644
index 0000000000..daa0a7ca2d
--- /dev/null
+++ b/comm/mail/test/browser/message-reader/browser_detectCharset.js
@@ -0,0 +1,99 @@
+/**
+ * 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 opening an .eml file the body of the message is correct,
+ * that it hasn't been UTF-8 mojibake'd.
+ */
+
+"use strict";
+
+var { get_about_message, open_message_from_file } = ChromeUtils.import(
+  "resource://testing-common/mozmill/FolderDisplayHelpers.jsm"
+);
+var { close_window } = ChromeUtils.import(
+  "resource://testing-common/mozmill/WindowHelpers.jsm"
+);
+
+var gReferenceTextContent;
+
+add_setup(async function () {
+  Services.prefs.setBoolPref("mailnews.display.prefer_plaintext", false);
+  Services.prefs.setIntPref("mailnews.display.html_as", 0);
+  Services.prefs.setIntPref("mailnews.display.disallow_mime_handlers", 0);
+
+  let { textContent } = await extract_eml_body_textcontent(
+    "./correctEncodingUTF8.eml",
+    false
+  );
+  gReferenceTextContent = textContent;
+});
+
+async function check_display_charset(eml, expectedCharset) {
+  let file = new FileUtils.File(getTestFilePath(`data/${eml}`));
+  let msgc = await open_message_from_file(file);
+  let aboutMessage = get_about_message(msgc.window);
+  is(aboutMessage.currentCharacterSet, expectedCharset);
+  close_window(msgc);
+}
+
+async function extract_eml_body_textcontent(eml, autodetect = true) {
+  let file = new FileUtils.File(getTestFilePath(`data/${eml}`));
+  let msgc = await open_message_from_file(file);
+  let aboutMessage = get_about_message(msgc.window);
+
+  if (autodetect) {
+    // Open other actions menu.
+    let popup = aboutMessage.document.getElementById("otherActionsPopup");
+    let popupShown = BrowserTestUtils.waitForEvent(popup, "popupshown");
+    EventUtils.synthesizeMouseAtCenter(
+      aboutMessage.document.getElementById("otherActionsButton"),
+      {},
+      aboutMessage
+    );
+    await popupShown;
+
+    // Click on the "Repair Text Encoding" item.
+    let hiddenPromise = BrowserTestUtils.waitForEvent(popup, "popuphidden");
+    let reloadPromise = BrowserTestUtils.browserLoaded(
+      aboutMessage.getMessagePaneBrowser()
+    );
+    EventUtils.synthesizeMouseAtCenter(
+      aboutMessage.document.getElementById("charsetRepairMenuitem"),
+      {},
+      aboutMessage
+    );
+    await hiddenPromise;
+    await reloadPromise;
+  }
+
+  let textContent =
+    aboutMessage.getMessagePaneBrowser().contentDocument.documentElement
+      .textContent;
+  let charset = aboutMessage.currentCharacterSet;
+  close_window(msgc);
+  return { textContent, charset };
+}
+
+/**
+ * Checks that the text content is equal for the .eml files and that
+ * the expected charset was detected.
+ */
+async function check_eml_textcontent(eml, expectedCharset) {
+  let { textContent, charset } = await extract_eml_body_textcontent(eml);
+  is(textContent, gReferenceTextContent);
+  is(charset, expectedCharset);
+}
+
+add_task(async function test_noCharset() {
+  await check_display_charset("./noCharsetKOI8U.eml", "KOI8-U");
+  await check_display_charset("./noCharsetWindows1252.eml", "windows-1252");
+});
+
+add_task(async function test_wronglyDeclaredCharset() {
+  await check_eml_textcontent("./wronglyDeclaredUTF8.eml", "UTF-8");
+  await check_eml_textcontent("./wronglyDeclaredShift_JIS.eml", "Shift_JIS");
+});