Fix test cases

esrakartalOpt · esrakartalOpt · commit 0a0cdefe688d · 2025-07-02T18:58:15.000+04:00
diff --git a/tests/test_event_builder.py b/tests/test_event_builder.py
@@ -86,6 +86,7 @@ def test_create_impression_event(self):
             'enrich_decisions': True,
             'anonymize_ip': False,
             'revision': '42',
+            'region': 'US',
         }
 
         with mock.patch('time.time', return_value=42.123), mock.patch(
@@ -141,6 +142,7 @@ def test_create_impression_event__with_attributes(self):
             'enrich_decisions': True,
             'anonymize_ip': False,
             'revision': '42',
+            'region': 'US',
         }
 
         with mock.patch('time.time', return_value=42.123), mock.patch(
@@ -194,6 +196,7 @@ def test_create_impression_event_when_attribute_is_not_in_datafile(self):
             'enrich_decisions': True,
             'anonymize_ip': False,
             'revision': '42',
+            'region': 'US',
         }
 
         with mock.patch('time.time', return_value=42.123), mock.patch(
@@ -250,6 +253,7 @@ def test_create_impression_event_calls_is_attribute_valid(self):
             'enrich_decisions': True,
             'anonymize_ip': False,
             'revision': '42',
+            'region': 'US',
         }
 
         def side_effect(*args, **kwargs):
@@ -328,6 +332,7 @@ def test_create_impression_event__with_user_agent_when_bot_filtering_is_enabled(
             'enrich_decisions': True,
             'anonymize_ip': False,
             'revision': '42',
+            'region': 'US',
         }
 
         with mock.patch('time.time', return_value=42.123), mock.patch(
@@ -392,6 +397,7 @@ def test_create_impression_event__with_empty_attributes_when_bot_filtering_is_en
             'enrich_decisions': True,
             'anonymize_ip': False,
             'revision': '42',
+            'region': 'US',
         }
 
         with mock.patch('time.time', return_value=42.123), mock.patch(
@@ -462,6 +468,7 @@ def test_create_impression_event__with_user_agent_when_bot_filtering_is_disabled
             'enrich_decisions': True,
             'anonymize_ip': False,
             'revision': '42',
+            'region': 'US',
         }
 
         with mock.patch('time.time', return_value=42.123), mock.patch(
diff --git a/tests/test_user_event_factory.py b/tests/test_user_event_factory.py
@@ -48,8 +48,10 @@ def test_impression_event_with_region_eu(self):
         experiment = self.project_config.get_experiment_from_key('test_experiment')
         user_id = 'test_user'
 
+        project_config.region = 'EU'
+
         impression_event = UserEventFactory.create_impression_event(
-            project_config, experiment, '111128', '', 'rule_key', 'rule_type', True, user_id, None, region='EU'
+            project_config, experiment, '111128', '', 'rule_key', 'rule_type', True, user_id, None
         )
 
         self.assertEqual(self.project_config.region, impression_event.event_context.region)
