如何对Spring批处理分区工作步骤进行单元测试？

@RunWith(SpringRunner.class)
@SpringBootTest
public class PartitionStepTest {

    @Autowired
    private JobLauncherTestUtils jobLauncherTestUtils;

    @MockBean
    private ItemReader<String> itemReader;

    @MockBean
    private ItemProcessor<String, String> itemProcessor;

    @MockBean
    private ItemWriter<String> itemWriter;

    @Test
    public void testPartitionStep() throws Exception {
        // 准备测试数据
        List<String> input = Arrays.asList("data1", "data2", "data3");
        List<String> expectedOutput = Arrays.asList("processedData1", "processedData2", "processedData3");
        when(itemReader.read()).thenReturn("data1", "data2", "data3");
        when(itemProcessor.process(anyString())).thenReturn("processedData1", "processedData2", "processedData3");

        // 执行测试
        JobExecution jobExecution = jobLauncherTestUtils.launchStep("partitionStep");

        // 验证输出结果
        List<String> actualOutput = new ArrayList<>();
        verify(itemWriter, times(3)).write(anyList());
        ArgumentCaptor<List<String>> argumentCaptor = ArgumentCaptor.forClass(List.class);
        verify(itemWriter, times(3)).write(argumentCaptor.capture());
        argumentCaptor.getAllValues().forEach(actualOutput::addAll);
        assertEquals(expectedOutput, actualOutput);
    }
}