Add PTY test

Author: edvilme

Lib/test/test_pyrepl/test_pyrepl.py

@@ -2102,6 +2102,31 @@ def setUp(self):
         self.addCleanup(patcher.stop)
         patcher.start()
 
+    def test_statement_submitted_hook_exception(self):
+        env = make_clean_env()
+        env["NO_COLOR"] = "1"
+        output, exit_code = self.run_repl(
+            [
+                "from _pyrepl import simple_interact as si",
+                "def hook(statement): raise RuntimeError('hook failed')",
+                "",
+                "si.statement_submitted_hook = hook",
+                "x = 1",
+                "print('executed:', x)",
+                "exit",
+            ],
+            env=env,
+            skip=True,
+        )
+
+        output = output.replace("\r\n", "\n")
+        self.assertEqual(exit_code, 0)
+        self.assertEqual(
+            output.count("Error in statement_submitted_hook:\n"), 2
+        )
+        self.assertEqual(output.count("RuntimeError: hook failed"), 2)
+        self.assertIn("\nexecuted: 1\n", output)
+
     @force_not_colorized
     def test_exposed_globals_in_repl(self):
         pre = "['__builtins__'"