Skip to content

Latest commit

 

History

History
107 lines (86 loc) · 2.98 KB

File metadata and controls

107 lines (86 loc) · 2.98 KB
name test-generator
description 自动生成单元测试:识别函数逻辑、覆盖边界条件、生成可运行测试代码

测试生成器

触发条件

当用户要求"生成测试"、"写单元测试"、"为 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. 渐进式:先覆盖核心路径,再补充边界条件