当前位置：首页 > news >正文

如何让AI生成项目的单元测试，propmt技巧详解

news 2026/6/2 1:41:04

比如你的项目是一个典型的Java (Spring Boot) + Vue3的前后端分离项目。

为了让 AI（无论是我、ChatGPT 还是其他代码模型）生成高质量的测试代码，你需要提供具体的上下文。AI 不能凭空猜你的数据库字段或接口参数。

以下是为你定制的Prompt 模板，你可以直接复制使用。为了获得最佳效果，建议你分模块、分文件地向 AI 提问，而不是一次性丢给它整个项目。

🚀 通用 Prompt 结构（万能公式）

无论你让 AI 写哪种测试，Prompt 最好包含以下 4 个要素：

角色设定：告诉 AI 它是谁。
上下文：提供代码或文档片段（你已经上传了文档，这很好）。
具体任务：明确要生成什么。
输出要求：指定格式（如 JUnit 5、Mockito、JSON 等）。

🧪 场景一：生成后端单元测试 (Java)

适用对象：你的Service层逻辑，比如CertificateTimeRangeFilterService（时间筛选逻辑）或CertificateGradeService（等级比较逻辑）。

你的输入：复制粘贴具体的 Java 类或方法代码。
使用的 Prompt：

角色：你是一个资深的 Java 架构师，擅长编写高质量的单元测试。
上下文：
这是一个 Spring Boot 项目中的证书管理模块。这是CertificateGradeService的代码（或者粘贴具体的代码片段）：
java
编辑
1// 在这里粘贴你的 Java 代码，例如： 2public enum CertificateGrade { ... } 3public int compareCertificateGrades(String grade1, String grade2) { ... }
任务：
请为这个类/方法生成 JUnit 5 单元测试用例。
要求：
使用@ExtendWith(MockitoExtension.class)。
覆盖所有分支逻辑（例如：grade1 为空、grade2 为空、正常比较）。
使用@ParameterizedTest和@ValueSource/@CsvSource来测试多种输入组合。
对于枚举类，测试所有枚举值的优先级排序是否正确。
断言要具体（使用assertEquals,assertTrue等）。

🐞 场景二：生成后端接口测试用例 (API Testing)

适用对象：你的Controller层接口，比如/project/certificate/list。

你的输入：提供 API 的定义或文档描述。
使用的 Prompt：

角色：你是一个专业的后端测试工程师。
上下文：
这是一个证书管理系统的 API 接口文档：
接口：GET /project/certificate/list
功能：分页查询证书列表，支持按名称、类型、等级、时间范围筛选。
参数：pageNum,pageSize,certificateName(模糊查询),certificateType,dateFilterType(GENERAL/PATENT),validStartDate,validEndDate。
任务：
请设计一份详细的接口测试用例表，并生成对应的 JSON 格式的请求示例。
要求：
正常场景：参数完整，查询有效数据。
边界值分析：
pageNum= 0 或负数（预期：默认为1）
pageSize= 1（最小页）或 1000（最大页）
时间范围：开始时间等于结束时间。
异常场景：
必填参数缺失（如不传 pageNum）。
参数类型错误（如传字符串给 pageNum）。
时间格式错误。
输出格式：Markdown 表格，包含“用例描述”、“请求参数”、“预期状态码”、“预期响应”。

🦋 场景三：生成前端单元测试 (Vue.js)

适用对象：你的 Vue 组件逻辑，比如index.vue中的calculateTableHeight或getDateFieldValue方法。

你的输入：复制粘贴<script setup>中的关键逻辑。
使用的 Prompt：

角色：你是一个前端测试专家，精通 Vue Test Utils 和 Jest。
上下文：
这是一个 Vue3 + Element Plus 的证书管理页面组件。这是<script setup>中的逻辑片段：
javascript
编辑
1// 在这里粘贴你的 JS 逻辑，例如： 2const getDateFieldValue = (row) => { ... } 3const calculateTableHeight = () => { ... }
任务：
请为这些函数/组件逻辑生成单元测试。
要求：
使用 Jest 框架。
模拟window.innerHeight的不同值（例如 800px, 600px），测试calculateTableHeight是否正确计算了减去预留高度后的值。
测试getDateFieldValue在不同queryParams.value.dateFilterType（PATENT, SOFTWARE_COPYRIGHT, GENERAL）下的返回值是否正确。
使用describe和test块组织代码。

💡 给你的特别建议（基于你的文档）

根据你的《证书管理模块开发方案》，有几个逻辑复杂点特别适合让 AI 帮你写测试，因为这些地方容易出 Bug：

时间字段的动态筛选逻辑：
- 代码位置：CertificateTimeRangeFilterService中的getEffectiveDateField。
- 提示词重点：请生成测试用例，验证当dateFilterType为PATENT时，SQL 条件是否正确指向authorization_announce_date字段；当为GENERAL时是否指向expiry_date。
证书等级的智能匹配：
- 代码位置：CertificateGradeService中的intelligentMatchGrade。
- 提示词重点：请生成测试数据，输入包含“国家”、“national”、“国”、“省”、“市”等关键词的字符串，验证是否能正确匹配到NATIONAL或PROVINCIAL枚举。
文件类型验证：
- 代码位置：validateCertificateFile。
- 提示词重点：请生成边界测试用例，例如上传.exe文件（预期拒绝）、超大尺寸文件（预期拒绝）、以及正常的.pdf和.jpg（预期通过）。