From 75d5e386d394e43f32a09ee9ddc79f1bf0a6a884 Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Sun, 6 Mar 2022 15:35:24 +0530
Subject: [PATCH] Make the integration tests more strict

Actually assert that there are/are not prompt marking bytes in the data
received by the pty
---
 kitty_tests/ssh.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/kitty_tests/ssh.py b/kitty_tests/ssh.py
index 4a5dacba8..113a84a4d 100644
--- a/kitty_tests/ssh.py
+++ b/kitty_tests/ssh.py
@@ -199,6 +199,7 @@ copy --exclude */w.* d1
                     elif login_shell == 'zsh':
                         pty.send_cmd_to_child('echo "login_shell=$ZSH_NAME"')
                         pty.wait_till(lambda: 'login_shell=zsh' in pty.screen_contents())
+                    self.assertIn(b'\x1b]133;', pty.received_bytes)
         # check that turning off shell integration works
         if ok_login_shell in ('bash', 'zsh'):
             for val in ('', 'no-rc', 'enabled no-rc'):
@@ -210,6 +211,7 @@ copy --exclude */w.* d1
                         pty.wait_till(lambda: 'kitty=fruity' in pty.screen_contents())
                         pty.wait_till(lambda: len(pty.screen_contents().splitlines()) >= num_lines + 2)
                         self.assertEqual(pty.screen.cursor.shape, 0)
+                        self.assertNotIn(b'\x1b]133;', pty.received_bytes)
 
     def check_bootstrap(self, sh, home_dir, login_shell='', SHELL_INTEGRATION_VALUE='enabled', test_script='', pre_data='', ssh_opts=None):
         ssh_opts = ssh_opts or {}