handle KeyError when get session (#641)

* handle KeyError when get session

* Run maintainer script

by blink1073

["", "pre-commit run --all-files"]

Co-authored-by: GitHub Action <41898282+github-actions[bot]@users.noreply.github.com>

Author: ccw630

jupyter_server/services/sessions/sessionmanager.py

@@ -216,7 +216,10 @@ async def get_session(self, **kwargs):
 
             raise web.HTTPError(404, u"Session not found: %s" % (", ".join(q)))
 
-        model = await self.row_to_model(row)
+        try:
+            model = await self.row_to_model(row)
+        except KeyError as e:
+            raise web.HTTPError(404, u"Session not found: %s" % str(e))
         return model
 
     async def update_session(self, session_id, **kwargs):

jupyter_server/tests/services/sessions/test_manager.py

@@ -91,7 +91,7 @@ async def test_get_session_dead_kernel(session_manager):
     )
     # Kill the kernel
     await session_manager.kernel_manager.shutdown_kernel(session["kernel"]["id"])
-    with pytest.raises(KeyError):
+    with pytest.raises(web.HTTPError):
         await session_manager.get_session(session_id=session["id"])
     # no session left
     listed = await session_manager.list_sessions()