bpo-40364: asyncio uses os.waitstatus_to_exitcode() (GH-23798)
test_unix_events.py no longer checks if waitstatus_to_exitcode() mock has been called or not to make the test more functional, rather than checking the exact implementation.
This commit is contained in:
parent
5f0fe8ec70
commit
99d28c5670
|
@ -44,6 +44,16 @@ def _sighandler_noop(signum, frame):
|
|||
pass
|
||||
|
||||
|
||||
def waitstatus_to_exitcode(status):
|
||||
try:
|
||||
return os.waitstatus_to_exitcode(status)
|
||||
except ValueError:
|
||||
# The child exited, but we don't understand its status.
|
||||
# This shouldn't happen, but if it does, let's just
|
||||
# return that status; perhaps that helps debug it.
|
||||
return status
|
||||
|
||||
|
||||
class _UnixSelectorEventLoop(selector_events.BaseSelectorEventLoop):
|
||||
"""Unix event loop.
|
||||
|
||||
|
@ -941,7 +951,7 @@ class PidfdChildWatcher(AbstractChildWatcher):
|
|||
" will report returncode 255",
|
||||
pid)
|
||||
else:
|
||||
returncode = _compute_returncode(status)
|
||||
returncode = waitstatus_to_exitcode(status)
|
||||
|
||||
os.close(pidfd)
|
||||
callback(pid, returncode, *args)
|
||||
|
@ -956,20 +966,6 @@ class PidfdChildWatcher(AbstractChildWatcher):
|
|||
return True
|
||||
|
||||
|
||||
def _compute_returncode(status):
|
||||
if os.WIFSIGNALED(status):
|
||||
# The child process died because of a signal.
|
||||
return -os.WTERMSIG(status)
|
||||
elif os.WIFEXITED(status):
|
||||
# The child process exited (e.g sys.exit()).
|
||||
return os.WEXITSTATUS(status)
|
||||
else:
|
||||
# The child exited, but we don't understand its status.
|
||||
# This shouldn't happen, but if it does, let's just
|
||||
# return that status; perhaps that helps debug it.
|
||||
return status
|
||||
|
||||
|
||||
class BaseChildWatcher(AbstractChildWatcher):
|
||||
|
||||
def __init__(self):
|
||||
|
@ -1080,7 +1076,7 @@ class SafeChildWatcher(BaseChildWatcher):
|
|||
# The child process is still alive.
|
||||
return
|
||||
|
||||
returncode = _compute_returncode(status)
|
||||
returncode = waitstatus_to_exitcode(status)
|
||||
if self._loop.get_debug():
|
||||
logger.debug('process %s exited with returncode %s',
|
||||
expected_pid, returncode)
|
||||
|
@ -1173,7 +1169,7 @@ class FastChildWatcher(BaseChildWatcher):
|
|||
# A child process is still alive.
|
||||
return
|
||||
|
||||
returncode = _compute_returncode(status)
|
||||
returncode = waitstatus_to_exitcode(status)
|
||||
|
||||
with self._lock:
|
||||
try:
|
||||
|
@ -1296,7 +1292,7 @@ class MultiLoopChildWatcher(AbstractChildWatcher):
|
|||
# The child process is still alive.
|
||||
return
|
||||
|
||||
returncode = _compute_returncode(status)
|
||||
returncode = waitstatus_to_exitcode(status)
|
||||
debug_log = True
|
||||
try:
|
||||
loop, callback, args = self._callbacks.pop(pid)
|
||||
|
@ -1399,7 +1395,7 @@ class ThreadedChildWatcher(AbstractChildWatcher):
|
|||
"Unknown child process pid %d, will report returncode 255",
|
||||
pid)
|
||||
else:
|
||||
returncode = _compute_returncode(status)
|
||||
returncode = waitstatus_to_exitcode(status)
|
||||
if loop.get_debug():
|
||||
logger.debug('process %s exited with returncode %s',
|
||||
expected_pid, returncode)
|
||||
|
|
|
@ -1,6 +1,5 @@
|
|||
"""Tests for unix_events.py."""
|
||||
|
||||
import collections
|
||||
import contextlib
|
||||
import errno
|
||||
import io
|
||||
|
@ -30,6 +29,15 @@ from test.test_asyncio import utils as test_utils
|
|||
MOCK_ANY = mock.ANY
|
||||
|
||||
|
||||
def EXITCODE(exitcode):
|
||||
return 32768 + exitcode
|
||||
|
||||
|
||||
def SIGNAL(signum):
|
||||
assert 1 <= signum <= 68
|
||||
return 32768 - signum
|
||||
|
||||
|
||||
def tearDownModule():
|
||||
asyncio.set_event_loop_policy(None)
|
||||
|
||||
|
@ -1125,15 +1133,6 @@ class BaseChildWatcherTests(unittest.TestCase):
|
|||
NotImplementedError, watcher._do_waitpid, f)
|
||||
|
||||
|
||||
WaitPidMocks = collections.namedtuple("WaitPidMocks",
|
||||
("waitpid",
|
||||
"WIFEXITED",
|
||||
"WIFSIGNALED",
|
||||
"WEXITSTATUS",
|
||||
"WTERMSIG",
|
||||
))
|
||||
|
||||
|
||||
class ChildWatcherTestsMixin:
|
||||
|
||||
ignore_warnings = mock.patch.object(log.logger, "warning")
|
||||
|
@ -1164,22 +1163,16 @@ class ChildWatcherTestsMixin:
|
|||
else:
|
||||
raise ChildProcessError()
|
||||
|
||||
def add_zombie(self, pid, returncode):
|
||||
self.zombies[pid] = returncode + 32768
|
||||
def add_zombie(self, pid, status):
|
||||
self.zombies[pid] = status
|
||||
|
||||
def WIFEXITED(self, status):
|
||||
return status >= 32768
|
||||
|
||||
def WIFSIGNALED(self, status):
|
||||
return 32700 < status < 32768
|
||||
|
||||
def WEXITSTATUS(self, status):
|
||||
self.assertTrue(self.WIFEXITED(status))
|
||||
return status - 32768
|
||||
|
||||
def WTERMSIG(self, status):
|
||||
self.assertTrue(self.WIFSIGNALED(status))
|
||||
return 32768 - status
|
||||
def waitstatus_to_exitcode(self, status):
|
||||
if status > 32768:
|
||||
return status - 32768
|
||||
elif 32700 < status < 32768:
|
||||
return status - 32768
|
||||
else:
|
||||
return status
|
||||
|
||||
def test_create_watcher(self):
|
||||
self.m_add_signal_handler.assert_called_once_with(
|
||||
|
@ -1191,19 +1184,13 @@ class ChildWatcherTestsMixin:
|
|||
return mock.patch(target, wraps=wrapper,
|
||||
new_callable=mock.Mock)
|
||||
|
||||
with patch('os.WTERMSIG', self.WTERMSIG) as m_WTERMSIG, \
|
||||
patch('os.WEXITSTATUS', self.WEXITSTATUS) as m_WEXITSTATUS, \
|
||||
patch('os.WIFSIGNALED', self.WIFSIGNALED) as m_WIFSIGNALED, \
|
||||
patch('os.WIFEXITED', self.WIFEXITED) as m_WIFEXITED, \
|
||||
with patch('asyncio.unix_events.waitstatus_to_exitcode', self.waitstatus_to_exitcode), \
|
||||
patch('os.waitpid', self.waitpid) as m_waitpid:
|
||||
func(self, WaitPidMocks(m_waitpid,
|
||||
m_WIFEXITED, m_WIFSIGNALED,
|
||||
m_WEXITSTATUS, m_WTERMSIG,
|
||||
))
|
||||
func(self, m_waitpid)
|
||||
return wrapped_func
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld(self, m):
|
||||
def test_sigchld(self, m_waitpid):
|
||||
# register a child
|
||||
callback = mock.Mock()
|
||||
|
||||
|
@ -1212,59 +1199,36 @@ class ChildWatcherTestsMixin:
|
|||
self.watcher.add_child_handler(42, callback, 9, 10, 14)
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# child is running
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# child terminates (returncode 12)
|
||||
self.running = False
|
||||
self.add_zombie(42, 12)
|
||||
self.add_zombie(42, EXITCODE(12))
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertTrue(m.WIFEXITED.called)
|
||||
self.assertTrue(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
callback.assert_called_once_with(42, 12, 9, 10, 14)
|
||||
|
||||
m.WIFSIGNALED.reset_mock()
|
||||
m.WIFEXITED.reset_mock()
|
||||
m.WEXITSTATUS.reset_mock()
|
||||
callback.reset_mock()
|
||||
|
||||
# ensure that the child is effectively reaped
|
||||
self.add_zombie(42, 13)
|
||||
self.add_zombie(42, EXITCODE(13))
|
||||
with self.ignore_warnings:
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
m.WIFSIGNALED.reset_mock()
|
||||
m.WIFEXITED.reset_mock()
|
||||
m.WEXITSTATUS.reset_mock()
|
||||
|
||||
# sigchld called again
|
||||
self.zombies.clear()
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld_two_children(self, m):
|
||||
def test_sigchld_two_children(self, m_waitpid):
|
||||
callback1 = mock.Mock()
|
||||
callback2 = mock.Mock()
|
||||
|
||||
|
@ -1275,10 +1239,6 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# register child 2
|
||||
with self.watcher:
|
||||
|
@ -1286,34 +1246,20 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# children are running
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# child 1 terminates (signal 3)
|
||||
self.add_zombie(43, -3)
|
||||
self.add_zombie(43, SIGNAL(3))
|
||||
self.watcher._sig_chld()
|
||||
|
||||
callback1.assert_called_once_with(43, -3, 7, 8)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertTrue(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertTrue(m.WTERMSIG.called)
|
||||
|
||||
m.WIFSIGNALED.reset_mock()
|
||||
m.WIFEXITED.reset_mock()
|
||||
m.WTERMSIG.reset_mock()
|
||||
callback1.reset_mock()
|
||||
|
||||
# child 2 still running
|
||||
|
@ -1321,40 +1267,25 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# child 2 terminates (code 108)
|
||||
self.add_zombie(44, 108)
|
||||
self.add_zombie(44, EXITCODE(108))
|
||||
self.running = False
|
||||
self.watcher._sig_chld()
|
||||
|
||||
callback2.assert_called_once_with(44, 108, 147, 18)
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertTrue(m.WIFEXITED.called)
|
||||
self.assertTrue(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
m.WIFSIGNALED.reset_mock()
|
||||
m.WIFEXITED.reset_mock()
|
||||
m.WEXITSTATUS.reset_mock()
|
||||
callback2.reset_mock()
|
||||
|
||||
# ensure that the children are effectively reaped
|
||||
self.add_zombie(43, 14)
|
||||
self.add_zombie(44, 15)
|
||||
self.add_zombie(43, EXITCODE(14))
|
||||
self.add_zombie(44, EXITCODE(15))
|
||||
with self.ignore_warnings:
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
m.WIFSIGNALED.reset_mock()
|
||||
m.WIFEXITED.reset_mock()
|
||||
m.WEXITSTATUS.reset_mock()
|
||||
|
||||
# sigchld called again
|
||||
self.zombies.clear()
|
||||
|
@ -1362,13 +1293,9 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld_two_children_terminating_together(self, m):
|
||||
def test_sigchld_two_children_terminating_together(self, m_waitpid):
|
||||
callback1 = mock.Mock()
|
||||
callback2 = mock.Mock()
|
||||
|
||||
|
@ -1379,10 +1306,6 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# register child 2
|
||||
with self.watcher:
|
||||
|
@ -1390,60 +1313,43 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# children are running
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# child 1 terminates (code 78)
|
||||
# child 2 terminates (signal 5)
|
||||
self.add_zombie(45, 78)
|
||||
self.add_zombie(46, -5)
|
||||
self.add_zombie(45, EXITCODE(78))
|
||||
self.add_zombie(46, SIGNAL(5))
|
||||
self.running = False
|
||||
self.watcher._sig_chld()
|
||||
|
||||
callback1.assert_called_once_with(45, 78, 17, 8)
|
||||
callback2.assert_called_once_with(46, -5, 1147, 18)
|
||||
self.assertTrue(m.WIFSIGNALED.called)
|
||||
self.assertTrue(m.WIFEXITED.called)
|
||||
self.assertTrue(m.WEXITSTATUS.called)
|
||||
self.assertTrue(m.WTERMSIG.called)
|
||||
|
||||
m.WIFSIGNALED.reset_mock()
|
||||
m.WIFEXITED.reset_mock()
|
||||
m.WTERMSIG.reset_mock()
|
||||
m.WEXITSTATUS.reset_mock()
|
||||
callback1.reset_mock()
|
||||
callback2.reset_mock()
|
||||
|
||||
# ensure that the children are effectively reaped
|
||||
self.add_zombie(45, 14)
|
||||
self.add_zombie(46, 15)
|
||||
self.add_zombie(45, EXITCODE(14))
|
||||
self.add_zombie(46, EXITCODE(15))
|
||||
with self.ignore_warnings:
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld_race_condition(self, m):
|
||||
def test_sigchld_race_condition(self, m_waitpid):
|
||||
# register a child
|
||||
callback = mock.Mock()
|
||||
|
||||
with self.watcher:
|
||||
# child terminates before being registered
|
||||
self.add_zombie(50, 4)
|
||||
self.add_zombie(50, EXITCODE(4))
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.watcher.add_child_handler(50, callback, 1, 12)
|
||||
|
@ -1452,14 +1358,14 @@ class ChildWatcherTestsMixin:
|
|||
callback.reset_mock()
|
||||
|
||||
# ensure that the child is effectively reaped
|
||||
self.add_zombie(50, -1)
|
||||
self.add_zombie(50, SIGNAL(1))
|
||||
with self.ignore_warnings:
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld_replace_handler(self, m):
|
||||
def test_sigchld_replace_handler(self, m_waitpid):
|
||||
callback1 = mock.Mock()
|
||||
callback2 = mock.Mock()
|
||||
|
||||
|
@ -1470,10 +1376,6 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# register the same child again
|
||||
with self.watcher:
|
||||
|
@ -1481,38 +1383,27 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# child terminates (signal 8)
|
||||
self.running = False
|
||||
self.add_zombie(51, -8)
|
||||
self.add_zombie(51, SIGNAL(8))
|
||||
self.watcher._sig_chld()
|
||||
|
||||
callback2.assert_called_once_with(51, -8, 21)
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertTrue(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertTrue(m.WTERMSIG.called)
|
||||
|
||||
m.WIFSIGNALED.reset_mock()
|
||||
m.WIFEXITED.reset_mock()
|
||||
m.WTERMSIG.reset_mock()
|
||||
callback2.reset_mock()
|
||||
|
||||
# ensure that the child is effectively reaped
|
||||
self.add_zombie(51, 13)
|
||||
self.add_zombie(51, EXITCODE(13))
|
||||
with self.ignore_warnings:
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback1.called)
|
||||
self.assertFalse(callback2.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld_remove_handler(self, m):
|
||||
def test_sigchld_remove_handler(self, m_waitpid):
|
||||
callback = mock.Mock()
|
||||
|
||||
# register a child
|
||||
|
@ -1521,30 +1412,22 @@ class ChildWatcherTestsMixin:
|
|||
self.watcher.add_child_handler(52, callback, 1984)
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# unregister the child
|
||||
self.watcher.remove_child_handler(52)
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# child terminates (code 99)
|
||||
self.running = False
|
||||
self.add_zombie(52, 99)
|
||||
self.add_zombie(52, EXITCODE(99))
|
||||
with self.ignore_warnings:
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld_unknown_status(self, m):
|
||||
def test_sigchld_unknown_status(self, m_waitpid):
|
||||
callback = mock.Mock()
|
||||
|
||||
# register a child
|
||||
|
@ -1553,10 +1436,6 @@ class ChildWatcherTestsMixin:
|
|||
self.watcher.add_child_handler(53, callback, -19)
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# terminate with unknown status
|
||||
self.zombies[53] = 1178
|
||||
|
@ -1564,24 +1443,18 @@ class ChildWatcherTestsMixin:
|
|||
self.watcher._sig_chld()
|
||||
|
||||
callback.assert_called_once_with(53, 1178, -19)
|
||||
self.assertTrue(m.WIFEXITED.called)
|
||||
self.assertTrue(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
callback.reset_mock()
|
||||
m.WIFEXITED.reset_mock()
|
||||
m.WIFSIGNALED.reset_mock()
|
||||
|
||||
# ensure that the child is effectively reaped
|
||||
self.add_zombie(53, 101)
|
||||
self.add_zombie(53, EXITCODE(101))
|
||||
with self.ignore_warnings:
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_remove_child_handler(self, m):
|
||||
def test_remove_child_handler(self, m_waitpid):
|
||||
callback1 = mock.Mock()
|
||||
callback2 = mock.Mock()
|
||||
callback3 = mock.Mock()
|
||||
|
@ -1602,9 +1475,9 @@ class ChildWatcherTestsMixin:
|
|||
self.assertFalse(self.watcher.remove_child_handler(55))
|
||||
|
||||
# all children terminate
|
||||
self.add_zombie(54, 0)
|
||||
self.add_zombie(55, 1)
|
||||
self.add_zombie(56, 2)
|
||||
self.add_zombie(54, EXITCODE(0))
|
||||
self.add_zombie(55, EXITCODE(1))
|
||||
self.add_zombie(56, EXITCODE(2))
|
||||
self.running = False
|
||||
with self.ignore_warnings:
|
||||
self.watcher._sig_chld()
|
||||
|
@ -1614,7 +1487,7 @@ class ChildWatcherTestsMixin:
|
|||
callback3.assert_called_once_with(56, 2, 3)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld_unhandled_exception(self, m):
|
||||
def test_sigchld_unhandled_exception(self, m_waitpid):
|
||||
callback = mock.Mock()
|
||||
|
||||
# register a child
|
||||
|
@ -1623,7 +1496,7 @@ class ChildWatcherTestsMixin:
|
|||
self.watcher.add_child_handler(57, callback)
|
||||
|
||||
# raise an exception
|
||||
m.waitpid.side_effect = ValueError
|
||||
m_waitpid.side_effect = ValueError
|
||||
|
||||
with mock.patch.object(log.logger,
|
||||
'error') as m_error:
|
||||
|
@ -1632,7 +1505,7 @@ class ChildWatcherTestsMixin:
|
|||
self.assertTrue(m_error.called)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld_child_reaped_elsewhere(self, m):
|
||||
def test_sigchld_child_reaped_elsewhere(self, m_waitpid):
|
||||
# register a child
|
||||
callback = mock.Mock()
|
||||
|
||||
|
@ -1641,19 +1514,15 @@ class ChildWatcherTestsMixin:
|
|||
self.watcher.add_child_handler(58, callback)
|
||||
|
||||
self.assertFalse(callback.called)
|
||||
self.assertFalse(m.WIFEXITED.called)
|
||||
self.assertFalse(m.WIFSIGNALED.called)
|
||||
self.assertFalse(m.WEXITSTATUS.called)
|
||||
self.assertFalse(m.WTERMSIG.called)
|
||||
|
||||
# child terminates
|
||||
self.running = False
|
||||
self.add_zombie(58, 4)
|
||||
self.add_zombie(58, EXITCODE(4))
|
||||
|
||||
# waitpid is called elsewhere
|
||||
os.waitpid(58, os.WNOHANG)
|
||||
|
||||
m.waitpid.reset_mock()
|
||||
m_waitpid.reset_mock()
|
||||
|
||||
# sigchld
|
||||
with self.ignore_warnings:
|
||||
|
@ -1667,7 +1536,7 @@ class ChildWatcherTestsMixin:
|
|||
callback.assert_called_once_with(58, 255)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_sigchld_unknown_pid_during_registration(self, m):
|
||||
def test_sigchld_unknown_pid_during_registration(self, m_waitpid):
|
||||
# register two children
|
||||
callback1 = mock.Mock()
|
||||
callback2 = mock.Mock()
|
||||
|
@ -1675,9 +1544,9 @@ class ChildWatcherTestsMixin:
|
|||
with self.ignore_warnings, self.watcher:
|
||||
self.running = True
|
||||
# child 1 terminates
|
||||
self.add_zombie(591, 7)
|
||||
self.add_zombie(591, EXITCODE(7))
|
||||
# an unknown child terminates
|
||||
self.add_zombie(593, 17)
|
||||
self.add_zombie(593, EXITCODE(17))
|
||||
|
||||
self.watcher._sig_chld()
|
||||
|
||||
|
@ -1688,7 +1557,7 @@ class ChildWatcherTestsMixin:
|
|||
self.assertFalse(callback2.called)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_set_loop(self, m):
|
||||
def test_set_loop(self, m_waitpid):
|
||||
# register a child
|
||||
callback = mock.Mock()
|
||||
|
||||
|
@ -1713,13 +1582,13 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
# child terminates
|
||||
self.running = False
|
||||
self.add_zombie(60, 9)
|
||||
self.add_zombie(60, EXITCODE(9))
|
||||
self.watcher._sig_chld()
|
||||
|
||||
callback.assert_called_once_with(60, 9)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_set_loop_race_condition(self, m):
|
||||
def test_set_loop_race_condition(self, m_waitpid):
|
||||
# register 3 children
|
||||
callback1 = mock.Mock()
|
||||
callback2 = mock.Mock()
|
||||
|
@ -1746,8 +1615,8 @@ class ChildWatcherTestsMixin:
|
|||
signal.SIGCHLD)
|
||||
|
||||
# child 1 & 2 terminate
|
||||
self.add_zombie(61, 11)
|
||||
self.add_zombie(62, -5)
|
||||
self.add_zombie(61, EXITCODE(11))
|
||||
self.add_zombie(62, SIGNAL(5))
|
||||
|
||||
# SIGCHLD was not caught
|
||||
self.assertFalse(callback1.called)
|
||||
|
@ -1773,7 +1642,7 @@ class ChildWatcherTestsMixin:
|
|||
|
||||
# child 3 terminates
|
||||
self.running = False
|
||||
self.add_zombie(622, 19)
|
||||
self.add_zombie(622, EXITCODE(19))
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.assertFalse(callback1.called)
|
||||
|
@ -1781,16 +1650,16 @@ class ChildWatcherTestsMixin:
|
|||
callback3.assert_called_once_with(622, 19)
|
||||
|
||||
@waitpid_mocks
|
||||
def test_close(self, m):
|
||||
def test_close(self, m_waitpid):
|
||||
# register two children
|
||||
callback1 = mock.Mock()
|
||||
|
||||
with self.watcher:
|
||||
self.running = True
|
||||
# child 1 terminates
|
||||
self.add_zombie(63, 9)
|
||||
self.add_zombie(63, EXITCODE(9))
|
||||
# other child terminates
|
||||
self.add_zombie(65, 18)
|
||||
self.add_zombie(65, EXITCODE(18))
|
||||
self.watcher._sig_chld()
|
||||
|
||||
self.watcher.add_child_handler(63, callback1)
|
||||
|
|
Loading…
Reference in New Issue