(Spring Boot) 대량의 테스트 케이스를 생성하는 방법 (feat. CommandLineRunner, ApplicationRunner)

🌟 공부하게 된 계기

기능을 구현하고 실행시켜볼 때마다 데이터값을 하나씩 넣는 것은 비효율적인 일이다. 특히 많은 양의 데이터를 처리하는 과정을 테스트 해야할 때는 말할 것도 없을 것이다. 테스트 케이스를 만드는 여러가지 방법을 정리해보고자 한다.

목표 : 100개의 데이터 만들기

1️⃣ Spring Test 활용

test 중 하나로 대량의 데이터를 생성하는 방법이 있을 수 있다.

@Test
@Rollback(false) // 테스트 후 DB 변경을 reset 하지 않고 유지
void 대량의_데이터_생성() {
    for (int i=1; i<=100; i++) {
        Question q = new Question();
        q.setSubject("제목 %d".formatted(i));
        q.setContent("내용 %d".formatted(i));
        q.setCreateDate(LocalDateTime.now());
        questionRepository.save(q);
    }
}

@Rollback 의 경우 여러 개의 테스트를 동시에 실행할 때 앞선 테스트로 인한 DB 변화가 뒤에 진행할 테스트에 영향을 주지 않도록 초기화 시키는 어노테이션인데 default 값이 True 이므로 테스트 케이스 작성을 위해서는 false 로 해두어야한다.

2️⃣ CommandLineRunner (또는 ApplicationRunner) 인터페이스 활용

위 두 인터페이가 어떤 인터페이스인지 궁금하다면 맨 아래 참고 파트에 있는 글을 읽어보면 좋을 것 같다.

① CommandLineRunner 활용

@Configuration
public class DevInitData {

    @Bean
    CommandLineRunner init(QuestionRepository questionRepository) {
        return args -> {
            for (int i=1; i<=100; i++) {
// Question.builder() 를 사용하기위해서는 Question 클래스에 @Builder 어노테이션이 추가되어있어야한다.
                questionRepository.save(Question.builder()
                    .subject("제목 %d".formatted(i))
                    .content("내용 %d".formatted(i))
                    .createDate(LocalDateTime.now())
                    .build());
            }
        };
    }
}

② ApplicationRunner 활용

CommandLineRunner 와 비슷하지만 좀 더 최근에 만들어졌다. 약간의 차이점은 있겠지만 여기서는 크게 다르지 않으므로 간단하게 정리하고 넘어가도록 하자.

@Configuration
public class DevInitData {
    @Bean
    ApplicationRunner init(QuestionRepository questionRepository) {
//			(생략)
    }
}

📚 참고

(1) CommandLineRunner 와 ApllicatioRunner: https://jeong-pro.tistory.com/206

스프링 부트 애플리케이션에서 초기화 코드를 넣는 3가지 방법 (by 토비님)