diff options
Diffstat (limited to 'tests/test_eventlet.py')
| -rw-r--r-- | tests/test_eventlet.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/test_eventlet.py b/tests/test_eventlet.py index 8105cf2..07dde9d 100644 --- a/tests/test_eventlet.py +++ b/tests/test_eventlet.py @@ -187,12 +187,26 @@ class WrapGreenthreadTests(tests.TestCase): def func(): return aiogreen.wrap_greenthread(gt) + self.loop.set_debug(False) gt = eventlet.spawn(func) fut1 = aiogreen.wrap_greenthread(gt) fut2 = self.loop.run_until_complete(fut1) fut3 = self.loop.run_until_complete(fut2) self.assertIs(fut3, fut2) + @tests.mock.patch('aiogreen.logger') + def test_wrap_greenthread_running_log(self, m_log): + def func(): + return aiogreen.wrap_greenthread(gt) + + self.loop.set_debug(True) + gt = eventlet.spawn(func) + fut1 = aiogreen.wrap_greenthread(gt) + fut2 = self.loop.run_until_complete(fut1) + m_log.warning.assert_called_with("wrap_greenthread() called on " + "a running greenthread") + + def test_wrap_greenthread_dead(self): def func(): return 'ok' @@ -218,6 +232,7 @@ class WrapGreenthreadTests(tests.TestCase): def coro_func(): pass coro_obj = coro_func() + self.addCleanup(coro_obj.close) self.assertRaises(TypeError, aiogreen.wrap_greenthread, coro_obj) |