Remove db requirement for champ UCR tests

dimagi · Jun 24, 2024 · ca7845f · ca7845f
1 parent 5081177
commit ca7845f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/custom/champ/tests/test_champ_cameroon.py b/custom/champ/tests/test_champ_cameroon.py
@@ -1,12 +1,14 @@
 from datetime import date
 
 from corehq.apps.userreports.specs import EvaluationContext
+from corehq.apps.users.tests.util import patch_user_data_db_layer
 from custom.champ.tests.utils import TestDataSourceExpressions
 from custom.champ.utils import POST_TEST_XMLNS, ACCOMPAGNEMENT_XMLNS, SUIVI_MEDICAL_XMLNS
 
 CHAMP_CAMEROON_DATA_SOURCE = 'champ_cameroon.json'
 
 
+@patch_user_data_db_layer()
 class TestEnhancedPeerMobilization(TestDataSourceExpressions):
 
     data_source_name = CHAMP_CAMEROON_DATA_SOURCE

diff --git a/custom/champ/tests/test_enhanced_peer_mobilization.py b/custom/champ/tests/test_enhanced_peer_mobilization.py
@@ -1,11 +1,14 @@
 from datetime import date
+
 from corehq.apps.userreports.specs import EvaluationContext
+from corehq.apps.users.tests.util import patch_user_data_db_layer
 from custom.champ.tests.utils import TestDataSourceExpressions
 from custom.champ.utils import PREVENTION_XMLNS, TARGET_XMLNS
 
 ENHANCED_PEER_MOBILIZATION_DATA_SOURCE = 'enhanced_peer_mobilization.json'
 
 
+@patch_user_data_db_layer()
 class TestEnhancedPeerMobilization(TestDataSourceExpressions):
 
     data_source_name = ENHANCED_PEER_MOBILIZATION_DATA_SOURCE