Ajout de tests

Author: raphodn

data/models/canteen.py

@@ -95,6 +95,9 @@ def publicly_hidden(self):
             else self.exclude(publication_status=Canteen.PublicationStatus.PUBLISHED)
         )
 
+    def is_satellite(self):
+        return self.filter(is_satellite_query())
+
     def get_satellites(self, central_producer_siret):
         return self.filter(is_satellite_query(), central_producer_siret=central_producer_siret)
 
@@ -124,6 +127,12 @@ def publicly_visible(self):
     def publicly_hidden(self):
         return self.get_queryset().publicly_hidden()
 
+    def is_satellite(self):
+        return self.get_queryset().is_satellite()
+
+    def get_satellites(self, central_producer_siret):
+        return self.get_queryset().get_satellites(central_producer_siret)
+
     def has_missing_data(self):
         return self.get_queryset().has_missing_data()
 

data/tests/test_canteen.py

@@ -89,6 +89,35 @@ def test_publicly_hidden_publish_by_default(self):
         self.assertEqual(qs.first(), self.canteen_published_armee)
 
 
+class TestCanteenSatelliteQuerySet(TestCase):
+    @classmethod
+    def setUpTestData(cls):
+        cls.canteen_central_1 = CanteenFactory(
+            siret="75665621899905", production_type=Canteen.ProductionType.CENTRAL, satellite_canteens_count=2
+        )  # 1 missing
+        cls.canteen_central_2 = CanteenFactory(
+            siret="75665621899906", production_type=Canteen.ProductionType.CENTRAL, satellite_canteens_count=2
+        )
+        cls.canteen_on_site_central_1 = CanteenFactory(
+            production_type=Canteen.ProductionType.ON_SITE_CENTRAL, central_producer_siret=cls.canteen_central_1.siret
+        )
+        cls.canteen_on_site_central_2 = CanteenFactory(
+            production_type=Canteen.ProductionType.ON_SITE_CENTRAL, central_producer_siret=cls.canteen_central_2.siret
+        )
+        cls.canteen_on_site_central_2 = CanteenFactory(
+            production_type=Canteen.ProductionType.ON_SITE_CENTRAL, central_producer_siret=cls.canteen_central_2.siret
+        )
+
+    def test_is_satellite(self):
+        self.assertEqual(Canteen.objects.count(), 5)
+        self.assertEqual(Canteen.objects.is_satellite().count(), 3)
+
+    def test_get_satellites(self):
+        self.assertEqual(Canteen.objects.count(), 5)
+        self.assertEqual(Canteen.objects.get_satellites(self.canteen_central_1.siret).count(), 1)
+        self.assertEqual(Canteen.objects.get_satellites(self.canteen_central_2.siret).count(), 2)
+
+
 class TestCanteenSiretOrSirenUniteLegaleQuerySet(TestCase):
     @classmethod
     def setUpTestData(cls):