From fee798ca9cc8bdaa8af10e9323ffa6b4f3629f0e Mon Sep 17 00:00:00 2001
From: Drew Larson <drew@drwlrsn.com>
Date: Sun, 10 Jul 2016 21:52:30 -0600
Subject: [PATCH] Test for registration models.

---
 bikeshop_project/registration/tests.py | 50 +++++++++++++++++++++++++-
 1 file changed, 49 insertions(+), 1 deletion(-)

diff --git a/bikeshop_project/registration/tests.py b/bikeshop_project/registration/tests.py
index 7ce503c..ded8282 100644
--- a/bikeshop_project/registration/tests.py
+++ b/bikeshop_project/registration/tests.py
@@ -1,3 +1,51 @@
 from django.test import TestCase
 
-# Create your tests here.
+
+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) as context_manager:
+            new_user = 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')