Misago/misago/users/tests/test_banadmin_views.py

from datetime import datetime, timedelta

from django.urls import reverse

from misago.admin.testutils import AdminTestCase
from misago.users.models import Ban


class BanAdminViewsTests(AdminTestCase):
    def test_link_registered(self):
        """admin nav contains bans link"""
        response = self.client.get(reverse('misago:admin:users:accounts:index'))

        response = self.client.get(response['location'])
        self.assertContains(response, reverse('misago:admin:users:bans:index'))

    def test_list_view(self):
        """bans list view returns 200"""
        response = self.client.get(reverse('misago:admin:users:bans:index'))
        self.assertEqual(response.status_code, 302)

        response = self.client.get(response['location'])
        self.assertEqual(response.status_code, 200)

    def test_mass_delete(self):
        """adminview deletes multiple bans"""
        test_date = datetime.now() + timedelta(days=180)

        for i in range(10):
            response = self.client.post(
                reverse('misago:admin:users:bans:new'),
                data={
                    'check_type': '1',
                    'banned_value': '%stest@test.com' % i,
                    'user_message': 'Lorem ipsum dolor met',
                    'staff_message': 'Sit amet elit',
                    'expires_on': test_date.isoformat(),
                },
            )
            self.assertEqual(response.status_code, 302)

        self.assertEqual(Ban.objects.count(), 10)

        bans_pks = []
        for ban in Ban.objects.iterator():
            bans_pks.append(ban.pk)

        response = self.client.post(
            reverse('misago:admin:users:bans:index'),
            data={
                'action': 'delete',
                'selected_items': bans_pks,
            },
        )
        self.assertEqual(response.status_code, 302)
        self.assertEqual(Ban.objects.count(), 0)

    def test_new_view(self):
        """new ban view has no showstoppers"""
        response = self.client.get(reverse('misago:admin:users:bans:new'))
        self.assertEqual(response.status_code, 200)

        test_date = datetime.now() + timedelta(days=180)

        response = self.client.post(
            reverse('misago:admin:users:bans:new'),
            data={
                'check_type': '1',
                'banned_value': 'test@test.com',
                'user_message': 'Lorem ipsum dolor met',
                'staff_message': 'Sit amet elit',
                'expires_on': test_date.isoformat(),
            },
        )
        self.assertEqual(response.status_code, 302)

        response = self.client.get(reverse('misago:admin:users:bans:index'))
        response = self.client.get(response['location'])
        self.assertEqual(response.status_code, 200)
        self.assertContains(response, 'test@test.com')

    def test_edit_view(self):
        """edit ban view has no showstoppers"""
        self.client.post(
            reverse('misago:admin:users:bans:new'),
            data={
                'check_type': '0',
                'banned_value': 'Admin',
            },
        )

        test_ban = Ban.objects.get(banned_value='admin')
        form_link = reverse(
            'misago:admin:users:bans:edit',
            kwargs={
                'pk': test_ban.pk,
            },
        )

        response = self.client.post(
            form_link,
            data={
                'check_type': '1',
                'banned_value': 'test@test.com',
                'user_message': 'Lorem ipsum dolor met',
                'staff_message': 'Sit amet elit',
                'expires_on': '',
            },
        )
        self.assertEqual(response.status_code, 302)

        response = self.client.get(reverse('misago:admin:users:bans:index'))
        response = self.client.get(response['location'])
        self.assertEqual(response.status_code, 200)
        self.assertContains(response, 'test@test.com')

    def test_delete_view(self):
        """delete ban view has no showstoppers"""
        self.client.post(
            reverse('misago:admin:users:bans:new'),
            data={
                'check_type': '0',
                'banned_value': 'TestBan',
            },
        )

        test_ban = Ban.objects.get(banned_value='testban')

        response = self.client.post(
            reverse(
                'misago:admin:users:bans:delete',
                kwargs={
                    'pk': test_ban.pk,
                },
            )
        )
        self.assertEqual(response.status_code, 302)

        response = self.client.get(reverse('misago:admin:users:bans:index'))
        self.client.get(response['location'])
        response = self.client.get(response['location'])

        self.assertEqual(response.status_code, 200)
        self.assertNotContains(response, test_ban.banned_value)