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)