TestGen AI 🤖 — 基于 Node.js 的智能单元测试生成工具
一款基于 Node.js 的智能单元测试生成工具,支持通过自然语言交互快速生成完善的测试用例。
功能特性 ✨
- 🚀 AI 智能生成:通过 OpenAI 模型自动生成测试代码
- 🔄 检测更新:仅生成新增/修改方法的测试用例
- 📁 配置管理:支持本地配置文件和环境变量
- ✅ 多框架支持:支持 Jest/Mocha 测试框架
- 💬 交互模式:提供友好的命令行交互界面
功能演示 🖥️
选择需要生成的JS或TS文件,例子:
// math.ts
export function getSizeImage(imgUrl: string, size: number):string {
return `${imgUrl}?param=${size}x${size}`;
}
export function getCount(count: number):number | string {
if (count < 0) return;
if (count < 10000) {
return count;
} else if (Math.floor(count / 10000) < 10000) {
return Math.floor(count / 1000) / 10 + "万";
} else {
return Math.floor(count / 10000000) / 10 + "亿";
}
}

选择需要生成的方法

等待AI生成

生成成功


第二次生成,生成过的方法不会再出现

第二次生成,代码会自动合并

安装使用指南 📦
前置要求
- Node.js v18+
- npm 6+
- 申请AI模型API-Key
安装方式
# 全局安装
npm install testgen-ai
使用方式
-
初始化,生成 testgen.config.json
npx testgen -i -
配置环境变量
- 访问bailian.console.aliyun.com/?spm=5176.2…
- 选取其中一个模型,打开 API调用示例 复制baseURL、apiKey、model到 testgen.config.json中


-
运行项目,根据命令操作
npx testgen
配置文件参数说明 📄
{
"openai": {
"apiKey": "",// api key
"baseURL": "", // 请求url
"model": "" // 模型
},
"config": {
"cache": true, // 为true 代表缓存这次生成用例的方法,下次会略过
"unit": "jest" // 单元测试框架
}
}
程序流程图

Github地址:github.com/JackPennnnn…