1 files changed, 253 insertions, 0 deletions
diff --git a/comm/mailnews/compose/test/unit/test_bug474774.js b/comm/mailnews/compose/test/unit/test_bug474774.js
new file mode 100644
index 0000000000..ba0c20667c
--- /dev/null
+++ b/comm/mailnews/compose/test/unit/test_bug474774.js
@@ -0,0 +1,253 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/**
+ * Tests bug 474774 - assertions when saving send later and when sending with
+ * FCC switched off.
+ */
+
+var { MailServices } = ChromeUtils.import(
+  "resource:///modules/MailServices.jsm"
+);
+
+var server;
+var smtpServer;
+var originalData;
+var finished = false;
+var identity = null;
+
+var testFile = do_get_file("data/429891_testcase.eml");
+
+var kTestFileSender = "from_A@foo.invalid";
+var kTestFileRecipient = "to_A@foo.invalid";
+
+var kIdentityMail = "identity@foo.invalid";
+
+var msgSendLater = Cc["@mozilla.org/messengercompose/sendlater;1"].getService(
+  Ci.nsIMsgSendLater
+);
+
+// This listener handles the post-sending of the actual message and checks the
+// sequence and ensures the data is correct.
+function msll() {}
+
+msll.prototype = {
+  _initialTotal: 0,
+
+  // nsIMsgSendLaterListener
+  onStartSending(aTotalMessageCount) {
+    this._initialTotal = 1;
+    Assert.equal(msgSendLater.sendingMessages, true);
+  },
+  onMessageStartSending(
+    aCurrentMessage,
+    aTotalMessageCount,
+    aMessageHeader,
+    aIdentity
+  ) {},
+  onMessageSendProgress(
+    aCurrentMessage,
+    aTotalMessageCount,
+    aMessageSendPercent,
+    aMessageCopyPercent
+  ) {
+    // XXX Enable this function
+  },
+  onMessageSendError(aCurrentMessage, aMessageHeader, aStatus, aMsg) {
+    do_throw(
+      "onMessageSendError should not have been called, status: " + aStatus
+    );
+  },
+  onStopSending(aStatus, aMsg, aTotalTried, aSuccessful) {
+    print("msll onStopSending\n");
+    try {
+      Assert.equal(aSuccessful, 1);
+      Assert.equal(aStatus, 0);
+      Assert.equal(aTotalTried, 1);
+      Assert.equal(this._initialTotal, 1);
+      Assert.equal(msgSendLater.sendingMessages, false);
+
+      do_check_transaction(server.playTransaction(), [
+        "EHLO test",
+        "MAIL FROM:<" +
+          kTestFileSender +
+          "> BODY=8BITMIME SIZE=" +
+          originalData.length,
+        "RCPT TO:<" + kTestFileRecipient + ">",
+        "DATA",
+      ]);
+
+      // Compare data file to what the server received
+      Assert.equal(originalData, server._daemon.post);
+
+      // Now wait till the copy is finished for the sent message
+      do_test_pending();
+    } catch (e) {
+      do_throw(e);
+    } finally {
+      server.stop();
+
+      var thread = gThreadManager.currentThread;
+      while (thread.hasPendingEvents()) {
+        thread.processNextEvent(true);
+      }
+    }
+    do_test_finished();
+  },
+};
+
+/* exported OnStopCopy */
+// for head_compose.js
+function OnStopCopy(aStatus) {
+  do_test_finished();
+
+  try {
+    Assert.equal(aStatus, 0);
+
+    // Check this is false before we start sending
+    Assert.equal(msgSendLater.sendingMessages, false);
+
+    let folder = msgSendLater.getUnsentMessagesFolder(identity);
+
+    // Check we have a message in the unsent message folder
+    Assert.equal(folder.getTotalMessages(false), 1);
+
+    // Now do a comparison of what is in the sent mail folder
+    let msgData = mailTestUtils.loadMessageToString(
+      folder,
+      mailTestUtils.firstMsgHdr(folder)
+    );
+
+    // Skip the headers etc that mailnews adds
+    var pos = msgData.indexOf("From:");
+    Assert.notEqual(pos, -1);
+
+    msgData = msgData.substr(pos);
+
+    // Check the data is matching.
+    Assert.equal(originalData, msgData);
+
+    do_test_pending();
+    sendMessageLater();
+  } catch (e) {
+    do_throw(e);
+  } finally {
+    server.stop();
+
+    var thread = gThreadManager.currentThread;
+    while (thread.hasPendingEvents()) {
+      thread.processNextEvent(true);
+    }
+
+    finished = true;
+  }
+}
+
+// This function does the actual send later
+function sendMessageLater() {
+  do_test_finished();
+
+  // Set up the SMTP server.
+  server = setupServerDaemon();
+
+  // Handle the server in a try/catch/finally loop so that we always will stop
+  // the server if something fails.
+  try {
+    // Start the fake SMTP server
+    server.start();
+    smtpServer.port = server.port;
+
+    // A test to check that we are sending files correctly, including checking
+    // what the server receives and what we output.
+    test = "sendMessageLater";
+
+    var messageListener = new msll();
+
+    msgSendLater.addListener(messageListener);
+
+    // Send the unsent message
+    msgSendLater.sendUnsentMessages(identity);
+
+    server.performTest();
+
+    do_timeout(10000, function () {
+      if (!finished) {
+        do_throw("Notifications of message send/copy not received");
+      }
+    });
+
+    do_test_pending();
+  } catch (e) {
+    do_throw(e);
+  } finally {
+    server.stop();
+
+    var thread = gThreadManager.currentThread;
+    while (thread.hasPendingEvents()) {
+      thread.processNextEvent(true);
+    }
+  }
+}
+
+add_task(async function run_the_test() {
+  // Test file - for bug 429891
+  originalData = await IOUtils.readUTF8(testFile.path);
+
+  // Ensure we have a local mail account, an normal account and appropriate
+  // servers and identities.
+  localAccountUtils.loadLocalMailAccount();
+
+  MailServices.accounts.setSpecialFolders();
+
+  let account = MailServices.accounts.createAccount();
+  let incomingServer = MailServices.accounts.createIncomingServer(
+    "test",
+    "localhost",
+    "pop3"
+  );
+
+  smtpServer = getBasicSmtpServer(0);
+  identity = getSmtpIdentity(kIdentityMail, smtpServer);
+
+  account.addIdentity(identity);
+  account.defaultIdentity = identity;
+  account.incomingServer = incomingServer;
+  MailServices.accounts.defaultAccount = account;
+
+  localAccountUtils.rootFolder.createLocalSubfolder("Sent");
+
+  identity.doFcc = false;
+
+  // Now prepare to actually "send" the message later, i.e. dump it in the
+  // unsent messages folder.
+
+  var compFields = Cc[
+    "@mozilla.org/messengercompose/composefields;1"
+  ].createInstance(Ci.nsIMsgCompFields);
+
+  // Setting the compFields sender and recipient to any value is required to
+  // survive mime_sanity_check_fields in nsMsgCompUtils.cpp.
+  // Sender and recipient are required for sendMessageFile but SMTP
+  // transaction values will be used directly from mail body.
+  compFields.from = "irrelevant@foo.invalid";
+  compFields.to = "irrelevant@foo.invalid";
+
+  var msgSend = Cc["@mozilla.org/messengercompose/send;1"].createInstance(
+    Ci.nsIMsgSend
+  );
+
+  msgSend.sendMessageFile(
+    identity,
+    "",
+    compFields,
+    testFile,
+    false,
+    false,
+    Ci.nsIMsgSend.nsMsgQueueForLater,
+    null,
+    copyListener,
+    null,
+    null
+  );
+
+  // Now we wait till we get copy notification of completion.
+  do_test_pending();
+});