phantom/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-04-29 14:59:39 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

test: add TestNode wait_until helper

Browse source
This commit is contained in:
Nikodemas Tuckus 2023-12-13 11:24:03 +01:00
parent d646ca35d9
commit bf0f7dbec6
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
test/functional/test_framework/test_node.py
View file

@ -260,7 +260,7 @@ class TestNode():
if self.version_is_at_least(190000): if self.version_is_at_least(190000):
# getmempoolinfo.loaded is available since commit # getmempoolinfo.loaded is available since commit
# bb8ae2c (version 0.19.0) # bb8ae2c (version 0.19.0)
wait_until_helper_internal(lambda: rpc.getmempoolinfo()['loaded'], timeout_factor=self.timeout_factor) self.wait_until(lambda: rpc.getmempoolinfo()['loaded'])
# Wait for the node to finish reindex, block import, and # Wait for the node to finish reindex, block import, and
# loading the mempool. Usually importing happens fast or # loading the mempool. Usually importing happens fast or
# even "immediate" when the node is started. However, there # even "immediate" when the node is started. However, there
@ -414,7 +414,7 @@ class TestNode():
def wait_until_stopped(self, *, timeout=BITCOIND_PROC_WAIT_TIMEOUT, expect_error=False, **kwargs): def wait_until_stopped(self, *, timeout=BITCOIND_PROC_WAIT_TIMEOUT, expect_error=False, **kwargs):
expected_ret_code = 1 if expect_error else 0 # Whether node shutdown return EXIT_FAILURE or EXIT_SUCCESS expected_ret_code = 1 if expect_error else 0 # Whether node shutdown return EXIT_FAILURE or EXIT_SUCCESS
wait_until_helper_internal(lambda: self.is_node_stopped(expected_ret_code=expected_ret_code, **kwargs), timeout=timeout, timeout_factor=self.timeout_factor) self.wait_until(lambda: self.is_node_stopped(expected_ret_code=expected_ret_code, **kwargs), timeout=timeout)
def replace_in_config(self, replacements): def replace_in_config(self, replacements):
""" """
@ -534,8 +534,7 @@ class TestNode():
initial_peer_id = get_highest_peer_id() initial_peer_id = get_highest_peer_id()
yield yield
wait_until_helper_internal(lambda: get_highest_peer_id() > initial_peer_id, self.wait_until(lambda: get_highest_peer_id() > initial_peer_id, timeout=timeout)
timeout=timeout, timeout_factor=self.timeout_factor)
@contextlib.contextmanager @contextlib.contextmanager
def profile_with_perf(self, profile_name: str): def profile_with_perf(self, profile_name: str):
@ -747,7 +746,7 @@ class TestNode():
p.peer_disconnect() p.peer_disconnect()
del self.p2ps[:] del self.p2ps[:]
wait_until_helper_internal(lambda: self.num_test_p2p_connections() == 0, timeout_factor=self.timeout_factor) self.wait_until(lambda: self.num_test_p2p_connections() == 0)
def bumpmocktime(self, seconds): def bumpmocktime(self, seconds):
"""Fast forward using setmocktime to self.mocktime + seconds. Requires setmocktime to have """Fast forward using setmocktime to self.mocktime + seconds. Requires setmocktime to have
@ -756,6 +755,9 @@ class TestNode():
self.mocktime += seconds self.mocktime += seconds
self.setmocktime(self.mocktime) self.setmocktime(self.mocktime)
def wait_until(self, test_function, timeout=60):
return wait_until_helper_internal(test_function, timeout=timeout, timeout_factor=self.timeout_factor)
class TestNodeCLIAttr: class TestNodeCLIAttr:
def __init__(self, cli, command): def __init__(self, cli, command):