From 6affff711d357155eae49f595a0112aaff04ed8b Mon Sep 17 00:00:00 2001 From: Lucas Hoffmann Date: Tue, 31 Jan 2017 11:34:33 +0100 Subject: Add tests for addressbook classes --- tests/addressbook/external_test.py | 55 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 tests/addressbook/external_test.py (limited to 'tests/addressbook/external_test.py') diff --git a/tests/addressbook/external_test.py b/tests/addressbook/external_test.py new file mode 100644 index 00000000..1459348d --- /dev/null +++ b/tests/addressbook/external_test.py @@ -0,0 +1,55 @@ +# 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 +from __future__ import absolute_import + +import unittest + +import mock + +from alot.addressbook import external + + +class TestExternalAddressbookGetContacts(unittest.TestCase): + + """Some test cases for + alot.addressbook.external.ExternalAddressbook.get_contacts""" + + regex = '(?P.*)\t(?P.*)' + + @staticmethod + def _patch_call_cmd(return_value): + return mock.patch('alot.addressbook.external.call_cmd', + mock.Mock(return_value=return_value)) + + def test_raises_if_external_command_exits_with_non_zero_status(self): + abook = external.ExternalAddressbook('foobar', '') + with self._patch_call_cmd(('', '', 42)): + with self.assertRaises(external.AddressbookError) as contextmgr: + abook.get_contacts() + expected = u'abook command "foobar" returned with return code 42' + self.assertEqual(contextmgr.exception.args[0], expected) + + def test_returns_empty_list_when_command_returns_no_output(self): + abook = external.ExternalAddressbook('foobar', self.regex) + with self._patch_call_cmd(('', '', 0)) as call_cmd: + actual = abook.get_contacts() + self.assertListEqual(actual, []) + call_cmd.assert_called_once_with(['foobar']) + + def test_splits_results_from_provider_by_regex(self): + abook = external.ExternalAddressbook('foobar', self.regex) + with self._patch_call_cmd( + ('me\t\nyou\t', '', 0)): + actual = abook.get_contacts() + expected = [('me', ''), ('you', '')] + self.assertListEqual(actual, expected) + + def test_returns_empty_list_if_regex_has_no_name_and_email_submatches(self): + abook = external.ExternalAddressbook( + 'foobar', self.regex.replace('name', 'xname').replace('email', + 'xemail')) + with self._patch_call_cmd( + ('me\t\nyou\t', '', 0)): + actual = abook.get_contacts() + self.assertListEqual(actual, []) -- cgit v1.2.3