Spring Boot测试中如何仅执行一次SQL脚本填充静态数据
在Spring Boot测试类CargoTest上使用@Sql({"/db/import_locations.sql"})注解时,SQL脚本会在每个测试方法执行时重复运行。由于import_locations.sql中的位置数据是静态且不会被测试修改的,需要实现仅在测试类初始化时执行一次该脚本,同时想了解是否需要自定义@BeforeAll执行器,以及现代的数据库填充解决方案。
方案1:通过@Sql注解原生配置实现类级别单次执行
Spring Test提供的@Sql注解支持通过executionPhase属性指定脚本执行时机,无需自定义执行器。只需修改测试类的注解配置:
@Sql( scripts = {"/db/import_locations.sql"}, executionPhase = Sql.ExecutionPhase.BEFORE_TEST_CLASS, config = @SqlConfig(transactionMode = SqlConfig.TransactionMode.ISOLATED) ) @SpringBootTest class CargoTest { // 原有测试代码保持不变 }
关键参数说明:
executionPhase = BEFORE_TEST_CLASS:指定脚本在测试类初始化前执行一次,而非每个测试方法执行前。transactionMode = ISOLATED:确保脚本执行在独立事务中,避免与测试方法的事务产生冲突,同时保证数据插入的原子性。
方案2:@BeforeAll自定义执行(灵活扩展场景)
如果需要添加额外逻辑(比如判断数据是否已存在,避免重复插入),可以使用@BeforeAll结合JdbcTemplate手动执行脚本:
@SpringBootTest class CargoTest { @Autowired JdbcTemplate jdbcTemplate; @Autowired LocationRepository locationRepository; @Autowired CargoRepository cargoRepository; @BeforeAll static void initLocations(@Autowired JdbcTemplate jdbcTemplate) throws IOException { // 先检查目标数据是否已存在 Integer existingCount = jdbcTemplate.queryForObject( "SELECT COUNT(*) FROM LOCATION WHERE CODE IN ('SYD', 'SEL', 'TYO', 'HKG')", Integer.class ); if (existingCount == 0) { // 读取并执行SQL脚本 try (InputStream is = new ClassPathResource("/db/import_locations.sql").getInputStream()) { String sql = new String(is.readAllBytes(), StandardCharsets.UTF_8); jdbcTemplate.execute(sql); } } } // 原有测试方法保持不变 }
1. Testcontainers:真实数据库容器化测试
对于需要贴近生产环境的测试场景,Testcontainers可以启动真实的数据库容器(如MySQL、PostgreSQL),在容器启动时一次性执行初始化脚本,确保测试环境的一致性。结合Spring Boot的@Testcontainers和@Container注解即可快速集成。
2. DataFaker:动态生成测试数据
如果需要大量动态测试数据,DataFaker可以生成各种类型的伪造数据(如名称、地址、日期等),配合@BeforeAll一次性插入到数据库中,适合需要多样化测试数据的场景。
3. Flyway/Liquibase:数据库版本化管理
如果项目本身使用Flyway或Liquibase进行数据库版本控制,可以在测试环境中配置这些工具自动执行初始化脚本,确保测试数据的结构与生产环境一致,且脚本仅执行一次。
4. Spring Test的@SqlGroup:复杂脚本组合
对于多脚本、多执行时机的复杂场景,可以使用@SqlGroup组合多个@Sql注解,配合@SqlMergeMode控制脚本的合并逻辑,实现更精细的执行顺序和时机控制。
内容的提问来源于stack exchange,提问作者Romillion

