| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 |
- """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
- from test.support import threading_helper
- LONGSLEEP = 2
- SHORTSLEEP = 0.5
- NUM_THREADS = 4
- class ForkWait(unittest.TestCase):
- def setUp(self):
- self._threading_key = threading_helper.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()
- threading_helper.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)
|
