엔티티에 @Builder와 @GeneratedValue를 같이 쓸 때 Id를 처리하는 방법

Question

@Entity @Getter @NoArgsConstructor(access = AccessLevel.PROTECTED) public class Member { @Id @GeneratedValue private Long id; private String name; private String info; @OneToMany(mappedBy = "member", cascade = CascadeType.REMOVE) private List clubMembers = new ArrayList<>(); @Builder public Member(String name, String info) { this.name = name; this.info = info; } }안녕하세요! 개별적으로 프로젝트를 진행중인 학생입니다.단위 테스트 코드를 작성하다가 문제가 생겨 질문드립니다.JPA를 사용하고 있고, 엔티티의 pk값을 자동생성하도록 @GeneratedValue를 사용했습니다. 그래서 외부에서 id값을 설정하지 못하도록 빌더에 id를 제외한 나머지 필드들만 포함을 시켰습니다.서비스 레이어에 대한 단위 테스트를 작성하면서 본인이 작성하지 않은 게시글을 수정하려 하면 예외가 발생하는 로직을 테스트하고 있는데, 레포지토리를 Mock객체로 사용했더니 member에 id값이 null인 상태로 남아서 본인확인 로직이 동작하지 않는 문제가 생겼습니다.(현재 본인확인은 게시글 작성자의 id와 로그인한 사용자의 id가 일치하면 성공하는 방식입니다.)해결 방법으로는빌더에 id필드를 추가한다레포지토리와 통합테스트를 수행한다사용자의 id로 회원을 조회하고 이를 비교하는 방식으로 본인확인 로직을 수정한다. (이 경우 memberRepository를 stub으로 사용할 수 있음)이렇게 세가지 정도로 추려봤는데 어떤 방법이 옳은 방법인지 잘 모르겠습니다.빌더에 id를 넣으면 개발자가 id값을 설정할 수 있어 테스트는 간편하지만 pk생성을 온전히 시스템에 위임하는 것이 아니라 위험해보입니다.본인확인 로직을 위와같이 수정하게 되면 db에 쿼리가 추가로 나가서 효율적이지는 않아보입니다.가장 합리적으로 보이는건 레포지토리와 통합테스트를 하는건데, 그러면 하나의 서비스에 어떤 메소드는 mock을 사용해서 단위테스트를 하고, 어떤 메소드는 레포지토리와 통합테스트를 하는 식이라 mock을 쓰는 의미가 있나 싶습니다.이런 경우에 실무에서는 어떤 방식으로 처리하는지 궁금합니다. 꼭 실무가 아니어도 어떤 방식이 좀 더 좋은 방법인지 의견 남겨주시면 감사하겠습니다!

짱구올시다 · Answer

저는 1번이 가장 적절하다고 생각해요. 왜냐하면 애초에 객체, 즉 클래스를 설계하는 단계부터 테스트하기 용이한지, 테스트 가능성을 고려해야 된다고 생각해요.2, 3번은 결국엔 객체 그 자체로는 테스트하기 어려우니 편법?과 같은 방법으로 테스트하려는건데 POJO 그 자체로도 테스트하기 용이해야 합니다.

kiel · Answer

제 경우는 id 는 빌더에서 제외시키고 테스트 할때ReflectionTestUtils.setField(..) 를 써서 id 를 임의로 생성해줍니다.참고 : https://dncjf64.tistory.com/314

엔티티에 @Builder와 @GeneratedValue를 같이 쓸 때 Id를 처리하는 방법

2개의 답변

#mockito 목록