@@ -260,7 +260,7 @@ def wait_for_rpc_connection(self):
260
260
if self .version_is_at_least (190000 ):
261
261
# getmempoolinfo.loaded is available since commit
262
262
# bb8ae2c (version 0.19.0)
263
- wait_until_helper_internal (lambda : rpc .getmempoolinfo ()['loaded' ], timeout_factor = self . timeout_factor )
263
+ self . wait_until (lambda : rpc .getmempoolinfo ()['loaded' ])
264
264
# Wait for the node to finish reindex, block import, and
265
265
# loading the mempool. Usually importing happens fast or
266
266
# even "immediate" when the node is started. However, there
@@ -414,7 +414,7 @@ def is_node_stopped(self, *, expected_stderr="", expected_ret_code=0):
414
414
415
415
def wait_until_stopped (self , * , timeout = BITCOIND_PROC_WAIT_TIMEOUT , expect_error = False , ** kwargs ):
416
416
expected_ret_code = 1 if expect_error else 0 # Whether node shutdown return EXIT_FAILURE or EXIT_SUCCESS
417
- wait_until_helper_internal (lambda : self .is_node_stopped (expected_ret_code = expected_ret_code , ** kwargs ), timeout = timeout , timeout_factor = self . timeout_factor )
417
+ self . wait_until (lambda : self .is_node_stopped (expected_ret_code = expected_ret_code , ** kwargs ), timeout = timeout )
418
418
419
419
def replace_in_config (self , replacements ):
420
420
"""
@@ -534,8 +534,7 @@ def get_highest_peer_id():
534
534
535
535
initial_peer_id = get_highest_peer_id ()
536
536
yield
537
- wait_until_helper_internal (lambda : get_highest_peer_id () > initial_peer_id ,
538
- timeout = timeout , timeout_factor = self .timeout_factor )
537
+ self .wait_until (lambda : get_highest_peer_id () > initial_peer_id , timeout = timeout )
539
538
540
539
@contextlib .contextmanager
541
540
def profile_with_perf (self , profile_name : str ):
@@ -747,7 +746,7 @@ def disconnect_p2ps(self):
747
746
p .peer_disconnect ()
748
747
del self .p2ps [:]
749
748
750
- wait_until_helper_internal (lambda : self .num_test_p2p_connections () == 0 , timeout_factor = self . timeout_factor )
749
+ self . wait_until (lambda : self .num_test_p2p_connections () == 0 )
751
750
752
751
def bumpmocktime (self , seconds ):
753
752
"""Fast forward using setmocktime to self.mocktime + seconds. Requires setmocktime to have
@@ -756,6 +755,9 @@ def bumpmocktime(self, seconds):
756
755
self .mocktime += seconds
757
756
self .setmocktime (self .mocktime )
758
757
758
+ def wait_until (self , test_function , timeout = 60 ):
759
+ return wait_until_helper_internal (test_function , timeout = timeout , timeout_factor = self .timeout_factor )
760
+
759
761
760
762
class TestNodeCLIAttr :
761
763
def __init__ (self , cli , command ):
0 commit comments