Java单元测试中可通过自定义@WithTestData注解结合JUnit 5扩展模型实现自动数据准备与清理：定义注解、编写实现BeforeEachCallback和AfterEachCallback的扩展类，按注解参数加载数据并操作数据库，支持全局或显式注册。
在 Java 单元测试中，可以通过自定义注解 + JUnit 5 的扩展模型（Extension API） 实现测试前自动准备数据、测试后自动清理数据，无需在每个测试方法里重复写 setup/teardown 逻辑。
定义自定义注解
先声明一个用于标记需要自动管理测试数据的注解，比如 @WithTestData：
@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface WithTestData {
String value() default "";
}
这个注解可作用于测试类或测试方法，支持按需指定数据集标识（如 "user_order"），方便扩展不同场景的数据初始化策略。
编写 JUnit 5 扩展类
实现 BeforeEachCallback 和 AfterEachCallback，在测试执行前后读取注解并操作数据：
Java Maven Common
下载
Java Maven 项目审查工作流的通用输入处理。当 Java Maven 项目以 ZIP压缩包或 GitLab 仓库 URL 形式提供且需要...时使用。
立即学习“Java免费学习笔记（深入）”；
- 检查当前测试方法或测试类是否标注了
@WithTestData - 根据注解参数（如
value()）加载对应 SQL 文件或调用预设的数据工厂（如TestDataFactory.create("user_order")） - 执行插入/清空数据库操作（推荐使用内存数据库如 H2，或事务回滚方式避免污染）
- 确保清理动作幂等：比如先 truncate 表，再 insert 测试数据；结束后再 truncate 或 rollback
注册扩展并使用
两种方式启用扩展：
-
全局注册：在
src/test/resources/META-INF/services/org.junit.jupiter.api.extension.Extension中写入扩展类全限定名 -
显式声明：在测试类上加
@ExtendWith(TestDataExtension.class)
然后直接在测试方法或类上使用你的注解：
@WithTestData("order_with_items")
@Test
void shouldCalculateTotalPrice() {
// 数据已自动插入：1个订单 + 2个商品项
var result = orderService.calculateTotal(1L);
assertEquals(BigDecimal.valueOf(199.98), result);
}
关键细节提醒
- 避免依赖真实数据库——优先用
@SpringBootTest配合@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.ANY)或纯 H2 + Flyway - 若用事务方式，确保测试运行在
@Transactional下，并配置@Rollback（Spring Test 默认开启） - 注解参数可扩展为枚举（如
DataSet.USERS_ONLY）或 JSON 字符串（动态构造数据），提升灵活性 - 扩展类中可通过
ExtensionContext.getRequiredTestMethod()或getRequiredTestClass()获取当前上下文，判断注解位置
