pythongh-118908: Protect the REPL subprocess with a timeout in tests

pablogsal · Jun 12, 2024 · 47e3511 · 47e3511
1 parent 7dd8c37
commit 47e3511
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Lib/test/test_pyrepl/test_pyrepl.py b/Lib/test/test_pyrepl/test_pyrepl.py
@@ -8,6 +8,7 @@
 from unittest import TestCase, skipUnless
 from unittest.mock import patch
 from test.support import force_not_colorized
+from test.support import SHORT_TIMEOUT
 
 from .support import (
     FakeConsole,
@@ -885,5 +886,10 @@ def run_repl(self, repl_input: str | list[str], env: dict | None = None) -> tupl
 
         os.close(master_fd)
         os.close(slave_fd)
-        exit_code = process.wait()
+        try:
+            exit_code = process.wait(timeout=SHORT_TIMEOUT)
+        except subprocess.TimeoutExpired:
+            process.terminate()
+            process.kill()
+            exit_code = process.returncode
         return "\n".join(output), exit_code