add login test

apps/authorization/tests.py

@@ -1,3 +0,0 @@
-from django.test import TestCase
-
-# Create your tests here.

apps/authorization/tests/tests.py

@@ -1,3 +1,25 @@
-from django.test import TestCase
-
+from rest_framework.test import APITestCase
+from account.models import User
 # Create your tests here.
+
+
+class AuthorizationTests(APITestCase):
+
+    def setUp(self):
+        self.username = 'sedragurda'
+        self.password = 'sedragurdaredips19'
+        self.email = 'sedragurda@desoz.com'
+        self.newsletter = True
+        self.user = User.objects.create_user(username=self.username, email=self.email, password=self.password)
+        self.tokkens = User.create_jwt_tokens(self.user)
+
+    def LoginTests(self):
+        data ={
+            'username_or_email': self.username,
+            'password': self.password,
+            'remember': True
+        }
+        response = self.client.post('/api/auth/login/', data=data)
+        self.assertEqual(response.data['access_token'], self.tokkens.get('access_token'))
+        self.assertEqual(response.data['refresh_token'], self.tokkens.get('refresh_token'))
+

apps/establishment/tests.py

@@ -1,4 +1,4 @@
-from rest_framework.test import APITestCase, APIClient
+from rest_framework.test import APITestCase
 from account.models import User
 from rest_framework import status
 from http.cookies import SimpleCookie
@@ -15,30 +15,14 @@ class BaseTestCase(APITestCase):
         self.user = User.objects.create_user(username=self.username, email=self.email, password=self.password)
         #get tokkens
         tokkens = User.create_jwt_tokens(self.user)
-        self.access_token = tokkens.get('access_token')
-        self.refresh_token = tokkens.get('refresh_token')
-
-        # print('access_token: ' + self.access_token)
+        self.client.cookies = SimpleCookie({'access_token': tokkens.get('access_token'),
+                                            'refresh_token': tokkens.get('refresh_token')})
 
 
-    # def getAuthInfo(self):
-    #     data ={
-    #         'username_or_email': self.username,
-    #         'password': self.password,
-    #         'remember': True
-    #     }
-    #     response = self.client.post('/api/auth/login/', data=data)
-    #
-    #     return {
-    #         'access_token': response.data['access_token'],
-    #         'refresh_token': response.data['refresh_token']
-    #     }
 
 
 class EmployeeTests(BaseTestCase):
-    def test_employee_create(self):
-        self.client.cookies = SimpleCookie({'access_token':self.access_token,
-                                            'refresh_token': self.refresh_token})
-
+    def test_employee_list(self):
         response = self.client.get('/api/back/establishments/employees/',  format='json')
-        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+