rest-app/smartpicasso/app/user/tests.py

@@ -1,11 +1,11 @@
 from django.test import TestCase
 from django.urls import reverse
+from rest_framework import serializers
 from rest_framework import status
 from rest_framework.test import APITestCase, APIClient
 
-from smartpicasso.app.user.models import User
+from smartpicasso.app.user.models import User, UserManager
-from smartpicasso.app.user.serializers import UserLoginSerializer
+from smartpicasso.app.user.serializers import UserLoginSerializer, UserRegistrationSerializer
-from rest_framework import serializers
 
 
 class UserApiTest(APITestCase):
@@ -46,3 +46,42 @@ class UserLoginSerializerTest(TestCase):
         result = self.serializer.validate(data)
         self.assertEqual(result['email'], 'test@test.com')
         self.assertIn('token', result)
+
+
+class UserRegistrationSerializerTest(TestCase):
+    serializer = UserRegistrationSerializer()
+
+    def test_create(self):
+        profile = {"first_name": "test", "last_name": "test"}
+        user = self.serializer.create({"email": "test@test.com", "password": "test", "profile": profile})
+
+        self.assertNotEqual(user, None)
+        self.assertEqual(user.email, "test@test.com")
+
+
+class UserManagerTest(TestCase):
+    manager = User.objects
+
+    def test_create_user_none_email(self):
+        email = None
+        self.assertRaises(ValueError, self.manager.create_user, email)
+
+    def test_create_user(self):
+        user = self.manager.create_user("test@test.pl", "test")
+        self.assertNotEqual(user, None)
+        self.assertEqual(user.email, "test@test.pl")
+        self.assertEqual(user.is_active, True)
+        self.assertEqual(user.is_superuser, False)
+        self.assertEqual(user.is_staff, False)
+
+    def test_create_superuser_none_password(self):
+        password = None
+        self.assertRaises(TypeError, self.manager.create_superuser, "super@test.pl", password)
+
+    def test_create_superuser(self):
+        user = self.manager.create_superuser("super@test.pl", "test")
+        self.assertNotEqual(user, None)
+        self.assertEqual(user.email, "super@test.pl")
+        self.assertEqual(user.is_active, True)
+        self.assertEqual(user.is_superuser, True)
+        self.assertEqual(user.is_staff, True)