dbunit使用

系统管理员 2021-05-12 11:50 506阅读 0赞

1.dbunit从名字上可以看出，是对有关数据库操作的测试。官网: [www.dbunit.org][] 从官网下载dbunit相关的jar包，dbunit还依赖于 [http://www.slf4j.org][http_www.slf4j.org] 和commons-logging.jar，所以也需要下载slf4j，使用它的slf4j-api-1.5.3.jar和slf4j-jcl-1.5.3.jar。   
DbUnit([http://dbunit.sourceforge.net/][http_dbunit.sourceforge.net] )  是专门针对数据库测试的对 JUnit  的一个扩展，它可以将测试对象数据库置于一个测试轮回之间的状态。  
dbunit的原理是：它会把数据库表里的数据和一个xml文件里表示的数据关联起来。也就是说  数据库表里的数据可以导出到一个对应的xml里，同时也可以将一个xml里的数据导入到数据库表里。是相互转换的。    
                  数据库表里的数据 <===> xml文件    
  
可以做个测试如下：

Java代码   ![收藏代码][icon_star.png]

1.  **package** com.test.dbunit;  
2.    
3.  **import** java.io.FileOutputStream;  
4.  **import** java.sql.Connection;  
5.  **import** java.sql.DriverManager;  
6.    
7.  **import** org.dbunit.database.DatabaseConnection;  
8.  **import** org.dbunit.database.IDatabaseConnection;  
9.  **import** org.dbunit.database.QueryDataSet;  
10. **import** org.dbunit.dataset.xml.FlatXmlDataSet;  
11.   
12. **public** **class** TestApp \{  
13.       
14.     **public** **static** **void** main(String\[\] args) **throws** Exception\{  
15.         Class.forName("com.mysql.jdbc.Driver");  
16.   
17.         Connection conn = DriverManager.getConnection(  
18.                 "jdbc:mysql://localhost/dbunit", "root", "");  
19.           
20.         IDatabaseConnection connection = **new** DatabaseConnection(conn);  
21.         QueryDataSet dataSet = **new** QueryDataSet(connection);  
22.         //将整个person表里的数据导出到 xml文件里  
23.         dataSet.addTable("person");  
24.         //将users表里符合条件的数据导出到xml文件里  
25.         dataSet.addTable("users","select \* from users where id < 4");  
26.         //导出到dbunit.xml文件里  
27.         FlatXmlDataSet.write(dataSet,**new** FileOutputStream("dbunit.xml"));  
28.     \}  
29.   
30. \}

如果想把某个数据库里的所有表里的数据全部导出到某个xml里,又不想通过addTable一个个来添加的话。则必须通过IDatabaseConnection的createDataSet()来创建IDataSet

Java代码   ![收藏代码][icon_star.png]

1.  **public** **static** **void** main(String\[\] args) **throws** Exception\{  
2.          Class.forName("com.mysql.jdbc.Driver");  
3.    
4.          Connection conn = DriverManager.getConnection(  
5.                  "jdbc:mysql://localhost/dbunit", "root", "");  
6.            
7.          IDatabaseConnection connection = **new** DatabaseConnection(conn);  
8.          //如果想把某个数据库里的所有表里的数据全部导出到某个xml里,又不想通过addTable一个个来添加的话。则必须通过IDatabaseConnection的createDataSet()来创建IDataSet  
9.          IDataSet dataSet = connection.createDataSet();  
10.         //导出到dbunit.xml文件里  
11.         FlatXmlDataSet.write(dataSet,**new** FileOutputStream("dbunit1.xml"));  
12.         //也可以用FlatDtdDataSet导出一个对应的dtd文件  
13.         FlatDtdDataSet.write(dataSet,**new** FileOutputStream("dbunit1.dtd"));  
14.     \}

生成的dbunit.xml内容如下：

Java代码   ![收藏代码][icon_star.png]

1.  <?xml version='1.0' encoding='UTF-8'?>  
2.  <dataset>  
3.    <person/>  
4.    <users id="1" username="zhangsan" password="123"/>  
5.    <users id="2" username="lisi" password="456"/>  
6.    <users id="3" username="wangwu" password="789"/>  
7.  </dataset>

2.用dbunit做测试，需要继承DBTestCase,并需要重写protected IDataSet getDataSet() throws Exception

Java代码   ![收藏代码][icon_star.png]

1.  **package** com.test.dbunit;  
2.    
3.  **import** java.io.File;  
4.  **import** java.io.FileInputStream;  
5.    
6.  **import** org.dbunit.Assertion;  
7.  **import** org.dbunit.DBTestCase;  
8.  **import** org.dbunit.PropertiesBasedJdbcDatabaseTester;  
9.  **import** org.dbunit.dataset.IDataSet;  
10. **import** org.dbunit.dataset.ITable;  
11. **import** org.dbunit.dataset.SortedTable;  
12. **import** org.dbunit.dataset.filter.DefaultColumnFilter;  
13. **import** org.dbunit.dataset.xml.FlatXmlDataSet;  
14. **import** org.dbunit.operation.DatabaseOperation;  
15. **public** **class** SimpleTest **extends** DBTestCase \{  
16.   
17.     **public** SimpleTest()  
18.     \{  
19. //      super();  
20.         System.setProperty(  
21.                 PropertiesBasedJdbcDatabaseTester.DBUNIT\_DRIVER\_CLASS,  
22.                 "com.mysql.jdbc.Driver");  
23.         System.setProperty(  
24.                 PropertiesBasedJdbcDatabaseTester.DBUNIT\_CONNECTION\_URL,  
25.                 "jdbc:mysql://localhost/dbunit");  
26.         System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT\_USERNAME,  
27.                 "root");  
28.         System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT\_PASSWORD,  
29.                 "");  
30.     \}  
31.       
32.     /\*\* 
33.      \* 给定数据集 
34.      \*/  
35.     @Override  
36.     **protected** IDataSet getDataSet() **throws** Exception \{  
37.         System.out.println("init...");  
38.         **return** **new** FlatXmlDataSet(**new** FileInputStream("dbunit.xml"));  
39.     \}  
40.       
41.     **public** **void** test1() **throws** Exception\{  
42.         IDataSet dbDataSet = getConnection().createDataSet();  
43.         ITable dbTable = dbDataSet.getTable("users");  
44.           
45.         IDataSet xmlDataSet = **new** FlatXmlDataSet(**new** FileInputStream("dbunit.xml"));  
46.         ITable xmlTable = xmlDataSet.getTable("users");  
47.           
48.         Assertion.assertEquals(xmlTable, dbTable);  
49.     \}  
50.     /\*\* 
51.      \* 表结构和dbunit1.xml结构不一致 
52.      \*/  
53.     **public** **void** test2() **throws** Exception\{  
54.         IDataSet dbDataSet = getConnection().createDataSet();  
55.         ITable dbTable = dbDataSet.getTable("users");  
56.           
57.         IDataSet xmlDataSet = **new** FlatXmlDataSet(**new** FileInputStream("dbunit1.xml"));  
58.         ITable xmlTable = xmlDataSet.getTable("users");  
59.         //第一种写法  
60. //      dbTable = DefaultColumnFilter.includedColumnsTable(dbTable,xmlTable.getTableMetaData().getColumns());  
61.         //第二种写法  
62.         dbTable = DefaultColumnFilter.includedColumnsTable(dbTable,xmlDataSet.getTableMetaData("users").getColumns());  
63.           
64.         Assertion.assertEquals(xmlTable, dbTable);  
65.     \}  
66.     /\*\* 
67.      \* 只比较username 
68.      \*/  
69.     **public** **void** test3() **throws** Exception\{  
70.         IDataSet dbDataSet = getConnection().createDataSet();  
71.         ITable dbTable = dbDataSet.getTable("users");  
72.           
73.         IDataSet xmlDataSet = **new** FlatXmlDataSet(**new** FileInputStream("dbunit1.xml"));  
74.         ITable xmlTable = xmlDataSet.getTable("users");  
75.           
76.         dbTable = DefaultColumnFilter.excludedColumnsTable(dbTable, **new** String\[\]\{ "id","password"\});  
77.         xmlTable = DefaultColumnFilter.excludedColumnsTable(dbTable, **new** String\[\]\{ "password"\});  
78.           
79.         Assertion.assertEquals(xmlTable, dbTable);  
80.     \}  
81.     /\*\* 
82.      \* 表里数据和dbunit2.xml里的数据的顺序不一致 
83.      \*/  
84.     **public** **void** test4() **throws** Exception\{  
85.         IDataSet dbDataSet = getConnection().createDataSet();  
86.         ITable dbTable = dbDataSet.getTable("users");  
87.           
88.         IDataSet xmlDataSet = **new** FlatXmlDataSet(**new** FileInputStream("dbunit2.xml"));  
89.         ITable xmlTable = xmlDataSet.getTable("users");  
90.           
91.         SortedTable dbSortedTable = **new** SortedTable(dbTable,**new** String\[\]\{ "id"\});  
92.         //按数据库里表结构字段类型来进行排序，如果是int则按int排序，不是依照字符串的形式排序  
93.         dbSortedTable.setUseComparable(**true**);  
94.           
95.         SortedTable xmlSortedTable = **new** SortedTable(xmlTable,**new** String\[\]\{ "id"\});  
96.         //按数据库里表结构字段类型来进行排序，如果是int则按int排序，不是依照字符串的形式排序  
97.         xmlSortedTable.setUseComparable(**true**);  
98.           
99.         Assertion.assertEquals(xmlSortedTable, dbSortedTable);  
100.     \}  
101.       
102.     /\*\* 
103.      \* 对Users表插入一条记录做测试 
104.      \* @throws Exception 
105.      \*/  
106.     **public** **void** test5() **throws** Exception\{  
107.         UsersDB db = **new** UsersDB();  
108.   
109.         Users users = **new** Users();  
110.         users.setId(9);  
111.         users.setUsername("hello");  
112.         users.setPassword("world");  
113.   
114.         db.save(users);  
115.   
116.         IDataSet dataSet = getConnection().createDataSet();  
117.         ITable actualTable = dataSet.getTable("users");  
118.   
119.         IDataSet dataSet2 = **new** FlatXmlDataSet(**new** File("expected.xml"));  
120.         ITable expectedTable = dataSet2.getTable("users");  
121.   
122.         Assertion.assertEquals(expectedTable, actualTable);  
123.     \}  
124.       
125.     /\*\* 
126.      \* 在把xml里的数据导入到db里之前，需要对数据库里的表的数据做的操作。 
127.      \*/  
128.     **protected** DatabaseOperation getSetUpOperation() **throws** Exception  
129.     \{  
130.         System.out.println("setup...");  
131.         **return** DatabaseOperation.CLEAN\_INSERT;  
132.     \}  
133.     /\*\* 
134.      \* 测试执行完以后，需要对数据库里的表里的数据做的操作 
135.      \*/  
136.     **protected** DatabaseOperation getTearDownOperation() **throws** Exception  
137.     \{  
138.         System.out.println("teardown...");  
139.         **return** DatabaseOperation.DELETE\_ALL;  
140.     \}  
141.   
142. \}

其中expected.xml内容为：

Java代码   ![收藏代码][icon_star.png]

[www.dbunit.org]: http://ttitfly.iteye.com/blog/www.dbunit.org
[http_www.slf4j.org]: http://www.slf4j.org/
[http_dbunit.sourceforge.net]: http://dbunit.sourceforge.net/
[icon_star.png]: /images/20210511/bcf5092309f74e208fdf045bcd2169a0.png