From e9957ddb35f1f3c8569506078a90851daefe289b Mon Sep 17 00:00:00 2001 From: Min RK Date: Mon, 14 Mar 2016 16:49:25 +0100 Subject: [PATCH] Handle coroutine possibility in test_sessionmanager --- notebook/services/sessions/tests/test_sessionmanager.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/notebook/services/sessions/tests/test_sessionmanager.py b/notebook/services/sessions/tests/test_sessionmanager.py index 0b31d97a4..14bda028b 100644 --- a/notebook/services/sessions/tests/test_sessionmanager.py +++ b/notebook/services/sessions/tests/test_sessionmanager.py @@ -175,6 +175,8 @@ class TestSessionManager(TestCase): # try to delete a session that doesn't exist ~ raise error sm = self.sm self.create_session(path='/path/to/test.ipynb', kernel_name='python') - self.assertRaises(TypeError, sm.delete_session, bad_kwarg='23424') # Bad keyword - self.assertRaises(web.HTTPError, sm.delete_session, session_id='23424') # nonexistant + with self.assertRaises(TypeError): + self.loop.run_sync(lambda : sm.delete_session(bad_kwarg='23424')) # Bad keyword + with self.assertRaises(web.HTTPError): + self.loop.run_sync(lambda : sm.delete_session(session_id='23424')) # nonexistent