Artillery项目中Playwright存储状态的动态加载实践

背景介绍

Artillery是一个现代化的负载测试工具，它支持使用Playwright进行浏览器级别的性能测试。在实际的性能测试场景中，经常需要模拟不同用户角色的登录状态，这时就需要动态加载不同的存储状态(storageState)文件。

问题分析

在Artillery配置文件中，开发者尝试通过参数化方式动态指定Playwright的storageState路径，期望从CSV文件中读取不同用户角色(userRole)对应的存储状态文件。初始配置中使用了storageState: "./auth/{{ userRole }}.json"的语法，但运行时却遇到了"undefined.json"文件不存在的错误。

解决方案探索

方案一：直接参数化路径（失败）

最初尝试在contextOptions中直接使用模板语法引用payload字段：

contextOptions: 
  storageState: "./auth/{{ userRole }}.json"

这种方法失败的原因是Artillery的模板替换机制在contextOptions配置层级不适用，导致userRole变量未被正确解析。

方案二：使用addCookies方法（成功）

最终解决方案是放弃直接配置storageState，转而使用Playwright提供的page.context().addCookies()方法动态添加认证信息。这种方法更加灵活，可以直接在测试脚本中控制认证流程。

实现建议

对于需要在Artillery中实现多用户角色测试的场景，推荐以下实践：

1. 准备阶段：

   • 为每个用户角色预先录制好登录状态，保存为单独的JSON文件
   • 创建CSV文件映射用户角色到对应的存储状态文件

2. 脚本编写：

   async function testflow1(page, context, user) {
     // 动态加载对应角色的cookies
     const cookies = require(`./auth/${user.userRole}.json`);
     await page.context().addCookies(cookies.cookies);
     
     // 继续测试流程
     // ...
   }
   

3. 配置优化：

   • 保持payload配置不变，确保userRole字段可用
   • 在测试脚本中实现存储状态的动态加载逻辑

技术要点

1. Playwright存储状态：存储状态文件包含了会话cookies、localStorage等信息，可以避免重复登录。

2. Artillery payload机制：支持从CSV文件读取测试数据，为每个虚拟用户提供不同的参数。

3. 上下文隔离：每个虚拟用户拥有独立的浏览器上下文，确保测试隔离性。

最佳实践

1. 对于简单的多用户测试，优先考虑使用addCookies方法
2. 对于复杂场景，可以结合Artillery的before钩子进行全局初始化
3. 存储状态文件应包含在版本控制中，但需注意排除敏感信息
4. 定期更新存储状态文件，避免会话过期导致测试失败

通过这种方式，可以有效地在Artillery性能测试中实现多用户角色的模拟，提高测试的真实性和覆盖范围。