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("已取消。")