如何对操作StepContext的Spring Batch独立Step进行单元测试？

问题原因 #

使用jobLauncherTestUtils.launchStep()测试时，Spring Batch的生命周期回调（如@BeforeStep）未被正确触发，导致Reader中的StepExecution实例未注入，进而引发空指针异常。这是因为单元测试环境的上下文初始化逻辑与生产环境存在差异，JobLauncherTestUtils并未完全模拟完整的Step执行生命周期。

可行解决方案 #

方案1：使用StepTestTemplate（推荐） #

StepTestTemplate是Spring Batch专为Step单元测试设计的工具，会自动处理所有Step生命周期回调，包括@BeforeStep的注入逻辑，无需手动绑定StepExecution。

示例代码：

@SpringBatchTest
@SpringBootTest
public class MyStepTest {

    @Autowired
    private StepTestTemplate stepTestTemplate;

    @Test
    void testStepExecutionContext() {
        // 执行Step，StepTestTemplate自动处理@BeforeStep注入
        stepTestTemplate.execute(stepExecution -> {
            // 可提前设置ExecutionContext初始值（按需）
            return null;
        });

        // 验证Reader存入的ExecutionContext数据
        StepExecution executedStep = stepTestTemplate.getStepExecution();
        assertEquals("预期值", executedStep.getExecutionContext().getString("你的KEY"));
        // 验证Step执行状态
        assertEquals(BatchStatus.COMPLETED, executedStep.getStatus());
    }
}

方案2：手动注入StepExecution #

若坚持使用JobLauncherTestUtils.launchStep()，可手动创建StepExecution并调用Reader的@BeforeStep方法完成注入：

示例代码：

@SpringBatchTest
@SpringBootTest
public class MyStepTest {

    @Autowired
    private JobLauncherTestUtils jobLauncherTestUtils;

    @Autowired
    private MyItemReader myItemReader; // 你的Reader实现类

    @Test
    void testStepWithManualInjection() throws Exception {
        // 创建StepExecution实例
        StepExecution stepExecution = jobLauncherTestUtils.createStepExecution();
        
        // 手动调用@BeforeStep方法注入StepExecution
        myItemReader.beforeStep(stepExecution);
        
        // 启动Step执行
        JobExecution jobExecution = jobLauncherTestUtils.launchStep("你的Step名称");
        
        // 验证结果
        assertEquals(BatchStatus.COMPLETED, jobExecution.getStatus());
        assertEquals("预期值", stepExecution.getExecutionContext().getString("你的KEY"));
    }
}

方案3：确保StepScope配置正确 #

若你的Reader是StepScope的Bean，需保证测试上下文正确加载StepScope代理逻辑：

• 测试类添加@SpringBatchTest注解（已包含@EnableBatchProcessing及Step测试相关配置）
• Reader的Bean定义需明确标注@StepScope：

@Bean
@StepScope
public ItemReader<String> myItemReader() {
    return new MyItemReader();
}

注意事项 #

• 测试类需加载完整的Step配置（Reader、Processor、Writer、Step定义）
• 若使用@StepScope，必须通过Spring容器注入Reader实例，避免直接实例化（保证代理生效）

内容的提问来源于stack exchange，提问作者Hesam Ghiasi