使用DBUnit做单元测试

作者：网络转载发布时间：[ 2013/11/22 16:01:07 ] 推荐标签：

　　DBUnit是一个方便的数据准备工具，方便于我们做单元测试的时候准备数据，它的数据准备是基于XML格式的，如下:

<?xmlversion='1.0'encoding='UTF-8'?>

<YourTableName_1Field_1="1"Field_2="f2"Field_3="f3"/>

<YourTableName_1Field_1="2"Field_2="f2_1"Field_3="f3_1"/>

<YourTableName_2Field_1="1"Field_2="2"/>

</dataset>

　　DBUnit的一个XML数据文件中，可以同时放多个表的数据，并且可以方便的把上面XML中准备的数据插入到数据库中. 只需要使用下面简单的代码可以做到：

protected ReplacementDataSet createDataSet(InputStream is) throws Exception {

return new ReplacementDataSet(new FlatXmlDataSetBuilder().build(is));

}

ReplacementDataSet createDataSet = createDataSet(Thread.currentThread().getContextClassLoader().getResourceAsStream("data.xml"));

DatabaseOperation.INSERT.execute(iconn， createDataSet);

　　注：准备这处XML数据文件时，一定要把同一个表中字段数多的记录放在前面，因为DBUnit在根据数据XML文件准备表的元数据字段的时候，是以当前表的第一记录为主的。如下面这个XML文件：

<?xml version='1.0' encoding='UTF-8'?>

<YourTableName_1 Field_1="1" Field_2="f2"/>

<YourTableName_1 Field_1="2" Field_2="f2_1" Field_3="f3_1"/>

<YourTableName_2 Field_1="1" Field_2="2"/>

</dataset>

　　Table YourTableName_1有三个字段，但是第三个字段Field_3是允许为空的，所以上面的XML是可以这样写的，并且DBUnit在执行插入的时候也不会报错，但是这里会出现一个问题，因为DBUnit在第一次分析到table YourTableName_1时，第一条记录只有两个字段，因而它在记录table YourTableName_1的字段的时候，只会记录两个到这个表的元数据信息，因而在对在对后面记录进行数据处理的时候，只会取后面记录的两个字段，而第三个字段是不会被插入到数据库中。解决这个问题很简单，把YourTableName_1的第二条记录和第一条记录给换一下位置好了。同理，在数据result数据文件的时候，也要遵循这样的规则，否则会得不到想要的结果的。这是经验总结。

　　你可能会担心如果是要准备的数据比较多是不是会比较麻烦，如上百条的数据准备，这个可以不用担心，因为使用DBUnit可以方便的从数据库中导出数据到指定的文件中，然后供我们使用，使用以下这个方法可以导出数据：

/**

* Export data for the table names by the given IDatabaseConnection into the resultFile.<br>

* The export data will be DBUnit format.

* @param conn

* @param tableNameList

* @param resultFile

* @throws SQLException

* @throws DatabaseUnitException

* @throws FileNotFoundException

* @throws IOException

public void exportData(IDatabaseConnection iconn， List<String> tableNameList， String resultFile) throws SQLException， DatabaseUnitException， FileNotFoundException， IOException {

QueryDataSet dataSet = null;

if (iconn == null) {

return;

}

if (tableNameList == null || tableNameList.size() == 0) {

return;

}

try {

dataSet = new QueryDataSet(iconn);

for (String tableName : tableNameList) {

dataSet.addTable(tableName);

}

} finally {

if (dataSet != null) {

FlatXmlDataSet.write(dataSet， new FileOutputStream(resultFile));

}

　　DBUnit的另一个非常有用的功能，是对执行结果进行比较，这样可以直接得到执行结果是否正确。操作方式是准备一个执行期待结果的XML文件，再准备一条从数据库查询结果的SQL。这里有一个经验非常重要，那是用于查询的执行结果的SQL文件，好是加上某个关键字段的ORDER BY语句，否则可能会因为记录的顺序而比较失败，因为DBUnit是把查询出来的结果和准备的结果进行一一对应的比较。当然，既然SQL查询都加上了排序，那我们的结果XML文件，也应该是根据关键字段排好序的结果的，否则也会因为记录的顺序问题而比较失败。

　　上面的是热身，该来点实际的东西了，弄个真实的实例来看看，下面是一个用于DBUnit测试的抽象类：

本文内容不用于商业目的，如涉及知识产权问题，请权利人联系SPASVO小编(021-61079698-8054)，我们将立即处理，马上删除。