From a43f61055519c9f51c6ceb13a71aad6a18db3d84 Mon Sep 17 00:00:00 2001 From: Kovid Goyal Date: Mon, 21 Feb 2022 18:18:31 +0530 Subject: [PATCH] Better error msg when cursor is not changed --- kitty_tests/shell_integration.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/kitty_tests/shell_integration.py b/kitty_tests/shell_integration.py index c03702733..0af158967 100644 --- a/kitty_tests/shell_integration.py +++ b/kitty_tests/shell_integration.py @@ -58,9 +58,12 @@ PS1="{ps1}" RPS1="{rps1}" ''') as pty: q = ps1 + ' ' * (pty.screen.columns - len(ps1) - len(rps1)) + rps1 - pty.wait_till(lambda: pty.screen.cursor.shape == CURSOR_BEAM) + try: + pty.wait_till(lambda: pty.screen.cursor.shape == CURSOR_BEAM) + except TimeoutError: + raise AssertionError(f'Cursor was not changed to beam. Screen contents: {repr(pty.screen_contents())}') self.ae(pty.screen_contents(), q) self.ae(pty.callbacks.titlebuf, '~') pty.send_cmd_to_child('mkdir test && ls -a') - pty.wait_till(lambda: pty.screen_contents().count('left>') == 2) + pty.wait_till(lambda: pty.screen_contents().count(ps1) == 2) self.ae(pty.last_cmd_output(), str(pty.screen.line(1)))