Spring-Boot 整合 DBUnit 单元测试

import com.github.springtestdbunit.bean.DatabaseConfigBean;
import com.github.springtestdbunit.bean.DatabaseDataSourceConnectionFactoryBean;
import org.dbunit.ext.mysql.MySqlMetadataHandler;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;

@Configuration
public class DbUnitConfig {
    @Bean("dbUnitConnection")
    public DatabaseDataSourceConnectionFactoryBean getTestConnection(DataSource dataSource) {
        DatabaseDataSourceConnectionFactoryBean bean = new DatabaseDataSourceConnectionFactoryBean();
        bean.setDataSource(dataSource);
        DatabaseConfigBean databaseConfigBean = new DatabaseConfigBean();
        databaseConfigBean.setMetadataHandler(new MySqlMetadataHandler());
        databaseConfigBean.setAllowEmptyFields(true);
        bean.setDatabaseConfig(databaseConfigBean);
        bean.setSchema("schema_name");
        return bean;
    }
}

import com.github.springtestdbunit.dataset.AbstractDataSetLoader;
import org.apache.commons.lang3.StringUtils;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.csv.CsvURLDataSet;
import org.dbunit.dataset.excel.XlsDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSetBuilder;
import org.springframework.core.io.Resource;

import java.io.FileNotFoundException;
import java.io.InputStream;

public class XXXDataSetLoader extends AbstractDataSetLoader {
    protected IDataSet createDataSet(Resource resource) throws Exception {
        String filename = resource.getFilename();
        if (StringUtils.isEmpty(filename)) {
            throw new FileNotFoundException("File name cannot be empty:" + resource.getURL());
        }
        if (filename.endsWith(".csv")) {
            return new CsvURLDataSet(resource.getURL());
        } else if (filename.endsWith(".xml")) {
            FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder();
            builder.setColumnSensing(true);
            try {
                return builder.build(resource.getURL());
            } catch (Exception ex) {
                try (InputStream inputStream = resource.getInputStream()) {
                    return builder.build(inputStream);
                }
            }
        } else if (filename.endsWith(".xls")) {
            return new XlsDataSet(resource.getFile());
        } else {
            throw new IllegalArgumentException("Unsupported file type: " + filename);
        }
    }
}

import com.github.springtestdbunit.DbUnitTestExecutionListener;
import com.github.springtestdbunit.annotation.DbUnitConfiguration;
import com.azh3ng.config.XXXDataSetLoader;
import lombok.extern.slf4j.Slf4j;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockitoTestExecutionListener;
import org.springframework.test.annotation.Rollback;
import org.springframework.test.context.TestExecutionListeners;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.context.support.DependencyInjectionTestExecutionListener;
import org.springframework.test.context.support.DirtiesContextTestExecutionListener;
import org.springframework.test.context.transaction.TransactionalTestExecutionListener;
import org.springframework.transaction.annotation.Transactional;

@SpringBootTest
@RunWith(SpringRunner.class)
@Slf4j
@Transactional
@Rollback(true)
/*
  配置数据加载器为自定义的 XXXDataSetLoader.class
  即可 以 Excel文件(仅支持 xls文件, 不支持 xlsx 文件), csv文件， xml文件作为初始化数据和验证结果
 */
@DbUnitConfiguration(dataSetLoader = XXXDataSetLoader.class, databaseConnection = "dbUnitConnection")
@TestExecutionListeners({DependencyInjectionTestExecutionListener.class,
        DirtiesContextTestExecutionListener.class,
        TransactionalTestExecutionListener.class,
        MockitoTestExecutionListener.class,
        DbUnitTestExecutionListener.class})
public class FooDaoTest {
    @Autowired
    private FooDao dao;

    @Test
    /*
      配置事务
      前置条件:需要 @TestExecutionListeners 中包含 TransactionalTestExecutionListener.class
      事务会在 @DatabaseSetup 之前启动，在 @DatabaseTearDown 和 @ExpectedDatabase 之后结束, 并回滚
     */
    @Transactional
    /*
      初始化数据
      1.到当前类的包下找 "FooDao" 包下的 Excel 文件 "FooDao_setup.xls",
        （文件中 sheet 名即为表名, 多 sheet 页即多表 , sheet 页内, 第一行为字段名）
      2.清空数据库中的 Excel 文件中包含的表 的数据,
      3.将 Excel 中的数据插入到数据库
     */
    @DatabaseSetup("FooDao/FooDao_setup.xls")
    /*
      验证结果
      1.当测试代码执行完成后，到当前类的包下找 "FooDao" 包下的 Excel 文件 "FooDao_expect.xls"
      2.比对数据库中数据和 Excel 中数据是否相同
      assertionMode = NON_STRICT_UNORDERED 表示比对时忽略 Excel 中没有的表和字段, 并且比对时忽略数据行的顺序
     */
    @ExpectedDatabase(value = "FooDao/FooDao_expect.xls", assertionMode = NON_STRICT_UNORDERED)
    /*
      teardown
      1.到当前类的包下找 "FooDao" 包下的Excel文件 "FooDao_setup.xls"
      2.清空数据库中的 Excel 文件中包含的表 的数据
      type = DatabaseOperation.DELETE_ALL 表示删除数据库中的 Excel文件中包含的表 的所有数据
     */
    @DatabaseTearDown(value = "FooDao/FooDao_setup.xls", type = DatabaseOperation.DELETE_ALL)
    public void test() {
        dao.doTest();
    }
}