39 - Testing User API

Author: codingforentrepreneurs

src/accounts/admin.py

@@ -1,3 +1,3 @@
 from django.contrib import admin
 
-# Register your models here.
+# Register your models here.

src/accounts/api/test_user_api.py

@@ -0,0 +1,116 @@
+from django.urls import reverse
+from rest_framework import status
+from rest_framework.reverse import reverse as api_reverse
+from rest_framework.test import APITestCase
+
+
+from django.contrib.auth import get_user_model
+
+User = get_user_model()
+
+
+class UserAPITestCase(APITestCase): 
+    def setUp(self): 
+        user =  User.objects.create(username='cfe', email='hello@cfe.com')
+        user.set_password("yeahhhcfe")
+        user.save()
+
+    def test_created_user(self):
+        qs = User.objects.filter(username='cfe')
+        self.assertEqual(qs.count(), 1)
+
+    def test_register_user_api_fail(self):
+        url = api_reverse('api-auth:register')
+        data = {
+            'username': 'cfe.doe',
+            'email': 'cfe.doe@gmail.com',
+            'password': 'learncode',
+        }
+        response = self.client.post(url, data, format='json')
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST) # 400
+        self.assertEqual(response.data['password2'][0], 'This field is required.')
+
+    def test_register_user_api(self):
+        url = api_reverse('api-auth:register')
+        data = {
+            'username': 'cfe.doe',
+            'email': 'cfe.doe@gmail.com',
+            'password': 'learncode',
+            'password2': 'learncode'
+        }
+        response = self.client.post(url, data, format='json')
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED) # 400
+        token_len = len(response.data.get("token", 0))
+        self.assertGreater(token_len, 0)
+
+
+    def test_login_user_api(self):
+        url = api_reverse('api-auth:login')
+        data = {
+            'username': 'cfe',
+            'password': 'yeahhhcfe',
+        }
+        response = self.client.post(url, data, format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK) # 400
+        token = response.data.get("token", 0)
+        token_len = 0
+        if token != 0:
+            token_len = len(token)
+        self.assertGreater(token_len, 0)
+
+
+
+    def test_login_user_api_fail(self):
+        url = api_reverse('api-auth:login')
+        data = {
+            'username': 'cfe.abc', # does not exist
+            'password': 'yeahhhcfe',
+        }
+        response = self.client.post(url, data, format='json')
+        self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED) # 400
+        token = response.data.get("token", 0)
+        token_len = 0
+        if token != 0:
+            token_len = len(token)
+        self.assertEqual(token_len, 0)
+    
+    def test_token_login_api(self):
+        url = api_reverse('api-auth:login')
+        data = {
+            'username': 'cfe',
+            'password': 'yeahhhcfe',
+        }
+        response = self.client.post(url, data, format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK) # 400
+        token = response.data.get("token", None)
+        self.client.credentials(HTTP_AUTHORIZATION='JWT ' + token)
+        response2 = self.client.post(url, data, format='json')
+        self.assertEqual(response2.status_code, status.HTTP_403_FORBIDDEN)
+
+
+    def test_token_register_api(self):
+        url = api_reverse('api-auth:login')
+        data = {
+            'username': 'cfe',
+            'password': 'yeahhhcfe',
+        }
+        response = self.client.post(url, data, format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK) # 400
+        token = response.data.get("token", None)
+        self.client.credentials(HTTP_AUTHORIZATION='JWT ' + token)
+
+        url2 = api_reverse('api-auth:register')
+        data2 = {
+            'username': 'cfe.doe',
+            'email': 'cfe.doe@gmail.com',
+            'password': 'learncode',
+            'password2': 'learncode'
+        }
+        response = self.client.post(url2, data2, format='json')
+        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN) # 403
+
+
+
+
+
+

src/accounts/api/urls.py

@@ -6,8 +6,8 @@
 
 from .views import AuthAPIView, RegisterAPIView
 urlpatterns = [
-    url(r'^$', AuthAPIView.as_view()),
-    url(r'^register/$', RegisterAPIView.as_view()),
+    url(r'^$', AuthAPIView.as_view(), name='login'),
+    url(r'^register/$', RegisterAPIView.as_view(), name='register'),
     url(r'^jwt/$', obtain_jwt_token),
     url(r'^jwt/refresh/$', refresh_jwt_token),
 ]

src/cfeapi/urls.py

@@ -26,7 +26,7 @@
 
 urlpatterns = [
     url(r'^admin/', admin.site.urls),
-    url(r'^api/auth/', include('accounts.api.urls')),
+    url(r'^api/auth/', include('accounts.api.urls', namespace='api-auth')),
     url(r'^api/user/', include('accounts.api.user.urls', namespace='api-user')),
     url(r'^api/status/', include('status.api.urls', namespace='api-status')),
     url(r'^api/updates/', include('updates.api.urls')), 

src/status/tests.py

@@ -1,3 +1,20 @@
 from django.test import TestCase
 
-# Create your tests here.
+from django.contrib.auth import get_user_model
+
+from .models import Status
+User = get_user_model()
+
+
+class StatusTestCase(TestCase): 
+    def setUp(self): 
+        user =  User.objects.create(username='cfe', email='hello@cfe.com')
+        user.set_password("yeahhhcfe")
+        user.save()
+
+    def test_creating_status(self):
+        user = User.objects.get(username='cfe')
+        obj = Status.objects.create(user=user, content='Some cool new content')
+        self.assertEqual(obj.id, 1)
+        qs = Status.objects.all()
+        self.assertEqual(qs.count(), 1)