Issue #10332: multiprocessing: fix a race condition when a Pool is closed

before all tasks have completed.
This commit is contained in:
Charles-François Natali 2011-10-24 18:43:51 +02:00
parent 22dabb6ffa
commit 46f990e58c
3 changed files with 22 additions and 1 deletions

View file

@ -1168,6 +1168,20 @@ class _TestPoolWorkerLifetime(BaseTestCase):
p.close()
p.join()
def test_pool_worker_lifetime_early_close(self):
# Issue #10332: closing a pool whose workers have limited lifetimes
# before all the tasks completed would make join() hang.
p = multiprocessing.Pool(3, maxtasksperchild=1)
results = []
for i in range(6):
results.append(p.apply_async(sqr, (i, 0.3)))
p.close()
p.join()
# check the results
for (j, res) in enumerate(results):
self.assertEqual(res.get(), sqr(j))
#
# Test that manager has expected number of shared objects left
#