3、IDP门户登录自动化：定位登录元素、输入凭证、处理验证码、登录成功断言

IDP自助门户的登录自动化是准入控制规则验证流程的起点。自动化脚本需要模拟真实用户操作，依次完成登录页面元素定位、凭证输入、验证码识别与填充，最后通过断言确认登录成功。本章将拆解每个步骤的技术实现与关键注意事项。

3.1 登录页面元素定位策略

IDP门户登录页面通常包含用户名输入框、密码输入框、验证码输入框、登录按钮以及验证码图片元素。定位策略需兼顾稳定性与执行效率，推荐优先级如下：

以下为典型登录元素定位代码示例（Python + Selenium）：

from selenium.webdriver.common.by import By

# 定位用户名输入框（ID 定位）
username_input = driver.find_element(By.ID, "username")

# 定位密码输入框（Name 定位）
password_input = driver.find_element(By.NAME, "password")

# 定位验证码输入框（CSS 选择器）
captcha_input = driver.find_element(By.CSS_SELECTOR, "input.captcha-input")

# 定位登录按钮（XPath 相对定位）
login_button = driver.find_element(By.XPATH, "//button[contains(@class, 'login-btn')]")

# 定位验证码图片元素
captcha_image = driver.find_element(By.ID, "captcha_img")

3.2 凭证输入与验证码处理

凭证输入需注意输入框的清空操作，避免残留字符导致输入错误。验证码处理是登录自动化的核心难点，常见方案包括：

• OCR 识别方案：使用 Tesseract、PaddleOCR 等引擎识别验证码图片文本，适用于简单数字/字母验证码。
• 第三方打码平台：对接打码 API（如超级鹰、打码兔），适用于复杂验证码或滑块验证码。
• 预置万能验证码：在测试环境配置固定验证码（如 "0000"），仅限非生产环境使用。
• 手动干预模式：脚本暂停等待人工输入验证码，适用于低频执行场景。

以下为凭证输入与验证码处理的完整代码片段：

import os
import time
from selenium.webdriver.common.keys import Keys

# 从环境变量读取凭证
username = os.getenv("IDP_USERNAME")
password = os.getenv("IDP_PASSWORD")

# 清空并输入用户名
username_input.clear()
username_input.send_keys(username)

# 清空并输入密码
password_input.clear()
password_input.send_keys(password)

# 处理验证码（OCR 识别方案示例）
captcha_text = recognize_captcha(captcha_image)  # 自定义 OCR 函数
captcha_input.clear()
captcha_input.send_keys(captcha_text)

# 点击登录按钮
login_button.click()

3.3 验证码识别流程（SVG 流程图）

验证码识别是登录自动化中不确定性最高的环节，其处理流程直接影响整体成功率。下图展示了从验证码图片获取到文本填充的完整链路：

3.4 登录成功断言

点击登录按钮后，需要通过断言确认登录是否成功。断言策略应覆盖多种验证维度，避免单一判断点失效导致误判。

以下为登录成功断言的完整实现：

import time
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# 等待页面跳转完成（最多等待 10 秒）
WebDriverWait(driver, 10).until(
    EC.url_contains("/dashboard")
)

# 断言 1：URL 跳转验证
assert "/dashboard" in driver.current_url, "登录失败：未跳转到仪表盘页面"

# 断言 2：页面元素验证（等待用户头像元素出现）
user_avatar = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.CLASS_NAME, "user-avatar"))
)
assert user_avatar.is_displayed(), "登录失败：用户头像未显示"

# 断言 3：Cookie 验证（检查认证 Cookie 是否存在）
auth_cookie = driver.get_cookie("session_id")
assert auth_cookie is not None, "登录失败：未生成认证 Cookie"
assert auth_cookie["value"] != "", "登录失败：认证 Cookie 值为空"

print("登录成功断言全部通过")

3.5 登录自动化整体架构（SVG 架构图）

将元素定位、凭证输入、验证码处理、断言验证整合为一个完整的自动化模块，其架构关系如下图所示：

3.6 常见问题与避坑指南

• 元素未加载完成：登录页面可能包含动态加载的组件，务必使用 WebDriverWait 等待元素可交互后再操作。
• 验证码刷新：每次登录失败后验证码会刷新，需重新获取图片并识别，避免使用过期验证码。
• 输入框残留：部分输入框有默认占位符或自动填充，使用 .clear() 方法清空后再输入。
• 登录按钮禁用状态：某些门户在表单未填完时禁用登录按钮，需确保所有输入框已填充后再点击。
• 多因素认证（MFA）：若 IDP 启用了 MFA，登录流程需额外处理短信/邮箱验证码或 TOTP 令牌。