Add test for address book completion

1 files changed, 86 insertions, 0 deletions

diff --git a/tests/completion_test.py b/tests/completion_test.py
new file mode 100644
index 00000000..817326b6
--- /dev/null
+++ b/tests/completion_test.py
@@ -0,0 +1,86 @@
+# encoding=utf-8
+# Copyright (C) 2017  Lucas Hoffmann
+# This file is released under the GNU GPL, version 3 or a later revision.
+# For further details see the COPYING file
+
+"""Tests for the alot.completion module."""
+from __future__ import absolute_import
+
+import unittest
+
+import mock
+
+from alot import completion
+
+
+class AbooksCompleterTest(unittest.TestCase):
+
+    @staticmethod
+    def _mock_lookup(query):
+        """Look up the query from fixed list of names and email addresses."""
+        abook = [
+            ("", "no-real-name@example.com"),
+            ("foo", "foo@example.com"),
+            ("comma, person", "comma@example.com"),
+            ("single 'quote' person", "squote@example.com"),
+            ('double "quote" person', "dquote@example.com"),
+            ("""all 'fanzy' "stuff" at, once""", "all@example.com")
+        ]
+        results = []
+        for name, email in abook:
+            if query in name or query in email:
+                results.append((name, email))
+        return results
+
+    @classmethod
+    def setUpClass(cls):
+        abook = mock.Mock()
+        abook.lookup = cls._mock_lookup
+        cls.empty_abook_completer = completion.AbooksCompleter([])
+        cls.example_abook_completer = completion.AbooksCompleter([abook])
+    def test_empty_address_book_returns_empty_list(self):
+        actual = self.__class__.empty_abook_completer.complete('real-name', 9)
+        expected = []
+        self.assertListEqual(actual, expected)
+
+    def _assert_only_one_list_entry(self, actual, expected):
+        """Check that the given lists are both of length 1 and the tuple at the
+        first positions are equal."""
+        self.assertEqual(len(actual), 1)
+        self.assertEqual(len(expected), 1)
+        self.assertTupleEqual(actual[0], expected[0])
+    def test_empty_real_name_returns_plain_email_address(self):
+        actual = self.__class__.example_abook_completer.complete("real-name", 9)
+        expected = [("no-real-name@example.com", 24)]
+        self._assert_only_one_list_entry(actual, expected)
+
+    def test_simple_address_with_real_name(self):
+        actual = self.__class__.example_abook_completer.complete("foo", 3)
+        expected = [("foo <foo@example.com>", 21)]
+        self.assertListEqual(actual, expected)
+
+    @unittest.expectedFailure
+    def test_real_name_with_comma(self):
+        actual = self.__class__.example_abook_completer.complete("comma", 5)
+        expected = [('"comma, person" <comma@example.com>', 32)]
+        self.assertListEqual(actual, expected)
+
+    @unittest.expectedFailure
+    def test_real_name_with_single_quotes(self):
+        actual = self.__class__.example_abook_completer.complete("squote", 6)
+        expected = [(""""single 'quote' person" <squote@example.com>""", 47)]
+        self._assert_only_one_list_entry(actual, expected)
+
+    @unittest.expectedFailure
+    def test_real_name_double_quotes(self):
+        actual = self.__class__.example_abook_completer.complete("dquote", 6)
+        expected = [("", 0)]
+        expected = [(""""double 'quote' person" <dquote@example.com>""", 47)]
+        self._assert_only_one_list_entry(actual, expected)
+
+    @unittest.expectedFailure
+    def test_real_name_with_quotes_and_comma(self):
+        actual = self.__class__.example_abook_completer.complete("all", 3)
+        expected = [(r""""all 'fanzy' \"stuff\" at, once" <all@example.com>""",
+                     50)]
+        self._assert_only_one_list_entry(actual, expected)