AI 测试工具平台

基于 AI 大模型 + LangGraph + Playwright 的智能化零代码全场景自动化测试平台，覆盖 UI 自动化与接口自动化双引擎，支持 Claude / DeepSeek / GPT / Gemini 等任意模型一键切换。

源码链接：https://github.com/ywq2019/ai_test_agent.git

快速部署

方式一：Docker 一键部署（推荐，适合服务器）

服务器已安装 Docker 和 Docker Compose 即可：

git clone https://github.com/ywq2019/ai_test_agent.git && cd ai_test_agent && docker compose up -d

启动后访问 http://服务器IP:4000，进入大模型配置页填写 API Key。

git pull && docker compose up -d --build  # 更新
docker compose logs -f                     # 查看日志
docker compose down                        # 停止

方式二：本地启动

环境要求：Python 3.11+、Node.js 18+

生产模式（单端口，前端打包后由后端托管）：

# 1. 安装依赖
pip install -r requirements.txt
playwright install chromium

# 2. 构建前端（编译产物由后端静态托管）
cd ui && npm install && npm run build && cd ..

# 3. 启动
python main.py

访问 http://localhost:4000，前端由后端统一托管，只需启动一个进程。

开发模式（双端口，支持热更新，适合开发调试）：

# 终端1：后端（API 服务，端口 4000）
python main.py

# 终端2：前端 Vite 开发服务器（端口 8090，自动代理 /api 请求到后端 4000）
cd ui && npm run dev

访问 http://localhost:8090，前端代码修改实时生效。

两个端口说明：生产模式下 FastAPI 直接托管 ui/dist/ 静态文件，访问 4000 即可；开发模式下 Vite 在 8090 启动独立服务并将 /api、/ws 等请求代理转发到后端 4000。

局域网共享（Windows 团队使用）

本机启动后，团队成员可通过内网 IP 直接访问，无需额外配置：

# 查看本机内网 IP（找以太网或 WLAN 的 IPv4 地址）
ipconfig

开放防火墙入站端口：

New-NetFirewallRule -DisplayName "AI测试平台" -Direction Inbound -Protocol TCP -LocalPort 4000 -Action Allow

团队成员访问 http://<本机IP>:4000 即可使用，例如 http://192.168.1.100:4000。

登录账号

平台内置 JWT 鉴权，首次启动自动创建默认账号：

登录后点击顶栏右上角「用户管理」可新建账号，普通用户只能登录和修改自己的密码。

核心功能

AI 用例生成（文档驱动）

上传需求文档，AI 按功能模块并行生成覆盖 6 种测试方法的高质量用例，导出 Markdown / XMind。

WebUI 自动化

Playwright 抓取页面元素，AI 生成含具体 selector 的可执行用例，直接运行自动化测试。

接口自动化

接口测试

测试计划

跨项目接口用例编排，共享变量上下文，实现端到端链路测试（如：登录 → 下单 → 查询）。

技术栈

后端

前端

AI 集成

所有 AI 功能统一读取「大模型配置」页的设置，自动判断 Anthropic / OpenAI 格式，支持一键切换：

所有 Prompt 统一管理在 skills/prompts/*.yaml，无需改代码即可调整 AI 生成效果。

项目结构

ai_test_agent/
├── main.py                 # 应用入口（含 JWT 鉴权中间件、默认账号初始化）
├── Dockerfile / docker-compose.yml
├── api/
│   ├── auth.py             # JWT 工具 + get_current_user 依赖注入
│   ├── routes.py           # 全部 REST 端点（含用户管理、增量更新、代码分析）
│   ├── websocket.py        # WebSocket 端点
│   └── websocket_manager.py
├── skills/
│   ├── ai_case_generator.py    # AI 文档驱动用例生成（分段 + 增量更新）
│   ├── case_generator.py       # UI 用例生成（分段 + 增量更新）
│   ├── api_case_generator.py   # 接口用例生成 + 代码分析
│   ├── api_executor.py         # 接口用例执行（断言 + 变量提取 + 前置依赖）
│   ├── api_load_tester.py      # 压力测试引擎
│   ├── param_resolver.py       # 参数解析（变量 / 函数 / 脚本）
│   ├── prompt_loader.py        # Prompt YAML 加载器
│   └── prompts/                # LLM Prompt 配置（YAML，无需改代码即可调整）
│       ├── ai_case_gen.yaml
│       ├── api_case_gen.yaml
│       ├── code_analyze.yaml
│       └── ui_case_gen.yaml
├── tools/
│   ├── database.py         # SQLAlchemy 模型（兼容 SQLite / PostgreSQL）
│   ├── config.py           # 环境变量（含 JWT 配置）
│   └── browser.py          # Playwright 封装（networkidle + 移动端 UA）
├── ui/src/
│   ├── api/index.js        # Axios 封装（JWT 拦截器 + 401 跳登录）
│   ├── stores/auth.js      # Pinia Auth Store（token 持久化）
│   ├── views/Login.vue     # 登录页
│   └── views/ApiTest/      # 接口测试组件（拆分后）
│       └── ScriptDialog.vue
└── tests/                  # 单元测试
    ├── test_param_resolver.py   # 27 条
    ├── test_api_executor.py     # 27 条
    └── test_incremental.py      # 12 条

接口说明

完整文档：http://localhost:4000/docs

认证

接口测试（部分）

WebSocket 频道

截图

许可证

本项目采用 MIT 许可证，详见 LICENSE 文件。