Misago/misago/users/tests/test_ban_model.py

from django.test import TestCase

from misago.users.models import Ban


class BansManagerTests(TestCase):
    def setUp(self):
        Ban.objects.bulk_create(
            [
                Ban(check_type=Ban.USERNAME, banned_value="bob"),
                Ban(check_type=Ban.EMAIL, banned_value="bob@test.com"),
                Ban(check_type=Ban.IP, banned_value="127.0.0.1"),
            ]
        )

    def test_get_ban_for_banned_name(self):
        """get_ban finds ban for given username"""
        self.assertIsNotNone(Ban.objects.get_ban(username="Bob"))
        with self.assertRaises(Ban.DoesNotExist):
            Ban.objects.get_ban(username="Jeb")

    def test_get_ban_for_banned_email(self):
        """get_ban finds ban for given email"""
        self.assertIsNotNone(Ban.objects.get_ban(email="bob@test.com"))
        with self.assertRaises(Ban.DoesNotExist):
            Ban.objects.get_ban(email="jeb@test.com")

    def test_get_ban_for_banned_ip(self):
        """get_ban finds ban for given ip"""
        self.assertIsNotNone(Ban.objects.get_ban(ip="127.0.0.1"))
        with self.assertRaises(Ban.DoesNotExist):
            Ban.objects.get_ban(ip="42.0.0.1")

    def test_get_ban_for_all_bans(self):
        """get_ban finds ban for given values"""
        valid_kwargs = {"username": "bob", "ip": "42.51.52.51"}
        self.assertIsNotNone(Ban.objects.get_ban(**valid_kwargs))

        invalid_kwargs = {"username": "bsob", "ip": "42.51.52.51"}
        with self.assertRaises(Ban.DoesNotExist):
            Ban.objects.get_ban(**invalid_kwargs)


class BanTests(TestCase):
    def test_check_value_literal(self):
        """ban correctly tests given values"""
        test_ban = Ban(banned_value="bob")

        self.assertTrue(test_ban.check_value("bob"))
        self.assertFalse(test_ban.check_value("bobby"))

    def test_check_value_starts_with(self):
        """ban correctly tests given values"""
        test_ban = Ban(banned_value="bob*")

        self.assertTrue(test_ban.check_value("bob"))
        self.assertTrue(test_ban.check_value("bobby"))

    def test_check_value_middle_match(self):
        """ban correctly tests given values"""
        test_ban = Ban(banned_value="b*b")

        self.assertTrue(test_ban.check_value("bob"))
        self.assertFalse(test_ban.check_value("bobby"))

    def test_check_value_ends_witch(self):
        """ban correctly tests given values"""
        test_ban = Ban(banned_value="*bob")

        self.assertTrue(test_ban.check_value("lebob"))
        self.assertFalse(test_ban.check_value("bobby"))