diff options
Diffstat (limited to 'service/test')
-rw-r--r-- | service/test/integration/mark_as_read_unread_test.py | 20 | ||||
-rw-r--r-- | service/test/support/integration_helper.py | 3 |
2 files changed, 19 insertions, 4 deletions
diff --git a/service/test/integration/mark_as_read_unread_test.py b/service/test/integration/mark_as_read_unread_test.py index a75e8c92..cdac78d5 100644 --- a/service/test/integration/mark_as_read_unread_test.py +++ b/service/test/integration/mark_as_read_unread_test.py @@ -30,21 +30,33 @@ class MarkAsReadTest(unittest.TestCase, SoledadTestBase): self.add_mail_to_inbox(input_mail) mails = self.get_mails_by_tag('inbox') - self.assertFalse('read' in mails[0].status) + self.assertNotIn('read', mails[0].status) self.mark_as_read(input_mail.ident) mails = self.get_mails_by_tag('inbox') - self.assertTrue('read' in mails[0].status) + self.assertIn('read', mails[0].status) def test_mark_single_as_unread(self): input_mail = MailBuilder().with_status('read').build_input_mail() + self.add_mail_to_inbox(input_mail) + self.mark_as_unread(input_mail.ident) + mails = self.get_mails_by_tag('inbox') - self.assertIn('read', mails[0].status) + self.assertNotIn('read', mails[0].status) - self.mark_as_unread(input_mail.ident) + def test_mark_many_mails_as_unread(self): + input_mail = MailBuilder().with_status('read').build_input_mail() + input_mail2 = MailBuilder().with_status('read').build_input_mail() + + self.add_mail_to_inbox(input_mail) + self.add_mail_to_inbox(input_mail2) + + self.mark_many_as_unread([input_mail.ident, input_mail2.ident]) mails = self.get_mails_by_tag('inbox') + self.assertNotIn('read', mails[0].status) + self.assertNotIn('read', mails[1].status) diff --git a/service/test/support/integration_helper.py b/service/test/support/integration_helper.py index c2f84157..8f42e335 100644 --- a/service/test/support/integration_helper.py +++ b/service/test/support/integration_helper.py @@ -186,6 +186,9 @@ class SoledadTestBase: def mark_as_unread(self, mail_ident): self.app.post('/mail/' + mail_ident + '/unread', content_type="application/json") + def mark_many_as_unread(self, idents): + self.app.post('/mails/unread', data={'idents': json.dumps(idents)}) + def add_mail_to_inbox(self, input_mail): mail = self.pixelated_mailboxes.inbox().add(input_mail) self.search_engine.index_mail(mail) |