auto-pytest-generator 1.0.0

Generate pytest tests from captured fetch/XHR traffic via mitmproxy

Smoke-Test-Recorder

本工具旨在为0自动化覆盖的团队快速创建一套基于API流量的冒烟测试用例。它通过启动一个本地代理，捕获您在浏览器中与被测系统交互时产生的fetch/XHR流量，并自动生成基于requests的pytest测试代码。

项目结构

auto-pytest-generator/
│
├── src/
│   └── auto_pytest_generator/
│       ├── __init__.py
│       ├── main.py             # CLI入口和mitmproxy启动逻辑
│       ├── dispatcher_addon.py # 网络流量分发器，过滤静态文件流量，分发流量对应的pytest_generator实例
|       ├── pytest_generator.py # 使用jinja2生成pytest用例的模板逻辑
|       ├── response_refresher.py # 用于更新已有用例的response断言部分
│       └── templates/
│           └── pytest_template.jinja2 # Jinja2模板文件
│
├── tests/
│   └── ...                     # 工具自身的单元测试
│
├── .gitignore
├── pyproject.toml              # 项目打包和依赖管理
└── README.md                   # 使用文档

特点

• 0配置启动：只需指定目标系统的URL前缀即可开始。
• 自动化生成：在浏览器中操作，用例自动生成。
• 基于真实数据断言：使用录制时的响应体做断言，无需依赖接口文档。
• 易于集成：生成的代码是标准的pytest格式，可直接运行。

安装

# 推荐使用uv或pipx进行安装，以避免污染全局环境
uv tool install auto-pytest-generator

# 或者使用pip
pip install auto-pytest-generator

使用

初次录制

• 启动代理服务 uv run apg proxy --url-prefix http://1.2.3.4:5678/
• 配置浏览器代理到apg代理服务
• 访问 http://mitm.it 检查是否代理成功
• 开始在客户端操作，用例被生成在./generated_tests/
• 拷贝所有*.py文件到测试用例运行环境(依赖pytest,requests)

二次录制

• 已有录制的用例在./tests/中
• (可选)修改./tests/config中修改运行环境和auth token
• 运行uv run apg response-refresh --origin_dir ./tests/ --target-dir ./refreshed_tests/
• apg会自动根据config自动重新运行接口用例，并将结果输出到./refreshed_tests/unchanged/ ./refreshed_tests/data_changed/ ./refreshed_tests/error/ 下