1 files changed, 144 insertions, 0 deletions
diff --git a/comm/mailnews/base/test/unit/test_viewSortByAddresses.js b/comm/mailnews/base/test/unit/test_viewSortByAddresses.js
new file mode 100644
index 0000000000..7235723bfd
--- /dev/null
+++ b/comm/mailnews/base/test/unit/test_viewSortByAddresses.js
@@ -0,0 +1,144 @@
+/* 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/. */
+
+/*
+ * Attempt to test nsMsgDBView's handling of sorting by sender/recipients
+ * when using a display name from the address book.
+ */
+
+/* import-globals-from ../../../test/resources/abSetup.js */
+load("../../../resources/abSetup.js");
+
+var { MailServices } = ChromeUtils.import(
+  "resource:///modules/MailServices.jsm"
+);
+var { MessageGenerator, SyntheticMessageSet } = ChromeUtils.import(
+  "resource://testing-common/mailnews/MessageGenerator.jsm"
+);
+var { MessageInjection } = ChromeUtils.import(
+  "resource://testing-common/mailnews/MessageInjection.jsm"
+);
+var { dump_view_contents } = ChromeUtils.import(
+  "resource://testing-common/mozmill/ViewHelpers.jsm"
+);
+
+var gMessageGenerator = new MessageGenerator();
+var messageInjection = new MessageInjection({ mode: "local" });
+
+Services.prefs.setBoolPref("mail.showCondensedAddresses", true);
+
+var gTestFolder;
+
+// Setup the display name to be opposite of alphabetic order of e-mail address.
+var cards = [
+  { email: "aaa@b.invalid", displayName: "4" },
+  { email: "ccc@d.invalid", displayName: "3" },
+  { email: "eee@f.invalid", displayName: "2" },
+  { email: "ggg@h.invalid", displayName: "1" },
+];
+
+add_setup(async function () {
+  // Ensure all the directories are initialised.
+  MailServices.ab.directories;
+
+  let ab = MailServices.ab.getDirectory(kPABData.URI);
+
+  function createAndAddCard(element) {
+    var card = Cc["@mozilla.org/addressbook/cardproperty;1"].createInstance(
+      Ci.nsIAbCard
+    );
+
+    card.primaryEmail = element.email;
+    card.displayName = element.displayName;
+
+    ab.addCard(card);
+  }
+
+  // Add address to addressbook so we can set display name and verify that
+  // the view uses the display name for display and sorting.
+  cards.forEach(createAndAddCard);
+
+  // build up a couple message with addresses in the ab.
+  let messages = [];
+  messages = messages.concat(
+    gMessageGenerator.makeMessage({
+      from: ["aaa", "aaa@b.invalid"],
+      to: [["ccc", "ccc@d.invalid"]],
+    })
+  );
+  messages = messages.concat(
+    gMessageGenerator.makeMessage({
+      from: ["eee", "eee@f.invalid"],
+      to: [["ggg", "ggg@h.invalid"]],
+    })
+  );
+
+  let msgSet = new SyntheticMessageSet(messages);
+  gTestFolder = await messageInjection.makeEmptyFolder();
+  await messageInjection.addSetsToFolders([gTestFolder], [msgSet]);
+});
+
+add_task(function test_view_sort_by_addresses() {
+  // - create the view
+  setup_view("threaded", Ci.nsMsgViewFlagsType.kNone);
+  // Check that sorting by sender uses the display name
+  gDBView.sort(Ci.nsMsgViewSortType.byAuthor, Ci.nsMsgViewSortOrder.ascending);
+  let sender1 = gDBView.cellTextForColumn(0, "senderCol");
+  let sender2 = gDBView.cellTextForColumn(1, "senderCol");
+
+  if (sender1 != 2) {
+    view_throw("expected sender 1 to be 2");
+  }
+  if (sender2 != 4) {
+    view_throw("expected sender 2 to be 4");
+  }
+
+  gDBView.sort(
+    Ci.nsMsgViewSortType.byRecipient,
+    Ci.nsMsgViewSortOrder.ascending
+  );
+  let recip1 = gDBView.cellTextForColumn(0, "recipientCol");
+  let recip2 = gDBView.cellTextForColumn(1, "recipientCol");
+
+  if (recip1 != 1) {
+    view_throw("expected recip 1 to be 1");
+  }
+  if (recip2 != 3) {
+    view_throw("expected recip 2 to be 3");
+  }
+});
+
+function view_throw(why) {
+  dump_view_contents();
+  do_throw(why);
+}
+var gDBView;
+var gTreeView;
+
+function setup_view(aViewType, aViewFlags, aTestFolder) {
+  let dbviewContractId = "@mozilla.org/messenger/msgdbview;1?type=" + aViewType;
+
+  if (aTestFolder == null) {
+    aTestFolder = gTestFolder;
+  }
+
+  // always start out fully expanded
+  aViewFlags |= Ci.nsMsgViewFlagsType.kExpandAll;
+
+  gDBView = Cc[dbviewContractId].createInstance(Ci.nsIMsgDBView);
+  gDBView.init(null, null, null);
+  var outCount = {};
+  gDBView.open(
+    aViewType != "search" ? aTestFolder : null,
+    Ci.nsMsgViewSortType.byDate,
+    aViewType != "search"
+      ? Ci.nsMsgViewSortOrder.ascending
+      : Ci.nsMsgViewSortOrder.descending,
+    aViewFlags,
+    outCount
+  );
+  dump("  View Out Count: " + outCount.value + "\n");
+
+  gTreeView = gDBView.QueryInterface(Ci.nsITreeView);
+}