handle KeyError when get session (#6245)

Author: ccw630

notebook/services/sessions/sessionmanager.py

@@ -189,7 +189,10 @@ def get_session(self, **kwargs):
 
             raise web.HTTPError(404, u'Session not found: %s' % (', '.join(q)))
 
-        model = yield maybe_future(self.row_to_model(row))
+        try:
+            model = yield maybe_future(self.row_to_model(row))
+        except KeyError as e:
+            raise web.HTTPError(404, u'Session not found: %s' % str(e))
         raise gen.Return(model)
 
     @gen.coroutine

notebook/services/sessions/tests/test_sessionmanager.py

@@ -94,7 +94,7 @@ def test_get_session_dead_kernel(self):
         session = self.create_session(path='/path/to/1/test1.ipynb', kernel_name='python')
         # kill the kernel
         sm.kernel_manager.shutdown_kernel(session['kernel']['id'])
-        with self.assertRaises(KeyError):
+        with self.assertRaises(web.HTTPError):
             self.loop.run_sync(lambda: sm.get_session(session_id=session['id']))
         # no sessions left
         listed = self.loop.run_sync(lambda: sm.list_sessions())