인프런 커뮤니티 질문&답변

멀티 스레드를 사용하는 테스트에서는 @Transactional 을 어떻게 사용해야 할까요..?

동시성 이슈를 확인하는 테스트에서는 주로 ExecutorService 를 통해 스레드 풀을 생성해서 동시에 서비스 로직을 호출하는 식으로 테스트를 진행하면서 여러 문제를 겪었는데 어떻게 해결하면 좋을지 고민을 하다가 토비님에게 질문드려봅니다 ㅠㅠ

1. 다른 스레드에서 save() 한 오브젝트를 조회할 수 없는 문제.

   1. 테스트 코드에 @Transactional 를 사용하게 되면 테스트 코드가 하나의 트랜잭션으로 묶이게 되는데, 다른 스레드에서 서비스를 호출 하기 전에 테스트를 위한 데이터를 save() 하고 다른 스레드가 서비스에서 save 한 데이터를 가져오려고 할 때 아직 DB에 commit 되지 않은 상태여서 가져오지 못하는 이슈가 있더라구요. -> db isolation level이 read uncommited 보다 높기 때문에 발생하는 것 같음.

          @Test
          @DisplayName("밖에서 저장한 데이터는 스레드 내부에서 가져올 수 없음")
          @Transactional
          void outsideTransaction() throws InterruptedException {
              Member member = new Member(1L, "테스트유저");
              memberRepository.saveAndFlush(member);
              entityManager.clear();
      
              TimeUnit.SECONDS.sleep(2);  // 일정 시간 대기.
      
              try (ExecutorService executorService = Executors.newFixedThreadPool(32)) {
                  executorService.submit(() -> {
                      long count = memberRepository.count();
                      Optional<Member> optionalMember = memberRepository.findById(1L);
                      System.out.println("member count: " + count);
                      System.out.println("optionalMember.isEmpty: " + optionalMember.isEmpty());
                  });
              }
              System.out.println("=================================");
              long count = memberRepository.count();
              Optional<Member> optionalMember = memberRepository.findById(1L);
              System.out.println("member count: " + count);
              System.out.println("optionalMember.isEmpty: " + optionalMember.isEmpty());
          }

      member count: 0
      optionalMember.isEmpty: true
      =================================
      member count: 1
      optionalMember.isEmpty: false

2. 테스트 코드에 @Transactional 을 선언해도 테스트 코드 내부에서 생성한 스레드에서는 트랜잭션이 존재하지 않는 문제.

       @Test
       @DisplayName("내부 스레드는 트랜잭션이 존재하지 않음.")
       @Transactional
       void activeTransaction() {
           System.out.println("쓰레드 밖에서 트랜잭션 상태: " + TransactionSynchronizationManager.isActualTransactionActive());
           try (ExecutorService executorService = Executors.newFixedThreadPool(32)) {
               executorService.submit(() -> {
                   System.out.println("쓰레드 안에서 트랜잭션 상태: " + TransactionSynchronizationManager.isActualTransactionActive());
               });
           }
       }

    위와 같은 테스트를 실행 했을 때, 출력되는 결과는 다음과 같았습니다.

   쓰레드 밖에서 트랜잭션 상태: true
   쓰레드 안에서 트랜잭션 상태: false

멀티 스레드를 사용하는 테스트에서 @Transactional 를 사용 했을 때 이런 문제점이 있었는데요.. 여러가지 시도해보다가 해결을 하지 못해서 결국 @Transactional 을 사용하지 않고.. 테스트가 끝날 때 마다 @AfterEach 통해서 DB를 직접 제거하는 방식으로 하고 있습니다... 이렇게 테스트는 @Transactional 을 사용하지 않고 서비스 로직에는 @Transactional 사용하고 있다보니, 실제 애플리케이션을 띄워서 api 부하 테스트를 해보면 원하는 결과가 나오지 않았습니다. 테스트 코드가 테스트 코드의 역할을 수행하고 있지 않네요

이런 경우에는 어떻게 해결하는 것이 좋을까요?

감사합니다.