test__api.py: update test for spawn_later

ce284562 · Denis Bilenko · dd96ee31 · ce284562
Commit ce284562 authored Aug 10, 2009 by Denis Bilenko
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 5 deletions

greentest/test__api.py greentest/test__api.py +8 -5

No files found.
--- a/greentest/test__api.py
+++ b/greentest/test__api.py
@@ -53,11 +53,14 @@ class Test(greentest.TestCase):
    def test_killing_not_yet_started(self):
        def func():
            pass
-        g = gevent.spawn_later(2000, func)
+        g, timer = gevent.spawn_later(2000, func)
-        assert not g.dead, g
+        try:
-        g.kill(block=True)
+            assert not g.dead, g
-        assert g.dead, g
+            g.kill(block=True)
-        self.assertRaises(Exception, g.get)
+            assert g.dead, g
+            self.assertRaises(Exception, g.get)
+        finally:
+            timer.cancel()
    def test_sleep_invalid_switch(self):
        p = gevent.spawn(util.wrap_errors(AssertionError, gevent.sleep), 2)