trace/services/class-monitor/tests/websocket_smoke.py

22 lines
560 B
Python

import asyncio
import os
import aiohttp
async def main() -> None:
url = "http://127.0.0.1:8765/onebot/v11/ws"
async with aiohttp.ClientSession() as session:
async with session.get(url) as response:
assert response.status == 401, response.status
async with session.ws_connect(
url,
headers={"Authorization": f"Bearer {os.environ['ONEBOT_ACCESS_TOKEN']}"},
) as ws:
await ws.close()
print("ONEBOT_WS_AUTH_HANDSHAKE=PASS")
if __name__ == "__main__":
asyncio.run(main())