dbunit 一篇文章

dbunit 在命令下

mvn dbunit:operation -Ddbunit.operation.type=DELETE_ALL

什麼是dbunit以及爲什麼要使用它?

dbunit是一個基於junit擴展的數據庫測試框架。它提供了大量的類對與數據庫相關的操作進行了抽象和封裝，雖然在80%的情況，你只需使用它極少的api。它通過使用用戶自定義的數據集以及相關操作使數據庫處於一種可知的狀態，從而使得測試自動化、可重複和相對獨立。雖然不用dbunit也可以達到這種目的，但是我們必須爲此付出代價（編寫大量代碼，測試及維護），既然有了這麼優秀的開源框架，我們又何必再造輪子。

 

dbunit的原理

 

dbunit的與單元測試相關的兩個最重要的核心是org.dbunit.database.IDatabaseConnection 和 org.dbunit.dataset.IDataSet ，

前者是產品代碼使用的數據庫連接的一個簡單的封裝

後者是對單元測試人員自定義的數據集（通常以xml文件的形式存在，且xml文件的格式也有好幾種）的封裝。

 

還有一個很重要的咚咚就是org.dbunit.operation.DatabaseOperation，該類是一個抽象類代表了對數據庫的操作，例如CUD以及其組合等， 它採用了退化的工廠模式，可直接通過它獲取其具體的子類（代表具體的某種操作）如下：

DatabaseOperation.UPDATE

DatabaseOperation.DELETE

DatabaseOperation.DELETE_ALL

DatabaseOperation.TRUNCATE

DatabaseOperation.REFRESH

DatabaseOperation.CLEAN_INSERT

DatabaseOperation.NONE

 

工作流程如下：

1)testcase.setup--->testcase.getConnection-->getDataSet----->operation.execute(
通常DatabaseOperation.CLEAN_INSERT）

2)testcase.testSomeMethod---->dao.someMethod

3)testcase.teardown---->operation.execute(
通常DatabaseOperation.DELETE_ALL或者DatabaseOperation.NONE）

 

 

實戰

以一個真實的系統（share@mofile)爲例，

1. 建立一個測試數據庫
2. 寫自定義的數據

1. <?xml version='1.0' encoding='UTF-8'?>
2.  
3. <dataset>
4.  
5. <FileType typeId='1' typeName='type1' />
6.  
7. <FileType typeId='2' typeName='type2' />
8.  
9. <FileType typeId='3' typeName='type3' />
10.  
11. <UserInSharedSystem userId='1' loginName='2' userNo='a' />
12.  
13. <DefaultFile fileId='1' fileName='a1' fileSize='10' fileStatus='1'
14. pickupCode='4414402888619758' releaseDate='2006-1-31 16:17:18'
15. storageFileHome='solar' storageFileId='1' userId='1' typeId='1'
16. description='ss' expiredDate='2006-1-31 16:17:18'
17. uploadDate='2006-1-31 16:17:18' extName='exe' totalPoints='0'
18. currentMonthPoints='0' lastMonthPoints='0' />
19.  
20. <DefaultFile fileId='2' fileName='a2' fileSize='20' fileStatus='1'
21. pickupCode='4414402888619759' releaseDate='2006-1-31 16:17:18'
22. storageFileHome='solar' storageFileId='2' userId='1' typeId='2'
23. description='ss' expiredDate='2006-1-31 16:17:18'
24. uploadDate='2006-1-31 16:17:18' extName='exe' totalPoints='0'
25. currentMonthPoints='0' lastMonthPoints='0' />
26.  
27. <DefaultFile fileId='3' fileName='a3' fileSize='30' fileStatus='1'
28. pickupCode='4414402888619768' releaseDate='2006-1-31 16:17:18'
29. storageFileHome='solar' storageFileId='3' userId='1' typeId='3'
30. description='ss' expiredDate='2006-1-31 16:17:18'
31. uploadDate='2006-1-31 16:17:18' extName='exe' totalPoints='0'
32. currentMonthPoints='0' lastMonthPoints='0' />
33.  
34. <FileTag tagId='1' tagName='t1' />
35.  
36. <FileTag tagId='2' tagName='t2' />
37.  
38. <FileTag tagId='3' tagName='t3' />
39.  
40.  
41. <FileTagRel fileId='1' tagId='1' />
42.  
43. <FileTagRel fileId='1' tagId='2' />
44.  
45. <FileTagRel fileId='2' tagId='1' />
46.  
47. <FileTagRel fileId='2' tagId='3' />
48.  
49. <FileTagRel fileId='3' tagId='2' />
50.  </dataset>

       注意數據的順序，否則會有約束違例，特別是外鍵約束

 

