Merge pull request #47 from kkroening/fix_hanging_select_test

Fix #45 test_raises_deadlock_if_no_goroutines hang due to zombie tasklet. Adds a tearDown check for zombies.
rgalanakis · Jun 9, 2017 · a75c5e8 · a75c5e8
2 parents b47c598 + e704d8c
commit a75c5e8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -20,6 +20,6 @@ class BaseTests(unittest.TestCase):
     def setUp(self):
         be.yield_()
 
-        def doyield():
-            be.yield_()
-        self.addCleanup(doyield)
+    def tearDown(self):
+        be.yield_()
+        self.assertTrue(be.would_deadlock(), 'Task(s) still running after test finished')
diff --git a/tests/test_select.py b/tests/test_select.py
@@ -53,6 +53,8 @@ def assert_default_readiness():
         assert_default_readiness()
         be.run(self.ch.send)
         self.assertFalse(self.ca.ready())
+        be.run(self.ch.recv)
+        assert_default_readiness()
 
     def test_executes(self):
         def recv():