python-tdd/src/apps/lyric/tests/integrated/test_models.py

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

from apps.lyric.models import Token, User


class UserModelTest(TestCase):
    def test_model_is_configured_for_django_auth(self):
        self.assertEqual(auth.get_user_model(), User)
    
    def test_user_is_valid_with_email_only(self):
        user = User(email="a@b.cde")
        user.set_unusable_password()
        user.full_clean() # should not raise

    def test_id_is_primary_key(self):
        user = User(id="123")
        self.assertEqual(user.pk, "123")

    def test_user_can_have_a_username(self):
        user = User.objects.create(email="a@b.cde")
        user.username = "stardust"
        user.save()
        self.assertEqual(User.objects.get(pk=user.pk).username, "stardust")

    def test_searchable_defaults_to_false(self):
        user = User.objects.create(email="a@b.cde")
        self.assertFalse(user.searchable)

class TokenModelTest(TestCase):
    def test_links_user_with_autogen_uid(self):
        token1 = Token.objects.create(email="a@b.cde")
        token2 = Token.objects.create(email="v@w.xyz")
        self.assertNotEqual(token1.pk, token2.pk)
        self.assertIsInstance(token1.pk, uuid.UUID)

class UserManagerTest(TestCase):
    def test_create_superuser_sets_is_staff_and_is_superuser(self):
        user = User.objects.create_superuser(
            email="admin@example.com",
            password="correct-password",
        )
        self.assertTrue(user.is_staff)
        self.assertTrue(user.is_superuser)

    def test_create_superuser_sets_usable_password(self):
        user = User.objects.create_superuser(
            email="admin@example.com",
            password="correct-password",
        )
        self.assertTrue(user.check_password("correct-password"))