본문 바로가기
Spring Batch

10.1. Spring Batch 단위 테스트 코드 - Reader 편

by 향로 (기억보단 기록을) 2019. 10. 20.
반응형

웹 애플리케이션을 개발하다보면 통합 테스트 보다, 단위 테스트가 훨씬 더 많이 작성됩니다.
단위 테스트로 많은 코드를 검증 후, 통합 테스트 코드를 통해 각 단위가 합쳐졌을때 잘 작동 되는지 검증하곤 하는데요.

스프링 배치를 이용한 배치 애플리케이션에서는 많은 분들이 통합 테스트만 작성할때가 많습니다.
전편에서 말씀드린것처럼 스프링 배치의 단위 테스트 작성이 통합 테스트 보다 복잡하기 때문입니다.

그래서 이번 챕터에서는 다음의 질문들에 대해 이야기해볼까 합니다.

  • Reader의 쿼리가 잘 작동되었는지는 어떻게 확인하지?
  • StepScope를 통한 JobParameter가 잘 할당 된다는 것은 어떻게 확인하지?

부분 부분을 잘개 쪼개서 테스트할 수 있는 방법들을 소개드리겠습니다

Reader의 단위 테스트는 다음을 보고 싶을때가 많습니다.

  • 내가 작성한 Reader의 Query가 잘 작동하는지

그래서 Reader의 단위 테스트 방법은 이 부분에 초점을 맞춰서 진행할 예정입니다.

테스트 방법은 총 2단계로 진행됩니다.
StepScope & JobScope가 없는 테스트와
StepScope & JobScope가 필요한 테스트.

차근차근 진행해보겠습니다.

참고로 아래 모든 단위 테스트들에는 H2 의존성이 필수입니다.
compile('com.h2database:h2')

10.1.1 StepScope 가 필요 없는 단위 테스트

Jdbc를 사용하는 배치를 만든다고 가정해보겠습니다.

전체 스프링 배치 코드를 작성하기까지 시간이 필요해 그전에 빠르게 JdbcItemReader의 쿼리만 검증하는 테스트 코드를 작성하고 싶을때가 많습니다.

전체 코드를 모두 작성하고 테스트를 시작하기에는 부담이 많이 되기 때문이죠.

그래서 최소한의 내용만 구현된 Reader 테스트 코드를 만들어보겠습니다.

참고로 이 방식은 JPA에서는 사용하기가 어렵습니다.
Spring Data Jpa를 통해 생성되는 여러 환경들을 본인이 직접 다 구성해야되기 때문입니다.
그래서 JPA를 쓰지 않고, JdbcTemplate로 배치 환경을 구성하시는 분들이 참고해보시면 좋을것 같습니다.

일단 테스트할 대상인 배치 코드입니다.

@Slf4j // log 사용을 위한 lombok 어노테이션
@RequiredArgsConstructor // 생성자 DI를 위한 lombok 어노테이션
@Configuration
public class BatchOnlyJdbcReaderTestConfiguration {
    public static final DateTimeFormatter FORMATTER = ofPattern("yyyy-MM-dd");
    public static final String JOB_NAME = "batchOnlyJdbcReaderTestJob";

    private final DataSource dataSource;

    private int chunkSize;

    @Value("${chunkSize:1000}")
    public void setChunkSize(int chunkSize) {
        this.chunkSize = chunkSize;
    }

    @Bean
    @StepScope
    public JdbcPagingItemReader<SalesSum> batchOnlyJdbcReaderTestJobReader(
            @Value("#{jobParameters[orderDate]}") String orderDate) throws Exception {

        Map<String, Object> params = new HashMap<>();

        params.put("orderDate", LocalDate.parse(orderDate, FORMATTER));

        SqlPagingQueryProviderFactoryBean queryProvider = new SqlPagingQueryProviderFactoryBean();
        queryProvider.setDataSource(dataSource);
        queryProvider.setSelectClause("order_date, sum(amount) as amount_sum");
        queryProvider.setFromClause("from sales");
        queryProvider.setWhereClause("where order_date =:orderDate");
        queryProvider.setGroupClause("group by order_date");
        queryProvider.setSortKey("order_date");

        return new JdbcPagingItemReaderBuilder<SalesSum>()
                .name("batchOnlyJdbcReaderTestJobReader")
                .pageSize(chunkSize)
                .fetchSize(chunkSize)
                .dataSource(dataSource)
                .rowMapper(new BeanPropertyRowMapper<>(SalesSum.class))
                .queryProvider(queryProvider.getObject())
                .parameterValues(params)
                .build();
    }
}

