Merge pull request #261 from American-Institutes-for-Research/HEA-947/add_french_translations

Hea 947/add french translations

Author: rhunwicks

apps/baseline/tests/test_viewsets.py

@@ -739,7 +739,11 @@ def setUp(self):
             common_name_en="common mukera",
             kcals_per_unit=550,
         )
-        LivelihoodStrategyFactory(product=self.product1, livelihood_zone_baseline=self.baseline1)
+        LivelihoodStrategyFactory(
+            product=self.product1,
+            livelihood_zone_baseline=self.baseline1,
+            strategy_type=LivelihoodStrategyType.FOOD_PURCHASE,
+        )
         self.characteristic1 = WealthCharacteristicFactory(description_en="my test")
         self.characteristic2 = WealthCharacteristicFactory(description_en="my mukera", description_fr="my test")
         WealthGroupCharacteristicValueFactory(
@@ -749,7 +753,11 @@ def setUp(self):
             wealth_group__livelihood_zone_baseline=self.baseline2, wealth_characteristic=self.characteristic2
         )
         self.characteristic3 = WealthCharacteristicFactory()
-        self.strategy = LivelihoodStrategyFactory(product=self.product1, livelihood_zone_baseline=self.baseline3)
+        self.strategy = LivelihoodStrategyFactory(
+            product=self.product1,
+            livelihood_zone_baseline=self.baseline3,
+            strategy_type=LivelihoodStrategyType.FOOD_PURCHASE,
+        )
         self.baseline = LivelihoodZoneBaselineFactory(main_livelihood_category=self.category1)
         self.url = reverse("livelihood-zone-baseline-faceted-search")
 
@@ -897,15 +905,6 @@ def test_search_with_livelihood_strategy_type(self):
         self.assertIn("livelihood_zone__code", baselines[0])
         self.assertIn("reference_year_end_date", baselines[0])
 
-        # Test that search "lait" with language=fr returns MilkProduction via French translation
-        response = self.client.get(self.url, {"search": "lait", "language": "fr"})
-        self.assertEqual(response.status_code, 200)
-        data = response.data
-        strategy_type_results = data["livelihood_strategy_types"]
-        milk_results = [r for r in strategy_type_results if r["value"] == "MilkProduction"]
-        self.assertEqual(len(milk_results), 1)
-        self.assertEqual(milk_results[0]["value_label"], "Production du lait")
-
         # Test that search "goat" returns multiple goat-related products
         response = self.client.get(self.url, {"search": "goat", "language": "en"})
         self.assertEqual(response.status_code, 200)
@@ -917,6 +916,29 @@ def test_search_with_livelihood_strategy_type(self):
         self.assertIn(goat_milk_product.cpc, product_cpcs)
         self.assertEqual(len(product_results), 3)
 
+        # Test that searching the French strategy type label "viande" matches MeatProduction
+        response = self.client.get(self.url, {"search": "viande", "language": "fr"})
+        self.assertEqual(response.status_code, 200)
+        data = response.data
+        strategy_type_results = data["livelihood_strategy_types"]
+        meat_results = [r for r in strategy_type_results if r["value"] == "MeatProduction"]
+        self.assertEqual(len(meat_results), 1)
+        self.assertEqual(meat_results[0]["value_label"], "Production de viande")
+        self.assertEqual(meat_results[0]["count"], 1)
+        baselines = meat_results[0]["livelihood_zone_baselines"]
+        self.assertEqual(len(baselines), 1)
+        self.assertEqual(baselines[0]["id"], self.baseline2.id)
+
+        # Test that searching "lait" with language=fr matches MilkProduction via French translation
+        response = self.client.get(self.url, {"search": "lait", "language": "fr"})
+        self.assertEqual(response.status_code, 200)
+        data = response.data
+        strategy_type_results = data["livelihood_strategy_types"]
+        milk_results = [r for r in strategy_type_results if r["value"] == "MilkProduction"]
+        self.assertEqual(len(milk_results), 1)
+        self.assertEqual(milk_results[0]["value_label"], "Production de lait")
+        self.assertEqual(milk_results[0]["count"], 1)
+
         # test taht strategy_type filter to baseline list endpoint
         baseline_url = reverse("livelihoodzonebaseline-list")
         response = self.client.get(baseline_url, {"strategy_type": "MilkProduction"})