diff --git a/services/web/server/tests/integration/01/test_garbage_collection.py b/services/web/server/tests/integration/01/test_garbage_collection.py index 7324b5ca59e9..4185a1848539 100644 --- a/services/web/server/tests/integration/01/test_garbage_collection.py +++ b/services/web/server/tests/integration/01/test_garbage_collection.py @@ -279,8 +279,12 @@ async def connect_to_socketio(client, user, socketio_client_factory: Callable): "user_id": str(user["id"]), "client_session_id": cur_client_session_id, } - assert await socket_registry.find_keys(("socket_id", sio.sid)) == [resource_key] - assert sio.sid in await socket_registry.find_resources(resource_key, "socket_id") + assert await socket_registry.find_keys(("socket_id", sio.get_sid())) == [ + resource_key + ] + assert sio.get_sid() in await socket_registry.find_resources( + resource_key, "socket_id" + ) assert len(await socket_registry.find_resources(resource_key, "socket_id")) == 1 sio_connection_data = sio, resource_key return sio_connection_data @@ -289,12 +293,12 @@ async def connect_to_socketio(client, user, socketio_client_factory: Callable): async def disconnect_user_from_socketio(client, sio_connection_data): """disconnect a previously connected socket.io connection""" sio, resource_key = sio_connection_data - sid = sio.sid + sid = sio.get_sid() socket_registry = get_registry(client.server.app) await sio.disconnect() assert not sio.sid await asyncio.sleep(0) # just to ensure there is a context switch - assert not await socket_registry.find_keys(("socket_id", sio.sid)) + assert not await socket_registry.find_keys(("socket_id", sio.get_sid())) assert not sid in await socket_registry.find_resources(resource_key, "socket_id") assert not await socket_registry.find_resources(resource_key, "socket_id")