mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-07-27 10:41:58 +02:00
Code Issues Packages Projects Releases Wiki Activity

Allow showing a more useful error message when wait_till fails

Browse Source
This commit is contained in:
Kovid Goyal
2024-05-11 08:33:47 +05:30
parent ab2a4f7de6
commit c17d7614e1
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
kitty_tests/__init__.py
View File

@@ -366,12 +366,15 @@ class PTY:
parse_bytes(self.screen, data) parse_bytes(self.screen, data)
return bytes_read return bytes_read
def wait_till(self, q, timeout=10): def wait_till(self, q, timeout=10, timeout_msg=None):
end_time = time.monotonic() + timeout end_time = time.monotonic() + timeout
while not q() and time.monotonic() <= end_time: while not q() and time.monotonic() <= end_time:
self.process_input_from_child(timeout=end_time - time.monotonic()) self.process_input_from_child(timeout=end_time - time.monotonic())
if not q(): if not q():
raise TimeoutError(f'The condition was not met. Screen contents: \n {repr(self.screen_contents())}') msg = 'The condition was not met'
if timeout_msg is not None:
msg = timeout_msg()
raise TimeoutError(f'Timed out: {msg}. Screen contents: \n {repr(self.screen_contents())}')
def wait_till_child_exits(self, timeout=30 if BaseTest.is_ci else 10, require_exit_code=None): def wait_till_child_exits(self, timeout=30 if BaseTest.is_ci else 10, require_exit_code=None):
end_time = time.monotonic() + timeout end_time = time.monotonic() + timeout