diff options
Diffstat (limited to 'service/test/unit/adapter')
-rw-r--r-- | service/test/unit/adapter/test_draft_service.py | 2 | ||||
-rw-r--r-- | service/test/unit/adapter/test_mail_service.py | 6 | ||||
-rw-r--r-- | service/test/unit/adapter/test_mailboxes.py | 1 |
3 files changed, 5 insertions, 4 deletions
diff --git a/service/test/unit/adapter/test_draft_service.py b/service/test/unit/adapter/test_draft_service.py index 0dd6cd2a..79eca5f6 100644 --- a/service/test/unit/adapter/test_draft_service.py +++ b/service/test/unit/adapter/test_draft_service.py @@ -12,7 +12,7 @@ class DraftServiceTest(unittest.TestCase): self.mailboxes = mock() self.drafts_mailbox = mock() self.draft_service = DraftService(self.mailboxes) - when(self.mailboxes).drafts().thenReturn(self.drafts_mailbox) + self.mailboxes.drafts = self.drafts_mailbox def test_add_draft(self): mail = InputMail() diff --git a/service/test/unit/adapter/test_mail_service.py b/service/test/unit/adapter/test_mail_service.py index f4b89f57..f5e29b0c 100644 --- a/service/test/unit/adapter/test_mail_service.py +++ b/service/test/unit/adapter/test_mail_service.py @@ -27,9 +27,9 @@ class TestMailService(unittest.TestCase): self.drafts = mock() self.querier = mock() self.mailboxes = mock() - self.mailboxes.drafts = lambda: self.drafts - self.mailboxes.trash = lambda: mock() - self.mailboxes.sent = lambda: mock() + self.mailboxes.drafts = self.drafts + self.mailboxes.trash = mock() + self.mailboxes.sent = mock() self.mail_sender = mock() self.search_engine = mock() diff --git a/service/test/unit/adapter/test_mailboxes.py b/service/test/unit/adapter/test_mailboxes.py index 5b4548eb..6ff3849b 100644 --- a/service/test/unit/adapter/test_mailboxes.py +++ b/service/test/unit/adapter/test_mailboxes.py @@ -23,6 +23,7 @@ from mock import MagicMock class PixelatedMailboxesTest(unittest.TestCase): + def setUp(self): self.querier = mock() self.search_engine = mock() |