In debug mode, wrap_greenthread() now logs a warning if the greenthead is

running

2 files changed, 19 insertions, 0 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
index 5843e5f..42a7e20 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -8,6 +8,10 @@ try:
     import unittest2 as unittest
 except ImportError:
     import unittest
+try:
+    from unittest.mock import mock
+except ImportError:
+    import mock
 
 class TestCase(unittest.TestCase):
     def setUp(self):
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)