java使用jUnit,jMockit,dbunit做DAO层单元测试方法

第一步：准备数据

先在MySQL数据库中填写测试数据和期望数据，然后导出到xml文件中。

全表导出：DBUnitHelper.exportTable("t_dp_content_02001", TEST_DATA_FILE);

条件查询导出：DBUnitHelper.exportQueryTable("t_dp_content_02001","select * from t_dp_content_02001 where id=123020011", TEST_DATA_FILE);

xml文件路径举例：private static final String TEST_DATA_FILE = "src/test/resources/testdata/t_dp_content_02001.xml";

第二步：单元测试前清理数据

在执行单元测试前，需要准备测试数据，清理库表：

调用DBUnitHelper.doCleanInsert(xml文件路径)方法将测试数据导入到数据库中，该方法不会删除测试数据以外的其它数据；

调用DBUnitHelper.deleteData(xml文件路径)方法，将测试过程中产生的新数据也删除；

也可以调用DBUnitHelper.doCleanTable(库表名)来清空库表。

代码举例：

@BeforeClass
public static void setUp() throws Exception {
Mockit.setUpMocks();

try {

// 删除并重新插入测试数据
DBUnitHelper.doCleanInsert(TEST_DATA_FILE);

// 删除测试过程中产生的新数据
DBUnitHelper.deleteData(EXPECT_DATA_FILE_1);

} catch (Throwable e) {

e.printStackTrace();
}
}

第三步：编写单元测试用例

调用DAO层的方法，然后调用BaseDaoTest.assertEquals(期望数据文件，SQL语句，库表名，忽略比较的字段)方法判断数据库的数据变化是否符合预期。

判断方法举例：

super.assertEquals(EXPECT_DATA_FILE_1,"select * from t_dp_content_02001 where id=3210200110", "t_dp_content_02001",new String[] {"logicKey", "lastUpdateTime"});

单元测试环境 pom.xml文件

<dependency>
<groupId>com.panguso</groupId>
<artifactId>pg-app-core</artifactId>
<version>2.0.0-SNAPSHOT</version>
</dependency>

<!-- 单元测试依赖jar包 -->

<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
<version>4.10</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>commons-dbcp</groupId>
<artifactId>commons-dbcp</artifactId>
<version>1.4</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.dbunit</groupId>
<artifactId>dbunit</artifactId>
<version>2.4.8</version>
<scope>test</scope>
</dependency>

dbunit.properties配置文件（src/test/resources目录下）

######################################################
drivers=com.mysql.jdbc.Driver
jdbc.url=jdbc:mysql://10.10.160.157:3306/opendata_unit?useUnicode=true&characterEncoding=utf8
jdbc.maxconn=5
jdbc.user=opendata_unit
jdbc.password=opendata_unit_123
schema=opendata_unit
######################################################
sqlMapFilePath=sqlMapConfig_unit.xml

代码：

BaseContentDaoTest .java

package com.panguso.op.resource.core.dao;

import junit.framework.Assert;
import mockit.Expectations;
import mockit.Mocked;
import mockit.Mockit;

import org.junit.BeforeClass;
import org.junit.Test;

import com.panguso.app.core.test.BaseDaoTest;
import com.panguso.app.core.test.DBUnitHelper;
import com.panguso.app.core.util.RandomUtil;
import com.panguso.op.resource.core.dao.impl.PlainSequenceGenerator;
import com.panguso.op.resource.core.po.Content;

