From 02e03dac79ebb3da42788430f25d196d8a7f83f7 Mon Sep 17 00:00:00 2001
From: shinmj <gkdlqkek@gmail.com>
Date: Mon, 15 Nov 2021 15:25:10 +0900
Subject: [PATCH] test code update

---
 .../domain/menu/MenuRepositoryTest.java          |  7 +++++++
 .../domain/policy/PolicyRepositoryTest.java      | 16 ++++++++--------
 2 files changed, 15 insertions(+), 8 deletions(-)

diff --git a/backend/portal-service/src/test/java/org/egovframe/cloud/portalservice/domain/menu/MenuRepositoryTest.java b/backend/portal-service/src/test/java/org/egovframe/cloud/portalservice/domain/menu/MenuRepositoryTest.java
index 5422f5b..b1bbaff 100644
--- a/backend/portal-service/src/test/java/org/egovframe/cloud/portalservice/domain/menu/MenuRepositoryTest.java
+++ b/backend/portal-service/src/test/java/org/egovframe/cloud/portalservice/domain/menu/MenuRepositoryTest.java
@@ -1,5 +1,6 @@
 package org.egovframe.cloud.portalservice.domain.menu;
 
+import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -42,6 +43,12 @@ class MenuRepositoryTest {
         siteRepository.saveAll(sites);
     }
 
+    @AfterEach
+    public void tearDown() {
+        menuRepository.deleteAll();
+        siteRepository.deleteAll();
+    }
+
     @Test
     public void 새로운메뉴_한건_등록() throws Exception {
         Site site = siteRepository.findAll().get(0);
diff --git a/backend/portal-service/src/test/java/org/egovframe/cloud/portalservice/domain/policy/PolicyRepositoryTest.java b/backend/portal-service/src/test/java/org/egovframe/cloud/portalservice/domain/policy/PolicyRepositoryTest.java
index 73507ee..e2bf786 100644
--- a/backend/portal-service/src/test/java/org/egovframe/cloud/portalservice/domain/policy/PolicyRepositoryTest.java
+++ b/backend/portal-service/src/test/java/org/egovframe/cloud/portalservice/domain/policy/PolicyRepositoryTest.java
@@ -134,6 +134,7 @@ class PolicyRepositoryTest {
 
     @Test
     public void ID로_한건조회() throws Exception {
+        Policy save = null;
         //given
         for (int i = 1; i <= 10; i++) {
             String title = "title_"+i;
@@ -143,20 +144,19 @@ class PolicyRepositoryTest {
                 type = "PP";
             }
 
-            policyRepository.save(Policy.builder()
-                    .type(type)
-                    .title(title)
-                    .contents(contentStr)
-                    .build());
+            save = policyRepository.save(Policy.builder()
+                .type(type)
+                .title(title)
+                .contents(contentStr)
+                .build());
         }
 
         //when
-        Optional<Policy> optionalTerms = policyRepository.findById(1L);
+        Optional<Policy> optionalTerms = policyRepository.findById(save.getId());
 
         //then
         System.out.println(optionalTerms.get());
-        assertThat(optionalTerms.get().getTitle()).isEqualTo("title_1");
-        assertThat(optionalTerms.get().getContents()).isEqualTo("contents 1");
+        assertThat(optionalTerms.get().getTitle()).isEqualTo(save.getTitle());
 
     }