1. 寫基本的測試類

 

1. package mofile.share.dao;
2.  
3. import javax.sql.DataSource;
4.  
5.  
6.  
7. import mofile.common.utils.SpringBeanProxy;
8.  
9.  
10. import org.dbunit.DatabaseTestCase;
11. import org.dbunit.database.DatabaseConnection;
12. import org.dbunit.database.IDatabaseConnection;
13. import org.dbunit.dataset.IDataSet;
14. import org.dbunit.operation.DatabaseOperation;
15. import org.springframework.context.ApplicationContext;
16. import org.springframework.context.support.ClassPathXmlApplicationContext;
17.  
18. /**
19. * 使用dbunit
20. *
21. * @author weip
22. * @time 2006-3-9 13:42:32
23. *
24. */
25. public abstract class BaseDaoTest extends DatabaseTestCase {
26.  
27. protected final static ApplicationContext ctx;
28.  
29. protected IDatabaseConnection conn = null;
30.  
31. protected IDataSet dataSet = null;
32.  
33. static {
34. // String pkg = ClassUtils.classPackageAsResourcePath(Constants.class);
35. String[] paths = { "classpath:applicationContext-share-database.xml",
36. "classpath:applicationContext-share-hibernate.xml",
37. "applicationContext-share-property.xml"/*
38. * ,
39. * "classpath:applicationContext-central-database.xml"
40. */};
41. ctx = new ClassPathXmlApplicationContext(paths);
42. SpringBeanProxy.setApplicationContext(ctx);
43. }
44.  
45. public BaseDaoTest(String methodName) {
46. super(methodName);
47. }
48.  
49. /**
50. *
51. * @author weip
52. * @time 2006-3-9 13:46:45 (non-Javadoc)
53. * @see org.dbunit.DatabaseTestCase#getConnection()
54. *
55. */
56.  
57. protected IDatabaseConnection getConnection() throws Exception {
58. DataSource ds = (DataSource) ctx.getBean("shareDataSource");
59.  
60. conn = new DatabaseConnection(ds.getConnection());
61.  
62. return conn;
63. }
64.  
65. /**
66. *
67. * @author weip
68. * @time 2006-3-9 13:52:27 (non-Javadoc)
69. * @see org.dbunit.DatabaseTestCase#getSetUpOperation()
70. *
71. */
72.  
73. protected DatabaseOperation getSetUpOperation() throws Exception {
74. return DatabaseOperation.CLEAN_INSERT;
75. }
76.  
77. /**
78. *
79. * @author weip
80. * @time 2006-3-9 13:52:31 (non-Javadoc)
81. * @see org.dbunit.DatabaseTestCase#getTearDownOperation()
82. *
83. */
84.  
85. protected DatabaseOperation getTearDownOperation() throws Exception {
86. //this.closeConnection(conn);
87. //conn=null;
88. return DatabaseOperation.NONE;
89. }
90.  
91.  
92. }

 

1. 寫具體的測試用例

 

