From e3871a45b584e8dcb633ba314e1ab1b8593c636b Mon Sep 17 00:00:00 2001 From: Duda Dornelles Date: Thu, 18 Sep 2014 18:30:28 -0300 Subject: Adding update_draft to mailboxes --- service/test/adapter/pixelated_mailboxes_test.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) (limited to 'service/test/adapter') diff --git a/service/test/adapter/pixelated_mailboxes_test.py b/service/test/adapter/pixelated_mailboxes_test.py index 15606b51..b112905c 100644 --- a/service/test/adapter/pixelated_mailboxes_test.py +++ b/service/test/adapter/pixelated_mailboxes_test.py @@ -14,6 +14,7 @@ # You should have received a copy of the GNU Affero General Public License # along with Pixelated. If not, see . import unittest +from pixelated.adapter.pixelated_mail import PixelatedMail from pixelated.adapter.pixelated_mailbox import PixelatedMailbox from pixelated.adapter.pixelated_mailboxes import PixelatedMailBoxes @@ -42,10 +43,23 @@ class PixelatedMailboxesTest(unittest.TestCase): self.assertEqual("mail", mails[0]) def test_add_draft(self): - mail = mock() + mail = PixelatedMail() when(self.drafts_mailbox).add(mail).thenReturn(1) self.mailboxes.add_draft(mail) verify(self.drafts_mailbox).add(mail) - verify(mail).set_ident('drafts', 1) + self.assertEqual('drafts', mail.mailbox_name) + self.assertEqual(1, mail.uid) + + def test_update_draft(self): + mail = PixelatedMail() + when(self.drafts_mailbox).add(mail).thenReturn(1) + + self.mailboxes.update_draft(mail) + + inorder.verify(self.drafts_mailbox).add(mail) + inorder.verify(self.drafts_mailbox).remove(mail) + + self.assertEqual('drafts', mail.mailbox_name) + self.assertEqual(1, mail.uid) -- cgit v1.2.3