diff options
Diffstat (limited to 'service/test/adapter/draft_service_test.py')
-rw-r--r-- | service/test/adapter/draft_service_test.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/service/test/adapter/draft_service_test.py b/service/test/adapter/draft_service_test.py new file mode 100644 index 00000000..92b25d36 --- /dev/null +++ b/service/test/adapter/draft_service_test.py @@ -0,0 +1,29 @@ +import unittest +from pixelated.adapter.pixelated_mail import InputMail +from pixelated.adapter.draft_service import DraftService +import test_helper +from mockito import * + + +class DraftServiceTest(unittest.TestCase): + + def setUp(self): + self.mailboxes = mock() + self.drafts_mailbox = mock() + self.draft_service = DraftService(self.mailboxes) + when(self.mailboxes).drafts().thenReturn(self.drafts_mailbox) + + def test_add_draft(self): + mail = InputMail() + self.draft_service.create_draft(mail) + + verify(self.drafts_mailbox).add(mail) + + def test_update_draft(self): + mail = test_helper.input_mail() + when(self.drafts_mailbox).add(mail).thenReturn(mail) + + self.draft_service.update_draft(mail.ident, mail) + + inorder.verify(self.drafts_mailbox).add(mail) + inorder.verify(self.drafts_mailbox).remove(mail.ident) |