Update tests

Update view

apps/collection/tests.py

@@ -89,17 +89,20 @@ class CollectionWebHomeTests(CollectionDetailTests):
         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,
-            slug="test"
-        )
+        for i in range(1, 5):
+            setattr(self, f"establishment{i}",
+                    Establishment.objects.create(
+                        name=f"Test establishment {i}",
+                        establishment_type_id=self.establishment_type.id,
+                        is_publish=True,
+                        slug=f"test-establishment-{i}"
+                    )
+                )
 
-        self.establishment.collections.add(self.collection)
-        self.establishment.save()
+            getattr(self, f"establishment{i}").collections.add(self.collection)
 
     def test_collection_list_filter(self):
-        print("========================================================================================")
-        response = self.client.get('/api/web/collections/', format='json')
-        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        response = self.client.get('/api/web/collections/?country_code=en', format='json')
+        data = response.json()
+        self.assertIn('count', data)
+        self.assertGreater(data['count'], 0)

apps/collection/views/common.py

@@ -46,8 +46,8 @@ class CollectionHomePageView(CollectionViewMixin, generics.ListAPIView):
         """Override get_queryset method"""
         queryset = models.Collection.objects.published()\
             .by_country_code(code=self.request.country_code)\
-            .annotate_related_objects_count() \
-            .filter_related_gt(3) \
+            .annotate_related_objects_count()\
+            .filter_related_gt(3)\
             .order_by('-on_top', '-created')
 
         return queryset