Test flushing bare Esc when timeout expires

Author: tanloong

Lib/test/test_pyrepl/test_unix_console.py

@@ -399,6 +399,20 @@ def test_restore_in_thread(self, _os_write):
         thread.start()
         thread.join()  # this should not raise
 
+    def test_escape_timeout(self, _os_write):
+        console = UnixConsole(term="xterm")
+        console.prepare()
+        console.event_queue.push(b"\x1b")
+        self.assertTrue(console.event_queue.pending())
+        # Simulate the read timing out before any follow-up bytes arrive.
+        console.wait = Mock(return_value=False)
+        event = console.get_event()
+        self.assertIsNotNone(event)
+        self.assertEqual(event.evt, "key")
+        self.assertEqual(event.data, "\x0b")
+        self.assertEqual(event.raw, b"\x1b")
+        console.restore()
+
 
 @unittest.skipIf(sys.platform == "win32", "No Unix console on Windows")
 class TestUnixConsoleEIOHandling(TestCase):

Lib/test/test_pyrepl/test_windows_console.py

@@ -648,6 +648,19 @@ def test_wait_not_empty(self):
         self.assertTrue(console.wait(0.0))
         self.assertEqual(console.wait_for_event.call_count, 0)
 
+    def test_escape_timeout(self):
+        console = WindowsConsole()
+        # Simulate the read timing out before any follow-up bytes arrive.
+        console.wait_for_event = MagicMock(return_value=False)
+        console.event_queue.push(b"\x1b")
+        self.assertTrue(console.event_queue.pending())
+        event = console.get_event()
+        console.restore()
+        self.assertIsNotNone(event)
+        self.assertEqual(event.evt, "key")
+        self.assertEqual(event.data, "\x1b")
+        self.assertEqual(event.raw, b"\x1b")
+
 
 if __name__ == "__main__":
     unittest.main()