보시면 딱 Reader 부분만 있는 상태입니다.
즉, Job / Step / Processor / Writer를 모두 구현하지 않은 상태이며 Reader쿼리가 정상이면 언제든 나머지 부분을 구현하면 되는 상태입니다.

모든 코드는 Github에 있습니다.

이 배치 코드를 테스트 한다면 다음과 같이 작성할 수 있습니다.

public class BatchNoSpringContextUnitTest2 {

    private DataSource dataSource;
    private JdbcTemplate jdbcTemplate;
    private ConfigurableApplicationContext context;
    private LocalDate orderDate;
    private BatchOnlyJdbcReaderTestConfiguration job;

    @Before
    public void setUp() {
        this.context = new AnnotationConfigApplicationContext(TestDataSourceConfiguration.class); // (1)
        this.dataSource = (DataSource) context.getBean("dataSource"); // (2)
        this.jdbcTemplate = new JdbcTemplate(this.dataSource); // (3)
        this.orderDate = LocalDate.of(2019, 10, 6);
        this.job = new BatchOnlyJdbcReaderTestConfiguration(dataSource); // (4)
        this.job.setChunkSize(10); // (5)
    }

    @After
    public void tearDown() {
        if (this.context != null) {
            this.context.close();
        }
    }

    @Test
    public void 기간내_Sales가_집계되어_SalesSum이된다() throws Exception {
        // given
        long amount1 = 1000;
        long amount2 = 100;
        long amount3 = 10;
        jdbcTemplate.update("insert into sales (order_date, amount, order_no) values (?, ?, ?)", orderDate, amount1, "1"); // (1)
        jdbcTemplate.update("insert into sales (order_date, amount, order_no) values (?, ?, ?)", orderDate, amount2, "2");
        jdbcTemplate.update("insert into sales (order_date, amount, order_no) values (?, ?, ?)", orderDate, amount3, "3");

        JdbcPagingItemReader<SalesSum> reader = job.batchOnlyJdbcReaderTestJobReader(orderDate.format(FORMATTER)); // (2)
        reader.afterPropertiesSet(); // (3)

        // when & then
        assertThat(reader.read().getAmountSum()).isEqualTo(amount1 + amount2 + amount3); // (4)
        assertThat(reader.read()).isNull(); //(5)
    }

    @Configuration
    public static class TestDataSourceConfiguration {

        // (1)
        private static final String CREATE_SQL =
                        "create table IF NOT EXISTS `sales` (id bigint not null auto_increment, amount bigint not null, order_date date, order_no varchar(255), primary key (id)) engine=InnoDB;";

        // (2)
        @Bean
        public DataSource dataSource() {
            EmbeddedDatabaseFactory databaseFactory = new EmbeddedDatabaseFactory();
            databaseFactory.setDatabaseType(H2);
            return databaseFactory.getDatabase();
        }

        // (3)
        @Bean
        public DataSourceInitializer initializer(DataSource dataSource) {
            DataSourceInitializer dataSourceInitializer = new DataSourceInitializer();
            dataSourceInitializer.setDataSource(dataSource);

            Resource create = new ByteArrayResource(CREATE_SQL.getBytes());
            dataSourceInitializer.setDatabasePopulator(new ResourceDatabasePopulator(create));

            return dataSourceInitializer;
        }
    }
}

코드가 너무 길어서 부분적으로 나눠서 설명 드리겠습니다.

setUp

먼저 setUp 부분입니다.

private DataSource dataSource;
private JdbcTemplate jdbcTemplate;
private ConfigurableApplicationContext context;
private LocalDate orderDate;
private BatchOnlyJdbcReaderTestConfiguration job;

