handle ctrl-c better when interrupting go test

2022-11-14 15:42:06 +05:30
parent 260249491d
commit f945ef8ee8
1 changed files with 6 additions and 1 deletions
@@ -158,7 +158,12 @@ def run_python_tests(args: Any, go_proc: 'Optional[subprocess.Popen[bytes]]' = N
    tests = find_all_tests()

    def print_go() -> None:
-        print(go_proc.stdout.read().decode(), end='', flush=True)
+        try:
+            print(go_proc.stdout.read().decode(), end='', flush=True)
+        except KeyboardInterrupt:
+            go_proc.terminate()
+            if go_proc.wait(0.1) is None:
+                go_proc.kill()
        go_proc.stdout.close()
        go_proc.wait()