JUnit 5测试Spring Batch作业遇数据源与Step Scope问题求助

问题场景 #

Spring Boot集成Spring Batch的应用，添加H2依赖后可正常运行，但编写JUnit 5测试时：

• 使用@SpringBatchTest和@SpringJUnitConfig注解后，出现找不到DataSource异常
• 同时存在Step Scope不活跃的问题
• 单独配置测试数据源会导致主应用运行失败

解决方案核心思路 #

通过测试专用配置+Profile隔离，既满足测试环境对DataSource和Step Scope的需求，又不影响主应用的生产配置：

1. 用@Profile("test")定义测试专属的数据源和Batch配置
2. 主应用的数据源配置用@Profile("!test")隔离
3. 借助@SpringBatchTest自动激活Step Scope并提供测试工具类

完整可运行代码示例 #

1. 测试专用Batch配置类 #

@Configuration
@Profile("test")
@EnableBatchProcessing
public class BatchTestConfig {

    // 配置H2内存数据源，自动初始化Spring Batch元数据表
    @Bean
    public DataSource dataSource() {
        return new EmbeddedDatabaseBuilder()
                .setType(EmbeddedDatabaseType.H2)
                .addScript("/org/springframework/batch/core/schema-h2.sql")
                .build();
    }

    // 配置事务管理器
    @Bean
    public PlatformTransactionManager transactionManager(DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }
}

2. 主应用Batch配置类（隔离测试配置） #

@Configuration
@Profile("!test")
@EnableBatchProcessing
public class BatchProductionConfig {

    // 生产环境数据源，读取application.yml配置
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource")
    public DataSource dataSource() {
        return DataSourceBuilder.create().build();
    }
}

3. JUnit 5测试类 #

@SpringBatchTest
@SpringJUnitConfig(classes = {BatchTestConfig.class, YourBatchJobConfig.class})
@ActiveProfiles("test")
class AggregationJobTest {

    @Autowired
    private JobLauncherTestUtils jobLauncherTestUtils;

    @Autowired
    private JobRepositoryTestUtils jobRepositoryTestUtils;

    // 测试前清理作业执行记录，保证测试独立性
    @BeforeEach
    void setUp() {
        jobRepositoryTestUtils.removeJobExecutions();
    }

    @Test
    void testBatchJobExecution() throws Exception {
        // 1. 准备测试输入数据（根据你的作业逻辑调整）
        // 例如：插入测试数据到H2内存库

        // 2. 启动作业
        JobExecution jobExecution = jobLauncherTestUtils.launchJob();

        // 3. 验证作业执行结果
        Assertions.assertEquals(BatchStatus.COMPLETED, jobExecution.getStatus());
        // 可添加更多断言：比如验证输出数据、步骤执行状态等
    }
}

4. 依赖配置（Maven示例） #

确保H2依赖仅在测试环境生效：

<dependency>
    <groupId>com.h2database</groupId>
    <artifactId>h2</artifactId>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.springframework.batch</groupId>
    <artifactId>spring-batch-test</artifactId>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-test</artifactId>
    <scope>test</scope>
</dependency>

关键注意事项 #

• @SpringBatchTest注解会自动注册StepScopeTestExecutionListener，解决Step Scope不活跃的问题，同时提供JobLauncherTestUtils、JobRepositoryTestUtils等测试工具
• 通过@ActiveProfiles("test")确保测试时加载测试专用配置，避免与生产数据源冲突
• 测试前调用jobRepositoryTestUtils.removeJobExecutions()清理历史执行记录，防止测试间互相干扰

内容的提问来源于stack exchange，提问作者Simeon Leyzerzon