catleop/llm-basic/code-review-test-gen

Code Review Test Generation

针对 PR 中新增的业务逻辑代码，自动生成单元测试，提升代码覆盖率。

触发条件

满足以下所有条件时触发：

1. changed_files 中包含业务逻辑文件（非 test_*.py/*_test.*/*.test.*/*.spec.*）
2. additions > 20（新增代码超过 20 行）
3. diff 中包含新增的函数或方法定义（def / function / const.*=> / class ）

输入

• diff：PR 的代码变更（来自 get_pull_request_files）
• framework_info：测试框架信息（来自 detect_test_framework）
• owner/repo：仓库标识
• pr_number：PR 编号

分析步骤

Step 1：识别新增函数/方法

从 diff 中提取所有新增（+ 开头）的函数/方法定义：

Python:  def function_name(...)
JS/TS:   function name(...) | const name = (...) => | async name(...)
Java:    public/private/protected returnType methodName(...)
Go:      func functionName(...)

Step 2：分析代码路径

对每个新增函数，识别需要测试的路径：

1. 正常路径：函数的主要功能（happy path）
2. 边界条件：空值、零值、最大值、最小值
3. 异常路径：错误输入、抛出异常的场景
4. 条件分支：if/else、try/except、switch 的各个分支

Step 3：生成测试代码

根据 framework_info.primary_framework 选择测试模板（参考 references/test-patterns.md）。

Prompt 结构

你是一个测试工程师。请为以下新增代码生成单元测试。
 
## 新增代码
{diff_additions}
 
## 测试框架
{framework}: {version}
 
## 要求
1. 覆盖正常路径、边界条件和异常路径
2. 每个测试用例有清晰的名称（描述测试场景）
3. 使用 {framework} 的标准断言方式
4. Mock 外部依赖（数据库、HTTP 请求、文件系统）
5. 测试文件路径：{test_file_path}
 
## 现有测试文件（如有）
{existing_tests}
 
输出格式：

{ "test_file_path": "tests/test_xxx.py", "test_code": "完整的测试文件内容", "test_count": 数字, "coverage_summary": "覆盖了哪些场景的简要说明" }

Step 4：确定测试文件路径

Step 5：提交测试文件

将生成的测试代码提交到 Fix PR（与自动修复共用同一 PR，或单独创建 test PR）：

# 先获取现有测试文件（如果存在）
file_info = get_file_contents(owner, repo, test_file_path, branch=fix_branch)
 
# 写入测试文件
create_or_update_file(
  owner=owner, repo=repo,
  path=test_file_path,
  content=test_code,
  sha=file_info.sha if file_info else None,  # 更新时需要 sha
  message=f"test: add unit tests for PR #{pr_number} changes",
  branch=fix_branch
)

输出

## 自动生成测试用例
 
**测试框架**: {framework}
**生成文件**: {test_file_path}
**测试用例数**: {test_count}
 
### 覆盖场景
{coverage_summary}
 
### 测试文件已添加到 Fix PR #{fix_pr_number}

注意事项

1. 不修改业务代码：只生成测试文件，不改动被测试的代码
2. 保持幂等：如果测试文件已存在，追加新测试用例而不是覆盖
3. Mock 外部依赖：数据库、HTTP、文件系统等必须 mock，避免测试依赖外部环境
4. 测试命名规范：test_<function_name>_<scenario> 格式

读取 references/test-patterns.md 了解各语言测试框架模板。