DbUnit 是在JUnit的基础上扩展而成的Java单元测试框架。
如果你的业务逻辑涉及到了对数据库中记录的增、删、改、查操作,而你又不想每次都手动到数据库里查询,来验证你的业务逻辑,那么DbUnit可以帮助你。实际上,如果你的单元测试中只有少量业务逻辑对数据库进行了操作,那么,DbUnit在单元测试中的优势还体现不出来,但是,如果你的单元测试用例中有大量的DAO操作,那么全凭手动执行数据库查询,讲造成工作效率降低,而且可能会遗漏功能点。而且,在回归测试中,验证DAO操作的工作不能重用。
现在就让我们开始认识DbUnit吧!首先介绍org.dbunit.DatabaseTestCase.java类,源码如下:
package org.dbunit; import junit.framework.TestCase; import org.dbunit.database.IDatabaseConnection; import org.dbunit.dataset.IDataSet; import org.dbunit.operation.DatabaseOperation; /** */ /** * @author Manuel Laflamme * @version $Revision: 1.11 $ * @since Feb 17, 2002 */ public abstract class DatabaseTestCase extends TestCase ... { public DatabaseTestCase() ...{ } public DatabaseTestCase(String name) ...{ super(name); } /** *//** * Returns the test database connection. */ protected abstract IDatabaseConnection getConnection() throws Exception; /** *//** * Returns the test dataset. */ protected abstract IDataSet getDataSet() throws Exception; /** *//** * Close the specified connection. Ovverride this method of you want to * keep your connection alive between tests. */ protected void closeConnection(IDatabaseConnection connection) throws Exception ...{ connection.close(); } /** *//** * Returns the database operation executed in test setup. */ protected DatabaseOperation getSetUpOperation() throws Exception ...{ return DatabaseOperation.CLEAN_INSERT; } /** *//** * Returns the database operation executed in test cleanup. */ protected DatabaseOperation getTearDownOperation() throws Exception ...{ return DatabaseOperation.NONE; } private void executeOperation(DatabaseOperation operation) throws Exception ...{ if (operation != DatabaseOperation.NONE) ...{ IDatabaseConnection connection = getConnection(); try ...{ operation.execute(connection, getDataSet()); } finally ...{ closeConnection(connection); } } } // TestCase class protected void setUp() throws Exception ...{ super.setUp(); executeOperation(getSetUpOperation()); } protected void tearDown() throws Exception ...{ super.tearDown(); executeOperation(getTearDownOperation()); }}