99 lines
3.5 KiB
Python
99 lines
3.5 KiB
Python
import uuid
|
|
from django.contrib import auth
|
|
from django.test import TestCase
|
|
from django.utils import timezone
|
|
|
|
from apps.lyric.models import LoginToken, Token, User, Wallet
|
|
|
|
|
|
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 LoginTokenModelTest(TestCase):
|
|
def test_links_user_with_autogen_uid(self):
|
|
login_token1 = LoginToken.objects.create(email="a@b.cde")
|
|
login_token2 = LoginToken.objects.create(email="v@w.xyz")
|
|
self.assertNotEqual(login_token1.pk, login_token2.pk)
|
|
self.assertIsInstance(login_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"))
|
|
|
|
class UserPaletteTest(TestCase):
|
|
def test_palette_field_defaults_to_palette_default(self):
|
|
user = User.objects.create(email="a@b.cde")
|
|
self.assertEqual(user.palette, "palette-default")
|
|
|
|
class WalletCreationTest(TestCase):
|
|
def setUp(self):
|
|
self.user = User.objects.create(email="capman@test.io")
|
|
|
|
def test_wallet_is_created_for_new_user(self):
|
|
self.assertTrue(Wallet.objects.filter(user=self.user).exists())
|
|
|
|
def test_new_wallet_has_144_writs(self):
|
|
wallet = Wallet.objects.get(user = self.user)
|
|
self.assertEqual(wallet.writs, 144)
|
|
|
|
def test_new_wallet_has_0_esteem(self):
|
|
wallet = Wallet.objects.get(user=self.user)
|
|
self.assertEqual(wallet.esteem, 0)
|
|
|
|
class TokenCreationTest(TestCase):
|
|
def setUp(self):
|
|
self.user = User.objects.create(email="capman@test.io")
|
|
|
|
def test_coin_on_a_string_created_for_new_user(self):
|
|
self.assertTrue(
|
|
Token.objects.filter(user=self.user, token_type=Token.COIN).exists()
|
|
)
|
|
|
|
def test_free_token_created_for_new_user(self):
|
|
self.assertTrue(
|
|
Token.objects.filter(user=self.user, token_type=Token.FREE).exists()
|
|
)
|
|
|
|
def test_coin_on_a_string_has_no_expiry(self):
|
|
coin = Token.objects.get(user=self.user, token_type=Token.COIN)
|
|
self.assertIsNone(coin.expires_at)
|
|
|
|
def test_free_token_has_expiry_within_7_days(self):
|
|
free = Token.objects.get(user=self.user, token_type=Token.FREE)
|
|
self.assertIsNotNone(free.expires_at)
|
|
delta = free.expires_at - timezone.now()
|
|
self.assertLessEqual(delta.days, 7)
|
|
self.assertGreater(delta.total_seconds(), 0)
|