async def ws(websocket: WebSocket, client_id: str):
conn = await manager.connect(websocket, client_id)
try:
while True:
await manager.manage(conn)
except WebSocketDisconnectError:
await manager.disconnect(conn)
Can you help me to figure out how to create the test using test_client for this case?