From 3095b5940f2a8afa41f75705e45fe2092b89f138 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=92=D0=B0=D1=88=C3=90=D0=92=D0=B8=D0=BA=D1=82=D0=BE?=
 =?UTF-8?q?=D1=80=20=D0=93=D0=BB=D0=B0=D0=B4=D0=BA=D0=B8=D1=85=C3=90?=
 <v.gladkikh@spider.ru>
Date: Thu, 19 Sep 2019 15:40:07 +0300
Subject: [PATCH] modify base test for route 'employees'

---
 apps/establishment/tests.py | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/apps/establishment/tests.py b/apps/establishment/tests.py
index b6a2b9c4..589f9f45 100644
--- a/apps/establishment/tests.py
+++ b/apps/establishment/tests.py
@@ -13,20 +13,22 @@ class BaseTestCase(APITestCase):
         self.newsletter = True
         self.user = User.objects.create_user(username=self.username, email=self.email, password=self.password)
 
-
-class EmployeeTests(BaseTestCase):
-    def test_employee_create(self):
-        self.data ={
-            'username_or_email': 'sedragurda',
-            'password': 'sedragurdaredips19',
+    def getAuthInfo(self):
+        data ={
+            'username_or_email': self.username,
+            'password': self.password,
             'remember': True
         }
+        response = self.client.post('/api/auth/login/', data=data)
 
-        response = self.client.post('/api/auth/login/',data=self.data)
-
-        auth = {
+        return {
             'access_token': response.data['access_token'],
             'refresh_token': response.data['refresh_token']
         }
-        response = self.client.get('/api/back/establishments/employees/',  data=auth, format='json')
+
+
+class EmployeeTests(BaseTestCase):
+    def test_employee_create(self):
+        data = self.getAuthInfo()
+        response = self.client.get('/api/back/establishments/employees/',  data=data, format='json')
         self.assertEqual(response.status_code, status.HTTP_200_OK)
\ No newline at end of file