class MessagesEndpointTest(APITestCase):
    def setUp(self):
        self.super_user_password = '12345678'
        self.super_user = User.objects.create_superuser(username='super', email='test@example.com', password=self.super_user_password)

    def test_list_messages(self):
        emails_user = create_user('test_user')

        deferred_emails = []
        deferred_emails.append(create_deferred_email(emails_user, message='test message1'))
        deferred_emails.append(create_deferred_email(emails_user, message='test message2', status=DeferredEmail.FAILED))
        deferred_emails.append(create_deferred_email(emails_user, message='test message3', status=DeferredEmail.SENT))

        is_logged_in = self.client.login(username=self.super_user.username, password=self.super_user_password)
        self.assertTrue(is_logged_in)

        response = self.client.get(reverse('deferred_email:deferredemail-list'))
        self.assertEqual(len(response.data['results']), len(deferred_emails))
        self.assertListEqual(response.data['results'], [OrderedDict([('message', de.message), ('status', de.status)])
                                                        for de in deferred_emails])

    def test_list_messages_filter_by_status(self):
        emails_user = create_user('test_user')

        waiting_deferred_emails = []
        waiting_deferred_emails.append(create_deferred_email(emails_user, message='test message1'))

        create_deferred_email(emails_user, message='test message2', status=DeferredEmail.FAILED)
        create_deferred_email(emails_user, message='test message3', status=DeferredEmail.SENT)

        is_logged_in = self.client.login(username=self.super_user.username, password=self.super_user_password)
        self.assertTrue(is_logged_in)

        response = self.client.get(reverse('deferred_email:deferredemail-list'), {'status': 'waiting'})
        self.assertEqual(len(response.data['results']), len(waiting_deferred_emails))
        self.assertListEqual(response.data['results'], [OrderedDict([('message', de.message), ('status', de.status)])
                                                        for de in waiting_deferred_emails])