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())