1. 项目背景与核心思路

在UI自动化测试领域，验证码登录一直是困扰测试工程师的典型难题。传统方案通常采用OCR识别或第三方打码平台，但这些方法存在识别率不稳定、额外成本高、响应速度慢等问题。我们团队在实际项目中探索出一种更可靠的解决方案：通过首次手动登录获取有效Cookie，后续测试直接复用认证状态。

这个方案的巧妙之处在于完全规避了验证码识别环节。第一次运行时，测试脚本会暂停在登录页面，等待人工完成验证码输入和登录操作。成功登录后，系统自动保存当前会话的Cookie信息到本地文件。后续测试运行时，脚本会先加载这些Cookie，直接跳过登录环节进入系统内部页面。

2. 技术实现详解

2.1 核心组件选型

我们选择Python+Selenium组合作为技术基础，主要考虑以下因素：

• Selenium的WebDriver API成熟稳定，支持所有主流浏览器
• Python生态有完善的Cookie处理库（如pickle）
• 易于集成到现有自动化测试框架中

关键依赖包：

from selenium import webdriver import pickle import time import os

2.2 Cookie存储与加载机制

首次运行时的Cookie保存流程：

def save_cookies(driver, path): with open(path, 'wb') as filehandler: pickle.dump(driver.get_cookies(), filehandler)

后续测试的Cookie加载流程：

def load_cookies(driver, path): with open(path, 'rb') as cookiesfile: cookies = pickle.load(cookiesfile) for cookie in cookies: driver.add_cookie(cookie)

2.3 完整工作流程实现

1. 初始化浏览器驱动

driver = webdriver.Chrome() driver.get("https://target-system.com/login")

2. 检查Cookie文件是否存在

cookie_file = 'cookies.pkl' if os.path.exists(cookie_file): load_cookies(driver, cookie_file) driver.refresh() # 刷新使Cookie生效 else: input("请手动完成登录后按回车继续...") save_cookies(driver, cookie_file)

3. 验证登录状态

if "dashboard" in driver.current_url: print("登录状态有效") else: print("需要重新获取Cookie") os.remove(cookie_file)

3. 关键问题与解决方案

3.1 Cookie时效性问题

我们发现不同系统的Cookie有效期差异很大：

• 银行类系统：通常15-30分钟
• SaaS平台：多数7-30天
• 内部系统：可能长期有效

解决方案：

# 在load_cookies前增加时效检查 def check_cookie_valid(cookie_file): if not os.path.exists(cookie_file): return False create_time = os.path.getmtime(cookie_file) return (time.time() - create_time) < 3600*24 # 24小时有效期

3.2 跨域Cookie限制

现代浏览器对Cookie的SameSite属性限制越来越严格。我们通过以下方式解决：

1. 确保测试域名与目标系统完全一致
2. 在ChromeOptions中添加：

options = webdriver.ChromeOptions() options.add_argument("--disable-web-security") options.add_argument("--disable-site-isolation-trials")

3.3 多用户场景处理

对于需要测试多账号的场景，我们改进存储策略：

def get_user_cookie_file(username): return f'cookies_{username}.pkl'

4. 实战优化技巧

4.1 自动化等待机制

在关键节点添加智能等待：

from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC WebDriverWait(driver, 10).until( EC.presence_of_element_located((By.ID, "main-content")) )

4.2 异常处理增强

增加网络波动时的重试机制：

max_retry = 3 for attempt in range(max_retry): try: driver.refresh() break except WebDriverException: if attempt == max_retry - 1: raise time.sleep(2)

4.3 无头模式适配

在CI/CD环境中使用时：

options.add_argument("--headless") options.add_argument("--disable-gpu")

5. 方案对比与选型建议

与传统验证码处理方案对比：

选择建议：

• 测试环境优先使用Cookie方案
• 预发布环境可结合OCR方案
• 生产验证必须使用人工验证

6. 扩展应用场景

这种技术方案还可应用于：

1. 需要定期抓取数据的后台系统
2. 微信小程序自动化测试（通过DevTools获取Cookie）
3. 需要保持登录状态的爬虫项目
4. 多系统间的SSO测试

对于需要更高安全性的系统，可以结合以下技术：

# 加密存储Cookie from cryptography.fernet import Fernet key = Fernet.generate_key() cipher_suite = Fernet(key) encrypted_cookie = cipher_suite.encrypt(pickle.dumps(cookies))

7. 安全注意事项

1. Cookie文件应加入.gitignore
2. 设置适当的文件权限（600）
3. 避免在共享环境中使用
4. 定期清理过期Cookie
5. 敏感系统建议使用临时测试账号

实现安全存储的改进方案：

import keyring def secure_save_cookies(driver, service_name): keyring.set_password(service_name, "cookies", pickle.dumps(driver.get_cookies())) def secure_load_cookies(driver, service_name): cookies = pickle.loads(keyring.get_password(service_name, "cookies")) for cookie in cookies: driver.add_cookie(cookie)

8. 性能优化实践

通过实测发现几个优化点：

1. Cookie加载时间对比：

   • 首次手动登录：平均12秒
   • Cookie复用登录：平均1.3秒

2. 内存占用优化：

# 减少不必要的Cookie属性 def clean_cookie(cookie): return {k: cookie[k] for k in ['name', 'value', 'domain', 'path']}

3. 并行测试支持：

# 使用多进程隔离Cookie from multiprocessing import Process def run_test(username): driver = webdriver.Chrome() load_cookies(driver, get_user_cookie_file(username)) # 执行测试用例

9. 企业级实施方案

在大型测试框架中的集成方式：

1. 创建Cookie管理中间件

class CookieManager: def __init__(self, config): self.storage_path = config['cookie_path'] def save(self, driver): # 实现版本控制 version = datetime.now().strftime("%Y%m%d_%H%M") path = f"{self.storage_path}_{version}.pkl" save_cookies(driver, path) def load_latest(self, driver): # 查找最新版本Cookie pass

2. 与PageObject模式结合：

class LoginPage: def __init__(self, driver): self.driver = driver self.cookie_manager = CookieManager() def bypass_login(self): if not self.cookie_manager.load_latest(self.driver): self.perform_manual_login()

10. 未来演进方向

1. 动态Cookie更新机制：

   • 定期自动刷新Cookie
   • 失效自动通知

2. 云存储方案：

   • 将Cookie存入AWS Secrets Manager
   • 实现团队共享

3. 智能失效检测：

def check_login_state(driver): try: driver.find_element(By.CSS_SELECTOR, ".user-avatar") return True except: return False

这套方案在我们团队已经稳定运行2年多，累计节省了超过500小时的手动验证时间。特别是在每日构建的回归测试中，将原本需要人工值守的验证环节完全自动化，使夜间测试成为可能。