@Before
public void setUp() {
    this.context = new AnnotationConfigApplicationContext(TestDataSourceConfiguration.class); // (1)
    this.dataSource = (DataSource) context.getBean("dataSource"); // (2)
    this.jdbcTemplate = new JdbcTemplate(this.dataSource); // (3)
    this.orderDate = LocalDate.of(2019, 10, 6);
    this.job = new BatchOnlyJdbcReaderTestConfiguration(dataSource); // (4)
    this.job.setChunkSize(10); // (5)
}

(1) new AnnotationConfigApplicationContext(...)

  • DataSource, JdbcTemplate, Reader 등이 실행될 수 있는 Context 를 생성합니다.
    • 해당 Context는 하단에서 별도로 구성한 TestDataSourceConfiguration 의 Bean과 Configuration을 받아서 생성됩니다.
    • TestDataSourceConfiguration 에 대한 자세한 설명은 하단에서 진행합니다.

(2) (DataSource) context.getBean("dataSource")

  • TestDataSourceConfiguration 를 통해 생성된 DataSource Bean을 가져옵니다.

(3) new JdbcTemplate(this.dataSource)

  • JdbcTemplate 의 경우 지정된 DataSource가 있어야 하며, 해당 DB에 쿼리를 실행합니다.
  • 지금 생성된 JdbcTemplate을 통해 create table, insert 등의 테스트 환경을 구축합니다.

(4) new BatchOnlyJdbcReaderTestConfiguration(dataSource)

  • 테스트할 대상인 Config에 (2) 에서 생성한 DataSource를 생성자 주입 합니다.
    • 해당 Job Config에서 Reader 인스턴스를 생성합니다.

(5) this.job.setChunkSize(10)

  • Reader의 PageSize / FetchSize를 결정하는 ChunkSize를 설정합니다.
    • 원래 ChunkSize와 Reader의 PageSize / FetchSize 는 목적이 조금 다르긴 합니다.
    • 다만 여러 이슈들 에 대한 영향도를 줄이기 위해 보통 이 3가지 값들을 다 일치해서 사용합니다.

여기서 (5)의 설정을 꼭 해야하는 것인가에 대해 궁금하실텐데요.
Reader를 생성하는 JdbcPagingItemReaderBuilder 에서는 pageSize가 지정되어 있지 않으면 에러를 발생시킵니다.

java.lang.IllegalArgumentException: pageSize must be greater than zero

그래서 여기서는 pageSize 값에 사용되는 chunkSizesetter 인잭션을 사용하도록 했습니다.

private int chunkSize;

@Value("${chunkSize:1000}") // setter 인잭션
public void setChunkSize(int chunkSize) {
    this.chunkSize = chunkSize;
}
...

@Bean
@StepScope
public JdbcPagingItemReader<SalesSum> batchJdbcUnitTestJobReader(
        @Value("#{jobParameters[orderDate]}") String orderDate) throws Exception {
    ...

    return ...
            .pageSize(chunkSize) // chunkSize와 일치
            .fetchSize(chunkSize) // chunkSize와 일치
            ...;
}

setter 대신에 생성자 인잭션을 사용하셔도 됩니다.
다만 그럴 경우 chunkSize를 가진 Spring Bean을 별도로 생성해야만 합니다.

테스트 메소드

다음으론 테스트 코드가 수행될 테스트 메소드입니다.

@Test
public void 기간내_Sales가_집계되어_SalesSum이된다() throws Exception {
    // given
    long amount1 = 1000;
    long amount2 = 100;
    long amount3 = 10;
    jdbcTemplate.update("insert into sales (order_date, amount, order_no) values (?, ?, ?)", orderDate, amount1, "1"); // (1)
    jdbcTemplate.update("insert into sales (order_date, amount, order_no) values (?, ?, ?)", orderDate, amount2, "2");
    jdbcTemplate.update("insert into sales (order_date, amount, order_no) values (?, ?, ?)", orderDate, amount3, "3");

    JdbcPagingItemReader<SalesSum> reader = job.batchOnlyJdbcReaderTestJobReader(orderDate.format(FORMATTER)); // (2)
    reader.afterPropertiesSet(); // (3)

    // when & then
    assertThat(reader.read().getAmountSum()).isEqualTo(amount1 + amount2 + amount3); // (4)
    assertThat(reader.read()).isNull(); //(5)
}

