1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
# 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
"""Test suite for alot.commands.thread module."""
import email
import unittest
import mock
from alot.commands import thread
from alot.account import Account
# Good descriptive test names often don't fit PEP8, which is meant to cover
# functions meant to be called by humans.
# pylint: disable=invalid-name
# These are tests, don't worry about names like "foo" and "bar"
# pylint: disable=blacklisted-name
class _AccountTestClass(Account):
"""Implements stubs for ABC methods."""
def send_mail(self, mail):
pass
class TestDetermineSender(unittest.TestCase):
header_priority = ["From", "To", "Cc", "Envelope-To", "X-Envelope-To",
"Delivered-To"]
mailstring = '\n'.join([
"From: from@example.com",
"To: to@example.com",
"Cc: cc@example.com",
"Envelope-To: envelope-to@example.com",
"X-Envelope-To: x-envelope-to@example.com",
"Delivered-To: delivered-to@example.com",
"Subject: Alot test",
"\n",
"Some content",
])
mail = email.message_from_string(mailstring)
def _test(self, accounts=(), expected=(), mail=None, header_priority=None,
force_realname=False, force_address=False):
"""This method collects most of the steps that need to be done for most
tests. Especially a closure to mock settings.get and a mock for
settings.get_accounts are set up."""
mail = self.mail if mail is None else mail
header_priority = self.header_priority if header_priority is None \
else header_priority
def settings_get(arg):
"""Mock function for setting.get()"""
if arg == "reply_account_header_priority":
return header_priority
elif arg.endswith('_force_realname'):
return force_realname
elif arg.endswith('_force_address'):
return force_address
with mock.patch('alot.commands.thread.settings.get_accounts',
mock.Mock(return_value=accounts)):
with mock.patch('alot.commands.thread.settings.get', settings_get):
actual = thread.determine_sender(mail)
self.assertTupleEqual(actual, expected)
def test_assert_that_some_accounts_are_defined(self):
with mock.patch('alot.commands.thread.settings.get_accounts',
mock.Mock(return_value=[])) as cm1:
with self.assertRaises(AssertionError) as cm2:
thread.determine_sender(None)
expected = ('no accounts set!',)
cm1.assert_called_once_with()
self.assertTupleEqual(cm2.exception.args, expected)
def test_default_account_is_used_if_no_match_is_found(self):
account1 = _AccountTestClass(address=u'foo@example.com')
account2 = _AccountTestClass(address=u'bar@example.com')
expected = (u'foo@example.com', account1)
self._test(accounts=[account1, account2], expected=expected)
def test_matching_address_and_account_are_returned(self):
account1 = _AccountTestClass(address=u'foo@example.com')
account2 = _AccountTestClass(address=u'to@example.com')
account3 = _AccountTestClass(address=u'bar@example.com')
expected = (u'to@example.com', account2)
self._test(accounts=[account1, account2, account3], expected=expected)
def test_force_realname_has_real_name_in_returned_address_if_defined(self):
account1 = _AccountTestClass(address=u'foo@example.com')
account2 = _AccountTestClass(address=u'to@example.com', realname='Bar')
account3 = _AccountTestClass(address=u'baz@example.com')
expected = (u'Bar <to@example.com>', account2)
self._test(accounts=[account1, account2, account3], expected=expected,
force_realname=True)
def test_doesnt_fail_with_force_realname_if_real_name_not_defined(self):
account1 = _AccountTestClass(address=u'foo@example.com')
account2 = _AccountTestClass(address=u'to@example.com')
account3 = _AccountTestClass(address=u'bar@example.com')
expected = (u'to@example.com', account2)
self._test(accounts=[account1, account2, account3], expected=expected,
force_realname=True)
def test_with_force_address_main_address_is_always_used(self):
# In python 3.4 this and the next test could be written as subtests.
account1 = _AccountTestClass(address=u'foo@example.com')
account2 = _AccountTestClass(address=u'bar@example.com',
aliases=[u'to@example.com'])
account3 = _AccountTestClass(address=u'bar@example.com')
expected = (u'bar@example.com', account2)
self._test(accounts=[account1, account2, account3], expected=expected,
force_address=True)
def test_without_force_address_matching_address_is_used(self):
# In python 3.4 this and the previous test could be written as
# subtests.
account1 = _AccountTestClass(address=u'foo@example.com')
account2 = _AccountTestClass(address=u'bar@example.com',
aliases=[u'to@example.com'])
account3 = _AccountTestClass(address=u'baz@example.com')
expected = (u'to@example.com', account2)
self._test(accounts=[account1, account2, account3], expected=expected,
force_address=False)
def test_uses_to_header_if_present(self):
account1 = _AccountTestClass(address=u'foo@example.com')
account2 = _AccountTestClass(address=u'to@example.com')
account3 = _AccountTestClass(address=u'bar@example.com')
expected = (u'to@example.com', account2)
self._test(accounts=[account1, account2, account3], expected=expected)
def test_header_order_is_more_important_than_accounts_order(self):
account1 = _AccountTestClass(address=u'cc@example.com')
account2 = _AccountTestClass(address=u'to@example.com')
account3 = _AccountTestClass(address=u'bcc@example.com')
expected = (u'to@example.com', account2)
self._test(accounts=[account1, account2, account3], expected=expected)
def test_accounts_can_be_found_by_alias_regex_setting(self):
account1 = _AccountTestClass(address=u'foo@example.com')
account2 = _AccountTestClass(address=u'to@example.com',
alias_regexp=r'to\+.*@example.com')
account3 = _AccountTestClass(address=u'bar@example.com')
mailstring = self.mailstring.replace(u'to@example.com',
u'to+some_tag@example.com')
mail = email.message_from_string(mailstring)
expected = (u'to+some_tag@example.com', account2)
self._test(accounts=[account1, account2, account3], expected=expected,
mail=mail)
|
