学步园

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());
    }
}