dbuntil2

在做 Java 企业程序的时候，不可避免地要和外部资源打交道，比如数据库， Http 请求等。对于这些外部资源的处理，我们可采取的操作或者是直接处理或者是模拟处理。当我们使用 Webwork ， Spring ， Hibernate 等框架时，我们要测试的并不仅仅是 Java 代码，我们还要测试依赖于这些框架的配置文件等等。因此，对于数据持久化的测试， Mock 方法是行不通的，我们需要真实地测试数据库操作。对于持久化测试来说，重要的是创造出已知的“干净的”的准备数据。如果我们在测试一个持久化方法前不能确定数据库到底存着什么数据，我们只能通过反复地查看数据库数据来验证测试方法的正确性了（这就是我和大多数人以前使用的最“直接”的方法）。现在就让我们使用 DbUnit ，来更好的更自动化的测试持久化操作吧！


先介绍一下 DbUnit 。 DbUnit 是一个 JUnit 扩展，适用于数据驱动的程序。使用 DbUnit ，可以在测试运行期间将数据库的数据处于已知状态，这样在测试时可以方便地写出测试断言，也能自动地完成对数据持久化方法的测试。在使用上， DbUnit 也很简单， 它提供了大量的类对与数据库相关的操作进行了抽象和封装，大多数情况下你只需要使用少量简单的 API 。

下面我通过一个实际的小例子，介绍一下如何使用 DbUnit 。我也是刚刚使用上了 DbUnit ，经验上不是很丰富，如果文中有不对的地方，也欢迎指正。这个例子很简单，我将较为详细地说明如何使用 Hibernate 和 DbUnit 进行测试。


测试第一步，准备数据集。操作的数据表就是如下的 Account 表（使用的数据库为Mysql）：


create table Account(
id bigint not null auto_increment,
name varchar ( 50 ) not null ,
primary key (id)
) character set gbk;


至于 Account 类，映射文件等这里就不给出了。AccountDAO 接口很简单，只有两个方法：


public interface AccountDAO {
void insert(Account a);
List < Account > findAll();
}


实现类如下：


public class AccountHibernateDAO implements AccountDAO{

public void insert(Account a){
Session s = HibernateSessionFactory.getSession();
s.save(a);
s.close();
}

public List < Account > findAll(){
Session s = HibernateSessionFactory.getSession();
List < Account > l = (List < Account > )s.createCriteria(Account. class ).list();
s.close();
return l;
}
}


在测试前，要准备出数据表中要装入的数据（也就是数据集），这里给出与Account表对应的数据集文件 Accout.xml 内容如下：


<? xml version='1.0' encoding='UTF-8' ?>
< dataset >
< Account id ="1" name ="kafka" />
< Account id ="2" name ="0102" />
</ dataset >



数据集就是一个 xml 文件， <dataset> 中的每个节点对应的就是一条表数据记录（一个 dataset文件可以对应多个数据表记录 ）。这里的 <Account> 节点对应的就是 Account 表，属性就是表中的字段，属性值就是字段值了。在做测试时，数据集中的内容可以手动敲进去，也可以通过工具将数据库中的数据导出来。 对于数据集的详细信息，可参考 http://dbunit.sourceforge.net/components.html#FlatXmlDataSet 。


测试第二步，扩展 DBTestCase 。 DBTestCase 是继承自 JUnit 的类，扩展它需要实现 getDataSet() 来提供数据集。另外，你也可以根据需要扩展继承于 DBTestCase 的子类 JdbcBasedDBTestCase ， DataSourceBasedDBTestCas ， JndiBasedDBTestCase 。下面是继承于 DBTestCase 的 AccountHibernateDAO 的测试类 AccountHibernateDAOTest ：

package hibernatesample.dao.impl;
import hibernatesample.domain.Account;
import hibernatesample.util.HibernateSessionFactory;
import java.io.File;
import java.io.InputStream;
import java.util.List;
import org.dbunit.Assertion;
import org.dbunit.DBTestCase;
import org.dbunit.PropertiesBasedJdbcDatabaseTester;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.ITable;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.dbunit.operation.DatabaseOperation;

public class AccountHibernateDAOTest extends DBTestCase {

private AccountHibernateDAO accountDAO;

public AccountHibernateDAOTest(){
System.setProperty( PropertiesBasedJdbcDatabaseTester.DBUNIT_DRIVER_CLASS, HibernateSessionFactory.getDriverClass());
System.setProperty( PropertiesBasedJdbcDatabaseTester.DBUNIT_CONNECTION_URL, HibernateSessionFactory.getConnectionURL());
System.setProperty( PropertiesBasedJdbcDatabaseTester.DBUNIT_USERNAME,HibernateSessionFactory.getUsername());
System.setProperty( PropertiesBasedJdbcDatabaseTester.DBUNIT_PASSWORD,HibernateSessionFactory.getPassword());
}

@Override
protected IDataSet getDataSet() throws Exception {
String path = " hibernatesample " + File.separator + " dao " + File.separator + " dataset " + File.separator + " Account.xml " ;
InputStream in = this .getClass().getClassLoader().getResourceAsStream(path);
return new FlatXmlDataSet(in);
}


@Override
protected DatabaseOperation getSetUpOperation() throws Exception {
return DatabaseOperation.CLEAN_INSERT;
}

@Override
protected DatabaseOperation getTearDownOperation() throws Exception {
return DatabaseOperation.NONE;
}

protected void setUp() throws Exception {
super .setUp();
accountDAO = new AccountHibernateDAO();
}

public void testInsert() {
Account a = new Account();
a.setName( " aa " );
accountDAO.insert(a);
List < Account > l = accountDAO.findAll();
assertEquals( 3 , l.size());
Account b = l.get( 2 );
assertEquals( " aa " , b.getName());
}

public void testFindAll() {
List < Account > l = accountDAO.findAll();
assertEquals( 2 , l.size());
Account a = l.get( 0 );
assertEquals( new Long( 1 ), a.getId());
assertEquals( " kafka " , a.getName());
Account b = l.get( 1 );
assertEquals( new Long( 2 ), b.getId());
assertEquals( " 0102 " , b.getName());
}

public void testDataset() throws Exception {
IDataSet databaseDataSet = getConnection().createDataSet();
ITable actualTable = databaseDataSet.getTable( " Account " );
IDataSet expectedDataSet = getDataSet();
ITable expectedTable = expectedDataSet.getTable( " Account " );
Assertion.assertEquals(expectedTable, actualTable);
}
}

上面的DBTestCase 依赖于 IDatabaseTester 接口完成工作，而 PropertiesBasedJdbcDatabaseTester 就是其使用的默认实现， AccountHibernateDAOTest 构造函数的作用是完成数据库连接参数的设置。 protected IDataSet getDataSet() 实现了装载数据集到 IDataSet 中。 getSetUpOperation 和 getTearDownOperation 是可选的方法，返回的 DatabaseOperation 为 DBTestCase 在 SetUp 和 TearDown 中将执行的操作， getSetUpOperation 默认的操作为 DatabaseOperation.CLEAN_INSERT ，也就是先清空数据表中的数据再插入数据集中的数据到数据表中。getTearDownOperation 默认的操作为 DatabaseOperation.NONE ，就是什么也不处理。可选的操作还有几个，可参考文档进行设置，但默认的设置是最通用的了。 testDataset （）只是测试数据集中的数据和装载到数据库中数据是否一致。


通过上面的设置，我们就可以测试dao方法的正确性了，而我们要做的只是准备dataset文件及使用少量的DbUnit API（可以将这些操作写到一个抽象类中，测试类都继承自这个抽象类）。