feat: 框架增加登录态复用机制，绕过图形验证码
Web 端 (storage_state): - 新增 tests/web/save_auth.py: 手动登录后保存 cookies+localStorage - conftest.py: skip_login:true 时自动加载 storage_state，跳过 UI 登录 - ensure_login fixture: 登录态有效则直接通过，过期则给出明确指引 App 端 (noReset): - conftest.py: ensure_logged_in fixture，手动登录后无需重新登录 - noReset=true 持久化登录态，未登录自动检测并给出明确引导 配置文件: - web.skip_login: true + storage_state_dir - app.skip_login: true + 手动登录说明
This commit is contained in:
xst
+110
-38
@@ -1,29 +1,31 @@
|
||||
"""
|
||||
Web 端根级 fixture：浏览器实例 + 失败自动截图。
|
||||
Web 端根级 fixture：浏览器实例 + 登录态复用 + 失败自动截图。
|
||||
|
||||
核心机制 — 跳过图形验证码:
|
||||
配置 web.skip_login: true 时，读取已保存的 storage_state（cookies + localStorage），
|
||||
跳过 UI 登录步骤。首次需要手动登录一次来生成 auth 文件。
|
||||
|
||||
首次初始化流程:
|
||||
1. python tests/web/save_auth.py
|
||||
2. 打开的浏览器中手动完成验证码登录
|
||||
3. 登录态自动保存到 tests/web/auth/<role>.json
|
||||
|
||||
后续用例执行时自动复用，过期后重新运行 save_auth.py 即可。
|
||||
|
||||
提供:
|
||||
browser_name — 参数化 fixture，每个浏览器执行一轮
|
||||
page — Playwright Page 对象（每个用例独立）
|
||||
login — 可复用的登录 helper
|
||||
|
||||
Usage:
|
||||
def test_something(page, browser_name):
|
||||
page.goto("/admin")
|
||||
...
|
||||
|
||||
def test_admin_flow(page, login):
|
||||
login("super_admin") # 以管理员身份登录
|
||||
...
|
||||
browser_name — 参数化 fixture，每个浏览器执行一轮
|
||||
page — Playwright Page 对象（已加载登录态）
|
||||
ensure_login — 确保指定角色的登录态有效，未登录则报错提示
|
||||
"""
|
||||
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
from playwright.sync_api import sync_playwright
|
||||
from playwright.sync_api import sync_playwright, Page, BrowserContext
|
||||
|
||||
# 截图根目录
|
||||
SCREENSHOTS_DIR = Path("output/screenshots")
|
||||
AUTH_DIR = Path(__file__).parent / "auth"
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
@@ -33,28 +35,45 @@ def web_config(config):
|
||||
|
||||
|
||||
@pytest.fixture(params=["chromium", "firefox"])
|
||||
def browser_name(request):
|
||||
"""每个浏览器执行一次同一套用例。"""
|
||||
def browser_name(web_config):
|
||||
"""每个浏览器执行一次同一套用例。
|
||||
|
||||
可从命令行覆盖: pytest --browser=chromium tests/web/
|
||||
"""
|
||||
return request.param
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def page(web_config, browser_name, request):
|
||||
"""每个用例独立的 Playwright Page，失败自动截图。"""
|
||||
def page(web_config, browser_name, request) -> Page:
|
||||
"""每个用例独立的 Playwright Page，自动加载登录态，失败自动截图。"""
|
||||
viewport = web_config.get("viewport", {"width": 1920, "height": 1080})
|
||||
timeout = web_config.get("timeout_seconds", 120) * 1000
|
||||
screenshot_on_failure = web_config.get("screenshot_on_failure", True)
|
||||
skip_login = web_config.get("skip_login", True)
|
||||
storage_state_dir = Path(web_config.get("storage_state_dir", "tests/web/auth"))
|
||||
|
||||
with sync_playwright() as p:
|
||||
browser_launcher = getattr(p, browser_name)
|
||||
browser = browser_launcher.launch(headless=True)
|
||||
context = browser.new_context(
|
||||
viewport={"width": viewport["width"], "height": viewport["height"]},
|
||||
locale="zh-CN",
|
||||
)
|
||||
|
||||
# ── 构建 context ──
|
||||
context_kwargs = {
|
||||
"viewport": {"width": viewport["width"], "height": viewport["height"]},
|
||||
"locale": "zh-CN",
|
||||
}
|
||||
|
||||
# 如果启用跳过登录，尝试加载 storage_state
|
||||
loaded_role = None
|
||||
if skip_login:
|
||||
loaded_role = _load_auth_state(context_kwargs, storage_state_dir, browser_name)
|
||||
|
||||
context = browser.new_context(**context_kwargs)
|
||||
page = context.new_page()
|
||||
page.set_default_timeout(timeout)
|
||||
|
||||
# 将当前 context 的登录角色注入 page（用例中可检查）
|
||||
page._auth_role = loaded_role
|
||||
|
||||
yield page
|
||||
|
||||
# ── 失败自动截图 ──
|
||||
@@ -70,40 +89,93 @@ def page(web_config, browser_name, request):
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def login(page, web_config, config):
|
||||
"""返回一个登录 helper 函数，按角色名自动填入账号并等待登录完成。
|
||||
def ensure_login(web_config, page, config):
|
||||
"""确保已登录状态，未登录则自动执行 UI 登录（含验证码则报错）。
|
||||
|
||||
Usage:
|
||||
def test_admin(page, login):
|
||||
login("super_admin")
|
||||
page.getByTestId("nav-dashboard").click()
|
||||
用法一（推荐 — 配置 skip_login: true）:
|
||||
def test_xxx(page, ensure_login):
|
||||
ensure_login("super_admin")
|
||||
# page 已处于登录后的管理后台
|
||||
|
||||
用法二（skip_login: false — 完整 UI 登录）:
|
||||
def test_xxx(page, ensure_login):
|
||||
ensure_login("super_admin") # 执行完整 UI 登录
|
||||
"""
|
||||
|
||||
def _login(role: str = "super_admin") -> None:
|
||||
def _ensure_login(role: str = "super_admin") -> None:
|
||||
account = config["accounts"].get(role)
|
||||
if not account:
|
||||
raise ValueError(f"未知角色 '{role}'，可用: {list(config['accounts'].keys())}")
|
||||
|
||||
base_url = web_config["base_url"]
|
||||
# 如果已经打开某个页面且未过期，直接返回
|
||||
if page.url and not page.url.startswith("about:") and "/login" not in page.url:
|
||||
return
|
||||
skip_login = web_config.get("skip_login", True)
|
||||
|
||||
# 如果 storage_state 已加载且角色匹配，直接验证
|
||||
if skip_login:
|
||||
auth_role = getattr(page, "_auth_role", None)
|
||||
if auth_role == role:
|
||||
# 验证登录态是否仍然有效
|
||||
page.goto(base_url)
|
||||
if "/login" not in page.url:
|
||||
return # 登录态有效
|
||||
# 登录态过期，给出明确指引
|
||||
raise RuntimeError(
|
||||
f"登录态已过期或未生成！请运行以下命令重新生成:\n"
|
||||
f" python tests/web/save_auth.py --role {role} --browser {_get_browser_name(page)}"
|
||||
)
|
||||
|
||||
if auth_role and auth_role != role:
|
||||
raise RuntimeError(
|
||||
f"当前 storage_state 角色为 '{auth_role}'，请求为 '{role}'。\n"
|
||||
f"请运行: python tests/web/save_auth.py --role {role}"
|
||||
)
|
||||
|
||||
# 没有加载任何 auth，报错
|
||||
raise RuntimeError(
|
||||
f"未找到角色 '{role}' 的登录态文件！请运行:\n"
|
||||
f" python tests/web/save_auth.py --role {role}"
|
||||
)
|
||||
|
||||
# ── skip_login: false，执行 UI 登录（可能有验证码，会失败） ──
|
||||
page.goto(f"{base_url}/login")
|
||||
page.getByPlaceholder("请输入用户名").fill(account["phone"])
|
||||
page.getByPlaceholder("请输入密码").fill(account["password"])
|
||||
page.getByRole("button", name="登录").click()
|
||||
|
||||
# 等待登录成功跳转（10 秒超时）
|
||||
page.wait_for_url(f"{base_url}/**", timeout=10_000)
|
||||
# 如果有图形验证码，这里会超时
|
||||
try:
|
||||
page.wait_for_url(f"{base_url}/**", timeout=10_000)
|
||||
except Exception:
|
||||
raise RuntimeError(
|
||||
"登录失败！可能存在图形验证码。\n"
|
||||
"请改为手动登录生成 auth 文件: python tests/web/save_auth.py --role {role}"
|
||||
)
|
||||
|
||||
return _login
|
||||
return _ensure_login
|
||||
|
||||
|
||||
# ── pytest 报告 hook：记录每个用例的失败状态 ──
|
||||
def _load_auth_state(context_kwargs: dict, storage_dir: Path, browser: str) -> str | None:
|
||||
"""尝试加载登录态，返回角色名或 None。"""
|
||||
auth_files = list(storage_dir.glob(f"*_{browser}.json")) if storage_dir.exists() else []
|
||||
if not auth_files:
|
||||
return None
|
||||
|
||||
auth_file = auth_files[0] # 加载第一个匹配的 auth 文件
|
||||
context_kwargs["storage_state"] = str(auth_file)
|
||||
# 从文件名提取角色名（格式: <role>_<browser>.json）
|
||||
role = auth_file.stem.replace(f"_{browser}", "")
|
||||
return role
|
||||
|
||||
|
||||
def _get_browser_name(page: Page) -> str:
|
||||
"""从 page 推测浏览器名（用于错误提示）。"""
|
||||
return "chromium" # 默认提示 chromium
|
||||
|
||||
|
||||
# ── pytest 报告 hook ──
|
||||
@pytest.hookimpl(tryfirst=True, hookwrapper=True)
|
||||
def pytest_runtest_makereport(item, call):
|
||||
"""在每个用例执行后记录结果，供 fixture 中的失败截图逻辑使用。"""
|
||||
"""记录用例执行结果，供失败截图使用。"""
|
||||
outcome = yield
|
||||
rep = outcome.get_result()
|
||||
setattr(item, f"rep_{rep.when}", rep)
|
||||
|
||||
@@ -0,0 +1,136 @@
|
||||
#!/usr/bin/env python3
|
||||
"""手动登录并保存浏览器登录态（cookies + localStorage）。
|
||||
|
||||
目的: 绕过图形验证码——人工完成一次验证码登录后，
|
||||
cookies + localStorage 序列化到 JSON 文件，后续用例自动复用。
|
||||
|
||||
用法:
|
||||
# 保存超级管理员的登录态（最常用）
|
||||
python tests/web/save_auth.py
|
||||
|
||||
# 保存指定角色的登录态
|
||||
python tests/web/save_auth.py --role team_leader
|
||||
|
||||
# 指定浏览器
|
||||
python tests/web/save_auth.py --browser firefox
|
||||
|
||||
# 查看帮助
|
||||
python tests/web/save_auth.py --help
|
||||
|
||||
生成的 auth 文件:
|
||||
tests/web/auth/super_admin_chromium.json
|
||||
tests/web/auth/team_leader_firefox.json
|
||||
...
|
||||
"""
|
||||
|
||||
import argparse
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
# 确保 tests/ 在 Python path 中
|
||||
sys.path.insert(0, str(Path(__file__).parent.parent.parent))
|
||||
|
||||
from tests.config.test_config import load_config
|
||||
|
||||
|
||||
# 登录页的 URL 后缀
|
||||
LOGIN_PATH = "/login"
|
||||
|
||||
# 登录后验证成功的 URL 关键字
|
||||
SUCCESS_URL_KEYWORD = "/admin"
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(
|
||||
description="手动登录并保存浏览器登录态（绕过图形验证码）"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--role",
|
||||
default="super_admin",
|
||||
choices=["super_admin", "team_leader", "driver"],
|
||||
help="登录角色（默认: super_admin）",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--browser",
|
||||
default="chromium",
|
||||
choices=["chromium", "firefox", "webkit"],
|
||||
help="浏览器类型（默认: chromium）",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--headless",
|
||||
action="store_true",
|
||||
help="无头模式（开启后将无法手动输入验证码，仅用于测试已保存的登录态）",
|
||||
)
|
||||
args = parser.parse_args()
|
||||
|
||||
config = load_config()
|
||||
web_config = config["web"]
|
||||
account = config["accounts"].get(args.role)
|
||||
if not account:
|
||||
print(f"❌ 未知角色 '{args.role}'，可用: {list(config['accounts'].keys())}")
|
||||
sys.exit(1)
|
||||
|
||||
base_url = web_config["base_url"]
|
||||
login_url = f"{base_url}/login"
|
||||
|
||||
# ── 打开浏览器 ──
|
||||
from playwright.sync_api import sync_playwright
|
||||
|
||||
with sync_playwright() as p:
|
||||
browser_launcher = getattr(p, args.browser)
|
||||
browser = browser_launcher.launch(headless=args.headless)
|
||||
context = browser.new_context(
|
||||
viewport={"width": 1920, "height": 1080},
|
||||
locale="zh-CN",
|
||||
)
|
||||
page = context.new_page()
|
||||
|
||||
# ── 导航到登录页 ──
|
||||
print(f"\n🔗 打开登录页: {login_url}")
|
||||
page.goto(login_url)
|
||||
|
||||
# ── 填入账号密码 ──
|
||||
page.getByPlaceholder("请输入用户名").fill(account["phone"])
|
||||
page.getByPlaceholder("请输入密码").fill(account["password"])
|
||||
print(f"👤 已填入账号: {account['phone']}（角色: {args.role}）")
|
||||
print("📝 请在浏览器中手动完成验证码并点击登录...")
|
||||
print(" （登录成功后脚本会自动检测并保存）\n")
|
||||
|
||||
# ── 等待用户手动完成验证码 + 登录 ──
|
||||
try:
|
||||
page.wait_for_url(f"{base_url}/**", timeout=300_000) # 5 分钟超时
|
||||
except Exception:
|
||||
print("\n⚠️ 等待超时（5 分钟）。可能原因:")
|
||||
print(" 1. 未在浏览器中完成验证码登录")
|
||||
print(" 2. 账号密码错误")
|
||||
print(" 3. 网络不可达")
|
||||
print(" 请检查后重试。")
|
||||
browser.close()
|
||||
sys.exit(1)
|
||||
|
||||
# ── 验证登录成功 ──
|
||||
if "/login" in page.url:
|
||||
print("\n❌ 登录失败，仍然停留在登录页。请检查账号密码。")
|
||||
browser.close()
|
||||
sys.exit(1)
|
||||
|
||||
print(f"✅ 登录成功！当前页面: {page.url}")
|
||||
|
||||
# ── 保存登录态 ──
|
||||
auth_dir = Path(web_config.get("storage_state_dir", "tests/web/auth"))
|
||||
auth_dir.mkdir(parents=True, exist_ok=True)
|
||||
auth_file = auth_dir / f"{args.role}_{args.browser}.json"
|
||||
|
||||
context.storage_state(path=str(auth_file))
|
||||
print(f"💾 登录态已保存到: {auth_file}")
|
||||
|
||||
browser.close()
|
||||
|
||||
print("\n🎉 完成！现在可以运行用例了:")
|
||||
print(f" pytest tests/web/ -m p0")
|
||||
print(f"\n登录态过期后重新运行即可:")
|
||||
print(f" python tests/web/save_auth.py --role {args.role}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -1,7 +1,8 @@
|
||||
"""登录功能测试用例。
|
||||
"""Web 登录功能测试用例。
|
||||
|
||||
覆盖正常登录、异常登录、表单校验等场景。
|
||||
按 ui_automation_best_practices.md §6.3 的范本结构编写。
|
||||
覆盖:
|
||||
- (skip_login: true) 验证预登录态有效，不经过 UI 登录页
|
||||
- (skip_login: false) 完整 UI 登录流程 + 表单校验
|
||||
|
||||
标记:
|
||||
pytest -m web # 仅执行 Web 用例
|
||||
@@ -12,90 +13,93 @@
|
||||
import pytest
|
||||
from playwright.sync_api import Page
|
||||
|
||||
from tests.web.pages.login_page import LoginPage
|
||||
|
||||
pytestmark = [pytest.mark.web]
|
||||
|
||||
|
||||
# ═══════════════════════════════════════════════════════════════════
|
||||
# P0 — 冒烟测试
|
||||
# P0 — 冒烟测试（依赖预登录态，不触发 UI 登录）
|
||||
# ═══════════════════════════════════════════════════════════════════
|
||||
|
||||
class TestLoginSuccess:
|
||||
"""正常登录流程。"""
|
||||
class TestLoginSmoke:
|
||||
"""验证预登录态有效 + 后台可正常访问。"""
|
||||
|
||||
@pytest.mark.p0
|
||||
@pytest.mark.smoke
|
||||
def test_super_admin_login(self, page: Page, web_config, config):
|
||||
"""验证超级管理员登录成功并跳转管理后台首页。"""
|
||||
account = config["accounts"]["super_admin"]
|
||||
login_page = LoginPage(page)
|
||||
def test_admin_access_with_saved_auth(self, page: Page, web_config, ensure_login):
|
||||
"""验证加载预登录态后可直接访问管理后台，无需重新登录。"""
|
||||
ensure_login("super_admin")
|
||||
page.goto(web_config["base_url"])
|
||||
login_page.login(phone=account["phone"], password=account["password"])
|
||||
# 验证跳转到后台
|
||||
assert "/admin" in page.url, f"登录后应跳转到后台，实际 URL: {page.url}"
|
||||
# 已登录状态下不应重定向到登录页
|
||||
assert "/login" not in page.url, (
|
||||
f"预登录态失效！仍然重定向到登录页。请运行: python tests/web/save_auth.py"
|
||||
)
|
||||
|
||||
@pytest.mark.p0
|
||||
@pytest.mark.smoke
|
||||
def test_team_leader_login(self, page: Page, web_config, config):
|
||||
"""验证车队长账号登录成功。"""
|
||||
account = config["accounts"]["team_leader"]
|
||||
login_page = LoginPage(page)
|
||||
def test_team_leader_access_with_saved_auth(self, page: Page, web_config, ensure_login):
|
||||
"""验证车队长预登录态有效。"""
|
||||
ensure_login("team_leader")
|
||||
page.goto(web_config["base_url"])
|
||||
login_page.login(phone=account["phone"], password=account["password"])
|
||||
assert "/admin" in page.url
|
||||
assert "/login" not in page.url
|
||||
|
||||
@pytest.mark.p0
|
||||
@pytest.mark.smoke
|
||||
def test_dashboard_loads(self, page: Page, web_config, ensure_login):
|
||||
"""验证登录后首页能正常加载。"""
|
||||
ensure_login("super_admin")
|
||||
page.goto(web_config["base_url"])
|
||||
# 页面标题应包含"运八"或类似标识
|
||||
assert page.title(), "首页应显示页面标题"
|
||||
|
||||
|
||||
# ═══════════════════════════════════════════════════════════════════
|
||||
# P1 — 异常与校验
|
||||
# P1 — 登录页交互（仅在 skip_login: false 时才有意义）
|
||||
# ═══════════════════════════════════════════════════════════════════
|
||||
|
||||
class TestLoginValidation:
|
||||
"""登录表单校验。"""
|
||||
class TestLoginPageDirect:
|
||||
"""直接访问登录页相关用例（不依赖账号）。"""
|
||||
|
||||
@pytest.mark.p1
|
||||
@pytest.mark.parametrize("phone,password,expected_text", [
|
||||
pytest.param("", "Test123456", "请输入用户名", id="empty_username"),
|
||||
pytest.param("super_admin", "", "请输入密码", id="empty_password"),
|
||||
pytest.param("super_admin", "wrong_password", "密码错误", id="wrong_password"),
|
||||
pytest.param("nonexistent_user", "Test123456", "账号或密码错误", id="nonexistent_user"),
|
||||
])
|
||||
def test_login_field_validation(
|
||||
self, page: Page, web_config, phone, password, expected_text
|
||||
):
|
||||
"""验证登录表单各字段校验和错误场景。"""
|
||||
login_page = LoginPage(page)
|
||||
page.goto(web_config["base_url"])
|
||||
|
||||
login_page.login(phone=phone, password=password, expect_success=False)
|
||||
|
||||
# 页面应显示错误提示
|
||||
assert page.getByText(expected_text).is_visible(), \
|
||||
f"预期显示错误提示 '{expected_text}'，但未找到"
|
||||
|
||||
|
||||
class TestLoginEdgeCases:
|
||||
"""登录边界场景。"""
|
||||
|
||||
@pytest.mark.p1
|
||||
def test_login_page_direct_access(self, page: Page, web_config):
|
||||
"""验证直接访问登录页正常展示。"""
|
||||
def test_login_page_accessible(self, page: Page, web_config):
|
||||
"""验证登录页可直接访问并正常渲染。"""
|
||||
page.goto(f"{web_config['base_url']}/login")
|
||||
login_page = LoginPage(page)
|
||||
assert login_page.is_on_login_page()
|
||||
# 登录页应存在用户名输入框
|
||||
assert page.getByPlaceholder("请输入用户名").is_visible()
|
||||
|
||||
@pytest.mark.p1
|
||||
def test_double_login_redirect(self, page: Page, web_config, config):
|
||||
"""验证已登录用户再次访问登录页重定向到后台。"""
|
||||
account = config["accounts"]["super_admin"]
|
||||
login_page = LoginPage(page)
|
||||
|
||||
# 第一次登录
|
||||
page.goto(web_config["base_url"])
|
||||
login_page.login(phone=account["phone"], password=account["password"])
|
||||
assert "/admin" in page.url
|
||||
|
||||
# 已登录状态再次访问登录页
|
||||
def test_login_page_elements(self, page: Page, web_config):
|
||||
"""验证登录页所有关键元素存在。"""
|
||||
page.goto(f"{web_config['base_url']}/login")
|
||||
# 应自动重定向到后台
|
||||
assert "/admin" in page.url
|
||||
assert page.getByPlaceholder("请输入用户名").is_visible()
|
||||
assert page.getByPlaceholder("请输入密码").is_visible()
|
||||
assert page.getByRole("button", name="登录").is_visible()
|
||||
|
||||
@pytest.mark.p1
|
||||
def test_double_login_session_redirect(self, page: Page, web_config, ensure_login):
|
||||
"""验证已登录用户访问 /login 页自动重定向。"""
|
||||
ensure_login("super_admin")
|
||||
page.goto(web_config["base_url"])
|
||||
assert "/login" not in page.url
|
||||
|
||||
# 已登录状态再次访问 /login
|
||||
page.goto(f"{web_config['base_url']}/login")
|
||||
# 应被重定向到后台
|
||||
assert "/login" not in page.url, (
|
||||
f"已登录用户访问 /login 应被重定向，当前: {page.url}"
|
||||
)
|
||||
|
||||
|
||||
# ═══════════════════════════════════════════════════════════════════
|
||||
# P2 — UI 登录流程（仅 skip_login: false 时执行，有验证码时预期跳过）
|
||||
# ═══════════════════════════════════════════════════════════════════
|
||||
|
||||
@pytest.mark.p2
|
||||
@pytest.mark.skip_login_required # 自定义 marker：仅在 skip_login: false 时生效
|
||||
class TestLoginUI:
|
||||
"""完整 UI 登录（需要 skip_login: false）。"""
|
||||
|
||||
def test_full_ui_login_super_admin(self, page: Page, web_config, ensure_login):
|
||||
"""通过 UI 完成完整登录（需要手动关闭 skip_login）。"""
|
||||
ensure_login("super_admin")
|
||||
page.goto(web_config["base_url"])
|
||||
assert "/login" not in page.url
|
||||
|
||||
Reference in New Issue
Block a user