[java] MyBatis의 테스트 케이스 작성 방법

21 Dec 2023

java

MyBatis는 Java에서 SQL을 효과적으로 다룰 수 있게 해주는 프레임워크이다. MyBatis를 사용하는 애플리케이션을 개발할 때, 테스트 케이스 작성은 매우 중요하다. 이 문서에서는 MyBatis의 테스트 케이스를 작성하는 방법에 대해 살펴볼 것이다.

1. MyBatis 테스트용 데이터베이스 설정

먼저, MyBatis의 테스트 케이스를 작성하기 위해서는 테스트용 데이터베이스가 필요하다. 대부분의 경우 H2 데이터베이스를 사용하여 메모리 기반의 테스트용 데이터베이스를 설정한다.

import org.apache.ibatis.jdbc.ScriptRunner;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;

import java.io.Reader;
import java.sql.Connection;

public class MyBatisTest {

    private static SqlSessionFactory sqlSessionFactory;

    @BeforeAll
    public static void setUp() throws Exception {
        try (Reader reader = Resources.getResourceAsReader("mybatis-config.xml")) {
            sqlSessionFactory = new SqlSessionFactoryBuilder().build(reader, "test");
        }

        try (SqlSession session = sqlSessionFactory.openSession()) {
            Connection conn = session.getConnection();
            try (Reader reader = Resources.getResourceAsReader("schema.sql")) {
                ScriptRunner runner = new ScriptRunner(conn);
                runner.setLogWriter(null);
                runner.runScript(reader);
            }
        }
    }

    @Test
    public void testMyBatis() {
        // 테스트 코드 작성
    }
}

위 코드에서는 H2 데이터베이스를 메모리에 생성하고, 초기화하기 위해 mybatis-config.xml 파일과 schema.sql 스크립트를 사용한다.

2. MyBatis 테스트 케이스 작성

MyBatis 테스트 케이스를 작성할 때에는 주로 JUnit을 사용한다. 테스트 케이스에서는 MyBatis의 매퍼 인터페이스를 이용하여 데이터베이스 작업을 수행한다.

import static org.junit.jupiter.api.Assertions.assertEquals;

public class MyBatisTest {

    // ...

    @Test
    public void testSelect() {
        try (SqlSession session = sqlSessionFactory.openSession()) {
            ProductMapper mapper = session.getMapper(ProductMapper.class);
            Product product = mapper.selectProduct(1);
            assertEquals("Sample Product", product.getName());
        }
    }

    @Test
    public void testInsert() {
        try (SqlSession session = sqlSessionFactory.openSession()) {
            ProductMapper mapper = session.getMapper(ProductMapper.class);
            Product newProduct = new Product(2, "New Product", 100.0);
            mapper.insertProduct(newProduct);
            session.commit();
        }
    }
}

위 코드에서는 ProductMapper 인터페이스를 통해 데이터베이스에서 제품을 선택하고 삽입하는 두 가지 테스트 메서드를 작성한다.

MyBatis 테스트 케이스를 작성하는 방법에 대해 간략히 살펴보았다. 효과적인 MyBatis 애플리케이션을 개발하기 위해서는 테스트 케이스 작성에 충분한 노력을 기울이는 것이 중요하다.

1. MyBatis 테스트용 데이터베이스 설정

2. MyBatis 테스트 케이스 작성

참고 자료