fix subprocesses

arihant2math · arihant2math · commit ce6585a15b94 · 2025-02-07T11:57:16.000-08:00
Signed-off-by: Ashwin Naren &lt;arihant2math@gmail.com&gt;
diff --git a/extra_tests/snippets/stdlib_subprocess.py b/extra_tests/snippets/stdlib_subprocess.py
@@ -27,8 +27,43 @@ def sleep(secs):
 assert p.poll() == 0
 assert p.returncode == 0
 
+p = subprocess.Popen(sleep(2))
+
+assert p.poll() is None
+
+with assert_raises(subprocess.TimeoutExpired):
+    assert p.wait(1)
+
+p.wait()
+
+assert p.returncode == 0
+
 p = subprocess.Popen(echo("test"), stdout=subprocess.PIPE)
 p.wait()
 
 
 assert p.stdout.read().strip() == b"test"
+
+p = subprocess.Popen(sleep(2))
+p.terminate()
+p.wait()
+if is_unix:
+    assert p.returncode == -signal.SIGTERM
+else:
+    assert p.returncode == 1
+
+p = subprocess.Popen(sleep(2))
+p.kill()
+p.wait()
+if is_unix:
+    assert p.returncode == -signal.SIGKILL
+else:
+    assert p.returncode == 1
+
+p = subprocess.Popen(echo("test"), stdout=subprocess.PIPE)
+(stdout, stderr) = p.communicate()
+assert stdout.strip() == b"test"
+
+p = subprocess.Popen(sleep(5), stdout=subprocess.PIPE)
+with assert_raises(subprocess.TimeoutExpired):
+    p.communicate(timeout=1)