(1) jdbcTemplate.update

  • insert 쿼리를 통해 테스트할 환경을 구축합니다.
  • 총 3개의 saels 데이터를 등록합니다.

(2) job.batchOnlyJdbcReaderTestJobReader

  • setUp 메소드에서 만든 Job에서 Reader를 가져옵니다.

(3) reader.afterPropertiesSet()

  • Reader의 쿼리를 생성합니다.
  • 이 메소드가 실행되지 않으면 Reader의 쿼리가 null입니다.

(4) assertThat(reader.read())

  • group by 결과로 원하는 값의 1개의 row가 반환되는지 검증합니다.

(5) assertThat(reader.read()).isNull()

  • 조회 결과가 1개의 row라서 다음으로 읽을 row는 없으니 null임을 검증한다.

테스트 Config

마지막으로 테스트 코드가 수행되는 환경을 만들어주는 TestDataSourceConfiguration을 보겠습니다.

@Configuration
public static class TestDataSourceConfiguration {

    // (1)
    private static final String CREATE_SQL =
                    "create table IF NOT EXISTS `sales` (id bigint not null auto_increment, amount bigint not null, order_date date, order_no varchar(255), primary key (id)) engine=InnoDB;";

    // (2)
    @Bean
    public DataSource dataSource() {
        EmbeddedDatabaseFactory databaseFactory = new EmbeddedDatabaseFactory();
        databaseFactory.setDatabaseType(H2);
        return databaseFactory.getDatabase();
    }

    // (3)
    @Bean
    public DataSourceInitializer initializer(DataSource dataSource) {
        DataSourceInitializer dataSourceInitializer = new DataSourceInitializer();
        dataSourceInitializer.setDataSource(dataSource);

        Resource create = new ByteArrayResource(CREATE_SQL.getBytes());
        dataSourceInitializer.setDatabasePopulator(new ResourceDatabasePopulator(create));

        return dataSourceInitializer;
    }
}

(1) create table

  • Reader의 쿼리가 수행될 테이블 (sales) 를 생성하는 쿼리입니다.
  • 제일 하단의 DataSourceInitializer 에서 *DB가 초기화 되는 시점에 실행될 예정입니다.

(2) @Bean dataSource

  • 테스트용 DB를 실행합니다.
    • @SpringBootTest, @DataJpaTest 등을 써보신 분들은 H2를 이용한 테스트 환경과 동일하다고 생각하시면 됩니다.
  • 인메모리 DB인 H2를 사용했기 때문에 편하게 실행/종료가 가능합니다.
    • Gradle / Maven에 H2 의존성이 꼭 있어야만 작동됩니다.

(3) @Bean initializer

  • (2)를 통해 생성된 DB의 초기 작업을 어떤걸 할지 결정합니다.
  • 여기서는 (1) 의 create table 쿼리를 (2) 의 DataBase에 실행하는 작업을 설정하였습니다.

Tip)
이 모든 과정을 @SpringBootTest가 자동으로 해줍니다.
다만 Spring에 관련된 모든 설정이 실행되다보니 한번 수행할때마다 오래걸립니다.
지금의 테스트는 순식간에 수행되니 속도 측면에서 충분히 장점이 있습니다.

자 그래서 실제로 이 테스트 코드를 수행해보시면!

1

아주 빠른속도로 잘 작동되는 것을 확인해볼 수 있습니다.

10.1.2 StepScope 가 필요한 단위 테스트

StepScope와 같이 스프링 배치만의 Scope가 있어야만 작동하는 스프링 배치 기능이 뭐가 있을까요?

대표적으로 JobParameter가 있습니다.

이전에 작성한 포스팅에 좀 더 자세하게 작성하였으니 참고해주세요.

JobParameter는 JobScope 혹은 StepScope가 있는 환경에서만 사용할 수 있다보니 앞에서 소개드린것과 같이 단순한 방식으로는 테스트 할 수가 없습니다.

그래서 이와 같이 StepScope 환경에서의 단위 테스트를 할 수 있는 방법을 소개드리겠습니다.
(JobScope 환경에서의 테스트 코드는 그냥 통합 테스트 사용하시면 됩니다.)

