Commit 24d931d4 authored by Denis Bilenko's avatar Denis Bilenko

test__pool.py: do not leak greenlets out of test case

--HG--
extra : transplant_source : %A2c%7CPD%F0%DC%2C9R6%10Ez%9E%0D%C7%D8%CB%DC
parent 9fc098c6
...@@ -225,6 +225,7 @@ class TestPool(greentest.TestCase): ...@@ -225,6 +225,7 @@ class TestPool(greentest.TestCase):
get = TimingWrapper(res.get) get = TimingWrapper(res.get)
self.assertRaises(gevent.Timeout, get, timeout=TIMEOUT2) self.assertRaises(gevent.Timeout, get, timeout=TIMEOUT2)
self.assertAlmostEqual(get.elapsed, TIMEOUT2, 1) self.assertAlmostEqual(get.elapsed, TIMEOUT2, 1)
self.pool.join()
def test_imap(self): def test_imap(self):
it = self.pool.imap(sqr, range(10)) it = self.pool.imap(sqr, range(10))
...@@ -248,10 +249,12 @@ class TestPool(greentest.TestCase): ...@@ -248,10 +249,12 @@ class TestPool(greentest.TestCase):
self.assertEqual(sorted(it), map(sqr, range(1000))) self.assertEqual(sorted(it), map(sqr, range(1000)))
def test_terminate(self): def test_terminate(self):
result = self.pool.map_async(gevent.sleep, [0.1] * 1000) result = self.pool.map_async(gevent.sleep, [0.1] * ((self.size or 10) * 2))
gevent.sleep(0.1)
kill = TimingWrapper(self.pool.kill) kill = TimingWrapper(self.pool.kill)
kill() kill()
assert kill.elapsed < 0.5, kill.elapsed assert kill.elapsed < 0.5, kill.elapsed
result.join()
def sleep(self, x): def sleep(self, x):
gevent.sleep(float(x) / 10.) gevent.sleep(float(x) / 10.)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment