Dbunit的使用---dataset

女爷i 2022-05-18 01:17 276阅读 0赞

关于IDataSet的使用，目前主要用到了FlatXmlDataSet ，XmlDataSet ，XlsDataset ,CompositeDataSet，ReplacementDataSet

**FlatXmlDataSet**

FlatXmlDataSet 是最常用的一种,文件格式是这样的

<TEST\_TABLE COL0="row 0 col 0" COL1="row 0 col 1" COL2="row 0 col 2"/>

<TEST\_TABLE COL1="row 1 col 1"/>

<SECOND\_TABLE COL0="row 0 col 0" COL1="row 0 col 1" />

<EMPTY\_TABLE/>

</dataset>

举例说明用法, 有这样一个xml文件: src\\test\\resources\\dataset\\invoke\_webservice\_log.xml

<dhzx\_webservice\_local  id="1" visit\_date ="\[NOW\]" status="1" remark="\[UUID\]" />

<dhzx\_webservice\_local  id="2" visit\_date ="\[NOW\]" status="1"  />

<dhzx\_webservice\_log id="1" visit\_date ="\[NOW\]" status="1" serial\_number="\[UUID\]" />

<dhzx\_invoke\_log id="1" create\_date="\[NOW\]" result="1" log\_id="\[LOGID\]" />

</dataset>

Java代码操作xml一般是这样的：

public IDataSet createIDataSet() {
    
    DataFileLoader loader = new FlatXmlDataFileLoader();
    
    IDataSet dataSet = loader.load("/dataset/invoke_webservice_log.xml");
    
    Map<String, Object> replacementObjects = Maps.newHashMap();
    
    replacementObjects.put("[NOW]", new Date());
    
    replacementObjects.put("[UUID]", uuid);
    
    loader.addReplacementObjects(replacementObjects);
    
    return dataSet;
    
    }

解释一下: 这里面用到了FlatXmlDataFileLoader，这个是dbUnit自带的解析dataset文件的利器，非常好用,他还集成了ReplacementDataSet的功能,可以做变量替换 。

一般情况下，用FlatXmlDataSet+FlatXmlDataFileLoader的方式就能解决大部分问题，但是官网推荐使用XmlDataSet，因为字段是由第一行数据决定的，比如表dhzx\_webservice\_local的第一条数据有A B C三个字段,而第二条数据有A B C D四个字段,最终XmlDataSet还是以A B C三个字段为准,因此使用FlatXmlDataSet的时候一定要保证字段最全的数据在最前面。

**XmlDataSet**

XmlDataSet的xml文件需要使用dtd约束规范(FlatXmlDataSet并没有强制要求，我估计FlatXmlDataSet是它的默认实现吧)，因此要先定义一个dtd文件,如下 dataset.dtd

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

<!ELEMENT dataset (table+)>

<!ELEMENT table (column\*, row\*)>

<!ATTLIST table name CDATA \#REQUIRED>

