mirror of
https://github.com/aimingmed/aimingmed-ai.git
synced 2026-01-19 13:23:23 +08:00
replace test that ping only
This commit is contained in:
parent
749da60a0e
commit
aaa03db4c7
@ -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
|
||||
Loading…
x
Reference in New Issue
Block a user