def test_reap_process_tree_plain(handler):
"""
Tests that process is killed when handling SIGTERM, times out, or ignores.
"""
proc = Process(target=noop_target, args=[handler])
try:
proc.start()
# Wait until ready
while not proc.pid:
time.sleep(0.1)
assert psutil.pid_exists(proc.pid)
base.reap_process_tree(proc.pid, wait_timeout=0.1)
assert not psutil.pid_exists(proc.pid)
finally:
# Clean up any potentially danging processp
if psutil.pid_exists(proc.pid):
os.kill(proc.pid, signal.SIGKILL)
assert False, 'KILLed process with pid={}'.format(proc.pid)
def test_reap_process_tree_children(handler):
"""
Tests recursive termination children with SIGTERM handlers.
"""
child_pid = Value('i', 0)
lock = Lock()
proc = Process(target=nested_target, args=[handler, child_pid, lock])
try:
proc.start()
while not proc.pid or not child_pid.value:
time.sleep(0.1)
pids = [proc.pid, child_pid.value]
assert all(psutil.pid_exists(p) for p in pids)
base.reap_process_tree(proc.pid, wait_timeout=1)
assert all(not psutil.pid_exists(p) for p in pids)
finally:
for pid in pids:
if psutil.pid_exists(proc.pid):
os.kill(proc.pid, signal.SIGKILL)
assert False, 'KILLed process with pid={}'.format(proc.pid)
"""
signal.signal(signal.SIGTERM, handler)
proc = Process(target=noop_target, args=[handler])
proc.start()
while not proc.pid:
time.sleep(0.1)
with lock:
child_pid.value = proc.pid
while True:
time.sleep(1)
@pytest.mark.parametrize("handler", [
lambda signum, frame: sys.exit(),
lambda signum, frame: None,
lambda signum, frame: base.reap_process_tree(os.getpid()) or sys.exit(),
])
def test_reap_process_tree_children(handler):
"""
Tests recursive termination children with SIGTERM handlers.
"""
child_pid = Value('i', 0)
lock = Lock()
proc = Process(target=nested_target, args=[handler, child_pid, lock])
try:
proc.start()
while not proc.pid or not child_pid.value:
time.sleep(0.1)
pids = [proc.pid, child_pid.value]
assert all(psutil.pid_exists(p) for p in pids)
base.reap_process_tree(proc.pid, wait_timeout=1)