@@ -46,7 +46,7 @@ def test_retry_success_before_max_attempts():
4646 # First server gives 2 BUSY responses then OK on the 3rd try
4747 response_sequences = [[busy_response , busy_response , ok_response , receipt_response ]]
4848
49- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ):
49+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ):
5050 # Configure client to allow 3 attempts - should succeed on the last try
5151 client .max_attempts = 3
5252
@@ -70,7 +70,7 @@ def test_retry_failure_after_max_attempts():
7070
7171 response_sequences = [[busy_response , busy_response ]]
7272
73- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ):
73+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ):
7474 client .max_attempts = 2
7575
7676 transaction = (
@@ -112,7 +112,7 @@ def test_node_switching_after_single_grpc_error():
112112 [error ],
113113 ]
114114
115- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ):
115+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ):
116116 transaction = (
117117 AccountCreateTransaction ()
118118 .set_key (PrivateKey .generate ().public_key ())
@@ -149,7 +149,7 @@ def test_node_switching_after_multiple_grpc_errors():
149149 [ok_response , receipt_response ],
150150 ]
151151
152- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ):
152+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ):
153153 transaction = (
154154 AccountCreateTransaction ()
155155 .set_key (PrivateKey .generate ().public_key ())
@@ -185,7 +185,7 @@ def test_transaction_with_expired_error_not_retried():
185185 [error_response ]
186186 ]
187187
188- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ):
188+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ):
189189 transaction = (
190190 AccountCreateTransaction ()
191191 .set_key (PrivateKey .generate ().public_key ())
@@ -216,7 +216,7 @@ def test_transaction_with_fatal_error_not_retried():
216216 [error_response ]
217217 ]
218218
219- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ):
219+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ):
220220 transaction = (
221221 AccountCreateTransaction ()
222222 .set_key (PrivateKey .generate ().public_key ())
@@ -248,7 +248,7 @@ def test_exponential_backoff_retry():
248248 response_sequences = [[busy_response , busy_response , busy_response , ok_response , receipt_response ]]
249249
250250 # Use a mock for time.sleep to capture the delay values
251- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ) as mock_sleep :
251+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ) as mock_sleep :
252252 client .max_attempts = 5
253253
254254 transaction = (
@@ -288,7 +288,7 @@ def test_retriable_error_does_not_switch_node():
288288 )
289289 )
290290 response_sequences = [[busy_response , ok_response , receipt_response ]]
291- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ):
291+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ):
292292 transaction = (
293293 AccountCreateTransaction ()
294294 .set_key (PrivateKey .generate ().public_key ())
@@ -333,7 +333,7 @@ def test_topic_create_transaction_retry_on_busy():
333333 [busy_response , ok_response , receipt_response ],
334334 ]
335335
336- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ) as mock_sleep :
336+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ) as mock_sleep :
337337 client .max_attempts = 3
338338
339339 tx = (
@@ -367,7 +367,7 @@ def test_topic_create_transaction_fails_on_nonretriable_error():
367367 [error_response ],
368368 ]
369369
370- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ):
370+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ):
371371 tx = (
372372 TopicCreateTransaction ()
373373 .set_memo ("Test with error" )
@@ -400,7 +400,7 @@ def test_transaction_node_switching_body_bytes():
400400 [ok_response , receipt_response ],
401401 ]
402402
403- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ):
403+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ):
404404 # We set the current node to 0
405405 client .network ._node_index = 0
406406 client .network .current_node = client .network .nodes [0 ]
@@ -467,8 +467,9 @@ def test_query_retry_on_busy():
467467 [ok_response ],
468468 ]
469469
470- with mock_hedera_servers (response_sequences ) as client , patch ('time.sleep' ) as mock_sleep :
470+ with mock_hedera_servers (response_sequences ) as client , patch ('hiero_sdk_python.executable. time.sleep' ) as mock_sleep :
471471 # We set the current node to the first node so we are sure it will return BUSY response
472+ client .network ._node_index = 0
472473 client .network .current_node = client .network .nodes [0 ]
473474
474475 query = CryptoGetAccountBalanceQuery ()
0 commit comments