diff options
author | Victor Stinner <victor.stinner@gmail.com> | 2014-11-24 14:40:49 +0100 |
---|---|---|
committer | Victor Stinner <victor.stinner@gmail.com> | 2014-11-24 14:40:49 +0100 |
commit | 0058f772bc161d5fcc475a6fcc5008a578126d91 (patch) | |
tree | e6d8f700724f296ebe768f635a87f9e36ceb5ef7 /tests/test_greenlet.py | |
parent | db9eb18c05f8ccbd44945df4cc6e0206419d88b6 (diff) | |
download | aioeventlet-0058f772bc161d5fcc475a6fcc5008a578126d91.tar.gz |
wrap_greenthread(): catch missing run attribute
Add 2 tests: no run attribute, wrap exception.
Diffstat (limited to 'tests/test_greenlet.py')
-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() |