test: 도감 테스트 추가 및 수정

Author: pizzazoa

src/test/java/org/devkor/apu/saerok_server/domain/dex/bird/BirdConservationGradeMigrationTest.java

@@ -0,0 +1,35 @@
+package org.devkor.apu.saerok_server.domain.dex.bird;
+
+import jakarta.persistence.EntityManager;
+import org.devkor.apu.saerok_server.testsupport.AbstractPostgresContainerTest;
+import org.junit.jupiter.api.DisplayName;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
+import org.springframework.test.context.ActiveProfiles;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+@DataJpaTest
+@ActiveProfiles("test")
+class BirdConservationGradeMigrationTest extends AbstractPostgresContainerTest {
+
+    @Autowired
+    EntityManager em;
+
+    @Test
+    @DisplayName("Flyway 보호등급 데이터가 현재 도감 seed 기준 카운트와 일치한다")
+    void conservationGradeCounts_matchSpreadsheet() {
+        Object[] row = (Object[]) em.createNativeQuery("""
+                SELECT
+                    COUNT(*) FILTER (WHERE conservation_grade = 'GRADE_I'),
+                    COUNT(*) FILTER (WHERE conservation_grade = 'GRADE_II'),
+                    COUNT(*) FILTER (WHERE conservation_grade = 'NONE')
+                FROM bird
+                """).getSingleResult();
+
+        assertThat(((Number) row[0]).longValue()).isEqualTo(16L);
+        assertThat(((Number) row[1]).longValue()).isEqualTo(58L);
+        assertThat(((Number) row[2]).longValue()).isEqualTo(512L);
+    }
+}

src/test/java/org/devkor/apu/saerok_server/domain/dex/bird/core/repository/BirdRepositoryTest.java

@@ -1,6 +1,7 @@
 package org.devkor.apu.saerok_server.domain.dex.bird.core.repository;
 
 import org.devkor.apu.saerok_server.domain.dex.bird.core.entity.Bird;
+import org.devkor.apu.saerok_server.domain.dex.bird.core.enums.ConservationGrade;
 import org.devkor.apu.saerok_server.domain.dex.bird.query.dto.BirdSearchDto;
 import org.devkor.apu.saerok_server.domain.dex.bird.query.dto.CmRangeDto;
 import org.devkor.apu.saerok_server.domain.dex.bird.query.enums.BirdSearchSortDirType;
@@ -46,6 +47,14 @@ void findById_returnsActiveBird() {
         assertThat(found.get().getId()).isEqualTo(bird.getId());
     }
 
+    @Test @DisplayName("findById returns conservation grade")
+    void findById_returnsConservationGrade() {
+        Optional<Bird> found = repo.findById(600L);
+
+        assertThat(found).isPresent();
+        assertThat(found.get().getConservationGrade()).isEqualTo(ConservationGrade.GRADE_I);
+    }
+
     @Test @DisplayName("findById - soft delete 된 새 제외")
     void findById_excludesSoftDeleted() {
         Bird bird = newBird("deleted-bird-" + System.nanoTime(), 10.0);

src/test/java/org/devkor/apu/saerok_server/domain/dex/bird/query/repository/BirdProfileViewRepositoryTest.java

@@ -1,6 +1,7 @@
 package org.devkor.apu.saerok_server.domain.dex.bird.query.repository;
 
 import org.devkor.apu.saerok_server.domain.dex.bird.core.entity.Bird;
+import org.devkor.apu.saerok_server.domain.dex.bird.core.enums.ConservationGrade;
 import org.devkor.apu.saerok_server.domain.dex.bird.core.repository.BirdRepository;
 import org.devkor.apu.saerok_server.domain.dex.bird.query.view.BirdProfileView;
 import org.devkor.apu.saerok_server.testsupport.AbstractPostgresContainerTest;
@@ -47,6 +48,16 @@ public class BirdProfileViewRepositoryTest extends AbstractPostgresContainerTest
         System.out.println(birdProfile.get().toSummaryString());
     }
 
+    @Test
+    void 새프로필에_보호등급이_포함된다() {
+        // when
+        Optional<BirdProfileView> birdProfile = birdProfileRepository.findById(600L);
+
+        // then
+        assertTrue(birdProfile.isPresent());
+        assertEquals(ConservationGrade.GRADE_I, birdProfile.get().getConservationGrade());
+    }
+
     @Test
     void 새프로필_전부_조회하기() {
         // when

src/test/java/org/devkor/apu/saerok_server/domain/dex/bird/query/view/builder/BirdProfileViewTestBuilder.java

@@ -3,6 +3,7 @@
 import org.devkor.apu.saerok_server.domain.dex.bird.core.entity.BirdDescription;
 import org.devkor.apu.saerok_server.domain.dex.bird.core.entity.BirdName;
 import org.devkor.apu.saerok_server.domain.dex.bird.query.view.BirdProfileView;
+import org.devkor.apu.saerok_server.domain.dex.bird.core.enums.ConservationGrade;
 import org.devkor.apu.saerok_server.domain.dex.bird.core.entity.BirdTaxonomy;
 import org.devkor.apu.saerok_server.domain.dex.bird.core.enums.HabitatType;
 import org.springframework.test.util.ReflectionTestUtils;
@@ -17,6 +18,7 @@ public class BirdProfileViewTestBuilder {
     private BirdDescription description = new BirdDescription();
     private Double bodyLengthCm = 25.0;
     private String nibrUrl = null;
+    private ConservationGrade conservationGrade = ConservationGrade.NONE;
     private List<HabitatType> habitats = List.of();
     private List<BirdProfileView.SeasonWithRarity> seasonsWithRarity = List.of();
     private List<BirdProfileView.Image> images = List.of();
@@ -30,6 +32,7 @@ public BirdProfileView build() {
         ReflectionTestUtils.setField(birdProfileView, "description", description);
         ReflectionTestUtils.setField(birdProfileView, "bodyLengthCm", bodyLengthCm);
         ReflectionTestUtils.setField(birdProfileView, "nibrUrl", nibrUrl);
+        ReflectionTestUtils.setField(birdProfileView, "conservationGrade", conservationGrade);
         ReflectionTestUtils.setField(birdProfileView, "habitats", habitats);
         ReflectionTestUtils.setField(birdProfileView, "seasonsWithRarity", seasonsWithRarity);
         ReflectionTestUtils.setField(birdProfileView, "images", images);