add get_tokkens and add cookies

apps/establishment/tests.py

@@ -1,6 +1,7 @@
 from rest_framework.test import APITestCase, APIClient
 from account.models import User
 from rest_framework import status
+from http.cookies import SimpleCookie
 # Create your tests here.
 
 
@@ -12,23 +13,32 @@ class BaseTestCase(APITestCase):
         self.email = 'sedragurda@desoz.com'
         self.newsletter = True
         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')
 
-    def getAuthInfo(self):
-        data ={
-            'username_or_email': self.username,
-            'password': self.password,
-            'remember': True
-        }
-        response = self.client.post('/api/auth/login/', data=data)
+        # print('access_token: ' + self.access_token)
 
-        return {
-            'access_token': response.data['access_token'],
-            'refresh_token': response.data['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):
-        data = self.getAuthInfo()
-        response = self.client.get('/api/back/establishments/employees/',  data=data, format='json')
+        self.client.cookies = SimpleCookie({'access_token':self.access_token,
+                                            'refresh_token': self.refresh_token})
+
+        response = self.client.get('/api/back/establishments/employees/',  format='json')
         self.assertEqual(response.status_code, status.HTTP_200_OK)