cleaning up child process tree spawned from test case

test_child_process.py

import time
import subprocess

import psutil, os

def kill_proc_tree(pid, including_parent=True):
    parent = psutil.Process(pid)
    children = parent.children(recursive=True)
    for child in children:
        child.kill()
    psutil.wait_procs(children, timeout=5)
    if including_parent:
        parent.kill()
        parent.wait(5)

def test_foo():
    try:
        p = subprocess.Popen(['./my-sleep.sh'])
        p.wait()
    except:
        print "cleaning up"
        #TODO: doesn't kill process tree
        #p.kill()
        kill_proc_tree(p.pid)
        raise