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还值得思考

 

我们在项目中为每个开发人员自建一个数据库解决并发问题,也许这个方案并非最佳,但实用