from __future__ import absolute_import
import os
import time
import unittest
import proctest
import signal
import mozunit
from mozprocess import processhandler
here = os.path.dirname(os.path.abspath(__file__))
class ProcTestKill(proctest.ProcTest):
def test_kill_before_run(self):
p = processhandler.ProcessHandler([self.python, '-V'])
self.assertRaises(RuntimeError, p.kill)
def test_process_kill(self):
p = processhandler.ProcessHandler([self.python, self.proclaunch,
"process_normal_finish.ini"],
cwd=here)
p.run()
p.kill()
self.determine_status(p, expectedfail=('returncode',))
def test_process_kill_deep(self):
p = processhandler.ProcessHandler([self.python, self.proclaunch,
"process_normal_deep.ini"],
cwd=here)
p.run()
p.kill()
self.determine_status(p, expectedfail=('returncode',))
def test_process_kill_deep_wait(self):
p = processhandler.ProcessHandler([self.python, self.proclaunch,
"process_normal_deep.ini"],
cwd=here)
p.run()
time.sleep(3)
p.kill()
self.determine_status(p, expectedfail=('returncode',))
def test_process_kill_broad(self):
p = processhandler.ProcessHandler([self.python, self.proclaunch,
"process_normal_broad.ini"],
cwd=here)
p.run()
p.kill()
self.determine_status(p, expectedfail=('returncode',))
def test_process_kill_broad_delayed(self):
p = processhandler.ProcessHandler([self.python, self.proclaunch,
"process_normal_broad.ini"],
cwd=here)
p.run()
time.sleep(3)
p.kill()
self.determine_status(p, expectedfail=('returncode',))
@unittest.skipUnless(processhandler.isPosix, "posix only")
def test_process_kill_with_sigterm(self):
script = os.path.join(here, 'scripts', 'infinite_loop.py')
p = processhandler.ProcessHandler([self.python, script])
p.run()
p.kill()
self.assertEquals(p.proc.returncode, -signal.SIGTERM)
@unittest.skipUnless(processhandler.isPosix, "posix only")
def test_process_kill_with_sigint_if_needed(self):
script = os.path.join(here, 'scripts', 'infinite_loop.py')
p = processhandler.ProcessHandler([self.python, script, 'deadlock'])
p.run()
time.sleep(1)
p.kill()
self.assertEquals(p.proc.returncode, -signal.SIGKILL)
if __name__ == '__main__':
mozunit.main()