이 역시 @SpringBatchTest가 등장하기 전 / 후로 분리해서 소개드리겠습니다.

아참!
테스트할 대상 코드는 10.1.1 코드에서 조금 변경이 있습니다.
Job Bean이 없으면 테스트가 불가능하기 때문입니다.

@Slf4j
@RequiredArgsConstructor
@Configuration
public class BatchJdbcTestConfiguration {
    public static final DateTimeFormatter FORMATTER = ofPattern("yyyy-MM-dd");
    public static final String JOB_NAME = "batchJdbcUnitTestJob";

    private final JobBuilderFactory jobBuilderFactory;
    private final StepBuilderFactory stepBuilderFactory;
    private final DataSource dataSource;

    private int chunkSize;

    @Value("${chunkSize:1000}")
    public void setChunkSize(int chunkSize) {
        this.chunkSize = chunkSize;
    }

    @Bean
    public Job batchJdbcUnitTestJob() throws Exception {
        return jobBuilderFactory.get(JOB_NAME)
                .start(batchJdbcUnitTestJobStep())
                .build();
    }

    @Bean
    public Step batchJdbcUnitTestJobStep() throws Exception {
        return stepBuilderFactory.get("batchJdbcUnitTestJobStep")
                .<SalesSum, SalesSum>chunk(chunkSize)
                .reader(batchJdbcUnitTestJobReader(null))
                .writer(batchJdbcUnitTestJobWriter())
                .build();
    }

    @Bean
    @StepScope
    public JdbcPagingItemReader<SalesSum> batchJdbcUnitTestJobReader(
            @Value("#{jobParameters[orderDate]}") String orderDate) throws Exception {

        Map<String, Object> params = new HashMap<>();
        params.put("orderDate", LocalDate.parse(orderDate, FORMATTER));

        SqlPagingQueryProviderFactoryBean queryProvider = new SqlPagingQueryProviderFactoryBean();
        queryProvider.setDataSource(dataSource);
        queryProvider.setSelectClause("order_date, sum(amount) as amount_sum");
        queryProvider.setFromClause("from sales");
        queryProvider.setWhereClause("where order_date =:orderDate");
        queryProvider.setGroupClause("group by order_date");
        queryProvider.setSortKey("order_date");

        return new JdbcPagingItemReaderBuilder<SalesSum>()
                .name("batchJdbcUnitTestJobReader")
                .pageSize(chunkSize)
                .fetchSize(chunkSize)
                .dataSource(dataSource)
                .rowMapper(new BeanPropertyRowMapper<>(SalesSum.class))
                .queryProvider(queryProvider.getObject())
                .parameterValues(params)
                .build();
    }

    @Bean
    public JdbcBatchItemWriter<SalesSum> batchJdbcUnitTestJobWriter() {
        return new JdbcBatchItemWriterBuilder<SalesSum>()
                .dataSource(dataSource)
                .sql("insert into sales_sum(order_date, amount_sum) values (:order_date, :amount_sum)")
                .beanMapped()
                .build();
    }
}
  • 10.1.1 의 코드에서 Job / Step / Writer가 추가 된 코드입니다.
  • 다만 실제로 수행될 코드는 Reader 뿐입니다.

4.0.x 이하 버전

4.0 이하 버전에서는 아래와 같이 작성됩니다.

@RunWith(SpringRunner.class)
@EnableBatchProcessing // (1)
@TestExecutionListeners( { // (2)
        DependencyInjectionTestExecutionListener.class,
        StepScopeTestExecutionListener.class })
@ContextConfiguration(classes={ // (3)
        BatchJdbcTestConfiguration.class,
        BatchJdbcUnitTestJobConfigurationLegacyTest.TestDataSourceConfiguration.class})
public class BatchJdbcUnitTestJobConfigurationLegacyTest {

    @Autowired private JdbcPagingItemReader<SalesSum> reader;
    @Autowired private DataSource dataSource;

    private JdbcOperations jdbcTemplate;
    private LocalDate orderDate = LocalDate.of(2019, 10, 6);

    // (4)
    public StepExecution getStepExecution() {
        JobParameters jobParameters = new JobParametersBuilder()
                .addString("orderDate", this.orderDate.format(FORMATTER))
                .toJobParameters();

        return MetaDataInstanceFactory.createStepExecution(jobParameters);
    }

