1 files changed, 74 insertions, 0 deletions
diff --git a/comm/mailnews/imap/test/unit/test_imapHdrStreaming.js b/comm/mailnews/imap/test/unit/test_imapHdrStreaming.js
new file mode 100644
index 0000000000..ca148dace6
--- /dev/null
+++ b/comm/mailnews/imap/test/unit/test_imapHdrStreaming.js
@@ -0,0 +1,74 @@
+/* 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/. */
+
+/* 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/. */
+
+/**
+ * This test checks if the imap message service code streams headers correctly.
+ * It checks that streaming headers for messages stored for offline use works.
+ * It doesn't test streaming messages that haven't been stored for offline use
+ * because that's not implemented yet, and it's unclear if anyone will want it.
+ */
+
+var { MessageGenerator } = ChromeUtils.import(
+  "resource://testing-common/mailnews/MessageGenerator.jsm"
+);
+var { PromiseTestUtils } = ChromeUtils.import(
+  "resource://testing-common/mailnews/PromiseTestUtils.jsm"
+);
+
+setupIMAPPump();
+
+var gMsgFile1 = do_get_file("../../../data/bugmail10");
+var gMsgId1 = "200806061706.m56H6RWT004933@mrapp54.mozilla.org";
+
+// Adds some messages directly to a mailbox (e.g. new mail).
+function addMessagesToServer(messages, mailbox) {
+  // For every message we have, we need to convert it to a file:/// URI.
+  messages.forEach(function (message) {
+    let URI = Services.io
+      .newFileURI(message.file)
+      .QueryInterface(Ci.nsIFileURL);
+    // Create the ImapMessage and store it on the mailbox.
+    mailbox.addMessage(new ImapMessage(URI.spec, mailbox.uidnext++, []));
+  });
+}
+
+add_setup(async function () {
+  // Add a couple of messages to the INBOX
+  //   this is synchronous, afaik.
+  addMessagesToServer(
+    [{ file: gMsgFile1, messageId: gMsgId1 }],
+    IMAPPump.daemon.getMailbox("INBOX")
+  );
+  Services.prefs.setBoolPref(
+    "mail.server.server1.autosync_offline_stores",
+    false
+  );
+  // Update IMAP Folder.
+  let listenerUpdate = new PromiseTestUtils.PromiseUrlListener();
+  IMAPPump.inbox.updateFolderWithListener(null, listenerUpdate);
+  await listenerUpdate.promise;
+  // Download all for offline.
+  let listenerDownload = new PromiseTestUtils.PromiseUrlListener();
+  IMAPPump.inbox.downloadAllForOffline(listenerDownload, null);
+  await listenerDownload.promise;
+});
+
+add_task(async function test_streamHeaders() {
+  let newMsgHdr = IMAPPump.inbox.GetMessageHeader(1);
+  let msgURI = newMsgHdr.folder.getUriForMsg(newMsgHdr);
+  let msgServ = MailServices.messageServiceFromURI(msgURI);
+  // We use this as a display consumer
+  let streamListener = new PromiseTestUtils.PromiseStreamListener();
+  msgServ.streamHeaders(msgURI, streamListener, null, true);
+  let data = await streamListener.promise;
+  Assert.ok(data.includes("Content-Type"));
+});
+
+add_task(async function endTest() {
+  teardownIMAPPump();
+});