Skip to content

Commit a704522

Browse files
committed
Update test to include strategy_type in the factory see HEA-947
1 parent 8b2f6f0 commit a704522

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

apps/baseline/tests/test_viewsets.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -739,7 +739,11 @@ def setUp(self):
739739
common_name_en="common mukera",
740740
kcals_per_unit=550,
741741
)
742-
LivelihoodStrategyFactory(product=self.product1, livelihood_zone_baseline=self.baseline1)
742+
LivelihoodStrategyFactory(
743+
product=self.product1,
744+
livelihood_zone_baseline=self.baseline1,
745+
strategy_type=LivelihoodStrategyType.FOOD_PURCHASE,
746+
)
743747
self.characteristic1 = WealthCharacteristicFactory(description_en="my test")
744748
self.characteristic2 = WealthCharacteristicFactory(description_en="my mukera", description_fr="my test")
745749
WealthGroupCharacteristicValueFactory(
@@ -749,7 +753,11 @@ def setUp(self):
749753
wealth_group__livelihood_zone_baseline=self.baseline2, wealth_characteristic=self.characteristic2
750754
)
751755
self.characteristic3 = WealthCharacteristicFactory()
752-
self.strategy = LivelihoodStrategyFactory(product=self.product1, livelihood_zone_baseline=self.baseline3)
756+
self.strategy = LivelihoodStrategyFactory(
757+
product=self.product1,
758+
livelihood_zone_baseline=self.baseline3,
759+
strategy_type=LivelihoodStrategyType.FOOD_PURCHASE,
760+
)
753761
self.baseline = LivelihoodZoneBaselineFactory(main_livelihood_category=self.category1)
754762
self.url = reverse("livelihood-zone-baseline-faceted-search")
755763

0 commit comments

Comments
 (0)