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

Make processing of input from child in test pty device a bit more robust

Browse source
This commit is contained in:
Kovid Goyal 2024-05-05 13:59:23 +05:30
parent 849dcd424d
commit 04af71696a
No known key found for this signature in database
GPG key ID: 06BC317B515ACE7C
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
kitty_tests/__init__.py
View file

@ -337,7 +337,7 @@ def send_cmd_to_child(self, cmd, flush=False):
self.write_to_child(cmd + '\r', flush=flush)
def process_input_from_child(self, timeout=10):
rd, wd, err = select.select([self.master_fd], [self.master_fd] if self.write_buf else [], [], timeout)
rd, wd, _ = select.select([self.master_fd], [self.master_fd] if self.write_buf else [], [], max(0, timeout))
while wd:
try:
n = os.write(self.master_fd, self.write_buf)
@ -363,7 +363,7 @@ def process_input_from_child(self, timeout=10):
def wait_till(self, q, timeout=10):
end_time = time.monotonic() + timeout
while not q() and time.monotonic() <= end_time:
self.process_input_from_child(timeout=max(0, end_time - time.monotonic()))
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())}')