1 files changed, 240 insertions, 0 deletions
diff --git a/comm/mail/test/browser/shared-modules/AttachmentHelpers.jsm b/comm/mail/test/browser/shared-modules/AttachmentHelpers.jsm
new file mode 100644
index 0000000000..971b25fa77
--- /dev/null
+++ b/comm/mail/test/browser/shared-modules/AttachmentHelpers.jsm
@@ -0,0 +1,240 @@
+/* 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";
+
+const EXPORTED_SYMBOLS = [
+  "create_body_part",
+  "create_deleted_attachment",
+  "create_detached_attachment",
+  "create_enclosure_attachment",
+  "gMockFilePicker",
+  "gMockFilePickReg",
+  "select_attachments",
+];
+
+var { MockObjectReplacer } = ChromeUtils.import(
+  "resource://testing-common/mozmill/MockObjectHelpers.jsm"
+);
+
+var gMockFilePickReg = new MockObjectReplacer(
+  "@mozilla.org/filepicker;1",
+  MockFilePickerConstructor
+);
+
+function MockFilePickerConstructor() {
+  return gMockFilePicker;
+}
+
+var gMockFilePicker = {
+  QueryInterface: ChromeUtils.generateQI(["nsIFilePicker"]),
+  defaultExtension: "",
+  filterIndex: null,
+  displayDirectory: null,
+  returnFiles: [],
+  addToRecentDocs: false,
+
+  get defaultString() {
+    throw Components.Exception("", Cr.NS_ERROR_FAILURE);
+  },
+
+  get fileURL() {
+    return null;
+  },
+
+  get file() {
+    if (this.returnFiles.length >= 1) {
+      return this.returnFiles[0];
+    }
+    return null;
+  },
+
+  get files() {
+    let self = this;
+    return {
+      index: 0,
+      QueryInterface: ChromeUtils.generateQI(["nsISimpleEnumerator"]),
+      hasMoreElements() {
+        return this.index < self.returnFiles.length;
+      },
+      getNext() {
+        return self.returnFiles[this.index++];
+      },
+      [Symbol.iterator]() {
+        return self.returnFiles.values();
+      },
+    };
+  },
+
+  init(aParent, aTitle, aMode) {},
+
+  appendFilters(aFilterMask) {},
+
+  appendFilter(aTitle, aFilter) {},
+
+  open(aFilePickerShownCallback) {
+    aFilePickerShownCallback.done(Ci.nsIFilePicker.returnOK);
+  },
+
+  set defaultString(aVal) {},
+};
+
+/**
+ * Create a body part with attachments for the message generator
+ *
+ * @param body the text of the main body of the message
+ * @param attachments an array of attachment objects (as strings)
+ * @param boundary an optional string defining the boundary of the parts
+ * @returns an object suitable for passing as the |bodyPart| for create_message
+ */
+function create_body_part(body, attachments, boundary) {
+  if (!boundary) {
+    boundary = "------------CHOPCHOP";
+  }
+
+  return {
+    contentTypeHeaderValue: 'multipart/mixed;\r\n boundary="' + boundary + '"',
+    toMessageString() {
+      let str =
+        "This is a multi-part message in MIME format.\r\n" +
+        "--" +
+        boundary +
+        "\r\n" +
+        "Content-Type: text/plain; charset=ISO-8859-1; " +
+        "format=flowed\r\n" +
+        "Content-Transfer-Encoding: 7bit\r\n\r\n" +
+        body +
+        "\r\n\r\n";
+
+      for (let i = 0; i < attachments.length; i++) {
+        str += "--" + boundary + "\r\n" + attachments[i] + "\r\n";
+      }
+
+      str += "--" + boundary + "--";
+      return str;
+    },
+  };
+}
+
+function help_create_detached_deleted_attachment(filename, type) {
+  return (
+    "You deleted an attachment from this message. The original MIME " +
+    "headers for the attachment were:\r\n" +
+    "Content-Type: " +
+    type +
+    ";\r\n" +
+    ' name="' +
+    filename +
+    '"\r\n' +
+    "Content-Transfer-Encoding: 7bit\r\n" +
+    "Content-Disposition: attachment;\r\n" +
+    ' filename="' +
+    filename +
+    '"\r\n\r\n'
+  );
+}
+
+/**
+ * Create the raw data for a detached attachment
+ *
+ * @param file an nsIFile for the external file for the attachment
+ * @param type the content type
+ * @returns a string representing the attachment
+ */
+function create_detached_attachment(file, type) {
+  let fileHandler = Services.io
+    .getProtocolHandler("file")
+    .QueryInterface(Ci.nsIFileProtocolHandler);
+  let url = fileHandler.getURLSpecFromActualFile(file);
+  let filename = file.leafName;
+
+  let str =
+    'Content-Type: text/plain;\r\n name="' +
+    filename +
+    '"\r\n' +
+    'Content-Disposition: attachment; filename="' +
+    filename +
+    '"\r\n' +
+    "X-Mozilla-External-Attachment-URL: " +
+    url +
+    "\r\n" +
+    'X-Mozilla-Altered: AttachmentDetached; date="' +
+    'Wed Oct 06 17:28:24 2010"\r\n\r\n';
+
+  str += help_create_detached_deleted_attachment(filename, type);
+  return str;
+}
+
+/**
+ * Create the raw data for a deleted attachment
+ *
+ * @param filename the "original" filename
+ * @param type the content type
+ * @returns a string representing the attachment
+ */
+function create_deleted_attachment(filename, type) {
+  let str =
+    'Content-Type: text/x-moz-deleted; name="Deleted: ' +
+    filename +
+    '"\r\n' +
+    "Content-Transfer-Encoding: 8bit\r\n" +
+    'Content-Disposition: inline; filename="Deleted: ' +
+    filename +
+    '"\r\n' +
+    'X-Mozilla-Altered: AttachmentDeleted; date="' +
+    'Wed Oct 06 17:28:24 2010"\r\n\r\n';
+  str += help_create_detached_deleted_attachment(filename, type);
+  return str;
+}
+
+/**
+ * Create the raw data for a feed enclosure attachment.
+ *
+ * @param filename the filename
+ * @param type the content type
+ * @param url the remote link url
+ * @param size the optional size (use > 1 for real size)
+ * @returns a string representing the attachment
+ */
+function create_enclosure_attachment(filename, type, url, size) {
+  return (
+    "Content-Type: " +
+    type +
+    '; name="' +
+    filename +
+    (size ? '"; size=' + size : '"') +
+    "\r\n" +
+    "X-Mozilla-External-Attachment-URL: " +
+    url +
+    "\r\n" +
+    'Content-Disposition: attachment; filename="' +
+    filename +
+    '"\r\n\r\n' +
+    "This MIME attachment is stored separately from the message."
+  );
+}
+
+/**
+ * A helper function that selects either one, or a continuous range
+ * of items in the attachment list.
+ *
+ * @param aController a composer window controller
+ * @param aIndexStart the index of the first item to select
+ * @param aIndexEnd (optional) the index of the last item to select
+ */
+function select_attachments(aController, aIndexStart, aIndexEnd) {
+  let bucket = aController.window.document.getElementById("attachmentBucket");
+  bucket.clearSelection();
+
+  if (aIndexEnd !== undefined) {
+    let startItem = bucket.getItemAtIndex(aIndexStart);
+    let endItem = bucket.getItemAtIndex(aIndexEnd);
+    bucket.selectItemRange(startItem, endItem);
+  } else {
+    bucket.selectedIndex = aIndexStart;
+  }
+
+  bucket.focus();
+  return [...bucket.selectedItems];
+}