DbUnit使用

 

没有UT的代码会让你有如芒在背的感觉。

 

DB的UT（或是集成测试）比较麻烦，尤其是测试数据的准备。

DbUnit就是解决这个问题。

 

一、资料

 

1. 网站

 

DbUnit的官方网站上的介绍的DbUnit的内容不多。

http://www.dbunit.org

 

DbUnit Wiki上的介绍的

http://dbunit.wikidot.com/

 

2. 文章

Effective Unit Testing with DbUnit

http://onjava.com/pub/a/onjava/2004/01/21/dbunit.html

 

DBUnit Made Easy

http://www.oreillynet.com/onjava/blog/2005/10/dbunit_made_easy.html

 

 

二、使用的代码例子

在DbUnit的官方网站的Get Start（http://www.dbunit.org/howto.html ）有简单的例子。

1. 使用使用DbUnit对TestCase扩展的子类来继承

官方网站有说明：

http://www.dbunit.org/howto.html#createtest

 

代码像这样：

public class SampleTest extends DBTestCase { public SampleTest(String name) { super( name ); System.setProperty( PropertiesBasedJdbcDatabaseTester.DBUNIT_DRIVER_CLASS, "org.hsqldb.jdbcDriver" ); System.setProperty( PropertiesBasedJdbcDatabaseTester.DBUNIT_CONNECTION_URL, "jdbc:hsqldb:sample" ); System.setProperty( PropertiesBasedJdbcDatabaseTester.DBUNIT_USERNAME, "sa" ); System.setProperty( PropertiesBasedJdbcDatabaseTester.DBUNIT_PASSWORD, "" ); // System.setProperty( PropertiesBasedJdbcDatabaseTester.DBUNIT_SCHEMA, "" ); } protected IDataSet getDataSet() throws Exception { return new FlatXmlDataSetBuilder().build(new FileInputStream("dataset.xml")); } }

 

2. 在自己的TestCase的子类中，加上DbUnit的支持

常常我们已经继承了Spring的TestCase的类来进行UT，就不能再继承DbUnit的TestCase子类了。

 

官方网站有说明：

http://www.dbunit.org/howto.html#noextend

 

下面是在一个TestCase上加上DbUnit支持的代码的例子：

 

package package1.service; import java.io.InputStream; import java.sql.Connection; import java.util.ArrayList; import java.util.List; import javax.sql.DataSource; import org.dbunit.database.DatabaseConfig; import org.dbunit.database.DatabaseConnection; import org.dbunit.database.IDatabaseConnection; import org.dbunit.dataset.IDataSet; import org.dbunit.dataset.xml.FlatXmlDataSet; import org.dbunit.ext.mysql.MySqlDataTypeFactory; import org.dbunit.ext.mysql.MySqlMetadataHandler; import org.dbunit.operation.DatabaseOperation; import org.springframework.jdbc.datasource.DataSourceUtils; import org.springframework.test.annotation.AbstractAnnotationAwareTransactionalTests; /** * 在Spring的<b>AnnotationAwareTransactional</b>的Test Case的基础上加上DbUnit的支持。<br> * 在Test Case上加上DbUnit的支持的文档：<br> * <ul> * <li><a href="http://www.dbunit.org/howto.html#noextend" mce_href="http://www.dbunit.org/howto.html#noextend">Database setup with * your own TestCase subclass</a> * <li><a href="http://www.dbunit.org/faq.html#NoSuchColumnException" mce_href="http://www.dbunit.org/faq.html#NoSuchColumnException">Why am I * getting a "NoSuchColumnException" but I am sure that the column actually * exists? </a> * </ul> * * @author ding.lid */ public abstract class DbUnitBaseTest extends AbstractAnnotationAwareTransactionalTests { /** * @return 初始化数据的xml配置文件地址 */ protected abstract String[] getDbUnitTestDataFiles(); @Override protected void onSetUpInTransaction() throws Exception { DataSource dataSource = jdbcTemplate.getDataSource(); Connection con = DataSourceUtils.getConnection(dataSource); IDatabaseConnection dbUnitConnection = new DatabaseConnection(con, "belfry"); dbUnitConnection.getConfig().setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new MySqlDataTypeFactory()); dbUnitConnection.getConfig().setProperty(DatabaseConfig.PROPERTY_METADATA_HANDLER, new MySqlMetadataHandler()); // 这两行是为MySQL数据库准备的，否则在数据库中写入测试数据会出异常！ List<IDataSet> dataSets = new ArrayList<IDataSet>(); String[] xmlTestFiles = getDbUnitTestDataFiles(); for (String file : xmlTestFiles) { InputStream is = this.getClass().getResourceAsStream(file); dataSets.add(new FlatXmlDataSet(is)); } try { for (IDataSet dataset : dataSets) { DatabaseOperation.CLEAN_INSERT.execute(dbUnitConnection, dataset); } } finally { DataSourceUtils.releaseConnection(con, dataSource); } } // Spring中方法 @Override protected String[] getConfigLocations() { return new String[] { "classpath:belfry_service_application_context.xml", }; } }

 

 

使用这个TestCase的例子：

import package1.service.DbUnitBaseTest; import package1.service.dao.interfaces.ContainerDao; import package1.service.dataobject.ContainerDo; /** * @author ding.lid */ public class ContainerDaoImplTest extends DbUnitBaseTest { @Override protected String[] getDbUnitTestDataFiles() { return new String[] { "/dbunit_data_container.xml", }; } private int getRowCount() { return countRowsInTable("CONTAINER"); } ContainerDao containerDao; @Autowired public void setContainerDao(ContainerDao containerDao) { this.containerDao = containerDao; } public void testGet() throws Exception { ContainerDo read1 = containerDao.read("container1"); assertEquals(1, read1.getId().intValue()); } }