    @Before // (5)
    public void setUp() throws Exception {
        this.reader.setDataSource(this.dataSource);
        this.jdbcTemplate = new JdbcTemplate(this.dataSource);
    }

    @After // (6)
    public void tearDown() throws Exception {
        this.jdbcTemplate.update("delete from sales");
    }

    @Test
    public void 기간내_Sales가_집계되어_SalesSum이된다() throws Exception {
        //given
        long amount1 = 1000;
        long amount2 = 500;
        long amount3 = 100;

        saveSales(amount1, "1");
        saveSales(amount2, "2");
        saveSales(amount3, "3");

        // when && then
        assertThat(reader.read().getAmountSum()).isEqualTo(amount1+amount2+amount3);
        assertThat(reader.read()).isNull();
    }

    private void saveSales(long amount, String orderNo) {
        jdbcTemplate.update("insert into sales (order_date, amount, order_no) values (?, ?, ?)", this.orderDate, amount, orderNo);
    }

    // (7)
    @Configuration
    public static class TestDataSourceConfiguration {

        private static final String CREATE_SQL =
                "create table IF NOT EXISTS `sales` (id bigint not null auto_increment, amount bigint not null, order_date date, order_no varchar(255), primary key (id)) engine=InnoDB;";

        @Bean
        public DataSource dataSource() {
            EmbeddedDatabaseFactory databaseFactory = new EmbeddedDatabaseFactory();
            databaseFactory.setDatabaseType(H2);
            return databaseFactory.getDatabase();
        }

        @Bean
        public DataSourceInitializer initializer(DataSource dataSource) {
            DataSourceInitializer dataSourceInitializer = new DataSourceInitializer();
            dataSourceInitializer.setDataSource(dataSource);

            Resource create = new ByteArrayResource(CREATE_SQL.getBytes());
            dataSourceInitializer.setDatabasePopulator(new ResourceDatabasePopulator(create));

            return dataSourceInitializer;
        }
    }
}

(1) @EnableBatchProcessing

  • 배치 환경을 자동 설정합니다.
  • 통합 테스트와 마찬가지로 별도의 테스트 설정 클래스 (ex: TestBatchConfig.class) 를 생성해 해당 클래스에 선언해서 import 해도 됩니다.

(2) @TestExecutionListeners(...)

  • TestContextManager어떤 TestExecutionListener들이 등록되어야 하는지 설정할 수 있게 지원합니다.
  • 즉, 스프링의 테스트 환경에서 내가 필요한 여러 리스너들을 사용할 수 있게 등록해주는 역할을 합니다.
  • 여기서는 2개의 리스너가 사용됩니다.
    • DependencyInjectionTestExecutionListener.class
    • 스프링 테스트 환경에서 일반적으로 사용되는 의존성 주입 (DI) 리스너입니다.
    • 테스트 인스턴스에 대한 의존성 주입 (DI) 을 제공합니다.
    • StepScopeTestExecutionListener.class
    • 이번 챕터의 가장 핵심이 되는 설정입니다.
    • 테스트 케이스에서 팩토리 메소드 (여기서는 getStepExecution()) 를 찾아서 팩토리 메소드에서 반환된 StepExecution각 테스트 메소드의 StepExecution 으로 사용합니다.
    • StepExecution를 반환 하는 메소드가 팩토리 메소드 입니다.
    • 팩토리 메소드가 없으면 기본 StepExecution이 생성됩니다.

(3) @ContextConfiguration(...)

  • 테스트를 수행할때 Import할 Config 클래스들을 등록합니다.
  • BatchJdbcTestConfiguration.class
    • 테스트 대상이 되는 Batch Job Config 클래스입니다.
  • BatchJdbcUnitTestJobConfigurationLegacyTest.TestDataSourceConfiguration.class
    • 10.1.1 에서 다룬 테스트 Config와 같은 역할을 합니다.
    • 테스트 코드가 수행되는 환경을 만들어주는 Config 클래스입니다.
    • DataSource Bean 생성
    • create table
    • 등을 담당합니다.

