From 49e5e4ea9141a035627c4aaf829e270dfd77d8a3 Mon Sep 17 00:00:00 2001
From: alex <birukov.system@gmail.com>
Date: Mon, 25 Nov 2019 09:38:50 +0300
Subject: [PATCH] test establishment and news carousel

---
 apps/establishment/tests.py | 25 +++++++++++++++++++++++++
 apps/news/tests.py          | 29 +++++++++++++++++++++++++++--
 2 files changed, 52 insertions(+), 2 deletions(-)

diff --git a/apps/establishment/tests.py b/apps/establishment/tests.py
index 3534608c..6d456a45 100644
--- a/apps/establishment/tests.py
+++ b/apps/establishment/tests.py
@@ -441,3 +441,28 @@ class EstablishmentWebFavoriteTests(ChildTestCase):
             f'/api/web/establishments/slug/{self.establishment.slug}/favorites/',
             format='json')
         self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+
+class EstablishmentCarouselTests(ChildTestCase):
+
+    def test_web_carousel_CR(self):
+        data = {
+            "object_id": self.establishment.id
+        }
+
+        response = self.client.post(f'/api/web/establishments/slug/{self.establishment.slug}/carousels/', data=data)
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.delete(f'/api/web/establishments/slug/{self.establishment.slug}/carousels/')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+    def test_mobile_carousel_CR(self):
+        data = {
+            "object_id": self.establishment.id
+        }
+
+        response = self.client.post(f'/api/mobile/establishments/slug/{self.establishment.slug}/carousels/', data=data)
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.delete(f'/api/mobile/establishments/slug/{self.establishment.slug}/carousels/')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
diff --git a/apps/news/tests.py b/apps/news/tests.py
index 532a6efc..a87e457a 100644
--- a/apps/news/tests.py
+++ b/apps/news/tests.py
@@ -30,12 +30,12 @@ class BaseTestCase(APITestCase):
              'refresh_token': tokens.get('refresh_token')})
         self.test_news_type = NewsType.objects.create(name="Test news type")
 
-        self.lang = Language.objects.get(
+        self.lang = Language.objects.create(
             title='Russia',
             locale='ru-RU'
         )
 
-        self.country_ru = Country.objects.get(
+        self.country_ru = Country.objects.create(
             name={"en-GB": "Russian"}
         )
 
@@ -128,3 +128,28 @@ class NewsTestCase(BaseTestCase):
 
         response = self.client.delete(f'/api/web/news/slug/{self.test_news.slug}/favorites/', format='json')
         self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+
+class NewsCarouselTests(BaseTestCase):
+
+    def test_web_carousel_CR(self):
+        data = {
+            "object_id": self.test_news.id
+        }
+
+        response = self.client.post(f'/api/web/news/slug/{self.test_news.slug}/carousels/', data=data)
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.delete(f'/api/web/news/slug/{self.test_news.slug}/carousels/')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+    def test_mobile_carousel_CR(self):
+        data = {
+            "object_id": self.test_news.id
+        }
+
+        response = self.client.post(f'/api/mobile/news/slug/{self.test_news.slug}/carousels/', data=data)
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.delete(f'/api/mobile/news/slug/{self.test_news.slug}/carousels/')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)