1 files changed, 109 insertions, 0 deletions
diff --git a/comm/mail/test/browser/composition/browser_messageBody.js b/comm/mail/test/browser/composition/browser_messageBody.js
new file mode 100644
index 0000000000..c9b16cc370
--- /dev/null
+++ b/comm/mail/test/browser/composition/browser_messageBody.js
@@ -0,0 +1,109 @@
+/* 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 related to message body.
+ */
+
+var { get_msg_source, open_compose_new_mail, setup_msg_contents } =
+  ChromeUtils.import("resource://testing-common/mozmill/ComposeHelpers.jsm");
+var {
+  be_in_folder,
+  get_special_folder,
+  get_about_message,
+  press_delete,
+  select_click_row,
+} = ChromeUtils.import(
+  "resource://testing-common/mozmill/FolderDisplayHelpers.jsm"
+);
+var { plan_for_window_close, wait_for_window_close } = ChromeUtils.import(
+  "resource://testing-common/mozmill/WindowHelpers.jsm"
+);
+
+var gOutboxFolder;
+
+add_setup(async function () {
+  gOutboxFolder = await get_special_folder(Ci.nsMsgFolderFlags.Queue);
+});
+
+/**
+ * Tests that sending link with invalid data uri works.
+ */
+add_task(async function test_invalid_data_uri() {
+  let cwc = open_compose_new_mail();
+  setup_msg_contents(
+    cwc,
+    "someone@example.com",
+    "Test sending link with invalid data uri",
+    ""
+  );
+
+  cwc.window
+    .GetCurrentEditor()
+    .insertHTML("<a href=data:1>invalid data uri</a>");
+  plan_for_window_close(cwc);
+  cwc.window.goDoCommand("cmd_sendLater");
+  wait_for_window_close();
+
+  await be_in_folder(gOutboxFolder);
+  let msgLoaded = BrowserTestUtils.waitForEvent(
+    get_about_message(),
+    "MsgLoaded"
+  );
+  let outMsg = select_click_row(0);
+  await msgLoaded;
+  let outMsgContent = await get_msg_source(outMsg);
+
+  ok(
+    outMsgContent.includes("invalid data uri"),
+    "message containing invalid data uri should be sent"
+  );
+
+  press_delete(); // Delete the msg from Outbox.
+});
+
+/**
+ * Tests that when converting <a href="$1">$2</a> to text/plain, if $1 matches
+ * with $2, $1 should be discarded to prevent duplicated links.
+ */
+add_task(async function test_freeTextLink() {
+  let prevSendFormat = Services.prefs.getIntPref("mail.default_send_format");
+  Services.prefs.setIntPref(
+    "mail.default_send_format",
+    Ci.nsIMsgCompSendFormat.PlainText
+  );
+  let cwc = open_compose_new_mail();
+  setup_msg_contents(cwc, "someone@example.com", "Test free text link", "");
+
+  let link1 = "https://example.com";
+  let link2 = "name@example.com";
+  let link3 = "https://example.net";
+  cwc.window
+    .GetCurrentEditor()
+    .insertHTML(
+      `<a href="${link1}/">${link1}</a> <a href="mailto:${link2}">${link2}</a> <a href="${link3}">link3</a>`
+    );
+  plan_for_window_close(cwc);
+  cwc.window.goDoCommand("cmd_sendLater");
+  wait_for_window_close();
+
+  await be_in_folder(gOutboxFolder);
+  let msgLoaded = BrowserTestUtils.waitForEvent(
+    get_about_message(),
+    "MsgLoaded"
+  );
+  let outMsg = select_click_row(0);
+  await msgLoaded;
+  let outMsgContent = await get_msg_source(outMsg);
+
+  Assert.equal(
+    getMessageBody(outMsgContent),
+    `${link1} ${link2} link3 <${link3}>\r\n`,
+    "Links should be correctly converted to plain text"
+  );
+
+  press_delete(); // Delete the msg from Outbox.
+
+  Services.prefs.setIntPref("mail.default_send_format", prevSendFormat);
+});