protagonistss/test-generation

单元测试生成技能

这个技能负责分析源代码并生成对应的单元测试用例。支持多种编程语言和测试框架。

核心能力

1. 代码分析

• AST解析，识别函数、类、方法
• 分析函数签名和参数类型
• 识别依赖关系和导入模块
• 检测异步函数和Promise使用

2. 测试场景生成

• 正常流程测试（Happy Path）
• 边界条件测试（Boundary Values）
• 异常情况测试（Error Cases）
• 参数验证测试

3. 测试代码生成

• 根据选择的框架生成测试代码
• 生成合适的测试名称和描述
• 添加必要的设置和清理代码
• 生成有意义的断言

支持的框架

JavaScript/TypeScript

• Jest - 最流行的JavaScript测试框架
• Vitest - 现代化的Vite原生测试框架
• Mocha - 灵活的测试框架
• Jasmine - 行为驱动的开发框架

Python

• pytest - 功能强大的测试框架
• unittest - Python标准库测试框架
• nose2 - unittest的扩展

Java

• JUnit 5 - 现代Java测试框架
• TestNG - 测试下一代框架
• Mockito - 强大的Mock框架

其他语言

• Go (testing)
• C# (xUnit, NUnit)
• Ruby (RSpec, Minitest)

使用方法

基础使用

# 生成单个函数的测试
/test src/utils/calculator.js --function add --framework vitest

# 生成整个文件的测试
/test src/services/userService.js --framework jest --mocks

高级选项

# 包含覆盖率分析
/test src/api/userController.js --framework jest --coverage

# 生成边界值测试
/test src/utils/validator.js --edge-cases

# 只生成异常测试
/test src/services/payment.js --error-only

测试模板示例

JavaScript/TypeScript - Vitest

import { describe, it, expect, beforeEach, vi } from 'vitest';
import { UserService } from '../src/services/UserService';

describe('UserService', () => {
  let userService: UserService;
  let mockDb: any;

  beforeEach(() => {
    mockDb = {
      find: vi.fn(),
      create: vi.fn(),
      update: vi.fn(),
      delete: vi.fn()
    };
    userService = new UserService(mockDb);
  });

  describe('getUserById', () => {
    it('should return user when found', async () => {
      // Arrange
      const userId = '123';
      const expectedUser = { id: userId, name: 'John' };
      mockDb.find.mockResolvedValue(expectedUser);

      // Act
      const result = await userService.getUserById(userId);

      // Assert
      expect(result).toEqual(expectedUser);
      expect(mockDb.find).toHaveBeenCalledWith({ id: userId });
    });

    it('should return null when user not found', async () => {
      // Arrange
      const userId = '999';
      mockDb.find.mockResolvedValue(null);

      // Act
      const result = await userService.getUserById(userId);

      // Assert
      expect(result).toBeNull();
    });
  });
});

Python - pytest

import pytest
from unittest.mock import Mock, patch
from services.user_service import UserService

class TestUserService:
    @pytest.fixture
    def user_service(self):
        with patch('services.user_service.Database') as mock_db:
            yield UserService(mock_db)

    def test_get_user_by_id_success(self, user_service):
        """Test getting user with valid ID"""
        # Arrange
        user_id = 123
        expected_user = {"id": user_id, "name": "John"}
        user_service.db.find.return_value = expected_user

        # Act
        result = user_service.get_user_by_id(user_id)

        # Assert
        assert result == expected_user
        user_service.db.find.assert_called_once_with({"id": user_id})

    def test_get_user_by_id_not_found(self, user_service):
        """Test getting non-existent user"""
        # Arrange
        user_id = 999
        user_service.db.find.return_value = None

        # Act
        result = user_service.get_user_by_id(user_id)

        # Assert
        assert result is None

测试数据生成

自动生成测试数据

// 根据类型生成测试数据
const testData = {
  string: ["hello", "", "a".repeat(255), "特殊字符"],
  number: [0, 1, -1, 100, Number.MAX_SAFE_INTEGER],
  boolean: [true, false],
  array: [[], [1], [1, 2, 3], Array(1000).fill(0)],
  object: [{}, { key: "value" }, null, undefined]
};

// 边界值生成
const boundaries = {
  string: ["", "a", "a".repeat(255), "a".repeat(256)],
  number: [Number.MIN_VALUE, -1, 0, 1, Number.MAX_VALUE],
  array: [[] , [1], [1000]],
};

Mock策略

自动Mock外部依赖

• 识别外部模块导入
• 自动生成Mock代码
• 配置Mock返回值
• 验证Mock调用

Mock示例

// 自动生成的Mock
jest.mock('../src/utils/logger', () => ({
  logger: {
    info: jest.fn(),
    error: jest.fn(),
    warn: jest.fn()
  }
}));

// 测试中验证Mock调用
expect(logger.info).toHaveBeenCalledWith('User created successfully');

覆盖率优化

智能测试补充

• 分析代码覆盖率报告
• 识别未测试的分支
• 自动生成补充测试用例
• 优化测试数据组合

覆盖率目标

• 语句覆盖率 > 90%
• 分支覆盖率 > 85%
• 函数覆盖率 > 95%
• 行覆盖率 > 90%

最佳实践

1. 测试命名规范

// ✅ 好的命名
it('should create user with valid data');
it('should throw error when email already exists');

// ❌ 避免的命名
it('test1');
it('user creation test');

2. AAA模式

it('should calculate discount correctly', () => {
  // Arrange - 准备测试数据
  const price = 100;
  const discountRate = 0.1;

  // Act - 执行被测代码
  const result = calculateDiscount(price, discountRate);

  // Assert - 验证结果
  expect(result).toBe(90);
});

3. 测试隔离

• 每个测试独立运行
• 使用beforeEach/afterEach清理
• 避免测试间的依赖

4. 有意义的断言

// ✅ 具体的断言
expect(user.email).toMatch(/^[^\s@]+@[^\s@]+\.[^\s@]+$/);

// ❌ 模糊的断言
expect(user).toBeDefined();

常见问题处理

1. 异步代码测试

// Promise
it('should resolve with data', async () => {
  const result = await fetchData();
  expect(result).toEqual(expectedData);
});

// Callback
it('should call callback', (done) => {
  fetchData((data) => {
    expect(data).toBeDefined();
    done();
  });
});

2. 错误处理测试

it('should throw error for invalid input', () => {
  expect(() => validateEmail('invalid')).toThrow();
});

it('should reject promise on error', async () => {
  await expect(asyncOperation()).rejects.toThrow('Error message');
});

3. 时间相关测试

// 使用假时间
beforeEach(() => {
  vi.useFakeTimers();
});

afterEach(() => {
  vi.useRealTimers();
});

it('should debounce function calls', async () => {
  const debouncedFn = debounce(originalFn, 100);

  debouncedFn();
  vi.advanceTimersByTime(50);
  expect(originalFn).not.toHaveBeenCalled();

  vi.advanceTimersByTime(50);
  expect(originalFn).toHaveBeenCalledTimes(1);
});

集成命令

这个技能与其他命令集成：

• /test - 生成测试用例
• /mock - 生成Mock数据
• /coverage - 分析测试覆盖率

通过智能分析和模板化生成，帮助开发者快速编写高质量的单元测试。