测试生成器

测试生成器

触发条件

当用户要求"生成测试"、"写单元测试"、"为 XX 函数写测试"、"增加测试覆盖率"时激活此技能。

工作流程

1. 分析目标代码

• 读取目标文件，识别函数/类的输入输出
• 分析参数类型、返回值、副作用
• 识别依赖关系（数据库、API、文件系统等）

2. 确定测试策略

• 纯函数：参数化测试，覆盖正常/边界/异常输入
• 类方法：测试初始化、正常调用、状态变化
• 异步函数：测试成功路径、超时、网络错误
• 含副作用的函数：使用 mock/​stub 隔离依赖

3. 生成测试用例

每个函数至少覆盖：

• ✅ 正常输入（happy path）
• ⚠️ 边界条件（空值、零值、极大值、空字符串/数组）
• ❌ 异常输入（类型错误、格式错误）
• 🔗 依赖失败（网络超时、数据库异常）

4. 输出格式

根据项目语言和测试框架生成对应代码：

Python (pytest)

import pytest
from mymodule import my_function

class TestMyFunction:
    def test_normal_input(self):
        assert my_function("hello") == "HELLO"
    
    def test_empty_string(self):
        assert my_function("") == ""
    
    def test_none_input(self):
        with pytest.raises(TypeError):
            my_function(None)
    
    def test_special_characters(self):
        assert my_function("你好!@#") == "你好!@#"

JavaScript (Jest)

const { myFunction } = require('./​mymodule');

describe('myFunction', () => {
  test('正常输入', () => {
    expect(myFunction('hello')).toBe('HELLO');
  });
  
  test('空字符串', () => {
    expect(myFunction('')).toBe('');
  });
  
  test('null 输入', () => {
    expect(() => myFunction(null)).toThrow(TypeError);
  });
});

Go (testing)

package mypackage

import "testing"

func TestMyFunction(t *testing.T) {
    tests := []struct {
        name  string
        input string
        want  string
    }{
        {"正常输入", "hello", "HELLO"},
        {"空字符串", "", ""},
    }
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            if got := MyFunction(tt.input); got != tt.want {
                t.Errorf("MyFunction(%q) = %q, want %q", tt.input, got, tt.want)
            }
        })
    }
}

测试命名规范

• 中文项目：test_函数名_场景描述 或 函数名_正常输入/边界条件/异常输入
• 英文项目：遵循项目现有命名风格

最佳实践

1. 先读再写：理解代码逻辑后再生成测试，不要盲目覆盖
2. 独立性：每个测试独立运行，不依赖执行顺序
3. 可读性：测试名即文档，一看就知道测什么
4. 适度 mock：外部依赖必须 mock，内部逻辑尽量不 mock
5. 渐进式：先覆盖核心路径，再补充边界条件