Files
SREBOT/test_spectra.py
T
FURRO404 2b399fdb81 add SREBOT, SHARED, TSSBOT contents (fixup for #1223)
PR #1223 only staged the deletions of the old paths because the new
top-level directories were still untracked when the commit was authored.
This commit adds the actual restructured tree: SREBOT/ (existing bot),
SHARED/ (vromfs, data_parser, ICONS/MAPS/FONTS, DAGOR_FILES,
update_game_files), and TSSBOT/ (skeleton).

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-13 23:17:02 -07:00

70 lines
2.4 KiB
Python

"""Quick test to check Spectra host connectivity — WS + HTTP."""
import asyncio
import os
import aiohttp
from dotenv import load_dotenv
from websockets.asyncio.client import connect
load_dotenv()
API_KEY = os.getenv("SPECTRA_API_KEY", "")
BASE_URL = os.getenv("SPECTRA_API_URL", "")
WS_URL = os.getenv("SPECTRA_WS_SQB_URL", "")
WS_GOB_URL = os.getenv("SPECTRA_WS_GOB_URL", "")
async def test_http():
"""Test HTTP API endpoint."""
url = f"{BASE_URL}/v1/replays/sort"
auth = API_KEY if API_KEY.startswith("Bearer ") else f"Bearer {API_KEY}"
headers = {"accept": "application/json", "Authorization": auth, "sortField": "sqb", "id": "463391108263081012"}
print(f"[HTTP] POST {url}")
try:
async with aiohttp.ClientSession() as session:
async with session.post(url, headers=headers, timeout=aiohttp.ClientTimeout(total=10)) as resp:
print(f"[HTTP] Status: {resp.status}")
text = await resp.text()
print(f"[HTTP] Body: {text[:500]}")
except Exception as e:
print(f"[HTTP] Failed: {type(e).__name__}: {e}")
async def test_ws_sqb():
"""Test SQB WebSocket connection."""
url = WS_URL
print(f"\n[WS-SQB] Connecting to {url}")
try:
async with connect(url, additional_headers={"Authorization": API_KEY}, open_timeout=15) as ws:
print("[WS-SQB] Connected! Waiting for message (10s)...")
msg = await asyncio.wait_for(ws.recv(), timeout=10)
print(f"[WS-SQB] Got message: {str(msg)[:300]}")
except Exception as e:
print(f"[WS-SQB] Failed: {type(e).__name__}: {e}")
async def test_ws_gob():
"""Test GOB WebSocket connection."""
url = WS_GOB_URL
print(f"\n[WS-GOB] Connecting to {url}")
try:
async with connect(url, additional_headers={"Authorization": API_KEY}, open_timeout=15) as ws:
print("[WS-GOB] Connected! Waiting for message (10s)...")
msg = await asyncio.wait_for(ws.recv(), timeout=10)
print(f"[WS-GOB] Got message ({len(msg)} bytes)")
except Exception as e:
print(f"[WS-GOB] Failed: {type(e).__name__}: {e}")
async def main():
print(f"API Key configured: {'Yes' if API_KEY else 'No'}")
print(f"Base URL: {BASE_URL}")
print(f"WS URL: {WS_URL}")
print(f"WS GOB URL: {WS_GOB_URL}\n")
await test_http()
await test_ws_sqb()
await test_ws_gob()
if __name__ == "__main__":
asyncio.run(main())