@@ -48,6 +48,7 @@ def setup_and_cleanup(
48
48
client_maint_events : Redis ,
49
49
fault_injector_client : FaultInjectorClient ,
50
50
endpoints_config : Dict [str , Any ],
51
+ endpoint_name : str ,
51
52
):
52
53
# Initialize cleanup flags first to ensure they exist even if setup fails
53
54
self ._migration_executed = False
@@ -70,7 +71,7 @@ def setup_and_cleanup(
70
71
71
72
try :
72
73
self .endpoint_id = ClusterOperations .find_endpoint_for_bind (
73
- fault_injector_client , endpoints_config
74
+ fault_injector_client , endpoints_config , endpoint_name
74
75
)
75
76
logging .info (f"Using endpoint: { self .endpoint_id } " )
76
77
except Exception as e :
@@ -396,7 +397,7 @@ def test_timeout_handling_during_migrating_and_moving(
396
397
397
398
"""
398
399
logging .info (f"Testing timeout handling for endpoint type: { endpoint_type } " )
399
- client = _get_client_maint_events (endpoints_config , endpoint_type )
400
+ client = _get_client_maint_events (endpoints_config = endpoints_config , endpoint_type = endpoint_type )
400
401
401
402
# Create three connections in the pool
402
403
logging .info ("Creating three connections in the pool." )
@@ -501,7 +502,7 @@ def test_new_connection_handling_during_migrating_and_moving(
501
502
endpoints_config : Dict [str , Any ],
502
503
):
503
504
logging .info (f"Testing timeout handling for endpoint type: { endpoint_type } " )
504
- client = _get_client_maint_events (endpoints_config , endpoint_type )
505
+ client = _get_client_maint_events (endpoints_config = endpoints_config , endpoint_type = endpoint_type )
505
506
506
507
logging .info ("Creating one connection in the pool." )
507
508
first_conn = client .connection_pool .get_connection ()
@@ -614,7 +615,7 @@ def test_disabled_handling_during_migrating_and_moving(
614
615
):
615
616
logging .info ("Creating client with disabled notifications." )
616
617
client = _get_client_maint_events (
617
- endpoints_config ,
618
+ endpoints_config = endpoints_config ,
618
619
enable_maintenance_events = False ,
619
620
)
620
621
@@ -741,7 +742,7 @@ def test_command_execution_during_migrating_and_moving(
741
742
socket_timeout = 0.5
742
743
743
744
client = _get_client_maint_events (
744
- endpoints_config ,
745
+ endpoints_config = endpoints_config ,
745
746
endpoint_type = endpoint_type ,
746
747
disable_retries = True ,
747
748
socket_timeout = socket_timeout ,
0 commit comments