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

from django.contrib import auth
from django.test import TestCase
from ..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.full_clean() # should not raise

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

class TokenModelTest(TestCase):
    def test_links_user_with_autogen_uid(self):
        token1 = Token.objects.create(id="123")
        token2 = Token.objects.create(id="124")
        self.assertNotEqual(token1.uid, token2.uid)