项目中首次使用DBUnit拾遗

参考

• 测试整合之王Unitils（单元测试系列）

正文

测试准备

测试数据准备

dev_equipment_info.xlsx - 测试数据

expected_dev_equipment_info.xlsx - 测试后数据

Excel数据准备注意点

• 日期/时间格式 使用 'yyyy-mm-dd hh:mm:ss’格式
• 每个sheet表的首行为数据表字段名称
• sheet名称需要与表名完全相同，也可以使用database.tableName格式的sheet名
• 一个excel数据表可以同时有多个数据表
• Excel最后一行数据之后，不能只是清除内容，导入时无法识别

Maven 包导入

在pom.xml的dependencies中，增加

<dependency>
    <groupId>com.github.yangjianzhou</groupId>
    <artifactId>spring-boot-unitils-starter</artifactId>
    <version>1.3.0.RELEASE</version>
</dependency>

Unitils额外配置（数据库连接）

application-ut.properties 存放于resources下,其内容可参考

database.driverClassName=com.mysql.jdbc.Driver
# 此数据库连接信息
database.url=jdbc:mysql://127.0.0.1/septictank_device
# 此数据库连接用户名
database.userName=root
# 此数据库连接用户密码
database.password=123456
# 此数据库连接的schema
database.schemaNames=septictank_device
# 此数据库数据库类型：oracle/mysql/postgres等
database.dialect=mysql
dataSource.beanName=druidDataSource

测试类配置 - BusDeviceTestApp

package com.citylink.device;

import com.unitils.boot.autoconfigure.ConfigurableApplicationContextAware;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.ComponentScan;

@SpringBootApplication
@ComponentScan(basePackages = {"com.citylink.device"}, basePackageClasses =
        {ConfigurableApplicationContextAware.class, DataSourcePostProcessorExt.class})
public class BusDeviceTestApp {
    public static void main(String[] args) {
        SpringApplication.run(BusDeviceTestApp.class, args);
    }
}

主要实现以下功能：

• basePackages = {“com.citylink.device”}，扫描需要注入上下文的业务bean
• ConfigurableApplicationContextAware.class，为测试提供上下文，以便@SpringBean，@SpringBeanByName，@SpringBeanByType等从上下文中注入bean
• DataSourcePostProcessorExt.class，为替换系统中原有的dataSource bean的类，如测试时一直使用系统原数据库，很有可能是相应的dataSource未被正常替换

package com.citylink.device;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.stereotype.Component;
import org.unitils.database.UnitilsDataSourceFactoryBean;

import javax.sql.DataSource;

@Component
public class DataSourcePostProcessorExt implements BeanPostProcessor {

    public DataSourcePostProcessorExt() {
    }

    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }

    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        if (bean instanceof DataSource) {
            try {
                return (new UnitilsDataSourceFactoryBean()).getObject();
            } catch (Exception var4) {
                throw new RuntimeException("replace database throw exception ,can not continue to process", var4);
            }
        } else {
            return bean;
        }
    }
}

测试类 - DeviceLogServiceTest

package com.citylink.device;

import com.citylink.device.pojo.Device;
import com.citylink.device.service.IDeviceService;
import com.citylink.security.common.msg.ObjectRestResponse;
import com.unitils.boot.util.UnitilsBootBlockJUnit4ClassRunner;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.unitils.dbunit.annotation.DataSet;
import org.unitils.dbunit.annotation.ExpectedDataSet;
import org.unitils.spring.annotation.SpringBeanByType;

import java.util.List;

@RunWith(UnitilsBootBlockJUnit4ClassRunner.class)
@SpringBootTest(classes = BusDeviceTestApp.class)
@DataSet("dev_equipment_info.xlsx")
public class DeviceLogServiceTest {
    @SpringBeanByType
    private IDeviceService deviceService;

    @Test
    @ExpectedDataSet("expected_dev_equipment_info.xlsx")
    public void testSelectAll() {
        ObjectRestResponse orr = deviceService.selectAll();
        Assert.assertTrue(orr.isRel());
        List<Device> deviceList = (List<Device>) orr.getData();
        System.out.println("deviceList.size()=" + deviceList.size());
    }
}