[JPA] 살려주세요

 

문제 상황을 간단하게 나타내 보도록 하겠습니다..

 

 

TestEntity

@Getter
@Setter
@NoArgsConstructor(access = AccessLevel.PROTECTED)
@Entity
public class TestEntity {
    @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
    @Column(name = "test_id")
    private Long id;


    private String name;

}

 

 

TestRepository

public interface TestRepository extends JpaRepository<TestEntity , Long> {
}

 

 

TestService

@Service
@RequiredArgsConstructor
@Transactional
public class TestService {

    private final TestRepository testRepository;

    public void test(){
        testRepository.findAll().forEach( i -> i.setName("멍청이"));
        throw new IllegalStateException("언첵 예외 발생");
    }

    public void add() {

        TestEntity test = new TestEntity();
        test.setName("바보");
        testRepository.save(test);
    }
}

 

 

 

무엇이 문제인가?

@SpringBootTest
@Transactional
class TestServiceTest {

    @Autowired TestService testService;
    @Autowired TestRepository testRepository;

    @Autowired EntityManager em;

    @Test
    public void rollBackTest() throws Exception {


        TestEntity test = new TestEntity();
        test.setName("바보");
        TestEntity save = testRepository.save(test);
        //'바보' 라는 TestEntity가 저장


        assertThrows(IllegalStateException.class, ()->testService.test());
        //'바보' 를 '멍청이'로 바꾸려는데 RunTimeException이 발생

        em.flush();
        em.clear();

        final TestEntity test1 = testRepository.findById(save.getId()).get();
        System.out.println(test1.getName());
        //'멍청이' 가 출력...ㅠ
    }

}

 

이유 예상:

테스트 클래스에 @Transactional 어노테이션이 붙어있으므로 rollBackTest()가 하나의 트랜잭션으로 관리

우선 영속성 컨텍스트에 '바보' -> '멍청이'로 이름이 변경된 상태로 저장
그 상태에서 RuntimeException이 발생하여 Rollback 마크가 표시된 후, assertThrows에서 catch로 예외를 잡음
이 상태에서는 아직 영속성 컨텍스트에는 멍청이로 이름이 저장된 상태.

그 상태에서 testRepository.findById(save.getId())로 조회하여 영속성 컨텍스트에 존재하는 '멍청이'가 조회됨.

이를 출력한 후 메서드가 끝났으나 Rollback 마크 표시가 되었으므로 롤백처리를 함.

따라서 실제로는 롤백이 되지만 테스트코드 상에서는 이를 확인할 수가 없음..

 

 

 

 

 

문제: 이를 테스트 코드에서 롤백처리가 된 것을 확인하기 위해서는 어떻게 해야하나?

1.

assertThrows(IllegalStateException.class, ()->testService.test()); 이후
em.flush();    

em.clear();로 영속성 컨텍스트를 비워준 후 조회하기

-> 실패

이유를 생각해보자면 롤백은 전체 트랜잭션이 끝나는 순간 롤백 마크를 확인하고 롤백시킴.

그러나 아직 전체 트랜잭션이 끝나지 않았으므로 em.flush로 밀어버리는 순간 데이터베이스에 쿼리가 날라감

그래서 이후 clear()를 하더라도 데이터베이스에는 멍청이로 저장되어 있음(물론 데이터베이스 커밋이나 롤백은 아직 트랜잭션이 종료되지 않았으므로 발생하지 않음)

 

 

 2.

assertThrows(IllegalStateException.class, ()->testService.test())를 새로운 메서드로 추출한 후
@Transactional(propagation = Propagation.REQUIRES_NEW) 을 적용하여 새로운 트랜잭션에서 실행되도록 함

-> 실패 (이유: 같은 클래스(빈)의 다른 메서드를 실행하면 적용되지 않음 - 스프링 AOP 프록시의 self invocation 문제  )

 

이를 해결하기 위해 언체크 예외가 발생하는 메서드를 새로운 클래스의 메서드로 만든 후, @Transactional(propagation = Propagation.REQUIRES_NEW) 옵션을 주면 해결되긴 함.

예시는 다음과 같음.

@Service
@RequiredArgsConstructor
@Transactional
public class TestExceptionService {
    private final TestRepository testRepository;

    @Transactional(propagation = Propagation.REQUIRES_NEW)
    public void test(){
        testRepository.findAll().forEach( i -> i.setName("멍청이"));
        throw new IllegalStateException("언첵 예외 발생");
    }

}

그러나 테스트코드를 위해 기존 코드를 바꾸는 것이 맞는 것일까..? 하는 의문..

 

 

 

 

 3.

TestService 자체에 @Transactional(propagation = Propagation.REQUIRES_NEW)를 적용

-> 성공
2번과 마찬가지로 테스트코드를 위해 기존 코드를 바꾸는 것이 맞는 것일까..? 하는 의문

 

 

4. 테스트 클래스에 존재하는 @Transactional을 제거

-> EntityManager를 사용할 수 없어서 실패

 

 

원하는 해결방법:

 테스트코드 내부에서 각각의 트랜잭션을 갖도록 만들 수 있는 방법이 있을까...?
 우선 클래스 단에 @Transactional 을 사용하였기에 안될거같다..
 만약 클래스 단에 @Transactional 을 제거한다면? -> EntityManager 를 사용할 수 없게 됨.

 

 

 

 

 

그런데 테스트 코드가 아닌 실제로 작동시켜본다면?

@RestController
@RequiredArgsConstructor
public class TestController {

    private final TestService testService;
    private final TestRepository testRepository;

    @GetMapping("/test")
    public void test(){
        testService.test();

        //바보라는 사람의 이름을 @@@@@로 바꾸려 했으나 RuntimeException이 발생하여 롤백처리.
        //결과는 바뀌지 않고 바보로 남아있음
    }




    @GetMapping("/test/add")
    public void add(){
        testService.add();

        //바보 라는 이름의 TestEntity를 DB에 저장
    }
}

 

 

먼저 /test/add 로 요청을 보내면 '바보'가 저장 

이후 /test 로 요청을 보내면 '멍청이'로 바꿨으나 런타임 예외가 발생하여 롤백 -> 디비에는 '바보'가 남아있음

 

 

 

 

 

결론

믿음으로 가자!!!!!!!!!!!!!!!!!

+ 같은 클래스 내에서 서로 다른 @Transactional 속성을 가진 메서드를 호출한다면 부모 @Transactional의 속성을 따라간다. 즉 다른 속성을 호출하더라도 아무 소용이 없다는 뜻!

 

 

 

 

 

 

 

✨ 2023 08 15 해결방법 추가

@FunctionalInterface
public interface TransactionalAssert {

    void execute();
}

@Component
public class TransactionHelper {

    @Transactional(readOnly = true)
    public void doAssert(final TransactionalAssert transactionalAssert) {
        transactionalAssert.execute();
    }
}

@SpringBootTest
// @Transactional 제거
class TestServiceTest {

    @Autowired TestService testService;
    @Autowired TestRepository testRepository;

    @Autowired TransactionHelper helper;

    @Test
    public void rollBackTest() throws Exception {


        TestEntity test = new TestEntity();
        test.setName("바보");
        TestEntity save = testRepository.save(test);
        //'바보' 라는 TestEntity가 저장


        assertThrows(IllegalStateException.class, ()->testService.test());

        helper.doAssert(() -> {
            final TestEntity test1 = testRepository.findById(save.getId()).get();
            System.out.println(test1.getName());
        });
    }
}