1. package mofile.share.dao;
2.  
3. import java.io.FileInputStream;
4. import java.util.List;
5.  
6. import junit.framework.Test;
7. import junit.framework.TestSuite;
8.  
9. import mofile.share.domain.DefaultFile;
10. import mofile.share.domain.DefaultUser;
11. import mofile.share.domain.FileTag;
12. import mofile.share.domain.FileType;
13. import mofile.share.vo.FileForm;
14.  
15. import org.dbunit.dataset.IDataSet;
16. import org.dbunit.dataset.xml.FlatXmlDataSet;
17. import org.dbunit.operation.DatabaseOperation;
18.  
19. /**
20. *
21. * @author weip
22. * @time 2006-3-9 11:13:34
23. *
24. */
25. public class FileDaoTest extends BaseDaoTest {
26. private FileDao fileDao;
27.  
28. protected void setUp() throws Exception {
29. super.setUp();
30.  
31. /*try {
32. DatabaseOperation.CLEAN_INSERT.execute(conn, dataSet);
33. } finally {
34. conn.close();
35. }*/
36. fileDao = (FileDao) ctx.getBean("fileDao");
37. }
38.  
39. protected void tearDown() throws Exception {
40. super.tearDown();
41. }
42.  
43. public FileDaoTest(String methodName) {
44. super(methodName);
45. }
46.  
47. public static Test suite() {
48. TestSuite suite = new TestSuite();
49.  
50. suite.addTest(new FileDaoTest("testQueryAllFiles"));
51. suite.addTest(new FileDaoTest("testQueryFilesByFileType"));
52.  
53. suite.addTest(new FileDaoTest("testAddFiles"));
54.  
55. suite.addTest(new FileDaoTest("testRemoveObject"));
56.  
57. return suite;
58. }
59.  
60. /**
61. *
62. *  @author weip
63. *  @time 22:53:34 2006-3-16
64. *  @see org.dbunit.DatabaseTestCase#getDataSet()
65. */
66. protected IDataSet getDataSet() throws Exception {
67. dataSet = new FlatXmlDataSet(new FileInputStream("d:/Projects/Share/Mofile_share/htdocs/WEB-INF/classes/mofile/share/dao/testFileDB.xml"));
68. return dataSet;
69.  
70. }
71.  
72. /**
73. *
74. * @author weip
75. * @time 22:53:17 2006-3-16
76. * @throws Exception
77. */
78. public void testAddFiles() throws Exception {
79. FileType fileType = new FileType();
80. fileType.setTypeId(new Integer(1));
81.  
82. FileTag fileTag = new FileTag();
83. fileTag.setTagName("dev3");
84.  
85. DefaultUser user = new DefaultUser();
86. // user.setLoginName("");
87. // user.setUserNo(new Long(1));
88. user.setUserId(new Long(1));
89.  
90. DefaultFile file = new DefaultFile();
91.  
92. file.setFileName("file001");
93. file.setFileSize(20);
94. file.setFileStatus(1);
95. file.setPickupCode("0001");
96. file.setStorageFileHome("xxx");
97. file.setStorageFileId(new Long(1));
98. file.setUser(user);
99. file.setFileType(fileType);
100. file.addTag(fileTag);
101.  
102. fileDao.saveObject(file);
103.  
104. DefaultFile fileafter=fileDao.getFileByStorageFileId("xxx",new Long(1));
105.  
106. assertNotNull(fileafter);
107.  
108. assertEquals("file001",fileafter.getFileName());
109.  
110.  
111. }
112.  
113. /**
114. *
115. * @author weip
116. * @time 2006-3-9 14:01:05
117. * @throws Exception
118. */
119. public void testQueryAllFiles() throws Exception {
120. FileForm frm = new FileForm();
121. frm.setFileStatus(-1);
122. frm.setFileType(-1);
123.  
124. List list = fileDao.queryFile(frm);
125.  
126. assertEquals(3, list.size());
127.  
128. }
129.  
130. /**
131. *
132. * @author weip
133. * @time 22:47:01 2006-3-16
134. * @throws Exception
135. */
136. public void testQueryFilesByFileType() throws Exception {
137. FileForm frm = new FileForm();
138. frm.setFileStatus(-1);
139. frm.setFileType(1);
140.  
141. List list = fileDao.queryFile(frm);
142.  
143. assertEquals(1, list.size());
144.  
145. }
146.  
147. /**
148. *
149. * @author weip
150. * @throws Exception
151. * @time 23:10:33 2006-3-16
152. */
153. public void testRemoveObject() throws Exception {
154.  
155. fileDao.removeObject(DefaultFile.class, new Long(2));
156.  
157. FileForm frm = new FileForm();
158. frm.setFileStatus(-1);
159. frm.setFileType(-1);
160.  
161. List list = fileDao.queryFile(frm);
162.  
163. assertEquals(2, list.size());
164.  
165. }
166.  
167.  
168. }

 

對比以下以前寫的測試用例

