使用junit測試用例

通常只會使用junit測試非main方法，在我眼里就是程序入口實現而已。今天，發現原來可以測試類。

針對mybatis練習。在需要測試的UserDaoImpl類上右鍵，新建一個junit case,位置可以放到新創建的source folder :test里面。

選擇需要測試的方法：

?

然后就會生成一個測試方法，自己補足測試方法就好：

package cn.mrf.mybatis.dao;

import static org.junit.Assert.*;

import java.io.InputStream;

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.Before;
import org.junit.Test;

import cn.mrf.mybatis.po.User;

public class UserDaoImplTest {
    
    private SqlSessionFactory sqlSessionFactory;
    
    //此方法是在執行testFindUserById之前執行
    @Before
    public void setUp() throws Exception{
        //創建sqlSessionFactory
        //mybatis配置文件
        String resource = "SqlMapConfig.xml";
        
        InputStream inputStream = Resources.getResourceAsStream(resource);
        //創建會話工廠,傳入mybatis的配置文件信息
        sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream); 
        
    }

    @Test
    public void testFindUserById() throws Exception {
        //創建UserDao對象
        UserDao userDao = new UserDaoImpl(sqlSessionFactory);
        //調用
        User user = userDao.findUserById(1);
        
        System.out.println(user);
    }

}

下面是被測試的類：

package cn.mrf.mybatis.dao;

import java.util.List;

import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;

import cn.mrf.mybatis.po.User;
/**
 * 
* @ClassName: UserDaoImpl 
* @Description: 接口實現類
* @author mrf
* @date 2015-9-19 下午05:57:03 
*
 */
public class UserDaoImpl implements UserDao {

    // 需要向dao實現類中注入SqlSessionFactory
    // 這里通過構造方法注入
    private SqlSessionFactory sqlSessionFactory;
    
    public UserDaoImpl(SqlSessionFactory sqlSessionFactory) {
        this.sqlSessionFactory = sqlSessionFactory;
    }

    
    @Override
    public User findUserById(int id) throws Exception {
        //sqlSession是線程不安全的，所以單獨使用
        SqlSession sqlSession = sqlSessionFactory.openSession();
        
        User user = sqlSession.selectOne("test.findUserById",id);
        
        //釋放資源
        sqlSession.close();
        return user;
    }

    @Override
    public List<User> findUserByName(String name) throws Exception {
        SqlSession sqlSession = sqlSessionFactory.openSession();
        
        List<User> list = sqlSession.selectList("test.findUserByName",name);
        
        //釋放資源
        return list;
    }

    @Override
    public void insertUser(User user) throws Exception {
        SqlSession sqlSession = sqlSessionFactory.openSession();
        //執行插入
        sqlSession.insert("test.insertUser",user);
        
        //提交事物
        sqlSession.commit();
        // 釋放資源
        sqlSession.close();


    }

    @Override
    public void deleteUser(int id) throws Exception {
        SqlSession sqlSession = sqlSessionFactory.openSession();

        //執行插入操作
        sqlSession.delete("test.deleteUser", id);

        // 提交事務
        sqlSession.commit();

        // 釋放資源
        sqlSession.close();
    }

}

?