def setUp(self):
self.wait = None
def _wait():
return self.wait()
self.reactor = reaperlib.SyncReactor(install=None,
run=None,
sleep=None,
wait=_wait)
def test_termination(self):
"""Terminating processes when we go down"""
handlers = {}
def _install(signum, value):
handlers[signum] = value
def _wait():
raise ValueError()
sleeps = []
def _sleep(tm):
sleeps.append(tm)
terminations = [0]
killings = [0]
polls = [None, None, 0, 0]
def _run(dummyCommand):
ret = DummyProcess()
def _terminate():
terminations[0] += 1
def _kill():
killings[0] += 1
def _poll():
return polls.pop(0)
ret.terminate = _terminate
ret.kill = _kill
ret.poll = _poll
return ret
reactor = reaperlib.SyncReactor(install=_install,
run=_run,
sleep=_sleep,
wait=_wait)
oldStderr = sys.stderr
def _cleanup():
sys.stderr = oldStderr
self.addCleanup(_cleanup)
sys.stderr = StringIO()
reaperlib.baseMain(reactor, [None, 'ls'])
self.assertEquals(terminations[0], 1)
self.assertEquals(killings[0], 0)
self.assertEquals(sleeps, [1, 1])
self.assertEquals(polls, [0])
def test_installSignals(self):
"""Signal installation adds handlers for all signals
Also, signal processor stops further processing"""
handlers = {}
def _install(signum, value):
handlers[signum] = value
reactor = reaperlib.SyncReactor(install=_install,
run=None,
sleep=None,
wait=None)
reaperlib.installSignals(reactor)
self.assertEquals(set(handlers),
set([signal.SIGTERM, signal.SIGINT, signal.SIGALRM]))
self.assertEquals(len(set(handlers.itervalues())), 1)
caller = next(handlers.itervalues())
with self.assertRaises(SystemError):
caller(12, None)
self.assertEquals(len(set(handlers.itervalues())), 1)
ignore = next(handlers.itervalues())
self.assertEquals(ignore, signal.SIG_IGN)
def test_value_error(self):
"""Random errors generate a traceback"""
handlers = {}
def _install(signum, value):
handlers[signum] = value
def _run(dummyCommand):
ret = DummyProcess()
ret.poll = lambda: 0
return ret
def _wait():
raise ValueError()
def _sleep(dummyTm):
pass
reactor = reaperlib.SyncReactor(install=_install,
run=_run,
sleep=_sleep,
wait=_wait)
oldStderr = sys.stderr
def _cleanup():
sys.stderr = oldStderr
self.addCleanup(_cleanup)
sys.stderr = StringIO()
reaperlib.baseMain(reactor, [None, 'ls'])
## pylint: disable=no-member
lines = sys.stderr.getvalue().splitlines()
## pylint: enable=no-member
_cleanup()
dummyBaseMain, = (line for line in lines
if line.endswith('in baseMain'))
def test_system_error(self):
"""System error (generated by signals) works"""
handlers = {}
def _install(signum, value):
handlers[signum] = value
def _run(dummyCommand):
ret = DummyProcess()
ret.poll = lambda: 0
return ret
def _wait():
raise SystemError()
def _sleep(dummyTm):
pass
reactor = reaperlib.SyncReactor(install=_install,
run=_run,
sleep=_sleep,
wait=_wait)
reaperlib.baseMain(reactor, [None, 'ls'])
self.assertTrue(True)
def test_ctrl_c(self):
"""KeyboardInterrupt works"""
handlers = {}
def _install(signum, value):
handlers[signum] = value
def _run(dummyCommand):
ret = DummyProcess()
ret.poll = lambda: 0
return ret
def _wait():
raise KeyboardInterrupt()
def _sleep(dummyTm):
pass
reactor = reaperlib.SyncReactor(install=_install,
run=_run,
sleep=_sleep,
wait=_wait)
reaperlib.baseMain(reactor, [None, 'ls'])
self.assertTrue(True)
def test_simple(self):
"""Happy path works"""
handlers = {}
def _install(signum, value):
handlers[signum] = value
args = []
def _run(command):
args[:] = command
ret = DummyProcess()
ret.poll = lambda: 0
return ret
waitCount = [0]
def _wait():
waitCount[0] += 1
return DummyProcess.pid, None
naps = []
def _sleep(tm):
naps.append(tm)
reactor = reaperlib.SyncReactor(install=_install,
run=_run,
sleep=_sleep,
wait=_wait)
reaperlib.baseMain(reactor, [None, 'ls'])
self.assertEquals(waitCount[0], 1)
self.assertEquals(naps, [])
self.assertEquals(set(handlers),
set([signal.SIGTERM, signal.SIGINT, signal.SIGALRM]))
self.assertEquals(args, ['ls'])