准备测试数据的利器DBUNIT介绍

在开展测试过程中，要考虑的第一环节就是测试数据的准备。毫无疑问的，我们可以采用最基本的JDBC连接Oracle数据库，用insert语句完成数据的插入。例如下面这段代码：
示例代码 插入一条记录
01 public void insertData() throws Exception{
02 String dbUrl = "jdbc:oracle:thin:@10.0.4.194:1521:test";
03 String user = "uid";
04 String passwd = "pwd";
05 Class.forName("oracle.jdbc.driver.OracleDriver");
06 Connection c = DriverManager.getConnection(dbUrl,user,passwd);
07 Statement s = c.createStatement();
08 String sql = "insert into TABLE (id, name) values('001', 'mike')";
09 s.executeUpdate(sql);
10 }

但是这种方式有着明显的局限性：
1. 不适合大批量的测试数据准备。
2. 测试数据与代码没有分离。
3. 在代码中维护表结构和表之间的关系非常不妥，当表结构发生变动，代码需要做大量重构工作。
4. 没有断言功能，需要自己编写代码检查测试结果。

我们使用DbUnit来解决上述问题。DbUnit是用于维护测试环境的自动化工具，它本身是一个开源项目。这里我们介绍DbUnit最常用的功能。
用DbUnit插入测试数据
DbUnit支持多种方式向数据库中插入数据，例如FlatXmlDataSet, DTDDataSet等等。这里采用最常用的FlatXmlDataSet。顾名思义，这种方式就是用xml的方式准备数据，DbUnit载入xml文件并完成插入数据库的操作。
首先需要准备一份xml的数据文件，格式如下：
示例代码 数据文件dataset.xml
1 <?xml version="1.0" encoding="GB2312"?>
2 <dataset>
3 <TABLE id="001" name="mike" />
4 <TABLE id="002" name="jack" />
5 </dataset>

第2行的dataset标签是xml的根节点，对应于DbUnit中的一个FlatXmlDataSet对象。
第3行表示要插入的一条记录。其中表名为TABLE，插入的字段为id、name，对应的值分别为“001”、“mike”。
整个xml文件一共插入两条记录。
注意
Xml文件中的值必须用双引号，DbUnit会根据实际的表结构进行类型转换。DbUnit无法插入空值，只能跳过该字段。
接下来要做的就是载入这份数据文件。
示例代码 载入xml文件中的数据
1 public IDataSet getDataSet(String path) {
2 FlatXmlDataSet dataSet = null;
3 try {
4 dataSet = new FlatXmlDataSet(new FileInputStream(new File(path)));
5 } catch (Exception e) {
6 e.printStackTrace();
7 }
8 return dataSet;
9 }

第2行声明一个FlatXmlDataSet对象用来装载测试数据。
第4行用FlatXmlDataSet的构造函数读取path指定的文件来初始化dataSet对象。
第8行返回载有数据的对象。

最后就是连接数据库，将对象中的数据插到数据库中。代码8.5给出了一份较完整的代码。
示例代码 DbUnit.java –insertData
01 package util;
02
03 ...
04
05 public class DbUnit {
06 private IDatabaseTester databaseTester;
07
08 private IDatabaseConnection connection;
09
10 private QueryDataSet queryDataSet;
11
12 private String driver;
13
14 private String url;
15
16 private String user;
17
18 private String password;
19
20 private Connection jdbcConnection;
21
22 public DbUnit() {}
23
24 public DbUnit(String driver, String url, String user, String password) {
25
26 try {
27 Class driverClass = Class.forName(driver);
28 jdbcConnection = DriverManager.getConnection(url, user, password);
29 } catch (Exception e) {
30 e.printStackTrace();
31 }
32 }
33
34 public void setSchema(String schema) throws SQLException {
35 connection = new DatabaseConnection(jdbcConnection, schema);
36 }
37
38 public void insertData(IDataSet dataSet) {
39 try {
40 //DatabaseOperation.DELETE.execute(connection, dataSet);
41 DatabaseOperation.INSERT.execute(connection, dataSet);
42 } catch (Exception e) {
43 e.printStackTrace();
44 }
45 }
46
47 public void deleteData(IDataSet dataSet) {
48 try {
49 DatabaseOperation.DELETE.execute(connection, dataSet);
50 } catch (Exception e) {
51 e.printStackTrace();
52 }
53 }
54
55 public void updateData(IDataSet dataSet) {
56 try {
57 DatabaseOperation.UPDATE.execute(connection, dataSet);
58 } catch (Exception e) {
59 e.printStackTrace();
60 }
61 }
62 }
63
64 public QueryDataSet getQueryDataSet() {
65 return queryDataSet;
66 }
67
68 public IDataSet getDataSet(String path) {
69 FlatXmlDataSet dataSet = null;
70 try {
71 dataSet = new FlatXmlDataSet(new FileInputStream(new File(path)));
72 } catch (Exception e) {
73 // TODO Auto-generated catch block
74 e.printStackTrace();
75 }
76 return dataSet;
77 }

第24-32行建立一个jdbc连接到指定的数据库，DbUnit将jdbcConnection对象封装成自身的通用数据库接口对象IDatabaseConnection。
第34-36行，定义setSchema方法完成不同schema之间的切换。
第38-45行，传入数据库连接和数据这个两参数，执行INSERT操作完成最终的插入动作。DatabaseOpeartion中还定义了UPDATE、DELETE等其他常用操作，这些操作都是DatabaseOperation类中的静态成员对象。DbUnit中的具体实现这里就不细讲了，有兴趣的可以参考DbUnit源代码。