更新管理登录功能

pytest/conftest.py

@@ -0,0 +1,34 @@
+import os
+import json
+import pytest
+import logging
+from websockets.asyncio.client import connect
+
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+WS_URL = os.environ.get("LMVPN_WS_URL", "ws://localhost:8080/ws")
+AUTH_USERNAME = os.environ.get("LMVPN_AUTH_USERNAME", "admin")
+AUTH_PASSWORD = os.environ.get("LMVPN_AUTH_PASSWORD", "admin123")
+
+
+@pytest.fixture
+async def raw_ws():
+    """仅建立 WebSocket 连接，不认证"""
+    logger.info("connecting to %s", WS_URL)
+    async with connect(WS_URL, ping_interval=None) as ws:
+        yield ws
+
+
+@pytest.fixture
+async def authenticated_ws():
+    """建立 WebSocket 连接并完成认证"""
+    logger.info("connecting to %s", WS_URL)
+    async with connect(WS_URL, ping_interval=None) as ws:
+        auth_msg = json.dumps({"type": "auth", "username": AUTH_USERNAME, "password": AUTH_PASSWORD})
+        await ws.send(auth_msg)
+        resp = await ws.recv()
+        resp_data = json.loads(resp)
+        assert resp_data.get("type") == "auth_ok", f"认证失败: {resp_data}"
+        logger.info("authenticated as %s", AUTH_USERNAME)
+        yield ws