Generate Unit Testing Server

MCP 单元测试生成服务

一个智能的 Model Context Protocol（MCP）服务，可使用 Jest、Karma/Jasmine 或其他测试框架，为 Angular、React 和 Next.js 项目自动生成全面的单元测试。

功能

• 🔍 自动检测项目 - 检测 Angular、React 或 Next.js 项目及其版本
• 📦 框架管理 - 检查测试框架，并提供自动安装选项
• 📚 实时获取文档 - 生成测试前获取最新的 Jest 和 Jasmine 文档
• 🤖 由 LLM 驱动的生成 - 使用 Claude Sonnet 4.5+ 智能生成测试
• 🎯 高覆盖率目标 - 生成以代码覆盖率 >85% 为目标的测试
• ✅ 所有公共方法 - 为所有导出的函数和公共类方法创建测试
• 🔄 并排验证 - 并排显示源代码和测试代码，方便审查
• 🛠️ 多种框架 - 支持 Jest、Karma/Jasmine 和 Vitest

安装

npm install
npm run build

使用方法

作为 MCP 服务

将其添加到 MCP 客户端配置中（例如，Claude Desktop、带 MCP 的 VS Code）：

{
  "mcpServers": {
    "generate-unit-testing": {
      "command": "node",
      "args": ["/path/to/GenerateUnitTesting/build/index.js"]
    }
  }
}

可用工具

1. detect_project

检测项目类型和测试框架。

{
  "filePath": "/path/to/your/project/src/component.ts"
}

2. install_test_framework

在获得用户许可后安装缺失的测试框架。

{
  "projectPath": "/path/to/your/project",
  "framework": "jest",
  "autoInstall": false
}

3. generate_unit_test

为源文件生成全面的单元测试。

{
  "filePath": "/path/to/your/project/src/component.ts",
  "framework": "jest",
  "coverageTarget": 85,
  "generateForAllPublicMethods": true,
  "includeEdgeCases": true
}

4. validate_test

运行生成的测试并计算覆盖率。

{
  "testFilePath": "/path/to/your/project/src/component.test.ts",
  "framework": "jest"
}

5. display_side_by_side

创建源代码和测试代码的并排对比。

{
  "sourceFilePath": "/path/to/your/project/src/component.ts",
  "testFilePath": "/path/to/your/project/src/component.test.ts"
}

可用资源

• docs://jest/latest - 最新 Jest 文档
• docs://jasmine/latest - 最新 Jasmine 文档

可用提示词

• generate-unit-test - 全面的单元测试生成提示词
• generate-edge-cases - 边界情况测试生成提示词
• generate-mocks - Mock 生成提示词

架构

src/
├── index.ts                 # Main MCP server entry point
├── types.ts                 # TypeScript type definitions
├── tools/
│   ├── detectProject.ts     # Project and framework detection
│   └── testGenerator.ts     # Test generation logic
├── resources/
│   └── documentationFetcher.ts  # Live doc fetching from Jest/Jasmine
├── templates/
│   └── testTemplates.ts     # Framework-specific test templates
├── prompts/
│   └── testPrompts.ts       # LLM prompts for test generation
└── utils/
    ├── fileUtils.ts         # File system utilities
    └── codeAnalyzer.ts      # AST-based code analysis

工作流

1. 项目检测 - 分析 package.json，识别项目类型和现有测试配置
2. 框架检查 - 验证测试框架是否已安装；如果缺失，则提示安装
3. 获取文档 - 从官方来源获取最新框架文档
4. 代码分析 - 使用 Babel 解析源代码，提取函数、类和方法
5. 测试生成 - 使用包含上下文（代码 + 文档 + 模板）的 LLM 生成全面的测试
6. 验证 - 运行测试并计算覆盖率
7. 显示 - 并排显示对比内容，供开发者审查

支持的框架

Jest

• React 和 Next.js 项目
• TypeScript/JavaScript
• 现代 async/await 模式
• Mock 函数和间谍

Karma/Jasmine

• Angular 项目
• 使用 TestBed 进行组件测试
• 服务和管道测试
• 传统的 describe/it 语法

覆盖率目标

该服务以 >85% 代码覆盖率 为目标，生成以下测试：

• 所有公共函数的测试
• 所有公共类方法的测试
• 边界情况场景
• 错误处理测试
• 异步操作测试
• 依赖项的 Mock 实现

输出示例

以 React 组件为例：

// Original: Button.tsx
export const Button = ({ onClick, label }) => {
  return <button onClick={onClick}>{label}</button>;
};

生成的测试：

// Generated: Button.test.tsx
import { render, fireEvent } from '@testing-library/react';
import { Button } from './Button';

describe('Button', () => {
  it('should render with label', () => {
    const { getByText } = render(<Button label="Click me" onClick={() => {}} />);
    expect(getByText('Click me')).toBeInTheDocument();
  });

  it('should call onClick when clicked', () => {
    const handleClick = jest.fn();
    const { getByText } = render(<Button label="Click me" onClick={handleClick} />);
    fireEvent.click(getByText('Click me'));
    expect(handleClick).toHaveBeenCalledTimes(1);
  });

  // ... more tests for edge cases;

要求

• Node.js 18+
• TypeScript 5+
• MCP 客户端（Claude Desktop、带 MCP 扩展的 VS Code 等）

开发

# Install dependencies
npm install

# Build
npm run build

# Watch mode
npm run watch

# Start server
npm start

配置

服务会自动检测：

• 包管理器（npm、yarn、pnpm）
• 项目类型（Angular、React、Next.js）
• 现有测试框架
• TypeScript/JavaScript

无需手动配置！

许可证

MIT

贡献

欢迎贡献！请确保：

• 代码遵循现有模式
• 为新功能添加测试
• 更新文档

支持

• 如有问题或疑问，请在代码仓库中提交 issue。
• 作者：Ahmadreza Shamimi (https://github.com/shentia)