Merge pull request #2 from YAPP-Github/test/setting

[Test] 각 레이어 별 테스트 환경 세팅

Author: leegwichan

build.gradle

@@ -43,9 +43,9 @@ dependencies {
     runtimeOnly 'com.mysql:mysql-connector-j'
 
     // Test
-    testImplementation 'org.springframework.boot:spring-boot-starter-test'
-    testImplementation 'org.springframework.restdocs:spring-restdocs-mockmvc'
     testRuntimeOnly 'org.junit.platform:junit-platform-launcher'
+    testImplementation 'org.springframework.boot:spring-boot-starter-test'
+    testImplementation 'io.rest-assured:rest-assured:5.5.0'
 }
 
 tasks.named('test', Test) {

src/test/java/baegam/sundae/DatabaseCleaner.java

@@ -0,0 +1,47 @@
+package baegam.sundae;
+
+import jakarta.persistence.EntityManager;
+import java.util.List;
+import org.junit.jupiter.api.extension.BeforeEachCallback;
+import org.junit.jupiter.api.extension.ExtensionContext;
+import org.springframework.context.ApplicationContext;
+import org.springframework.test.context.junit.jupiter.SpringExtension;
+import org.springframework.transaction.support.TransactionTemplate;
+
+public class DatabaseCleaner implements BeforeEachCallback {
+
+    @Override
+    public void beforeEach(ExtensionContext extensionContext) {
+        ApplicationContext context = SpringExtension.getApplicationContext(extensionContext);
+        cleanup(context);
+    }
+
+    private void cleanup(ApplicationContext context) {
+        EntityManager em = context.getBean(EntityManager.class);
+        TransactionTemplate transactionTemplate = context.getBean(TransactionTemplate.class);
+
+        transactionTemplate.execute(action -> {
+            em.clear();
+            truncateTables(em);
+            return null;
+        });
+    }
+
+    private void truncateTables(EntityManager em) {
+        em.createNativeQuery("SET REFERENTIAL_INTEGRITY FALSE").executeUpdate();
+        for (String tableName : findTableNames(em)) {
+            em.createNativeQuery("TRUNCATE TABLE %s RESTART IDENTITY".formatted(tableName)).executeUpdate();
+        }
+        em.createNativeQuery("SET REFERENTIAL_INTEGRITY TRUE").executeUpdate();
+    }
+
+    @SuppressWarnings("unchecked")
+    private List<String> findTableNames(EntityManager em) {
+        String tableNameSelectQuery = """
+                SELECT TABLE_NAME
+                FROM INFORMATION_SCHEMA.TABLES
+                WHERE TABLE_SCHEMA = 'PUBLIC'
+                """;
+        return em.createNativeQuery(tableNameSelectQuery).getResultList();
+    }
+}

src/test/java/baegam/sundae/SundaeApplicationTests.java

@@ -9,5 +9,4 @@ class SundaeApplicationTests {
 	@Test
 	void contextLoads() {
 	}
-
 }

src/test/java/baegam/sundae/controller/BaseControllerTest.java

@@ -0,0 +1,38 @@
+package baegam.sundae.controller;
+
+import baegam.sundae.DatabaseCleaner;
+import io.restassured.RestAssured;
+import io.restassured.builder.RequestSpecBuilder;
+import io.restassured.filter.Filter;
+import io.restassured.filter.log.RequestLoggingFilter;
+import io.restassured.filter.log.ResponseLoggingFilter;
+import io.restassured.specification.RequestSpecification;
+import java.util.List;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.boot.test.web.server.LocalServerPort;
+
+@ExtendWith(DatabaseCleaner.class)
+@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
+public class BaseControllerTest {
+
+    private static final List<Filter> SPEC_FILTERS = List.of(new RequestLoggingFilter(), new ResponseLoggingFilter());
+
+    @LocalServerPort
+    private int port;
+
+    private RequestSpecification spec;
+
+    @BeforeEach
+    void setEnvironment() {
+        RestAssured.port = port;
+        spec = new RequestSpecBuilder()
+                .addFilters(SPEC_FILTERS)
+                .build();
+    }
+
+    protected final RequestSpecification given() {
+        return RestAssured.given(spec);
+    }
+}

src/test/java/baegam/sundae/repository/BaseRepositoryTest.java

@@ -0,0 +1,8 @@
+package baegam.sundae.repository;
+
+import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
+
+@DataJpaTest
+public abstract class BaseRepositoryTest {
+
+}

src/test/java/baegam/sundae/service/BaseServiceTest.java

@@ -0,0 +1,11 @@
+package baegam.sundae.service;
+
+import baegam.sundae.DatabaseCleaner;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.springframework.boot.test.context.SpringBootTest;
+
+@ExtendWith(DatabaseCleaner.class)
+@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE)
+public abstract class BaseServiceTest {
+
+}

src/test/resources/application.yml

@@ -0,0 +1,25 @@
+spring:
+  profiles:
+    active: test
+
+---
+
+spring:
+  config:
+    activate:
+      on-profile: test
+  datasource:
+    driver-class-name: org.h2.Driver
+    url: jdbc:h2:mem:database
+    username: sa
+    password:
+  jpa:
+    show-sql: true
+    properties:
+      hibernate:
+        format_sql: true
+    hibernate:
+      ddl-auto: create-drop
+    defer-datasource-initialization: true
+  flyway:
+    enabled: false