diff options
Diffstat (limited to 'service/test/integration/mark_as_read_unread_test.py')
| -rw-r--r-- | service/test/integration/mark_as_read_unread_test.py | 19 |
1 files changed, 9 insertions, 10 deletions
diff --git a/service/test/integration/mark_as_read_unread_test.py b/service/test/integration/mark_as_read_unread_test.py index c04cdbc6..55467e9e 100644 --- a/service/test/integration/mark_as_read_unread_test.py +++ b/service/test/integration/mark_as_read_unread_test.py @@ -13,9 +13,8 @@ # # You should have received a copy of the GNU Affero General Public License # along with Pixelated. If not, see <http://www.gnu.org/licenses/>. -import unittest -from test.support.integration_helper import MailBuilder, SoledadTestBase +from test.support.integration import * from pixelated.adapter.status import Status @@ -29,7 +28,7 @@ class MarkAsReadUnreadTest(SoledadTestBase): def test_mark_single_as_read(self): input_mail = MailBuilder().build_input_mail() - self.add_mail_to_inbox(input_mail) + self.client.add_mail_to_inbox(input_mail) mails = self.get_mails_by_tag('inbox') self.assertNotIn('read', mails[0].status) @@ -41,7 +40,7 @@ class MarkAsReadUnreadTest(SoledadTestBase): def test_mark_single_as_unread(self): input_mail = MailBuilder().with_status([Status.SEEN]).build_input_mail() - self.add_mail_to_inbox(input_mail) + self.client.add_mail_to_inbox(input_mail) self.mark_as_unread(input_mail.ident) mail = self.get_mails_by_tag('inbox')[0] @@ -52,8 +51,8 @@ class MarkAsReadUnreadTest(SoledadTestBase): input_mail = MailBuilder().with_status([Status.SEEN]).build_input_mail() input_mail2 = MailBuilder().with_status([Status.SEEN]).build_input_mail() - self.add_mail_to_inbox(input_mail) - self.add_mail_to_inbox(input_mail2) + self.client.add_mail_to_inbox(input_mail) + self.client.add_mail_to_inbox(input_mail2) self.mark_many_as_unread([input_mail.ident, input_mail2.ident]) @@ -66,8 +65,8 @@ class MarkAsReadUnreadTest(SoledadTestBase): input_mail = MailBuilder().build_input_mail() input_mail2 = MailBuilder().build_input_mail() - self.add_mail_to_inbox(input_mail) - self.add_mail_to_inbox(input_mail2) + self.client.add_mail_to_inbox(input_mail) + self.client.add_mail_to_inbox(input_mail2) mails = self.get_mails_by_tag('inbox') @@ -86,8 +85,8 @@ class MarkAsReadUnreadTest(SoledadTestBase): input_mail = MailBuilder().build_input_mail() input_mail2 = MailBuilder().with_status([Status.SEEN]).build_input_mail() - self.add_mail_to_inbox(input_mail) - self.add_mail_to_inbox(input_mail2) + self.client.add_mail_to_inbox(input_mail) + self.client.add_mail_to_inbox(input_mail2) mails = self.get_mails_by_tag('inbox') |