41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
import asyncio
|
|
import json
|
|
from websockets import connect
|
|
import httpx
|
|
|
|
async def test():
|
|
print("=== Test v3 Simple ===")
|
|
|
|
async with httpx.AsyncClient() as client:
|
|
r = await client.post('http://localhost:3030/auth/login',
|
|
json={'username': 'admin', 'password': 'admin'})
|
|
token = r.json()['token']
|
|
|
|
async with connect('ws://localhost:3030/ws/chat') as ws:
|
|
await ws.send(json.dumps({'token': token}))
|
|
await ws.recv() # init
|
|
|
|
await ws.send(json.dumps({'type': 'create_session', 'name': 't'}))
|
|
created = json.loads(await ws.recv())
|
|
sid = created['session_id']
|
|
|
|
await ws.send(json.dumps({'type': 'connect_session', 'session_id': sid}))
|
|
await ws.recv() # connected
|
|
|
|
print("Enviando mensaje...")
|
|
await ws.send(json.dumps({'type': 'message', 'content': 'di OK'}))
|
|
|
|
# Esperar respuesta con mucho timeout
|
|
print("Esperando (30s)...")
|
|
try:
|
|
for i in range(30):
|
|
try:
|
|
msg = await asyncio.wait_for(ws.recv(), timeout=1.0)
|
|
print(f"RECIBIDO: {msg}")
|
|
except asyncio.TimeoutError:
|
|
print(f" {i+1}s...", end="\r")
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
|
|
asyncio.run(test())
|