From ac95f3004f0aaa2985984b5285c16c8c17c4bcbd Mon Sep 17 00:00:00 2001
From: xst <xst@qafleet.dev>
Date: Fri, 4 Sep 2026 15:20:01 +0800
Subject: [PATCH] =?UTF-8?q?fix(save=5Fauth):=20=E7=99=BB=E5=BD=95=E6=A3=80?=
 =?UTF-8?q?=E6=B5=8B=E6=94=B9=E4=B8=BA=E6=8E=A2=E6=B5=8B=E9=A1=B5=E5=AE=9E?=
 =?UTF-8?q?=E6=B5=8B=E4=BC=9A=E8=AF=9D=E6=9C=89=E6=95=88=E6=80=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

不再读取用户标签页的 URL 判定——SPA hash 跳转后脚本读到的
可能是旧值, 登录界面也可能在其它标签页, 两种情况都会把成功
登录误判为失败。

改为: 回车后在同一 context 新开探测页访问首页, 由应用自己
回答是否已登录(有会话落 #/home, 无会话被重定向回 #/login),
与测试用例 ensure_login 的判定方式一致; 探测完即关, 不动
用户标签页。路由判断用精确 hash/path 解析, 参数含 login
字样不再误报。

验证: 用用户误报场景下保存的登录态实测, 访问直达 #/home,
证实会话判定有效。
---
 tests/web/admin/save_auth.py | 38 +++++++++++++++++++++++++++++++-----
 1 file changed, 33 insertions(+), 5 deletions(-)

diff --git a/tests/web/admin/save_auth.py b/tests/web/admin/save_auth.py
index 9025193..a5e5658 100644
--- a/tests/web/admin/save_auth.py
+++ b/tests/web/admin/save_auth.py
@@ -69,6 +69,22 @@ def _fill_credentials(page, account: dict) -> bool:
     return filled_user and filled_pwd
 
 
+def _is_on_login_page(url: str) -> bool:
+    """精确判断是否仍在登录页：只看路由本身（hash 或 path）。
+
+    不做子串匹配——登录成功后的 URL 可能带 login 字样的参数
+    （如 ?from=login、#/loginLog 菜单），子串判断会把成功登录误判为失败。
+    """
+    from urllib.parse import urlparse
+
+    parsed = urlparse(url)
+    frag = (parsed.fragment or "").lstrip("/")   # #/login → "login"
+    path = (parsed.path or "").rstrip("/")       # /admin/login → "/admin/login"
+    frag_is_login = frag == "login" or frag.startswith("login/") or frag.startswith("login?")
+    path_is_login = path == "/login" or path.endswith("/login")
+    return frag_is_login or path_is_login
+
+
 def main():
     parser = argparse.ArgumentParser(
         description="手动登录并保存浏览器登录态（绕过图形验证码）"
@@ -118,11 +134,23 @@ def main():
 
         input(">>> 按 Enter 保存登录态...")
 
-        # 检查是否真的登录成功了
-        current = page.url
-        if "/login" in current or "/#/login" in current:
-            print(f"\n[警告] 当前仍在登录页: {current}")
-            print("   要保存吗？(y/n): ", end="")
+        # 不依赖用户标签页的 URL（SPA hash 跳转后读到的可能是旧值, 登录
+        # 界面也可能在别的标签页）——在 context 里新开探测页访问首页,
+        # 由应用自己回答"是否已登录": 有会话落在 #/home, 无会话被重定向
+        # 回 #/login。探测完成即关闭, 不动用户的标签页。
+        # (与测试用例 ensure_login 的判定方式一致)
+        print("[检测] 验证登录状态...")
+        probe = context.new_page()
+        try:
+            probe.goto(base_url, wait_until="domcontentloaded")
+            probe.wait_for_timeout(3000)  # SPA hash 路由重定向的 settle
+            current = probe.url
+        finally:
+            probe.close()
+
+        if _is_on_login_page(current):
+            print(f"\n[警告] 登录态无效，访问首页被重定向回登录页: {current}")
+            print("   登录可能未成功（验证码错误/未点击登录）。仍要保存吗？(y/n): ", end="")
             choice = input().strip().lower()
             if choice != "y":
                 print("已取消。")