Mercurial > public > mercurial-scm > hg-stable
diff tests/killdaemons.py @ 17464:eddfb9a550d0
run-tests: do not duplicate killdaemons() code
author | Patrick Mezard <patrick@mezard.eu> |
---|---|
date | Sun, 19 Aug 2012 16:41:09 +0200 |
parents | 13a1b2fb7ef2 |
children | 2d4a096e213c |
line wrap: on
line diff
--- a/tests/killdaemons.py Fri Aug 31 13:17:16 2012 -0500 +++ b/tests/killdaemons.py Sun Aug 19 16:41:09 2012 +0200 @@ -2,24 +2,39 @@ import os, time, errno, signal -# Kill off any leftover daemon processes -try: - fp = open(os.environ['DAEMON_PIDS']) - for line in fp: - try: - pid = int(line) - except ValueError: - continue - try: - os.kill(pid, 0) - os.kill(pid, signal.SIGTERM) - for i in range(10): - time.sleep(0.05) +def killdaemons(pidfile, tryhard=True, remove=False, logfn=None): + if not logfn: + logfn = lambda s: s + # Kill off any leftover daemon processes + try: + fp = open(pidfile) + for line in fp: + try: + pid = int(line) + except ValueError: + continue + try: os.kill(pid, 0) - os.kill(pid, signal.SIGKILL) - except OSError, err: - if err.errno != errno.ESRCH: - raise - fp.close() -except IOError: - pass + logfn('# Killing daemon process %d' % pid) + os.kill(pid, signal.SIGTERM) + if tryhard: + for i in range(10): + time.sleep(0.05) + os.kill(pid, 0) + else: + time.sleep(0.1) + os.kill(pid, 0) + logfn('# Daemon process %d is stuck - really killing it' % pid) + os.kill(pid, signal.SIGKILL) + except OSError, err: + if err.errno != errno.ESRCH: + raise + fp.close() + if remove: + os.unlink(pidfile) + except IOError: + pass + +if __name__ == '__main__': + killdaemons(os.environ['DAEMON_PIDS']) +