From eebd66fa275c4ab45816ca97939fa9ade24048f6 Mon Sep 17 00:00:00 2001
From: littlewolf <nickptic@code-artel.com>
Date: Tue, 24 Sep 2019 13:35:05 +0300
Subject: [PATCH] Add tests to created_by and modified_by

---
 apps/utils/tests.py | 43 +++++++++++++++++++++++++++++++++++++------
 1 file changed, 37 insertions(+), 6 deletions(-)

diff --git a/apps/utils/tests.py b/apps/utils/tests.py
index 1c9fa71d..786ed940 100644
--- a/apps/utils/tests.py
+++ b/apps/utils/tests.py
@@ -8,6 +8,8 @@ from http.cookies import SimpleCookie
 from account.models import User
 from news.models import News, NewsType
 
+from establishment.models import Establishment, EstablishmentType, Employee
+
 
 class BaseTestCase(APITestCase):
 
@@ -28,6 +30,12 @@ class BaseTestCase(APITestCase):
              'locale': "en"
              })
 
+
+class TranslateFieldTests(BaseTestCase):
+
+    def setUp(self):
+        super().setUp()
+
         self.news_type = NewsType.objects.create(name="Test news type")
 
         self.news_item = News.objects.create(
@@ -45,15 +53,9 @@ class BaseTestCase(APITestCase):
             news_type=self.news_type
         )
 
-
-class TranslateFieldModel(BaseTestCase):
-
     def test_model_field(self):
         self.assertIsNotNone(getattr(self.news_item, "title_translated", None))
 
-
-class TranslateFieldReview(BaseTestCase):
-
     def test_read_locale(self):
         response = self.client.get(f"/api/web/news/{self.news_item.id}/", format='json')
         self.assertEqual(response.status_code, status.HTTP_200_OK)
@@ -62,3 +64,32 @@ class TranslateFieldReview(BaseTestCase):
         self.assertIn("title_translated", news_data)
 
         self.assertEqual(news_data['title_translated'], "Test news item")
+
+
+class BaseAttributeTests(BaseTestCase):
+
+    def setUp(self):
+        super().setUp()
+
+        self.establishment_type = EstablishmentType.objects.create(name="Test establishment type")
+        self.establishment = Establishment.objects.create(
+            name="Test establishment",
+            establishment_type_id=self.establishment_type.id,
+            is_publish=True
+        )
+
+    def test_base_attr_api(self):
+        data = {
+            'user': self.user.id,
+            'name': 'Test name'
+        }
+
+        response = self.client.post('/api/back/establishments/employees/', data=data)
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response_data = response.json()
+        self.assertIn("id", response_data)
+
+        employee = Employee.objects.get(id=response_data['id'])
+        self.assertEqual(self.user, employee.created_by)
+        self.assertEqual(self.user, employee.modified_by)