83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
|
"""This test case provides support for checking forking and wait behavior.
|
||
|
|
|
||
|
|
To test different wait behavior, override the wait_impl method.
|
||
|
|
|
||
|
|
We want fork1() semantics -- only the forking thread survives in the
|
||
|
|
child after a fork().
|
||
|
|
|
||
|
|
On some systems (e.g. Solaris without posix threads) we find that all
|
||
|
|
active threads survive in the child after a fork(); this is an error.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os, sys, time, unittest
|
||
|
|
import threading
|
||
|
|
from test import support
|
||
|
|
|
||
|
|
|
||
|
|
LONGSLEEP = 2
|
||
|
|
SHORTSLEEP = 0.5
|
||
|
|
NUM_THREADS = 4
|
||
|
|
|
||
|
|
class ForkWait(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self._threading_key = support.threading_setup()
|
||
|
|
self.alive = {}
|
||
|
|
self.stop = 0
|
||
|
|
self.threads = []
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
# Stop threads
|
||
|
|
self.stop = 1
|
||
|
|
for thread in self.threads:
|
||
|
|
thread.join()
|
||
|
|
thread = None
|
||
|
|
self.threads.clear()
|
||
|
|
support.threading_cleanup(*self._threading_key)
|
||
|
|
|
||
|
|
def f(self, id):
|
||
|
|
while not self.stop:
|
||
|
|
self.alive[id] = os.getpid()
|
||
|
|
try:
|
||
|
|
time.sleep(SHORTSLEEP)
|
||
|
|
except OSError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def wait_impl(self, cpid, *, exitcode):
|
||
|
|
support.wait_process(cpid, exitcode=exitcode)
|
||
|
|
|
||
|
|
def test_wait(self):
|
||
|
|
for i in range(NUM_THREADS):
|
||
|
|
thread = threading.Thread(target=self.f, args=(i,))
|
||
|
|
thread.start()
|
||
|
|
self.threads.append(thread)
|
||
|
|
|
||
|
|
# busy-loop to wait for threads
|
||
|
|
deadline = time.monotonic() + support.SHORT_TIMEOUT
|
||
|
|
while len(self.alive) < NUM_THREADS:
|
||
|
|
time.sleep(0.1)
|
||
|
|
if deadline < time.monotonic():
|
||
|
|
break
|
||
|
|
|
||
|
|
a = sorted(self.alive.keys())
|
||
|
|
self.assertEqual(a, list(range(NUM_THREADS)))
|
||
|
|
|
||
|
|
prefork_lives = self.alive.copy()
|
||
|
|
|
||
|
|
if sys.platform in ['unixware7']:
|
||
|
|
cpid = os.fork1()
|
||
|
|
else:
|
||
|
|
cpid = os.fork()
|
||
|
|
|
||
|
|
if cpid == 0:
|
||
|
|
# Child
|
||
|
|
time.sleep(LONGSLEEP)
|
||
|
|
n = 0
|
||
|
|
for key in self.alive:
|
||
|
|
if self.alive[key] != prefork_lives[key]:
|
||
|
|
n += 1
|
||
|
|
os._exit(n)
|
||
|
|
else:
|
||
|
|
# Parent
|
||
|
|
self.wait_impl(cpid, exitcode=0)
|