在我們使用JUnit單元測試框架編寫單元測試的時候,少不免要對資料庫進行操作,但請試想一下,當我要編寫一個獲取使用者的單元測試時,資料庫是不存在該記錄的,那麼我要測試獲取使用者時就需要往資料庫新增一條使用者記錄,但當獲取使用者的單元測試完成併成功後,此測試並沒有清理現場(刪除插入資料庫的記錄),那樣當我們再有單元測試需要插入記錄時,就會造成ID衝突的情況,少量的單元測試還可以避免此種情況,但當單元測試的資料龐大時,就會出現各種各樣的問題,而DBUnit可以幫助我們解決這型別的問題

1. <dependency>
2. <groupId>org.dbunit</groupId>
3. <artifactId>dbunit</artifactId>
4. <version>2.4.9</version>
5. <scope>test</scope>
6. </dependency>
7. <dependency>
8. <groupId>org.slf4j</groupId>
9. <artifactId>slf4j-api</artifactId>
10. <version>1.7.5</version>
11. <scope>compile</scope>
12. </dependency
    >

      在Maven管理的專案src/test/resource/下建立default-data.xml 

1. <?xmlversion="1.0"encoding="UTF-8"?>
2. <dataset>
3. <!-- 注意,此處的user代表表名(database table name) -->
4. <userid="1"username="123"password="456"/>
5. <userid="2"username="123"password="456"/>
6. <userid="3"username="123"password="456"
   />
7. <userid="4"username="123"password="456"/>
8. <userid="5"username="123"password="456"/>
9. </dataset>

      以下程式碼為提取後的BaseTest 

1. package com.accentrix.ray;  
2. import java.io.File;  
3. import java.io.FileInputStream;  
4. import java.io.FileWriter;  
5. import java.io.InputStream;  
6. import javax.sql.DataSource;  
7. import org.dbunit.database.DatabaseDataSourceConnection;  
8. import org.dbunit.database.IDatabaseConnection;  
9. import org.dbunit.database.QueryDataSet;  
10. import org.dbunit.dataset.DataSetException;  
11. import org.dbunit.dataset.IDataSet;  
12. import org.dbunit.dataset.xml.FlatXmlDataSet;  
13. import org.dbunit.dataset.xml.FlatXmlProducer;  
14. import org.dbunit.operation.DatabaseOperation;  
15. import org.junit.AfterClass;  
16. import org.junit.BeforeClass;  
17. import org.xml.sax.InputSource;  
18. publicclass BaseTest {  
19. publicstatic IDatabaseConnection connection;  
20. publicstatic DataSource dataSource;  
21. private File temp;  
22. @BeforeClass
23. publicstaticvoid init() throws Exception {  
24. // 通過DataSource獲取DataBaseSourceConnection
25.         connection = new DatabaseDataSourceConnection(dataSource);  
26.     }  
27. @AfterClass
28. publicstaticvoid destroy() throws Exception {  
29. if (connection != null) {  
30.             connection.close();  
31.         }  
32.     }  
33. protected IDataSet getDataSet(String name) throws DataSetException {  
34. // 通過類載入器獲取default-data.xml檔案的內容
35.         InputStream is = this.getClass().getClassLoader()  
36.                 .getResourceAsStream(name + ".xml");  
37. // IDataSet就類似是一個數據的容器,把default-data.xml內容
38. // 轉換成了DBUnit可識別的資料返回出去
39. returnnew FlatXmlDataSet(new FlatXmlProducer(new InputSource(is)));  
40.     }  
41. protectedvoid backupAll() throws Exception {  
42. // createDataSet代表從資料庫中獲取到DataSet,此時DataSet為資料庫的內容
43.         IDataSet ds = connection.createDataSet();  
44. // 建立臨時檔案
45.         temp = File.createTempFile("temp", "xml");  
46. // 將資料庫內容的DataSet寫入臨時檔案當中
47.         FlatXmlDataSet.write(ds, new FileWriter(temp), "UTF-8");  
48.     }  
49. protectedvoid backupCustom(String... tableName) throws Exception {  
50. // 此種形式能儲存某幾張表的的資料,而不需要全部備份
51.         QueryDataSet qds = new QueryDataSet(connection);  
52. for (String str : tableName) {  
53.             qds.addTable(str);  
54.         }  
55.         temp = File.createTempFile("temp", "xml");  
56.         FlatXmlDataSet.write(qds, new FileWriter(temp), "UTF-8");  
57.     }  
58. protectedvoid recover() throws Exception {  
59. // 獲取資料庫內容的臨時檔案生成DataSet
60.         IDataSet ds = new FlatXmlDataSet(new FlatXmlProducer(new InputSource(  
61. new FileInputStream(temp))));  
62. // 使用DatabaseOperation的列舉方法先清空資料庫內容再還原資料庫
63.         DatabaseOperation.CLEAN_INSERT.execute(connection, ds);  
64.     }  
65. }  

      以下為單元測試類 

1. package com.accentrix.ray;  
2. importstatic org.junit.Assert.assertEquals;  
3. importstatic org.junit.Assert.assertNotNull;  
4. import org.dbunit.dataset.IDataSet;  
5. import org.dbunit.operation.DatabaseOperation;  
6. import org.junit.Test;  
7. publicclass TestUser extends BaseTest {  
8. @Test
9. publicvoid testGetUser() throws Exception {  
10. // 首先獲取到default-data.xml檔案中的內容
11.         IDataSet ds = getDataSet("default-data.xml");  
12. // 呼叫backupAll方法儲存資料庫表
13.         backupAll();  
14. // 將default-data.xml中的內容替換成資料庫的內容
15.         DatabaseOperation.CLEAN_INSERT.execute(connection, ds);  
16. // 進行單元測試邏輯的判斷
17.         User user = userService.get(1);  
18.         assertNotNull(user);  
19.         assertEquals(user.getId(), new Integer(1));  
20.         assertEquals(user.getPassword(), "456");  
21.         assertEquals(user.getUsername(), "123");  
22. // 最後進行資料庫的恢復
23. // 當然我們也可以將backupAll和recover放在@Before和@After當中
24.         recover();  
25.     }  
26. }  

總結: 
      使用了DBUnit後可以實現了對資料庫的隔離,成功彌補了JUnit單元測試不清理資料現場的缺憾,實際上DBUnit只是簡單的在單元測試前把資料庫的資料進行了備份然後插入xml中配置好的資料,在測試結束後再用備份好的原資料庫資料填充回資料庫.但當面對複雜的表關係和大資料量的時候,每次進行測試都進行資料的備份,也是一個很大的負擔,而且把全部的測試資料都編寫在xml當中也是很大的工作量