BrowserUse自动化测试效率优化：Cookie复用实现登录状态持久化

BrowserUse自动化测试效率优化：Cookie复用实现登录状态持久化

在Web自动化测试中，登录状态的维护一直是影响测试效率的关键因素。本文将详细介绍如何利用BrowserUse工具的特性，通过Cookie复用机制跳过重复登录步骤，大幅提升测试用例的执行效率和可维护性。

测试场景的痛点分析

在网站功能测试过程中，我们经常面临这样的困境：

• 所有测试用例都依赖登录状态，但登录功能本身并非测试重点
• 每个用例单独执行登录步骤会导致测试时间呈线性增长
• 集中式前置登录会造成用例耦合度高，故障排查困难
• 频繁的登录操作可能触发网站的安全验证机制（如验证码）

传统解决方案要么牺牲测试效率，要么降低用例独立性。而通过复用浏览器Cookie实现登录状态持久化，成为平衡效率与独立性的理想选择。

BrowserUse的状态管理方案

BrowserUse作为现代化的浏览器自动化工具，提供了storage_state参数专门用于管理浏览器的存储状态，其核心特性包括：

• 支持一次性加载Cookies、localStorage等状态信息
• 无需重复执行登录流程即可恢复用户会话
• 兼容主流浏览器的存储机制
• 可与user_data_dir配合实现更复杂的状态管理

关键参数配置示例：

browser = Browser(
    window_size={'width': 1920, 'height': 1440},
    storage_state="./login_cookies.json",  # 指定状态文件路径
    headless=False,
    keep_alive=True  # 保持会话连续性
)

完整解决方案实施步骤

1. 导出登录状态的Cookie

使用浏览器插件（如Get cookies.txt）获取登录后的Cookie信息：

1. 手动登录目标网站（如WPS会员中心）
2. 启动插件，选择当前网站域名
3. 导出Cookie为JSON格式文件
4. 保存文件（如original_cookies.json）

注意：导出时确保包含所有与身份验证相关的Cookie，尤其是包含session、token等关键词的条目

2. Cookie格式转换工具

浏览器插件导出的Cookie格式与BrowserUse所需格式存在差异，需进行转换。以下是优化后的转换脚本：

import json
import os

def load_json_file(file_path):
    """加载JSON文件并返回数据"""
    if not os.path.exists(file_path):
        raise FileNotFoundError(f"文件 {file_path} 不存在")
    
    with open(file_path, 'r', encoding='utf-8') as f:
        try:
            return json.load(f)
        except json.JSONDecodeError:
            raise ValueError(f"文件 {file_path} 不是有效的JSON格式")

def save_json_file(data, file_path, indent=2):
    """将数据保存为JSON文件"""
    with open(file_path, 'w', encoding='utf-8') as f:
        json.dump(data, f, ensure_ascii=False, indent=indent)
    print(f"成功保存文件: {file_path}")
    return True

def convert_to_browseruse_format(original_cookies):
    """将原始Cookie转换为BrowserUse兼容格式"""
    converted = []
    for cookie in original_cookies:
        # 映射关键字段并处理默认值
        converted_cookie = {
            "name": cookie.get("name", ""),
            "value": cookie.get("value", ""),
            "domain": cookie.get("domain", ""),
            "path": cookie.get("path", "/"),
            "expires": int(cookie["expirationDate"]) if "expirationDate" in cookie else -1,
            "httpOnly": cookie.get("httpOnly", False),
            "secure": cookie.get("secure", False),
            "sameSite": cookie.get("sameSite", "Lax")
        }
        # 处理会话Cookie（无过期时间）
        if converted_cookie["expires"] == -1:
            converted_cookie["session"] = True
        converted.append(converted_cookie)
    return converted

def main():
    # 配置文件路径
    input_file = 'original_cookies.json'
    output_file = 'browseruse_cookies.json'

    try:
        # 读取原始Cookie
        original_cookies = load_json_file(input_file)
        
        # 转换格式
        converted_cookies = convert_to_browseruse_format(original_cookies)
        
        # 构建完整的storage_state结构
        storage_state = {
            "cookies": converted_cookies,
            "origins": []  # 预留localStorage等存储区域
        }
        
        # 保存转换结果
        save_json_file(storage_state, output_file)
        print("Cookie格式转换完成，已生成BrowserUse兼容文件")

    except Exception as e:
        print(f"处理失败: {str(e)}")

if __name__ == "__main__":
    main()

转换工具解决了以下关键问题：

• 字段映射：将插件导出的expirationDate转换为BrowserUse需要的expires
• 类型处理：确保日期字段为整数类型
• 会话Cookie标识：为无过期时间的Cookie添加session: true标记
• 结构补全：生成包含cookies和origins的完整状态结构

3. 测试脚本集成与验证

在测试脚本中集成转换后的Cookie文件，实现登录状态自动恢复：

import asyncio
from browser_use import Agent, Browser
from browser_use.llm import ChatDeepSeek
from dotenv import load_dotenv
import os

async def main():
    # 加载环境变量
    load_dotenv()
    deepseek_api_key = os.getenv('DEEPSEEK_API_KEY')
    if not deepseek_api_key:
        print('未找到DEEPSEEK_API_KEY环境变量，请设置后重试')
        return

    # 初始化LLM
    llm = ChatDeepSeek(
        base_url='https://api.deepseek.com/v1',
        model='deepseek-chat',
        api_key=deepseek_api_key
    )

    # 配置浏览器，加载转换后的Cookie
    browser = Browser(
        window_size={'width': 1920, 'height': 1440},
        storage_state="./browseruse_cookies.json",  # 使用转换后的Cookie文件
        headless=False,
        keep_alive=True
    )

    # 创建测试代理
    agent = Agent(
        task="访问WPS会员中心，验证登录状态并查看会员信息",
        llm=llm,
        browser=browser
    )

    # 执行测试任务
    await agent.run()
    
    # 关闭浏览器
    await browser.stop()

if __name__ == "__main__":
    asyncio.run(main())

实施效果与最佳实践

显著收益

• 测试效率提升：单个用例执行时间减少40%-60%（省去登录环节）
• 用例独立性增强：每个用例可单独运行，互不干扰
• 维护成本降低：登录逻辑变更时只需更新Cookie转换规则
• 稳定性提高：减少因频繁登录导致的验证码触发风险

注意事项

1. Cookie有效期管理：

   • 定期更新Cookie文件（根据网站会话有效期）
   • 实现Cookie自动检测机制，过期时自动重新生成

2. 安全性保障：

   • 加密存储敏感Cookie文件
   • 测试完成后自动清理本地Cookie文件
   • 避免在公共环境中使用真实用户Cookie

3. 跨环境适配：

   • 为不同测试环境（开发/测试/预生产）维护独立Cookie
   • 转换工具中添加环境标识字段

通过BrowserUse的storage_state特性与Cookie复用机制，我们成功解决了自动化测试中登录状态管理的痛点问题。这种方案既保持了测试用例的独立性，又大幅提升了执行效率，特别适合需要频繁访问登录后页面的测试场景。随着Web应用安全机制的不断升级，我们还需要持续优化Cookie管理策略，确保自动化测试的稳定性和可靠性。