44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
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.
|
|
|
|
|
|
class BaseTestCase(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)
|
|
#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)
|
|
|
|
|
|
# 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})
|
|
|
|
response = self.client.get('/api/back/establishments/employees/', format='json')
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK) |