diff --git a/apps/establishment/tests.py b/apps/establishment/tests.py index 589f9f45..b6d72564 100644 --- a/apps/establishment/tests.py +++ b/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) \ No newline at end of file