<!ELEMENT column (\#PCDATA)>

<!ELEMENT row (value | null | none)\*>

<!ELEMENT value (\#PCDATA)>

<!ELEMENT null EMPTY>

<!ELEMENT none EMPTY>

然后再使用xml文件 ，文件名为dhzx\_invoke\_local\_log.xml （注意其中的dataset.dtd即为上面定义的dtd文件）

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

<!DOCTYPE dataset SYSTEM "dataset.dtd">

<column>create\_date</column>

<column>result</column>

<row>

<value>\[LOGID\]</value>

<value>\[LOGID\]</value>

</row>

</table>

</dataset>

Java代码是这样的(文件都存放在 src\\test\\resources\\dataset目录下)

public IDataSet createIDataSet() throws Exception {
    
    InputStream inputStream = this.getClass().getResource("/dataset/dhzx_invoke_local_log.xml").openStream();
    
    ReplacementDataSet dataSet = new ReplacementDataSet(new XmlDataSet(inputStream));
    
    dataSet.addReplacementObject("[NOW]", new Date());
    
    dataSet.addReplacementObject("[LOGID]", logId);
    
    return dataSet;
    
    }

解释一下:这里主要使用了XmlDataSet和ReplacementDataSet ，XmlDataSet就是针对xml+dtd的解析,而ReplacementDataSet就是为了做文件中的标识替换

**XlsDataSet**

XlsDataSet使用excel作为数据源,每个sheet表示一个表,sheet名为表名，第一行为字段名，其它行表示字段值，如下图所示

![WEBRESOURCEa20a831fd48133176b0c28808780842c][]

初始化代码如下:

private static IDataSet createDataSet() throws Exception {
    
    XlsDataFileLoader fileLoader = new XlsDataFileLoader();
    
    IDataSet dataSet = fileLoader.load("/datasets/data.xlsx");
    
    ReplacementDataSet replacementDataSet = new ReplacementDataSet(dataSet);
    
    replacementDataSet.addReplacementObject("[NOW]", DateFormatUtils.format(new Date(), "yyyy-MM-dd"));
    
    replacementDataSet.addReplacementSubstring("[UUID]", UUID.randomUUID().toString());
    
    return replacementDataSet;
    
    }

**CompositeDataSet**

顾名思义，这是一个组合式的dataset，即它可以把多个dataset组合为一个，而每个dataset对应一个xml，因此也可以说CompositeDataSet可以组合多个dataset的xml

![WEBRESOURCEcc5e8c2726f447e88f1e266068f54c68][]

具体例子不再细说，我们可以把图中的4个xml组合为一个,当然这里的4个xml用的式XmlDataset的方式,具体java代码式这样的

public IDataSet createIDataSet() throws Exception{
    
    IDataSet[] iDataSets = new IDataSet[4] ;
    
    InputStream inputStream = this.getClass().getResource("/dataset/dhzx_webservice_local_log.xml")
    
    .openStream();
    
    iDataSets[0] = new XmlDataSet(inputStream);
    
    
    
    inputStream = this.getClass().getResource("/dataset/dhzx_invoke_local_log.xml")
    
    .openStream();
    
    iDataSets[1] = new XmlDataSet(inputStream);
    
    
    
    inputStream = this.getClass().getResource("/dataset/dhzx_webservice_log.xml")
    
    .openStream();
    
    iDataSets[2] = new XmlDataSet(inputStream);
    
    
    
    inputStream = this.getClass().getResource("/dataset/dhzx_invoke_log.xml")
    
    .openStream();
    
    iDataSets[3] = new XmlDataSet(inputStream);
    
    
    
    //将4个dataset组合为一个compositeDataSet
    
    CompositeDataSet compositeDataSet = new CompositeDataSet(iDataSets) ;
    
    //将compositeDataSet包装为replacementDataSet,便于进行变量替换
    
    ReplacementDataSet replacementDataSet = new ReplacementDataSet(compositeDataSet);
    
    replacementDataSet.addReplacementObject("[NOW]", new Date());
    
    replacementDataSet.addReplacementObject("[UUID]", uuid);
    
    return replacementDataSet;
    
    }

具体就不解释了，代码就是最好的注释

**ReplacementDataSet**

通过以上的几个例子就可以看出来，ReplacementDataSet是做文件中变量替换的，它的构造方法是直接以其他dataset为入参的，构造起来非常方便

关于dataset的更详细说明，可参考官方文档： [http://dbunit.sourceforge.net/components.html][http_dbunit.sourceforge.net_components.html]

[WEBRESOURCEa20a831fd48133176b0c28808780842c]: https://note.youdao.com/yws/res/1167/WEBRESOURCEa20a831fd48133176b0c28808780842c
[WEBRESOURCEcc5e8c2726f447e88f1e266068f54c68]: http://note.youdao.com/yws/res/788/WEBRESOURCEcc5e8c2726f447e88f1e266068f54c68
[http_dbunit.sourceforge.net_components.html]: http://dbunit.sourceforge.net/components.html