#432 - Adding test_mark_as_read and refactoring existing test_mark_as_read in TestMailService - Gil/Shyba

1 files changed, 14 insertions, 2 deletions

diff --git a/service/test/unit/adapter/test_mail_service.py b/service/test/unit/adapter/test_mail_service.py
index 7dd68706..c1450625 100644
--- a/service/test/unit/adapter/test_mail_service.py
+++ b/service/test/unit/adapter/test_mail_service.py
@@ -104,14 +104,26 @@ class TestMailService(unittest.TestCase):
 
     @defer.inlineCallbacks
     def test_mark_as_read(self):
-        mail = LeapMail('id', 'INBOX')
-        when(self.mail_store).get_mail(ANY(), include_body=True).thenReturn(mail)
+        mail = LeapMail(1, 'INBOX')
+        when(self.mail_store).get_mail(1, include_body=True).thenReturn(mail)
         yield self.mail_service.mark_as_read(1)
 
         self.assertIn(Status.SEEN, mail.flags)
         verify(self.mail_store).update_mail(mail)
 
     @defer.inlineCallbacks
+    def test_mark_as_unread(self):
+        mail = LeapMail(1, 'INBOX')
+        mail.flags.add(Status.SEEN)
+
+        when(self.mail_store).get_mail(1, include_body=True).thenReturn(mail)
+        yield self.mail_service.mark_as_unread(1)
+
+        verify(self.mail_store).update_mail(mail)
+
+        self.assertNotEqual(mail.status, Status.SEEN)
+
+    @defer.inlineCallbacks
     def test_delete_mail(self):
         mail_to_delete = LeapMail(1, 'INBOX')
         when(self.mail_store).get_mail(1, include_body=True).thenReturn(defer.succeed(mail_to_delete))