mirror of https://github.com/fspc/workstand.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.6 KiB
50 lines
1.6 KiB
from django.test import TestCase
|
|
|
|
from ..models import CustomUser, Member
|
|
|
|
|
|
class TestCustomUserManager(TestCase):
|
|
def test_create_user(self):
|
|
new_user = CustomUser.objects.create_user('test@example.com')
|
|
self.assertTrue(new_user.pk)
|
|
|
|
def test_create_user_no_email(self):
|
|
with self.assertRaises(ValueError):
|
|
CustomUser.objects.create_user(email=None)
|
|
|
|
def test_create_superuser(self):
|
|
new_user = CustomUser.objects\
|
|
.create_superuser(email='super@example.com', password='password')
|
|
self.assertTrue(new_user.is_admin)
|
|
self.assertTrue(new_user.is_staff)
|
|
self.assertTrue(new_user.check_password('password'))
|
|
self.assertTrue(new_user.pk)
|
|
|
|
|
|
class TestCustomUser(TestCase):
|
|
def setUp(self):
|
|
self.new_user = CustomUser.objects.create_user('test@example.com')
|
|
|
|
def test_get_short_name(self):
|
|
self.assertEqual(self.new_user.get_short_name(), 'test@example.com')
|
|
|
|
def test_get_full_name(self):
|
|
self.assertEqual(self.new_user.get_full_name(), 'test@example.com')
|
|
|
|
|
|
class TestMember(TestCase):
|
|
def setUp(self):
|
|
self.new_member = Member.objects.create(
|
|
first_name='First',
|
|
last_name='Last',
|
|
post_code='H0H0H0'
|
|
)
|
|
|
|
def test_get_full_name(self):
|
|
self.assertEqual(self.new_member.get_full_name(), 'First Last')
|
|
self.assertEqual(self.new_member.get_short_name(), 'Last')
|
|
|
|
# add email to instance
|
|
self.new_member.email = 'member@example.com'
|
|
self.assertEqual(self.new_member.get_short_name(),
|
|
'member@example.com')
|
|
|