/**
*
* @author yangjianfei
* @date 2012-11-22
*/
public class BaseContentDaoTest extends BaseDaoTest {
private static BaseContentDao contentDao;
private static final String TEST_DATA_FILE = "src/test/resources/testdata/t_dp_content_02001.xml";
private static final String EXPECT_DATA_FILE_1 = "src/test/resources/testdata/expectation/t_dp_content_02001_1.xml";
private static final Long ID = 3210200110L;

@BeforeClass
public static void setUp() throws Exception {
Mockit.setUpMocks();

try {
ISequenceGenerator generator = new PlainSequenceGenerator(sqlMap);
contentDao = new BaseContentDao(sqlMap, generator);

DBUnitHelper.doCleanInsert(TEST_DATA_FILE);
DBUnitHelper.deleteData(EXPECT_DATA_FILE_1);
} catch (Throwable e) {
e.printStackTrace();
}
}

@Mocked
PlainSequenceGenerator sequenceGenerator;

@Test
public void testCRUD() {
new Expectations() {
{
sequenceGenerator.getNextSequence(anyString);
result = ID;
times = 1;
}
};

Content content = contentDao.getBaseInfoById(123020011L);
content.setId(null);
content.setLogicKey(RandomUtil.generateLetterStr(10));

// 插入数据
contentDao.saveBaseInfo(content);
Content content1 = contentDao.getBaseInfoById(ID);
Assert.assertEquals(content.getContentState(), content1.getContentState());
super.assertEquals(EXPECT_DATA_FILE_1,
"select * from t_dp_content_02001 where id=3210200110", "t_dp_content_02001",
new String[] {"logicKey", "lastUpdateTime"});

}

public static void main(String[] args) throws Exception {
//      DBUnitHelper.exportTable("t_dp_content_02001", TEST_DATA_FILE);
DBUnitHelper.exportQueryTable("t_dp_content_02001",
"select * from t_dp_content_02001 where id=123020011", TEST_DATA_FILE);
//      DBUnitHelper.doCleanInsert(TEST_DATA_FILE);
//      DBUnitHelper.deleteData(TEST_DATA_FILE);
//      DBUnitHelper.doInsert(TEST_DATA_FILE);
}
}

sqlMapConfig.xml文件：

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE sqlMapConfig
PUBLIC "-//ibatis.apache.org//DTD SQL Map Config 2.0//EN"
"http://ibatis.apache.org/dtd/sql-map-config-2.dtd">
<sqlMapConfig>
<settings cacheModelsEnabled="true" enhancementEnabled="true"
lazyLoadingEnabled="true" errorTracingEnabled="false" maxRequests="32"
maxSessions="10" maxTransactions="5" useStatementNamespaces="true" />
<transactionManager type="JDBC" commitRequired="false">
<dataSource type="dbcp">
<property name="JDBC.Driver" value="com.mysql.jdbc.Driver" />
<property name="JDBC.ConnectionURL" value="jdbc:mysql://10.10.160.157:3306/opendata_unit?useUnicode=true&characterEncoding=utf8" />
<property name="JDBC.Username" value="opendata_unit" />
<property name="JDBC.Password" value="opendata_unit_123" />
<property name="Pool.MaximumActiveConnections" value="2" />
<property name="Pool.MaximumIdleConnections" value="1" />
<property name="Pool.MaximumCheckoutTime" value="120000" />
<property name="Pool.TimeToWait" value="500" />
<property name="Pool.PingQuery" value="select 1 from t_userinfo" />
<property name="Pool.PingEnabled" value="false" />
<property name="Pool.PingConnectionsOlderThan" value="1" />
<property name="Pool.PingConnectionsNotUsedFor" value="1" />
</dataSource>
</transactionManager>

<sqlMap resource="com/panguso/op/resource/core/dao/sql/ResourceSQL.xml" />
<sqlMap resource="com/panguso/op/resource/core/dao/sql/ContentSQL.xml" />
<sqlMap resource="com/panguso/op/resource/core/dao/sql/SequenceSQL.xml" />
<sqlMap resource="com/panguso/op/resource/core/dao/sql/MetaDataSQL.xml" />
<sqlMap resource="com/panguso/op/resource/core/dao/sql/DataPartnerSQL.xml" />
<sqlMap resource="com/panguso/op/resource/core/dao/sql/ResourceContentListenerSQL.xml" />
</sqlMapConfig>