24.01.02 TIL

CardService 클래스에 대한 테스트 코드를 짜는데, 워낙 의존하고 있는 클래스가 많아 Mockito를 활용하기가 너무 어려웟다...
given-willReturn 이 너무 많이 사용되고, 상위 엔티티의 id를 참조하는 경우가 많기 때문에,
차라리 H2를 이용하여 실제 값을 넣도록 진행했다.

값들을 저장하는 것은 문제가 Never 없었다. 하지만!! ⚠️ 한 개의 테스트가 끝날 때마다 데이터 값들을 모두 지워줘야 했는데 다른 엔티티의 repository를 직접 사용하는 것이 아니라, service 단을 거쳐서 사용하기 때문에 deleteAll 메서드를 사용할 수 없었다.
그래서 다시 mockito 를 적용해보려하다 장렬히 실패.......

다시 기존으로 돌아와서 여러 서치를 해본 결과 Database를 지워주는 클래스를 만들면 되었다.

초기에 다른 분들의 코드를 참조한 결과는 밑의 코드

@Component
@Profile("test")
public class DatabaseCleaner implements InitializingBean {
    @PersistenceContext
    private EntityManager entityManager;

    private List<String> tableNames;

    @Override
    public void afterPropertiesSet() {
        tableNames = entityManager.getMetamodel().getEntities().stream()
                .filter(e -> e.getJavaType().getAnnotation(Entity.class) != null)
                .map(e -> CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, e.getName()))
                .collect(Collectors.toList());
    }

    @Transactional
    public void execute() {
        entityManager.flush();
        entityManager.createNativeQuery("SET REFERENTIAL_INTEGRITY FALSE").executeUpdate();

        for (String tableName : tableNames) {
            System.out.println(tableName);
            entityManager.createNativeQuery("TRUNCATE TABLE " + tableName).executeUpdate();
            entityManager.createNativeQuery("ALTER TABLE " + tableName + " ALTER COLUMN ID RESTART WITH 1").executeUpdate();
        }

        entityManager.createNativeQuery("SET REFERENTIAL_INTEGRITY TRUE").executeUpdate();
    }

그런데 여기서 문제는 tableNames를 가져올 때, DB 내의 이름이 아니라, Entity Class 자체의 이름을 가져오는 문제가 발생했다.
그래서 수정된 코드

@Component
@Profile("test")
public class DatabaseCleaner implements InitializingBean {
    @PersistenceContext
    private EntityManager entityManager;

    private List<String> tableNames = List.of("comment", "checklists", "cards", "card_worker", "columns", "boards", "board_worker", "users");

    @Override
    public void afterPropertiesSet() {
    }

    @Transactional
    public void execute() {
        entityManager.flush();
        entityManager.createNativeQuery("SET REFERENTIAL_INTEGRITY FALSE").executeUpdate();

        for (String tableName : tableNames) {
            System.out.println(tableName);
            entityManager.createNativeQuery("TRUNCATE TABLE " + tableName).executeUpdate();
            entityManager.createNativeQuery("ALTER TABLE " + tableName + " ALTER COLUMN ID RESTART WITH 1").executeUpdate();
        }

        entityManager.createNativeQuery("SET REFERENTIAL_INTEGRITY TRUE").executeUpdate();
    }
}

그냥 넣어줬다 ㅎ......;;;
잘 동작하면 됐다!~!ㅎ