public class FileDaoImplTestCase extends BaseDaoTestCase {

private FileDao fileDao;

 

protected void setUp() throws Exception {

super.setUp();

fileDao = (FileDao) ctx.getBean("fileDao");

}

 

protected void tearDown() throws Exception {

super.tearDown();

}

 

public FileDaoImplTestCase(String methodName) {

super(methodName);

}

 

public static Test suite() {

TestSuite suite = new TestSuite();

// suite.addTest(new FileDaoImplTest("testRemoveObject"));

 

// suite.addTest(new UserDaoImplTest("testSaveObject"));

// suite.addTest(new FileDaoImplTestCase("testSaveObject"));

// suite.addTest(new FileDaoImplTestCase("testQueryFileByTag"));

//suite.addTest(new FileDaoImplTestCase("testQueryFile4Pagenation"));

suite.addTest(new FileDaoImplTestCase("testUpdateLastMonthPoints"));

 

return suite;

}

 

/*

* Test method for

* 'mofile.share.dao.impl.BaseHibernateDaoImpl.getObject(Class,

* Serializable)'

*/

public void testGetObject() {

 

}

 

/**

* 插入一條文件記錄 ，注意插入記錄需檢查是否存在重複的filetag

*

* @author weip

* @time 16:42:20 2006-1-15

* @throws Exception

*

* Hibernate: insert into DefaultFile (fileName, fileSize, fileStatus,

* pickupCode, releaseDate, storageFileHome, storageFileId, userId, typeId)

* values (?, ?, ?, ?, ?, ?, ?, ?, ?) Hibernate: insert into FileTag

* (tagName) values (?) Hibernate: insert into FileTagRel (fileId, tagId)

* values (?, ?)

*

*/

public void testSaveObject() throws Exception {

FileType fileType = new FileType();

fileType.setTypeId(new Integer(1));

 

FileTag fileTag = new FileTag();

fileTag.setTagName("dev3");

 

DefaultUser user = new DefaultUser();

// user.setLoginName("");

// user.setUserNo(new Long(1));

user.setUserId(new Long(1));

 

DefaultFile file = new DefaultFile();

 

file.setFileName("file001");

file.setFileSize(20);

file.setFileStatus(1);

file.setPickupCode("0001");

file.setStorageFileHome("xxx");

file.setStorageFileId(new Long(1));

file.setUser(user);

file.setFileType(fileType);

file.addTag(fileTag);

 

fileDao.saveObject(file);

 

}

 

/**

*

* 刪除一條文件記錄

*

* @author weip

* @time 16:47:29 2006-1-15

*

* Hibernate: select defaultfil0_.fileId as fileId2_, defaultfil0_.fileName

* as fileName2_, defaultfil0_.fileSize as fileSize2_,

* defaultfil0_.fileStatus as fileStatus2_, defaultfil0_.pickupCode as

* pickupCode2_, defaultfil0_.releaseDate as releaseD6_2_,

* defaultfil0_.storageFileHome as storageF7_2_, defaultfil0_.storageFileId

* as storageF8_2_, defaultfil0_.userId as userId2_, defaultfil0_.typeId as

* typeId2_, defaultuse1_.userId as userId0_, defaultuse1_.loginName as

* loginName0_, defaultuse1_.userNo as userNo0_, filetype2_.typeId as

* typeId1_, filetype2_.typeName as typeName1_ from DefaultFile defaultfil0_

* left outer join UseInSharedSystem defaultuse1_ on

* defaultfil0_.userId=defaultuse1_.userId left outer join FileType

* filetype2_ on defaultfil0_.typeId=filetype2_.typeId where

* defaultfil0_.fileId=? Hibernate: select tagset0_.fileId as fileId__,

* tagset0_.tagId as tagId__, filetag1_.tagId as tagId0_, filetag1_.tagName

* as tagName0_ from FileTagRel tagset0_ inner join FileTag filetag1_ on

* tagset0_.tagId=filetag1_.tagId where tagset0_.fileId=? Hibernate: delete

* from FileTagRel where fileId=? Hibernate: delete from DefaultFile where

* fileId=?

*/

public void testRemoveObject() {

 

fileDao.removeObject(DefaultFile.class, new Long(2));

 

}

 

/**

* 測試按標籤獲取文件

*

* @author weip

* @throws Exception

* @time 2006-1-24 20:12:07

*/

public void testQueryFileByTag() throws Exception {

 

List list = fileDao.queryFileByTag(54, 0, 10);

 

assertNotNull(list);

 

DefaultFile file = null;

if (list.size() > 0)

file = (DefaultFile) list.get(0);

}

 

public void testQueryFile4Pagenation() throws Exception {

FileForm frm = new FileForm();

frm.setFileStatus(-1);

frm.setFileType(-1);

fileDao.queryFileSum(frm);

}

 

/**

* 測試 批量更新上月得分總數

*

* @author weip

* @throws Exception

* @time 2006-2-21 16:07:32

*/

public void testUpdateLastMonthPoints() throws Exception {

fileDao.updateLastMonthPoints();

assertTrue(true);

 

}

 

}

 

 

• 總結

 

通過與以前的例子比較發現：使用dbunit的testcase更自動化和可重複，以前寫的testcase與數據庫中的數據嚴重耦合，所以一般都不敢寫斷言，寫了之後怕數據又發生變化，所以測試也是不可重複，並且也不是自動化，因爲沒有斷言，你不得不測試完之後還得檢查數據庫。

 

當然dbunit也許並不是銀彈，它在併發測試的時候得表現我沒有實踐過，也不敢妄下斷言，而且是不是應該另外再建一個同樣的數據專門測試dao還值得思考

 

我們在項目中爲每個開發人員自建一個數據庫解決併發問題,也許這個方案並非最佳,但實用