Misago/misago/users/tests/test_user_getters.py

from django.contrib.auth import get_user_model
from django.test import TestCase

User = get_user_model()


class UserGettersTests(TestCase):
    def test_get_user_by_username(self):
        user = User.objects.create_user("User", "test@example.com")
        assert User.objects.get_by_username("User") == user

    def test_getting_user_by_username_is_case_insensitive(self):
        user = User.objects.create_user("User", "test@example.com")
        assert User.objects.get_by_username("uSeR") == user

    def test_getting_user_by_username_raises_does_not_exist_for_no_result(self):
        with self.assertRaises(User.DoesNotExist):
            User.objects.get_by_username("user")

    def test_getting_user_by_username_supports_diacritics(self):
        with self.assertRaises(User.DoesNotExist):
            User.objects.get_by_username("łóć")

    def test_getting_user_by_username_is_not_doing_fuzzy_matching(self):
        user = User.objects.create_user("User", "test@example.com")
        with self.assertRaises(User.DoesNotExist):
            User.objects.get_by_username("usere")

    def test_get_user_by_email(self):
        user = User.objects.create_user("User", "test@example.com")
        assert User.objects.get_by_email("test@example.com") == user

    def test_getting_user_by_email_is_case_insensitive(self):
        user = User.objects.create_user("User", "test@example.com")
        assert User.objects.get_by_email("tEsT@eXaMplE.com") == user

    def test_getting_user_by_email_supports_diacritics(self):
        user = User.objects.create_user("User", "łóć@łexĄmple.com")
        assert User.objects.get_by_email("łÓć@ŁexĄMple.com") == user

    def test_getting_user_by_email_raises_does_not_exist_for_no_result(self):
        with self.assertRaises(User.DoesNotExist):
            User.objects.get_by_email("test@example.com")

    def test_get_user_by_username_using_combined_getter(self):
        user = User.objects.create_user("User", "test@example.com")
        assert User.objects.get_by_username_or_email("user") == user

    def test_get_user_by_email_using_combined_getter(self):
        user = User.objects.create_user("User", "test@example.com")
        assert User.objects.get_by_username_or_email("test@example.com") == user

    def test_combined_getter_handles_username_slug_and_email_collision(self):
        email_match = User.objects.create_user("Bob", "test@test.test")
        slug_match = User.objects.create_user("TestTestTest", "bob@test.com")

        assert User.objects.get_by_username_or_email("test@test.test") == email_match
        assert User.objects.get_by_username_or_email("TestTestTest") == slug_match

    def test_combined_getter_raises_does_not_exist_for_no_result(self):
        with self.assertRaises(User.DoesNotExist):
            User.objects.get_by_username_or_email("User")