22from unittest .mock import patch
33
44import pytest
5+ from werkzeug .datastructures import MultiDict
56
67from common .entities import DB
78from common .entity_services import EventService
@@ -22,14 +23,14 @@ def test_get_events_with_rules_filters(
2223):
2324 filter_cases = (
2425 ({}, (event_entity , event_entity_2 )),
25- ({"component_id" : [ str (pipeline .id )] }, (event_entity ,)),
26- ({"instance_id" : [ str (instance_instance_set .instance_id )] }, (event_entity ,)),
27- ({"journey_id" : [ str (instance_instance_set .instance .journey_id )] }, (event_entity ,)),
28- ({"event_type" : [ "BATCH_PIPELINE_STATUS" ] }, (event_entity ,)),
29- ({"event_type" : [ "DATASET_OPERATION" ] }, (event_entity_2 ,)),
30- ({"event_id" : [ str (event_entity_2 .id )] }, (event_entity_2 ,)),
31- ({"run_id" : [ str (run .id )] }, (event_entity ,)),
32- ({"task_id" : [ str (task .id )] }, (event_entity ,)),
26+ ({"component_id" : str (pipeline .id )}, (event_entity ,)),
27+ ({"instance_id" : str (instance_instance_set .instance_id )}, (event_entity ,)),
28+ ({"journey_id" : str (instance_instance_set .instance .journey_id )}, (event_entity ,)),
29+ ({"event_type" : "BATCH_PIPELINE_STATUS" }, (event_entity ,)),
30+ ({"event_type" : "DATASET_OPERATION" }, (event_entity_2 ,)),
31+ ({"event_id" : str (event_entity_2 .id )}, (event_entity_2 ,)),
32+ ({"run_id" : str (run .id )}, (event_entity ,)),
33+ ({"task_id" : str (task .id )}, (event_entity ,)),
3334 ({"date_range_start" : "2024-01-20T09:56:00" }, (event_entity ,)),
3435 ({"date_range_end" : "2024-01-20T09:59:10" }, (event_entity , event_entity_2 )),
3536 (
@@ -42,7 +43,7 @@ def test_get_events_with_rules_filters(
4243
4344 rules = ListRules ()
4445 for filter_params , expected_result in filter_cases :
45- filters = ProjectEventFilters .from_params (params = filter_params , project_ids = [project .id ])
46+ filters = ProjectEventFilters .from_params (params = MultiDict ( filter_params . items ()) , project_ids = [project .id ])
4647 page = EventService .get_events_with_rules (rules = rules , filters = filters )
4748 assert set (page .results ) == set (expected_result ), filter_params
4849
@@ -51,7 +52,7 @@ def test_get_events_with_rules_filters(
5152@pytest .mark .parametrize ("sort_order,reverse" , ((SortOrder .ASC , False ), (SortOrder .DESC , True )))
5253def test_get_events_with_rules_sort (sort_order , reverse , event_entity , event_entity_2 , project ):
5354 rules = ListRules (sort = sort_order )
54- filters = ProjectEventFilters .from_params (params = {} , project_ids = [project .id ])
55+ filters = ProjectEventFilters .from_params (params = MultiDict ({}) , project_ids = [project .id ])
5556
5657 page = EventService .get_events_with_rules (rules = rules , filters = filters )
5758
@@ -62,7 +63,7 @@ def test_get_events_with_rules_sort(sort_order, reverse, event_entity, event_ent
6263@pytest .mark .integration
6364def test_get_events_with_rules_prefetch (event_entity , project , instance ):
6465 rules = ListRules ()
65- filters = ProjectEventFilters .from_params (params = {} , project_ids = [project .id ])
66+ filters = ProjectEventFilters .from_params (params = MultiDict ({}) , project_ids = [project .id ])
6667
6768 with assert_num_queries (3 ):
6869 (result_event ,) = EventService .get_events_with_rules (rules = rules , filters = filters )
0 commit comments