1 files changed, 90 insertions, 0 deletions
diff --git a/comm/mailnews/imap/test/unit/test_preserveDataOnMove.js b/comm/mailnews/imap/test/unit/test_preserveDataOnMove.js
new file mode 100644
index 0000000000..a1f08cc8f0
--- /dev/null
+++ b/comm/mailnews/imap/test/unit/test_preserveDataOnMove.js
@@ -0,0 +1,90 @@
+/* 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 tests that arbitrary message header properties are preserved
+//  during online move of an imap message.
+
+var { PromiseTestUtils } = ChromeUtils.import(
+  "resource://testing-common/mailnews/PromiseTestUtils.jsm"
+);
+var { MailServices } = ChromeUtils.import(
+  "resource:///modules/MailServices.jsm"
+);
+
+var gMessage = "bugmail10"; // message file used as the test message
+var gSubfolder;
+
+add_setup(function () {
+  setupIMAPPump();
+  Services.prefs.setBoolPref(
+    "mail.server.default.autosync_offline_stores",
+    false
+  );
+});
+
+add_task(async function createSubfolder() {
+  IMAPPump.incomingServer.rootFolder.createSubfolder("Subfolder", null);
+  await PromiseTestUtils.promiseFolderAdded("Subfolder");
+  gSubfolder = IMAPPump.incomingServer.rootFolder.getChildNamed("Subfolder");
+  Assert.ok(gSubfolder instanceof Ci.nsIMsgImapMailFolder);
+  let listener = new PromiseTestUtils.PromiseUrlListener();
+  gSubfolder.updateFolderWithListener(null, listener);
+  await listener.promise;
+});
+
+// load and update a message in the imap fake server
+add_task(async function loadImapMessage() {
+  IMAPPump.mailbox.addMessage(
+    new ImapMessage(specForFileName(gMessage), IMAPPump.mailbox.uidnext++, [])
+  );
+  IMAPPump.inbox.updateFolder(null);
+  await PromiseTestUtils.promiseFolderNotification(IMAPPump.inbox, "msgAdded");
+  Assert.equal(1, IMAPPump.inbox.getTotalMessages(false));
+  let msgHdr = mailTestUtils.firstMsgHdr(IMAPPump.inbox);
+  Assert.ok(msgHdr instanceof Ci.nsIMsgDBHdr);
+
+  // set an arbitrary property
+  msgHdr.setStringProperty("testprop", "somevalue");
+});
+
+// move the message to a subfolder
+add_task(async function moveMessageToSubfolder() {
+  let msgHdr = mailTestUtils.firstMsgHdr(IMAPPump.inbox);
+  let copyListener = new PromiseTestUtils.PromiseCopyListener();
+  MailServices.copy.copyMessages(
+    IMAPPump.inbox, // srcFolder
+    [msgHdr], // messages
+    gSubfolder, // dstFolder
+    true, // isMove
+    copyListener, // listener
+    null, // msgWindow
+    false // allowUndo
+  );
+  await copyListener.promise;
+});
+
+add_task(async function testPropertyOnMove() {
+  let listener = new PromiseTestUtils.PromiseUrlListener();
+  gSubfolder.updateFolderWithListener(null, listener);
+  await PromiseTestUtils.promiseFolderNotification(gSubfolder, "msgAdded");
+  await listener.promise;
+  let msgHdr = mailTestUtils.firstMsgHdr(gSubfolder);
+  Assert.equal(msgHdr.getStringProperty("testprop"), "somevalue");
+});
+
+// Cleanup
+add_task(function endTest() {
+  teardownIMAPPump();
+});
+
+/*
+ * helper functions
+ */
+
+// given a test file, return the file uri spec
+function specForFileName(aFileName) {
+  let file = do_get_file("../../../data/" + aFileName);
+  let msgfileuri = Services.io.newFileURI(file).QueryInterface(Ci.nsIFileURL);
+  return msgfileuri.spec;
+}