Added waiting method back to pool.Pool that was lost, plus a unit test so it doesn't get lost again.
This commit is contained in:
@@ -84,6 +84,14 @@ class Pool(object):
|
||||
and return the result."""
|
||||
return self.results.wait()
|
||||
|
||||
def waiting(self):
|
||||
"""Return the number of coroutines waiting to execute.
|
||||
"""
|
||||
if self.sem.balance < 0:
|
||||
return -self.sem.balance
|
||||
else:
|
||||
return 0
|
||||
|
||||
def killall(self):
|
||||
""" Kill every running coroutine as immediately as possible."""
|
||||
return self.procs.killall()
|
||||
|
@@ -21,6 +21,30 @@ class TestCoroutinePool(LimitedTestCase):
|
||||
worker = pool.execute(some_work)
|
||||
self.assertEqual(value, worker.wait())
|
||||
|
||||
def test_waiting(self):
|
||||
pool = self.klass(0,1)
|
||||
done = coros.event()
|
||||
def consume():
|
||||
done.wait()
|
||||
def waiter(pool):
|
||||
evt = pool.execute(consume)
|
||||
evt.wait()
|
||||
|
||||
waiters = []
|
||||
waiters.append(coros.execute(waiter, pool))
|
||||
api.sleep(0)
|
||||
self.assertEqual(pool.waiting(), 0)
|
||||
waiters.append(coros.execute(waiter, pool))
|
||||
api.sleep(0)
|
||||
self.assertEqual(pool.waiting(), 1)
|
||||
waiters.append(coros.execute(waiter, pool))
|
||||
api.sleep(0)
|
||||
self.assertEqual(pool.waiting(), 2)
|
||||
done.send(None)
|
||||
for w in waiters:
|
||||
w.wait()
|
||||
self.assertEqual(pool.waiting(), 0)
|
||||
|
||||
def test_multiple_coros(self):
|
||||
evt = coros.event()
|
||||
results = []
|
||||
|
Reference in New Issue
Block a user