(4) getStepExecution()

  • StepScopeTestExecutionListener 가 사용하는 팩토리 메소드입니다.
  • 여기서 반환하는 StepExecution 가 테스트에서 사용되는 Reader의 Step 환경이 됩니다.
  • 아래에서 좀 더 자세히 설명드리겠습니다.

(5) setUp()

  • Bean으로 등록된 Reader와 insert 를 비롯한 쿼리를 수행할 JdbcOperations 에 동일한 DataSource를 등록합니다.

(6) tearDown()

  • 여러 테스트들이 동시에 수행될때를 대비해 insert로 데이터가 저장된 sales 테이블을 초기화합니다.
  • 이렇게 해야만 테스트가 끝날때마다 sales 테이블이 깨끗하게 비어있어 다른 테스트에 영향이 없습니다

(7) @Configuration TestDataSourceConfiguration

  • 10.1.1 에서 사용된 테스트 환경 Config 클래스와 같은 역할을 합니다.

자 이제 이 테스트 코드를 수행해보면!?

2

정상적으로 테스트가 수행되고 통과 되는 것을 알 수 있습니다.

그렇다면 한가지 의문이 있습니다.

StepScopeTestExecutionListener 클래스는 getStepExecution() 를 어떻게 찾고, StepExecution 를 교체하는걸까요?

StepScopeTestExecutionListener?

StepScopeTestExecutionListener 클래스의 코드를 열어보시면 아래와 같이 getStepExecution 메소드가 있습니다.
저희가 테스트 클래스에서 작성한 메소드와 같은 이름이죠?

이 메소드의 코드는 다음과 같습니다.

public class StepScopeTestExecutionListener implements TestExecutionListener {
    ...
    protected StepExecution getStepExecution(TestContext testContext) {
        Object target;

        try {
            Method method = TestContext.class.getMethod(GET_TEST_INSTANCE_METHOD);
            target = ReflectionUtils.invokeMethod(method, testContext);
        } catch (NoSuchMethodException e) {
            throw new IllegalStateException("No such method " + GET_TEST_INSTANCE_METHOD + " on provided TestContext", e);
        }

        ExtractorMethodCallback method = new ExtractorMethodCallback(StepExecution.class, "getStepExecution");
        ReflectionUtils.doWithMethods(target.getClass(), method);
        if (method.getName() != null) {
            HippyMethodInvoker invoker = new HippyMethodInvoker();
            invoker.setTargetObject(target);
            invoker.setTargetMethod(method.getName());
            try {
                invoker.prepare();
                return (StepExecution) invoker.invoke();
            }
            catch (Exception e) {
                throw new IllegalArgumentException("Could not create step execution from method: " + method.getName(),
                        e);
            }
        }

        return MetaDataInstanceFactory.createStepExecution();
    }
    ...
}

여기서 중요한 코드는 ReflectionUtils.doWithMethods(target.getClass(), method) 입니다.

테스트 코드를 수행해서 디버깅을 진행해보면 위 메소드를 통해 팩토리 메소드를 찾는다는 것을 알 수 있습니다.

3

그리고 이렇게 찾은 메소드를 Invoker를 이용해 실행합니다.

4

자 여기서 그럼 한가지 추가로 의문이 있습니다.
팩토리 메소드를 찾는 것은 얼핏보면 preferredName 을 통해하는것 같습니다.

new ExtractorMethodCallback(StepExecution.class, "getStepExecution")

즉, getStepExecution 이라는 메소드 이름으로 찾는것 같다는 이야기죠.

만약 getStepExecution 이란 이름이 아니면 어떻게 될까요?

아래와 같이 테스트 클래스의 메소드 명을 getStepExecution 에서 getStepExecution2 로 변경해보겠습니다.

public StepExecution getStepExecution2() {
    JobParameters jobParameters = new JobParametersBuilder()
            .addString("orderDate", this.orderDate.format(FORMATTER))
            .toJobParameters();

    return MetaDataInstanceFactory.createStepExecution(jobParameters);
}

그리고 테스트 코드를 다시 수행해서 디버깅 해보면!

5

실제 테스트 코드도 잘 수행된다는 것을 확인할 수 있습니다.

6

