ayo/kitty
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
No known key found for this signature in database
GPG key ID: 06BC317B515ACE7C
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
kitty_tests/__init__.py
View file

@ -366,12 +366,15 @@ def process_input_from_child(self, timeout=10):
parse_bytes(self.screen, data)
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
while not q() and time.monotonic() <= end_time:
self.process_input_from_child(timeout=end_time - time.monotonic())
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):
end_time = time.monotonic() + timeout