diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_greenlet.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/test_greenlet.py b/tests/test_greenlet.py index 67cde61..8ee8a06 100644 --- a/tests/test_greenlet.py +++ b/tests/test_greenlet.py @@ -14,6 +14,21 @@ class WrapGreenletTests(tests.TestCase): result = self.loop.run_until_complete(fut) self.assertEqual(result, 15) + def test_wrap_greenlet_exc(self): + def func(): + raise ValueError(7) + + gl = greenlet.greenlet(func) + fut = aiogreen.wrap_greenthread(gl) + gl.switch() + self.assertRaises(ValueError, self.loop.run_until_complete, fut) + + def test_wrap_greenlet_no_run_attr(self): + gl = greenlet.greenlet() + msg = "wrap_greenthread: the run attribute of the greenlet is not set" + self.assertRaisesRegexp(RuntimeError, msg, + aiogreen.wrap_greenthread, gl) + def test_wrap_greenlet_running(self): def func(value): gl = greenlet.getcurrent() |