즉, 메소드 이름과 무관하게 StepExecution를 반환 하는 메소드를 팩토리 메소드로 사용한다는 것을 알 수 있습니다.

4.1.x 이상 버전

4.1.x 에서는 @TestExecutionListeners 를 통해 하던 설정이 모두 @SpringBatchTest로 교체됩니다.

@RunWith(SpringRunner.class)
@EnableBatchProcessing
@SpringBatchTest // (1)
@ContextConfiguration(classes={
        BatchJdbcTestConfiguration.class,
        BatchJdbcUnitTestJobConfigurationNewTest.TestDataSourceConfiguration.class})
public class BatchJdbcUnitTestJobConfigurationNewTest {
    .... // 4.0 이하 버전과 동일

}

(1) @SpringBatchTest

  • @TestExecutionListeners로 등록된 StepScopeTestExecutionListener.class 이 자동 지원됩니다.
  • 그외 나머지 설정등을 기존과 동일하게 하시면 됩니다.

마찬가지로 테스트는 정상적으로 통과 됩니다.

7

@SpringBatchTest@SpringBootTest를 대신하진 않습니다.
다만, 그간 해왔던 귀찮은 설정들인 JobLauncherTestUtils, StepScopeTestExecutionListener 등을 간단하게 대신해줍니다.

그간 스프링 배치 테스트 설정이 힘드셨던 분들이라면 4.1 버전으로 올려 편하게 테스트 코드를 작성해보시는걸 추천드립니다.

10.1.3 JPA에서의 Reader 테스트

Jdbc로만 테스트하면 아쉬우실까봐 JPA에서의 Reader 테스트도 첨부합니다.
사실 JPA는 @SpringBootTest자동으로 해주는 설정들이 많아 좀 더 편하게 테스트 코드를 작성할 수 있습니다.

@RunWith(SpringRunner.class)
@SpringBatchTest
@SpringBootTest(classes={BatchJpaTestConfiguration.class, TestBatchConfig.class})
public class BatchJpaUnitTestJobConfigurationTest {

    @Autowired private JpaPagingItemReader<SalesSum> reader;
    @Autowired private SalesRepository salesRepository;
    @Autowired private SalesSumRepository salesSumRepository;

    private static final LocalDate orderDate = LocalDate.of(2019,10,6);

    @After
    public void tearDown() throws Exception {
        salesRepository.deleteAllInBatch();
        salesSumRepository.deleteAllInBatch();
    }

    public StepExecution getStepExecution() {
        JobParameters jobParameters = new JobParametersBuilder()
                .addString("orderDate", orderDate.format(FORMATTER))
                .toJobParameters();

        return MetaDataInstanceFactory.createStepExecution(jobParameters);
    }

    @Test
    public void 기간내_Sales가_집계되어_SalesSum이된다() throws Exception {
        //given
        int amount1 = 1000;
        int amount2 = 500;
        int amount3 = 100;

        saveSales(amount1, "1");
        saveSales(amount2, "2");
        saveSales(amount3, "3");

        reader.open(new ExecutionContext());

        //when & then
        assertThat(reader.read().getAmountSum()).isEqualTo(amount1+amount2+amount3);
        assertThat(reader.read()).isNull(); // 더이상 읽을게 없어 null
    }

    private Sales saveSales(long amount, String orderNo) {
        return salesRepository.save(new Sales(orderDate, amount, orderNo));
    }
}

JPA 테스트 역시 Jdbc와 크게 다르지 않습니다.
getStepExecution 메소드를 이용해 테스트용 StepExecution을 생성하고 Reader의 결과를 검증하시면 됩니다.

아! 물론 @After를 통해 사용된 테이블은 깨끗하게 비우시는것 잊지마시구요.

10.1 마무리

이번 10.1에서는 다음을 배웠습니다.

  • JdbcReader 환경에서의 StepScope 없이 진행하는 Reader 단위 테스트 방법
  • JdbcReader 환경에서의 StepScope 를 활성화시킨 Reader 단위 테스트 방법
  • getStepExecution() 의 사용 방법 및 구동 원리
  • JPA 에서의 Reader 단위 테스트 방법

다음은 Processor 단위 테스트 를 진행하겠습니다.

긴 글 끝까지 봐주셔서 감사합니다.


반응형