Fixed handler, and unit tests

Author: 3coins

jupyter_ai_router/extension.py

@@ -99,7 +99,7 @@ async def _get_chat(self, room_id: str):
             self.log.error(f"Error getting chat document for {room_id}: {e}")
             return None
 
-    def stop_extension(self):
+    async def stop_extension(self):
         """Clean up router when extension stops."""
         try:
             if hasattr(self, "router"):

jupyter_ai_router/tests/test_handlers.py

@@ -5,9 +5,15 @@
 and does not expose HTTP endpoints.
 """
 
-def test_no_handlers():
-    """Test that extension loads without HTTP handlers."""
-    from jupyter_ai_router.extension import RouterExtension
-    
-    ext = RouterExtension()
-    assert ext.handlers == []
+import json
+
+async def test_get_health(jp_fetch):
+    # When
+    response = await jp_fetch("jupyter-ai-router", "health")
+
+    # Then
+    assert response.code == 200
+    payload = json.loads(response.body)
+    assert payload == {
+        "data": "JupyterLab extension @jupyter-ai/router is activated!"
+    }