From d30d68bf17841cc873d8953b534398126dacb093 Mon Sep 17 00:00:00 2001
From: alex <birukov.system@gmail.com>
Date: Fri, 22 Nov 2019 10:32:01 +0300
Subject: [PATCH] partner test

---
 apps/partner/serializers/back.py | 13 +++++++++++-
 apps/partner/tests.py            | 36 ++++++++++++++------------------
 apps/partner/views/back.py       |  4 ++--
 project/urls/back.py             |  1 +
 4 files changed, 31 insertions(+), 23 deletions(-)

diff --git a/apps/partner/serializers/back.py b/apps/partner/serializers/back.py
index 94a7c4b9..d011e058 100644
--- a/apps/partner/serializers/back.py
+++ b/apps/partner/serializers/back.py
@@ -6,4 +6,15 @@ from partner.models import Partner
 class BackPartnerSerializer(serializers.ModelSerializer):
     class Meta:
         model = Partner
-        fields = '__all__'
+        fields = (
+            'id',
+            'old_id',
+            'name',
+            'url',
+            'image',
+            'establishment',
+            'type',
+            'starting_date',
+            'expiry_date',
+            'price_per_month',
+        )
diff --git a/apps/partner/tests.py b/apps/partner/tests.py
index 97931a8a..3233950b 100644
--- a/apps/partner/tests.py
+++ b/apps/partner/tests.py
@@ -65,30 +65,26 @@ class PartnerBackTestCase(BaseTestCase):
         response = self.client.get('/api/back/partner/')
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
-    # def test_partner_post(self):
-    #     test_inquiry = {
-    #         'review': self.test_review.pk,
-    #         'author': self.user.pk,
-    #         'comment': 'New test comment',
-    #     }
-    #     response = self.client.post('/api/back/partner/', data=test_inquiry)
-    #     self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+    def test_partner_post(self):
+        test_partner = {
+            'url': 'http://google.com',
+        }
+        response = self.client.post('/api/back/partner/', data=test_partner, format='json')
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
 
     def test_partner_detail(self):
         response = self.client.get(f'/api/back/partner/{self.partner.id}/')
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
-    # def test_partner_detail_put(self):
-    #     data = {
-    #         'id': self.test_inquiry.id,
-    #         'review': self.test_review.pk,
-    #         'author': self.user.pk,
-    #         'comment': 'New test comment 2',
-    #     }
-    #
-    #     response = self.client.put(f'/api/back/partner/{self.partner.id}/', data=data)
-    #     self.assertEqual(response.status_code, status.HTTP_200_OK)
+    def test_partner_detail_put(self):
+        data = {
+            'url': 'http://yandex.com',
+            'name': 'Yandex',
+        }
+
+        response = self.client.put(f'/api/back/partner/{self.partner.id}/', data=data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
 
     def test_partner_delete(self):
-        response = self.client.get(f'/api/back/partner/{self.partner.id}/')
-        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        response = self.client.delete(f'/api/back/partner/{self.partner.id}/')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
diff --git a/apps/partner/views/back.py b/apps/partner/views/back.py
index e0727a7c..1033d0ee 100644
--- a/apps/partner/views/back.py
+++ b/apps/partner/views/back.py
@@ -10,9 +10,9 @@ class PartnerLstView(generics.ListCreateAPIView):
     """Partner list create view."""
     queryset = Partner.objects.all()
     serializer_class = serializers.BackPartnerSerializer
+    pagination_class = None
     permission_classes = [permissions.IsAdminUser | IsEstablishmentManager]
-    filter_backends = (DjangoFilterBackend, filters.OrderingFilter)
-    ordering_fields = '__all__'
+    filter_backends = (DjangoFilterBackend,)
     filterset_fields = (
         'establishment',
         'type',
diff --git a/project/urls/back.py b/project/urls/back.py
index fdd3d10a..e7e2b43b 100644
--- a/project/urls/back.py
+++ b/project/urls/back.py
@@ -16,4 +16,5 @@ urlpatterns = [
     path('re_blocks/', include(('advertisement.urls.back', 'advertisement'),
                                namespace='advertisement')),
     path('main/', include('main.urls.back')),
+    path('partner/', include('partner.urls.back')),
 ]