@@ -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,9 @@ 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 (
401
+ endpoints_config = endpoints_config , endpoint_type = endpoint_type
402
+ )
400
403
401
404
# Create three connections in the pool
402
405
logging .info ("Creating three connections in the pool." )
@@ -501,7 +504,9 @@ def test_new_connection_handling_during_migrating_and_moving(
501
504
endpoints_config : Dict [str , Any ],
502
505
):
503
506
logging .info (f"Testing timeout handling for endpoint type: { endpoint_type } " )
504
- client = _get_client_maint_events (endpoints_config , endpoint_type )
507
+ client = _get_client_maint_events (
508
+ endpoints_config = endpoints_config , endpoint_type = endpoint_type
509
+ )
505
510
506
511
logging .info ("Creating one connection in the pool." )
507
512
first_conn = client .connection_pool .get_connection ()
@@ -614,7 +619,7 @@ def test_disabled_handling_during_migrating_and_moving(
614
619
):
615
620
logging .info ("Creating client with disabled notifications." )
616
621
client = _get_client_maint_events (
617
- endpoints_config ,
622
+ endpoints_config = endpoints_config ,
618
623
enable_maintenance_events = False ,
619
624
)
620
625
@@ -741,7 +746,7 @@ def test_command_execution_during_migrating_and_moving(
741
746
socket_timeout = 0.5
742
747
743
748
client = _get_client_maint_events (
744
- endpoints_config ,
749
+ endpoints_config = endpoints_config ,
745
750
endpoint_type = endpoint_type ,
746
751
disable_retries = True ,
747
752
socket_timeout = socket_timeout ,
0 commit comments