replace test that ping only

This commit is contained in:
leehk 2025-04-18 16:25:40 +08:00
parent 749da60a0e
commit aaa03db4c7

View File

@ -1,24 +1,11 @@
import pytest
import json
import websockets
@pytest.mark.asyncio
async def test_chatbot_integration():
# Send a request to the chatbot endpoint
async def test_websocket_connection():
url = "ws://backend-aimingmedai:80/ws"
data = [{"content": "Hello"}]
try:
async with websockets.connect(url) as websocket:
await websocket.send(json.dumps(data))
response = await websocket.recv()
assert response is not None
try:
response_json = json.loads(response)
assert "type" in response_json
assert "payload" in response_json
assert response_json["payload"] == ""
except json.JSONDecodeError:
assert False, "Invalid JSON response"
except Exception as e:
pytest.fail(f"Request failed: {e}")
async with websockets.connect(url):
assert True # If the connection is established, the test passes
except Exception:
assert False # If any exception occurs, the test fails