From fbdde65a7f5eab74d2334481cee52ba429910c22 Mon Sep 17 00:00:00 2001 From: Anton Khirnov Date: Thu, 6 Feb 2020 14:15:56 +0100 Subject: db/message: forbid instantiating Message with thread = None It should always be instantiated from a Thread instance. --- tests/db/test_message.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) (limited to 'tests') diff --git a/tests/db/test_message.py b/tests/db/test_message.py index adc099f6..953d0fe4 100644 --- a/tests/db/test_message.py +++ b/tests/db/test_message.py @@ -65,7 +65,8 @@ class TestMessage(unittest.TestCase): email address is provided. """ msg = message.Message(mock.Mock(), - MockNotmuchMessage({'From': 'user@example.com'})) + MockNotmuchMessage({'From': 'user@example.com'}), + mock.Mock()) self.assertEqual(msg.get_author(), ('', 'user@example.com')) def test_get_author_name_and_email(self): @@ -74,7 +75,8 @@ class TestMessage(unittest.TestCase): """ msg = message.Message( mock.Mock(), - MockNotmuchMessage({'From': '"User Name" '})) + MockNotmuchMessage({'From': '"User Name" '}), + mock.Mock()) self.assertEqual(msg.get_author(), ('User Name', 'user@example.com')) def test_get_author_sender(self): @@ -83,7 +85,8 @@ class TestMessage(unittest.TestCase): """ msg = message.Message( mock.Mock(), - MockNotmuchMessage({'Sender': '"User Name" '})) + MockNotmuchMessage({'Sender': '"User Name" '}), + mock.Mock()) self.assertEqual(msg.get_author(), ('User Name', 'user@example.com')) def test_get_author_no_name_draft(self): @@ -96,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(), MockNotmuchMessage(tags=['draft']), mock.Mock()) self.assertEqual(msg.get_author(), ('User Name', 'user@example.com')) def test_get_author_no_name(self): @@ -108,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()) + msg = message.Message(mock.Mock(), MockNotmuchMessage(), mock.Mock()) self.assertEqual(msg.get_author(), ('Unknown', '')) -- cgit v1.2.3