for #167, deduplicate contacts by largest

2 files changed, 58 insertions, 0 deletions

diff --git a/service/test/integration/test_contacts.py b/service/test/integration/test_contacts.py
index 4974c471..c5baa094 100644
--- a/service/test/integration/test_contacts.py
+++ b/service/test/integration/test_contacts.py
@@ -57,3 +57,26 @@ class ContactsTest(SoledadTestBase):
             self.assertFalse('recipient@trash.com' in contacts)
         d.addCallback(_assert)
         return d
+
+    def test_deduplication_on_same_mail_address_using_largest(self):
+        input_mail = MailBuilder().with_tags(['important']).build_input_mail()
+
+        formatted_input_mail = MailBuilder().with_tags(['important'])
+        formatted_input_mail.with_to('Recipient Principal <recipient@to.com>')
+        formatted_input_mail.with_cc('Recipient Copied <recipient@cc.com>')
+        formatted_input_mail.with_bcc('Recipient Carbon <recipient@bcc.com>')
+        formatted_input_mail = formatted_input_mail.build_input_mail()
+
+        self.client.add_mail_to_inbox(input_mail)
+        self.client.add_mail_to_inbox(formatted_input_mail)
+
+        d = self.get_contacts(query='Recipient')
+
+        def _assert(contacts):
+            print contacts
+            self.assertEquals(3, len(contacts))
+            self.assertTrue('Recipient Principal <recipient@to.com>' in contacts)
+            self.assertTrue('Recipient Copied <recipient@cc.com>' in contacts)
+            self.assertTrue('Recipient Carbon <recipient@bcc.com>' in contacts)
+        d.addCallback(_assert)
+        return d
diff --git a/service/test/unit/adapter/test_contacts.py b/service/test/unit/adapter/test_contacts.py
new file mode 100644
index 00000000..ee9bf618
--- /dev/null
+++ b/service/test/unit/adapter/test_contacts.py
@@ -0,0 +1,35 @@
+#
+# Copyright (c) 2014 ThoughtWorks, Inc.
+#
+# Pixelated is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Pixelated is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
+import unittest
+
+from pixelated.adapter.contacts import address_duplication_filter
+from pixelated.adapter.contacts import extract_mail_address
+
+
+class TestContacts(unittest.TestCase):
+
+    def test_contacts_filter_duplication_by_largest(self):
+        contacts = ['John Large Name <john@name.example.com>', 'john@name.example.com', 'dont.delete@example.com']
+        contacts_filtered = address_duplication_filter(contacts)
+        self.assertIn('dont.delete@example.com', contacts_filtered)
+        self.assertIn('John Large Name <john@name.example.com>', contacts_filtered)
+        self.assertNotIn('john@name.example.com', contacts_filtered)
+
+    def test_extract_mail_address_from_contact(self):
+        full_address = 'John Large Name <john@name.example.com>'
+        mail_address = 'john@name.example.com'
+
+        self.assertEquals(mail_address, extract_mail_address(full_address))