summaryrefslogtreecommitdiff
path: root/tests/db/test_message.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/db/test_message.py')
-rw-r--r--tests/db/test_message.py16
1 files changed, 8 insertions, 8 deletions
diff --git a/tests/db/test_message.py b/tests/db/test_message.py
index 43b06e19..cd50ab45 100644
--- a/tests/db/test_message.py
+++ b/tests/db/test_message.py
@@ -64,9 +64,9 @@ class TestMessage(unittest.TestCase):
"""Message._from is populated using the 'From' header when only an
email address is provided.
"""
- msg = message.Message(mock.Mock(),
+ msg = message.Message(mock.Mock(), mock.Mock(),
MockNotmuchMessage({'From': 'user@example.com'}),
- mock.Mock(), mock.Mock())
+ mock.Mock())
self.assertEqual(msg.get_author(), ('', 'user@example.com'))
def test_get_author_name_and_email(self):
@@ -74,9 +74,9 @@ class TestMessage(unittest.TestCase):
name are provided.
"""
msg = message.Message(
- mock.Mock(),
+ mock.Mock(), mock.Mock(),
MockNotmuchMessage({'From': '"User Name" <user@example.com>'}),
- mock.Mock(), mock.Mock())
+ mock.Mock())
self.assertEqual(msg.get_author(), ('User Name', 'user@example.com'))
def test_get_author_sender(self):
@@ -84,9 +84,9 @@ class TestMessage(unittest.TestCase):
header is present.
"""
msg = message.Message(
- mock.Mock(),
+ mock.Mock(), mock.Mock(),
MockNotmuchMessage({'Sender': '"User Name" <user@example.com>'}),
- mock.Mock(), mock.Mock())
+ mock.Mock())
self.assertEqual(msg.get_author(), ('User Name', 'user@example.com'))
def test_get_author_no_name_draft(self):
@@ -99,7 +99,7 @@ class TestMessage(unittest.TestCase):
with mock.patch('alot.db.message.settings.get_accounts',
mock.Mock(return_value=[acc])):
msg = message.Message(
- mock.Mock(), MockNotmuchMessage(tags=['draft']), mock.Mock(), mock.Mock())
+ mock.Mock(), mock.Mock(), MockNotmuchMessage(tags=['draft']), mock.Mock())
self.assertEqual(msg.get_author(), ('User Name', 'user@example.com'))
def test_get_author_no_name(self):
@@ -111,5 +111,5 @@ class TestMessage(unittest.TestCase):
acc.realname = 'User Name'
with mock.patch('alot.db.message.settings.get_accounts',
mock.Mock(return_value=[acc])):
- msg = message.Message(mock.Mock(), MockNotmuchMessage(), mock.Mock(), mock.Mock())
+ msg = message.Message(mock.Mock(), mock.Mock(), MockNotmuchMessage(), mock.Mock())
self.assertEqual(msg.get_author(), ('Unknown', ''))