fix more tests

unicef · Dec 19, 2024 · 6221b0a · 6221b0a
1 parent 2b87e59
commit 6221b0a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/unit/apps/registration_datahub/test_rdi_merge.py b/tests/unit/apps/registration_datahub/test_rdi_merge.py
@@ -447,18 +447,21 @@ def test_merge_rdi_create_collections(self, household_representation_exists: boo
         if False, another household representation exists, but it does not have collection,
         if None, household representation does not exist in another program
         """
+        program_2 = ProgramFactory(business_area=self.rdi.business_area)
         self.rdi.data_source = RegistrationDataImport.PROGRAM_POPULATION
         self.rdi.save()
         imported_household = HouseholdFactory(
             collect_individual_data=COLLECT_TYPE_FULL,
             registration_data_import=self.rdi,
             unicef_id="HH-9",
             rdi_merge_status=MergeStatusModel.PENDING,
+            program=self.rdi.program,
         )
         self.set_imported_individuals(imported_household)
         individual_without_collection = IndividualFactory(
             unicef_id="IND-9",
             business_area=self.rdi.business_area,
+            program=program_2,
             household=None,
         )
         individual_without_collection.individual_collection = None
@@ -468,6 +471,7 @@ def test_merge_rdi_create_collections(self, household_representation_exists: boo
         IndividualFactory(
             unicef_id="IND-8",
             business_area=self.rdi.business_area,
+            program=program_2,
             individual_collection=individual_collection,
             household=None,
         )
@@ -477,6 +481,7 @@ def test_merge_rdi_create_collections(self, household_representation_exists: boo
             household = HouseholdFactory(
                 head_of_household=individual_without_collection,
                 business_area=self.rdi.business_area,
+                program=program_2,
                 unicef_id="HH-9",
             )
             household.household_collection = None