一个DAO测试基类

 package com.taobao.dbunit.dao;

 import java.sql.SQLException;

 import javax.sql.DataSource;

 import org.dbunit.Assertion;
 import org.dbunit.database.DatabaseConnection;
 import org.dbunit.database.IDatabaseConnection;
 import org.dbunit.dataset.DataSetException;
 import org.dbunit.dataset.DefaultDataSet;
 import org.dbunit.dataset.DefaultTable;
 import org.dbunit.dataset.IDataSet;
 import org.dbunit.dataset.xml.FlatXmlDataSet;
 import org.dbunit.operation.DatabaseOperation;
 import org.junit.Assert;
 import org.junit.Before;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.core.io.ClassPathResource;
 import org.springframework.jdbc.datasource.DataSourceUtils;
 import org.springframework.test.context.ContextConfiguration;
 import org.springframework.test.context.junit4.AbstractTransactionalJUnit4SpringContextTests;
 import org.springframework.test.context.transaction.TransactionConfiguration;

 @ContextConfiguration(locations = { "classpath:testApplicationContext.xml" })
 @TransactionConfiguration(defaultRollback = true)
 public class BaseDaoTest extends AbstractTransactionalJUnit4SpringContextTests {

     @Autowired
     private DataSource dataSource;

     private IDatabaseConnection conn;

     @Before
     public void initDbunit() throws Exception {
         conn = new DatabaseConnection(DataSourceUtils.getConnection(dataSource));
    }

     /**
      * 清空file中包含的表中的数据，并插入file中指定的数据
      *
      * @param file
      * @throws Exception
      */
     protected void setUpDataSet(String file) throws Exception {
         IDataSet dataset = new FlatXmlDataSet(new ClassPathResource(file)
                .getFile());
        DatabaseOperation.CLEAN_INSERT.execute(conn, dataset);
    }

     /**
      * 验证file中包含的表中的数据和数据库中的相应表的数据是否一致
      *
      * @param file
      * @throws Exception
      */
     protected void verifyDataSet(String file) throws Exception {
         IDataSet expected = new FlatXmlDataSet(new ClassPathResource(file)
                .getFile());
        IDataSet dataset = conn.createDataSet();

         for (String tableName : expected.getTableNames()) {
            Assertion.assertEquals(expected.getTable(tableName), dataset
                    .getTable(tableName));
        }

    }

     /**
      * 清空指定的表中的数据
      *
      * @param tableName
      * @throws Exception
      */
     protected void clearTable(String tableName) throws Exception {
         DefaultDataSet dataset = new DefaultDataSet();
         dataset.addTable(new DefaultTable(tableName));
        DatabaseOperation.DELETE_ALL.execute(conn, dataset);
    }

     /**
      * 验证指定的表为空
      *
      * @param tableName
      * @throws DataSetException
      * @throws SQLException
      */
     protected void verifyEmpty(String tableName) throws DataSetException,
            SQLException {
         Assert.assertEquals(0, conn.createDataSet().getTable(tableName)
                .getRowCount());
    }
}

 @Test
 public void updateUser() throws Exception {

•     setUpDataSet("com/taobao/dbunit/dao/user001.xml");   
•        
•     User user = new User();   
•     user.setNick("user001");   
•     user.setPassword("password002");   
•     userDao.update(user);   
•   
•     verifyDataSet("com/taobao/dbunit/dao/user001_updated.xml");   
• }