如何单元测试Spring Batch分区中的Worker Step并传入StepExecutionContext？

当然可以单独测试分区里的Worker Step！其实Spring Batch的Worker Step本质上就是普通的Step组件，只是在分区场景下会被分区器分配带有特定上下文的StepExecution。你完全不用跑整个分区步骤，下面给你两种实用的测试方式：

方法一：手动构建StepExecution直接执行Worker Step #

这是最直接的方式，我们自己模拟分区器传递的StepExecutionContext，然后调用Step的execute方法即可。

步骤示例： #

1. 在测试类中注入你要测试的Worker Step
2. 构建基础的JobExecution（包含JobParameters和Job级别的上下文）
3. 创建StepExecution并设置分区相关的上下文参数
4. 执行Worker Step并断言结果

@SpringBootTest
public class MyWorkerStepTest {

    @Autowired
    private Step myWorkerStep; // 注入你的Worker Step

    @Test
    void testWorkerStepWithPartitionContext() {
        // 1. 构建JobExecution，模拟Job级别的参数和上下文
        JobParameters jobParams = new JobParametersBuilder()
                .addString("batchDate", "2024-05-20")
                .toJobParameters();
        JobExecution jobExecution = new JobExecution(1L, jobParams);
        jobExecution.getExecutionContext().put("globalConfig", "globalValue");

        // 2. 构建StepExecution，设置分区传递的上下文参数
        StepExecution stepExecution = new StepExecution("myWorkerStep", jobExecution);
        // 这里模拟分区器给Worker Step分配的参数，比如分片ID、数据范围等
        stepExecution.getExecutionContext().put("partitionStartId", 1);
        stepExecution.getExecutionContext().put("partitionEndId", 100);

        // 3. 执行Worker Step
        myWorkerStep.execute(stepExecution);

        // 4. 断言执行结果
        // 检查Step执行状态
        Assertions.assertEquals(BatchStatus.COMPLETED, stepExecution.getStatus());
        // 检查Step上下文里的业务结果（如果有的话）
        Assertions.assertEquals(100, stepExecution.getExecutionContext().getInt("processedCount"));
    }
}

方法二：扩展JobLauncherTestUtils适配StepExecutionContext #

如果你习惯用JobLauncherTestUtils，也可以通过手动创建StepExecution并关联到JobExecution，再调用测试工具的方法。不过本质上和第一种方法类似，只是借助了测试工具的封装：

@SpringBootTest
public class WorkerStepWithTestUtilsTest {

    @Autowired
    private JobLauncherTestUtils jobLauncherTestUtils;

    @Autowired
    private Step myWorkerStep;

    @Test
    void testWorkerStepUsingTestUtils() throws Exception {
        // 1. 获取JobExecution
        JobParameters jobParams = new JobParametersBuilder()
                .addString("param", "test")
                .toJobParameters();
        JobExecution jobExecution = jobLauncherTestUtils.getJobExecution(jobParams);

        // 2. 创建并配置StepExecution
        StepExecution stepExecution = new StepExecution(myWorkerStep.getName(), jobExecution);
        stepExecution.getExecutionContext().put("partitionKey", "partitionValue");

        // 3. 执行Step（这里可以用JobLauncherTestUtils的executeStep方法，或者直接调用step.execute）
        myWorkerStep.execute(stepExecution);

        // 4. 断言结果
        assertThat(stepExecution).hasStatus(BatchStatus.COMPLETED);
    }
}

关键注意点： #

• 模拟真实上下文：一定要把分区器会传递给Worker Step的所有参数都放到StepExecutionContext里，比如分片的范围、特定标识等，确保测试场景和生产一致。
• 依赖注入完整：测试类要加载足够的Spring上下文，保证Worker Step的Reader、Processor、Writer等组件都能被正确注入（用@SpringBootTest或者@ContextConfiguration指定配置类）。
• 简化断言：可以用Spring Batch提供的org.springframework.batch.test.Assertions类，比如assertThat(stepExecution).hasStatus(BatchStatus.COMPLETED).hasExitStatus(ExitStatus.COMPLETED);，让断言更简洁。

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