Fix test_fork_handler

jupyterlab · Nov 29, 2024 · 47d71f8 · 47d71f8
1 parent 7a5613f
commit 47d71f8
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_handlers.py b/tests/test_handlers.py
@@ -252,9 +252,8 @@ def _on_root_change(topic: str, event: Any) -> None:
     file_id = data["fileId"]
     root_roomid = f"text:file:{file_id}"
 
-    async with await rtc_connect_doc_client("text", "file", path) as ws, WebsocketProvider(
-        root_ydoc.ydoc, ws
-    ):
+    websocket, room_name = await rtc_connect_doc_client("text", "file", path)
+    async with websocket as ws, WebsocketProvider(root_ydoc.ydoc, Websocket(ws, room_name)):
         await root_connect_event.wait()
 
         resp = await rtc_create_fork_client(root_roomid, False, "my fork0", "is awesome0")