diff options
author | James O'Beirne <james.obeirne@pm.me> | 2021-12-10 14:26:04 -0500 |
---|---|---|
committer | James O'Beirne <james.obeirne@pm.me> | 2021-12-29 13:04:34 -0500 |
commit | a8ffbc01db85d918b8d1406b178b3c6f09123cf1 (patch) | |
tree | f3f3a91c305bd63d44dc94487b8a043de69d8805 | |
parent | a2fb62b632d39e468f59e7e0b6890f82d0f5eebb (diff) | |
download | bitcoin-a8ffbc01db85d918b8d1406b178b3c6f09123cf1.tar.xz |
test: add TestNode.wait_for_debug_log
-rwxr-xr-x | test/functional/test_framework/test_node.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/test/functional/test_framework/test_node.py b/test/functional/test_framework/test_node.py index 49d885f793..0b9154a030 100755 --- a/test/functional/test_framework/test_node.py +++ b/test/functional/test_framework/test_node.py @@ -422,6 +422,42 @@ class TestNode(): time.sleep(0.05) self._raise_assertion_error('Expected messages "{}" does not partially match log:\n\n{}\n\n'.format(str(expected_msgs), print_log)) + def wait_for_debug_log(self, expected_msgs, timeout=10, ignore_case=False) -> int: + """ + Block until we see a particular debug log message fragment or until we exceed the timeout. + Return: + the number of log lines we encountered when matching + """ + time_end = time.time() + timeout * self.timeout_factor + prev_size = self.debug_log_bytes() + re_flags = re.MULTILINE | (re.IGNORECASE if ignore_case else 0) + + while True: + found = True + with open(self.debug_log_path, encoding='utf-8') as dl: + dl.seek(prev_size) + log = dl.read() + + for expected_msg in expected_msgs: + if re.search(re.escape(expected_msg), log, flags=re_flags) is None: + found = False + + if found: + num_logs = len(log.splitlines()) + return num_logs + + if time.time() >= time_end: + print_log = " - " + "\n - ".join(log.splitlines()) + break + + # No sleep here because we want to detect the message fragment as fast as + # possible. + + self._raise_assertion_error( + 'Expected messages "{}" does not partially match log:\n\n{}\n\n'.format( + str(expected_msgs), print_log)) + return -1 # useless return to satisfy linter + @contextlib.contextmanager def profile_with_perf(self, profile_name: str): """ |