Skip to content

Commit f68d678

Browse files
committed
Fix test cases
1 parent 0a0cdef commit f68d678

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

tests/test_event_factory.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def test_create_impression_event(self):
118118
log_event = EventFactory.create_log_event(event_obj, self.logger)
119119

120120
self._validate_event_object(
121-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
121+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
122122
)
123123

124124
def test_create_impression_event__with_attributes(self):
@@ -182,7 +182,7 @@ def test_create_impression_event__with_attributes(self):
182182
log_event = EventFactory.create_log_event(event_obj, self.logger)
183183

184184
self._validate_event_object(
185-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
185+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
186186
)
187187

188188
def test_create_impression_event_when_attribute_is_not_in_datafile(self):
@@ -244,7 +244,7 @@ def test_create_impression_event_when_attribute_is_not_in_datafile(self):
244244
log_event = EventFactory.create_log_event(event_obj, self.logger)
245245

246246
self._validate_event_object(
247-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
247+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
248248
)
249249

250250
def test_create_impression_event_calls_is_attribute_valid(self):
@@ -323,7 +323,7 @@ def side_effect(*args, **kwargs):
323323

324324
self._validate_event_object(
325325
log_event,
326-
EventFactory.EVENT_ENDPOINT,
326+
EventFactory.EVENT_ENDPOINTS.get('US'),
327327
expected_params,
328328
EventFactory.HTTP_VERB,
329329
EventFactory.HTTP_HEADERS,
@@ -399,7 +399,7 @@ def test_create_impression_event__with_user_agent_when_bot_filtering_is_enabled(
399399
log_event = EventFactory.create_log_event(event_obj, self.logger)
400400

401401
self._validate_event_object(
402-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
402+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
403403
)
404404

405405
def test_create_impression_event__with_empty_attributes_when_bot_filtering_is_enabled(self,):
@@ -471,7 +471,7 @@ def test_create_impression_event__with_empty_attributes_when_bot_filtering_is_en
471471
log_event = EventFactory.create_log_event(event_obj, self.logger)
472472

473473
self._validate_event_object(
474-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
474+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
475475
)
476476

477477
def test_create_impression_event__with_user_agent_when_bot_filtering_is_disabled(self,):
@@ -549,7 +549,7 @@ def test_create_impression_event__with_user_agent_when_bot_filtering_is_disabled
549549
log_event = EventFactory.create_log_event(event_obj, self.logger)
550550

551551
self._validate_event_object(
552-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
552+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
553553
)
554554

555555
def test_create_conversion_event(self):
@@ -595,7 +595,7 @@ def test_create_conversion_event(self):
595595
log_event = EventFactory.create_log_event(event_obj, self.logger)
596596

597597
self._validate_event_object(
598-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
598+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
599599
)
600600

601601
def test_create_conversion_event__with_attributes(self):
@@ -643,7 +643,7 @@ def test_create_conversion_event__with_attributes(self):
643643
log_event = EventFactory.create_log_event(event_obj, self.logger)
644644

645645
self._validate_event_object(
646-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
646+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
647647
)
648648

649649
def test_create_conversion_event__with_user_agent_when_bot_filtering_is_enabled(self,):
@@ -700,7 +700,7 @@ def test_create_conversion_event__with_user_agent_when_bot_filtering_is_enabled(
700700
log_event = EventFactory.create_log_event(event_obj, self.logger)
701701

702702
self._validate_event_object(
703-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
703+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
704704
)
705705

706706
def test_create_conversion_event__with_user_agent_when_bot_filtering_is_disabled(self,):
@@ -762,7 +762,7 @@ def test_create_conversion_event__with_user_agent_when_bot_filtering_is_disabled
762762
log_event = EventFactory.create_log_event(event_obj, self.logger)
763763

764764
self._validate_event_object(
765-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
765+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
766766
)
767767

768768
def test_create_conversion_event__with_event_tags(self):
@@ -817,7 +817,7 @@ def test_create_conversion_event__with_event_tags(self):
817817
log_event = EventFactory.create_log_event(event_obj, self.logger)
818818

819819
self._validate_event_object(
820-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
820+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
821821
)
822822

823823
def test_create_conversion_event__with_invalid_event_tags(self):
@@ -870,7 +870,7 @@ def test_create_conversion_event__with_invalid_event_tags(self):
870870
log_event = EventFactory.create_log_event(event_obj, self.logger)
871871

872872
self._validate_event_object(
873-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
873+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
874874
)
875875

876876
def test_create_conversion_event__when_event_is_used_in_multiple_experiments(self):
@@ -925,5 +925,5 @@ def test_create_conversion_event__when_event_is_used_in_multiple_experiments(sel
925925
log_event = EventFactory.create_log_event(event_obj, self.logger)
926926

927927
self._validate_event_object(
928-
log_event, EventFactory.EVENT_ENDPOINT, expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
928+
log_event, EventFactory.EVENT_ENDPOINTS.get('US'), expected_params, EventFactory.HTTP_VERB, EventFactory.HTTP_HEADERS,
929929
)

0 commit comments

Comments
 (0)