File: test_killsubprocess.py

package info (click to toggle)
pypy 7.0.0%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 107,216 kB
  • sloc: python: 1,201,787; ansic: 62,419; asm: 5,169; cpp: 3,017; sh: 2,534; makefile: 545; xml: 243; lisp: 45; awk: 4
file content (34 lines) | stat: -rw-r--r-- 1,118 bytes parent folder | download | duplicates (9)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import sys, time
import subprocess
from rpython.tool.killsubprocess import killsubprocess

def waitdead(process):
    for i in range(50):
        time.sleep(0.1)
        if process.poll() is not None:
            break       # ok
    else:
        raise AssertionError("the subprocess did not die within 5 seconds")

def test_killsubprocess():
    popen = subprocess.Popen([sys.executable, '-c', 'raw_input()'],
                             stdin=subprocess.PIPE)
    time.sleep(0.9)
    assert popen.poll() is None
    assert popen.poll() is None
    killsubprocess(popen)
    waitdead(popen)

def test_already_dead_but_no_poll():
    popen = subprocess.Popen([sys.executable, '-c', 'pass'],
                             stdin=subprocess.PIPE)
    time.sleep(3)    # a safe margin to be sure the subprocess is already dead
    killsubprocess(popen)
    assert popen.poll() is not None

def test_already_dead_and_polled():
    popen = subprocess.Popen([sys.executable, '-c', 'pass'],
                             stdin=subprocess.PIPE)
    waitdead(popen)
    killsubprocess(popen)
    assert popen.poll() is not None