Spring Batch Java配置类是否需要编写单元测试？

可以对这类Spring Batch配置类进行单元测试，虽然集成测试在Spring Batch场景中更常见，但单元测试能快速验证配置逻辑的正确性，提前发现配置层面的问题。

一、单元测试的核心验证点 #

• Bean创建逻辑：验证Job、Step、Reader、Writer等Bean是否能正确实例化，依赖注入是否正常。
• 配置参数正确性：比如Reader的行映射规则、分片大小（chunk(2)）、SQL语句是否符合预期。
• 组件关联关系：确认Step是否正确关联了指定的Reader、Processor、Writer，Job是否正确包含目标Step。

二、具体测试实现思路 #

1. 依赖模拟 #

使用Mockito模拟配置类的依赖：JobBuilderFactory、StepBuilderFactory、EmployeeProcessor、DataSource，避免依赖真实的Spring上下文或外部资源。

2. 测试示例代码 #

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.batch.core.Job;
import org.springframework.batch.core.Step;
import org.springframework.batch.item.database.JdbcBatchItemWriter;
import org.springframework.batch.item.file.FlatFileItemReader;

import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.*;

@ExtendWith(MockitoExtension.class)
class Demo9Test {

    @Mock
    private JobBuilderFactory jobBuilderFactory;

    @Mock
    private StepBuilderFactory stepBuilderFactory;

    @Mock
    private EmployeeProcessor employeeProcessor;

    @Mock
    private DataSource dataSource;

    @InjectMocks
    private Demo9 demo9;

    @Test
    void demo9Job_ShouldCreateJobWithStep1() throws Exception {
        // 模拟JobBuilder的行为
        JobBuilder jobBuilder = mock(JobBuilder.class);
        when(jobBuilderFactory.get("demo9")).thenReturn(jobBuilder);

        JobFlowBuilder jobFlowBuilder = mock(JobFlowBuilder.class);
        when(jobBuilder.start(any(Step.class))).thenReturn(jobFlowBuilder);
        when(jobFlowBuilder.build()).thenReturn(mock(Job.class));

        Job job = demo9.demo9Job();
        assertNotNull(job);
        verify(jobBuilderFactory).get("demo9");
        verify(jobBuilder).start(any(Step.class));
    }

    @Test
    void employeeDBWriterDefault_ShouldConfigureCorrectSqlAndDataSource() {
        JdbcBatchItemWriter<Employee> writer = demo9.employeeDBWriterDefault();
        assertEquals(dataSource, writer.getDataSource());
        assertEquals("insert into employee (employee_id, first_name, last_name, email, age) values (:employeeId, :firstName, :lastName, :email, :age)", writer.getSql());
        assertTrue(writer.getItemSqlParameterSourceProvider() instanceof BeanPropertyItemSqlParameterSourceProvider);
    }

    @Test
    void employeeReader_ShouldSetCorrectConfiguration() throws Exception {
        FlatFileItemReader<EmployeeDTO> reader = demo9.employeeReader();
        assertEquals(1, reader.getLinesToSkip());
        assertTrue(reader.getSkippedLinesCallback() instanceof SkipRecordCallback);
        // 可进一步验证LineMapper的配置，比如字段名称、列范围等
    }
}

三、注意事项 #

• StepScope Bean的处理：对于带有@StepScope的Bean（如inputFileResource、employeeReader），单元测试时可直接调用方法，@StepScope在单元测试环境下不会触发代理逻辑，直接返回实例。若需测试Job参数注入场景，可手动传入参数调用inputFileResource方法。
• 避免过度测试：单元测试重点在配置逻辑，无需验证Reader/Writer的业务功能（比如读取文件内容、写入数据库的正确性），这些交给集成测试或组件测试更合适。
• 结合Spring Test简化测试：如果希望在轻量Spring上下文里测试配置，可使用@SpringBootTest配合@TestConfiguration，但这更偏向集成测试范畴，单元测试优先使用纯Mock方式。

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