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:
+110
-10
@@ -1,14 +1,23 @@
|
||||
"""
|
||||
App 端根级 fixture:Appium Driver 实例 + 失败自动截图。
|
||||
App 端根级 fixture:Appium Driver 实例 + 登录态复用 + 失败自动截图。
|
||||
|
||||
核心机制 — 手动登录后复用:
|
||||
Appium noReset=true 确保 app 数据不重置,登录态持久保留在设备上。
|
||||
只需手动在设备上登录一次,后续用例通过 ensure_logged_in 检查首页是否已登录态。
|
||||
如果检测到未登录(在登录页),给出明确指引让用户先去设备上手动登录。
|
||||
|
||||
配置 app.skip_login: true 时:
|
||||
- 用例通过 ensure_logged_in() 检查是否已登录
|
||||
- 未登录则抛出明确错误,引导用户先去设备上手动登录
|
||||
- 已登录则直接开始执行
|
||||
|
||||
配置 app.skip_login: false 时:
|
||||
- 每次用例尝试完整 UI 登录(有验证码时会失败)
|
||||
|
||||
提供:
|
||||
app_config — session 级 App 配置
|
||||
driver — Appium WebDriver(每个用例独立)
|
||||
platform — 当前平台 android / ios
|
||||
|
||||
Usage:
|
||||
def test_app_login(driver, platform):
|
||||
driver.find_element(AppiumBy.ACCESSIBILITY_ID, "login_btn").click()
|
||||
app_config — session 级 App 配置
|
||||
driver — Appium WebDriver(每个用例独立,失败自动截图)
|
||||
platform — 当前平台
|
||||
"""
|
||||
|
||||
from datetime import datetime
|
||||
@@ -27,7 +36,7 @@ def app_config(config):
|
||||
|
||||
@pytest.fixture
|
||||
def platform(app_config) -> str:
|
||||
"""当前执行的平台(默认 android,后续可参数化)。"""
|
||||
"""当前执行的平台。"""
|
||||
return app_config.get("platforms", ["android"])[0]
|
||||
|
||||
|
||||
@@ -35,7 +44,8 @@ def platform(app_config) -> str:
|
||||
def driver(app_config, platform, request):
|
||||
"""每个用例独立的 Appium driver,失败自动截图。
|
||||
|
||||
注意:运行前需确保 Appium Server 已启动(appium --log-level info)。
|
||||
前提: Appium Server 已启动(appium --log-level info)。
|
||||
noReset=true → app 不重置 → 登录态持久保留。
|
||||
"""
|
||||
try:
|
||||
from appium import webdriver
|
||||
@@ -89,6 +99,96 @@ def driver(app_config, platform, request):
|
||||
driver.quit()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def ensure_logged_in(driver, platform, app_config, config):
|
||||
"""确保 APP 已处于登录状态。
|
||||
|
||||
使用方式:
|
||||
def test_xxx(driver, ensure_logged_in):
|
||||
ensure_logged_in() # 检查登录态,未登录则报错
|
||||
# 后面直接写业务断言
|
||||
"""
|
||||
|
||||
def _ensure_logged_in(timeout_s: float = 10.0) -> None:
|
||||
skip_login = app_config.get("skip_login", True)
|
||||
|
||||
from selenium.webdriver.support.ui import WebDriverWait
|
||||
from selenium.webdriver.support import expected_conditions as EC
|
||||
from appium.webdriver.common.appiumby import AppiumBy
|
||||
|
||||
# ⚠️ home_tab 的 accessibility_id 需根据实际 APP UI 确认
|
||||
HOME_INDICATOR_ID = "tab_home" # 首页 tab 的 accessibility_id
|
||||
|
||||
wait = WebDriverWait(driver, timeout=timeout_s)
|
||||
|
||||
try:
|
||||
wait.until(
|
||||
EC.visibility_of_element_located((AppiumBy.ACCESSIBILITY_ID, HOME_INDICATOR_ID))
|
||||
)
|
||||
# 已登录,无需操作
|
||||
return
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# ── 未检测到首页元素,判断是否在登录页 ──
|
||||
try:
|
||||
# 尝试找登录页元素
|
||||
login_input = driver.find_element(AppiumBy.ACCESSIBILITY_ID, "login_phone_input")
|
||||
if login_input.is_displayed():
|
||||
if skip_login:
|
||||
raise RuntimeError(
|
||||
"APP 当前在登录页,未检测到登录态!\n"
|
||||
"请在设备上手动完成一次登录(包含验证码),之后重新运行用例即可。\n"
|
||||
"如需通过自动化登录,请设置 app.skip_login: false(不推荐,有验证码)。"
|
||||
)
|
||||
# skip_login: false — 走自动化登录
|
||||
account = config["accounts"].get("driver")
|
||||
if not account:
|
||||
raise RuntimeError("配置中未找到 driver 账号")
|
||||
|
||||
login_input.send_keys(account["phone"])
|
||||
driver.find_element(AppiumBy.ACCESSIBILITY_ID, "login_password_input").send_keys(
|
||||
account["password"]
|
||||
)
|
||||
driver.find_element(AppiumBy.ACCESSIBILITY_ID, "login_submit_btn").click()
|
||||
|
||||
# 等待登录完成
|
||||
wait.until(
|
||||
EC.visibility_of_element_located(
|
||||
(AppiumBy.ACCESSIBILITY_ID, HOME_INDICATOR_ID)
|
||||
)
|
||||
)
|
||||
return
|
||||
except RuntimeError:
|
||||
raise # 重新抛出明确的错误提示
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
raise RuntimeError(
|
||||
f"无法确定 APP 的登录状态!请确保 APP 已启动且首页或登录页可见。\n"
|
||||
f"当前平台: {platform}"
|
||||
)
|
||||
|
||||
return _ensure_logged_in
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def is_logged_in(driver, platform, app_config) -> bool:
|
||||
"""快速检查当前是否已登录(不执行登录操作)。"""
|
||||
from selenium.webdriver.support.ui import WebDriverWait
|
||||
from selenium.webdriver.support import expected_conditions as EC
|
||||
from appium.webdriver.common.appiumby import AppiumBy
|
||||
|
||||
try:
|
||||
wait = WebDriverWait(driver, timeout=5)
|
||||
wait.until(
|
||||
EC.visibility_of_element_located((AppiumBy.ACCESSIBILITY_ID, "tab_home"))
|
||||
)
|
||||
return True
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
|
||||
# ── pytest 报告 hook ──
|
||||
@pytest.hookimpl(tryfirst=True, hookwrapper=True)
|
||||
def pytest_runtest_makereport(item, call):
|
||||
|
||||
Reference in New Issue
Block a user