aboutsummaryrefslogtreecommitdiff
path: root/test/functional/test_framework
diff options
context:
space:
mode:
authorJames O'Beirne <james.obeirne@pm.me>2021-12-10 14:26:04 -0500
committerJames O'Beirne <james.obeirne@pm.me>2021-12-29 13:04:34 -0500
commita8ffbc01db85d918b8d1406b178b3c6f09123cf1 (patch)
treef3f3a91c305bd63d44dc94487b8a043de69d8805 /test/functional/test_framework
parenta2fb62b632d39e468f59e7e0b6890f82d0f5eebb (diff)
test: add TestNode.wait_for_debug_log
Diffstat (limited to 'test/functional/test_framework')
-rwxr-xr-xtest/functional/test_framework/test_node.py36
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):
"""