Spring + Mybatis + JUnit Test를 위한 예제

mybatistest-context.xml
Mybatis 설정을 위한 dataSource와 transactionManager, Mybatis관련 설정을 아래와 같이 한다.

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xmlns:aop="http://www.springframework.org/schema/aop"
    xmlns:tx="http://www.springframework.org/schema/tx"
    xmlns:jdbc="http://www.springframework.org/schema/jdbc"
xmlns:context="http://www.springframework.org/schema/context"
xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-3.1.xsd
http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-3.1.xsd
http://www.springframework.org/schema/jdbc http://www.springframework.org/schema/jdbc/spring-jdbc-3.1.xsd
      http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-3.1.xsd
      http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-3.1.xsd">

<bean id="dataSource" class="org.springframework.jdbc.datasource.SimpleDriverDataSource">
<property name="driverClass" value="com.mysql.jdbc.Driver" />
<property name="url" value="jdbc:mysql://localhost/springbook?characterEncoding=UTF-8" />
<property name="username" value="spring" />
<property name="password" value="book" />
</bean>

<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
<property name="dataSource" ref="dataSource" />
</bean>

<!-- enable autowire -->
    <context:annotation-config />

    <!-- enable transaction demarcation with annotations -->
    <tx:annotation-driven />
 
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dataSource" />
<!-- property name="configLocation" value="springbook/learningtest/spring/mybatis/mybatis-config.xml" /-->
<property name="mapperLocations" value="springbook/learningtest/spring/mybatis/*-sqlmap.xml" />
</bean>

<bean id="sqlSession" class="org.mybatis.spring.SqlSessionTemplate">
<constructor-arg index="0" ref="sqlSessionFactory" />
</bean>

<bean id="mybatisDao" class="springbook.learningtest.spring.mybatis.MybatisTest$MybatisDAO">
<property name="sqlSession" ref="sqlSession" />
</bean>
</beans>
--------------------------------------------------------------------------

mybatis-config.xml
이 부분은 상기의 mybatistest-context.xml에 있는 sqlSessionFactory부분의 주석처리되어 있는 configLocation 태그를 활성화 했을 경우에 필요한 파일.
이때 당연 mapperLocations 태그 부분을 주석처리해야 한다.

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE configuration PUBLIC "-//mybatis.org//DTD Config 3.0//EN" "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
<mappers>
<mapper resource="springbook/learningtest/spring/mybatis/user-sqlmap.xml" />
</mappers>
</configuration>
----------------------------------------------------------------------------

user-sqlmap.xml
수행되어야 할 쿼리문을 아래와 같이 작성.

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="springbook.learningtest.spring.mybatis.ExampleDAO">
 
<select id="getUserName" parameterType="String" resultType="string">
    SELECT NAME FROM USERS WHERE ID = #{id}
</select>

<select id="getUserList" resultType="springbook.learningtest.spring.mybatis.Users">
SELECT id, name, password, level, login, recommend, email
FROM users
order by id, name
</select>
</mapper>
----------------------------------------------------------------------------

ExampleDAO.java

/**
 *
 */
package springbook.learningtest.spring.mybatis;

import java.util.List;

import org.apache.ibatis.annotations.Param;

/**
 * @author
 *
 */
public interface ExampleDAO {
public String getUserName(@Param("id") String userId);
public List<Users> getUserList();
}

-----------------------------------------------------------------------------

Users.java

/**
 *
 */
package springbook.learningtest.spring.mybatis;

import java.io.Serializable;

/**
 * @author
 *
 */
public class Users implements Serializable {
/**
*
*/
private static final long serialVersionUID = 1L;

private String id;
private String name;
private String password;
private String level;
private String login;
private String recommend;
private String email;

public String getId() {
return id;
}
public void setId(String id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getPassword() {
return password;
}
public void setPassword(String password) {
this.password = password;
}
public String getLevel() {
return level;
}
public void setLevel(String level) {
this.level = level;
}
public String getLogin() {
return login;
}
public void setLogin(String login) {
this.login = login;
}
public String getRecommend() {
return recommend;
}
public void setRecommend(String recommend) {
this.recommend = recommend;
}
public String getEmail() {
return email;
}
public void setEmail(String email) {
this.email = email;
}
public String toString() {
return getId() + getName();
}
}

-----------------------------------------------------------------------------

MybatisTest.java

/**
 *
 */
package springbook.learningtest.spring.mybatis;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

import java.util.List;

import org.apache.ibatis.session.SqlSession;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

/**
 * @author
 *
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration
public class MybatisTest {
@Autowired MybatisDAO mybatisDao;

@Test
public void getUserName() {
String id = "gyumee";
String name = mybatisDao.getUserName(id);

assertThat(name, is("박성철"));
}

@Test
public void getUserList() {
List<Users> userList = mybatisDao.getUserList();

assertThat(true, is(userList.size() > 1));
}

public static class MybatisDAO {
private SqlSession sqlSession;

public void setSqlSession(SqlSession sqlSession) {
this.sqlSession = sqlSession;
}

public String getUserName(String id) {
ExampleDAO exampleDao = sqlSession.getMapper(ExampleDAO.class);
return exampleDao.getUserName(id);
}

public List<Users> getUserList() {
ExampleDAO exampleDao = sqlSession.getMapper(ExampleDAO.class);
List<Users> list = exampleDao.getUserList();
return list;
}
}
}