基于Unitils和Spring解决一些单元测试的常见问题

在具体做单元测试的过程中，会遇到一些小问题，比如：

• 断言 JavaBean 或集合类
• 依赖 DB 数据
• 单元测试的数据清理
• Mock 依赖方
• 彻底排除第三方环境对单元测试的影响

 

Unitils 提供的特性和 Spring 的一些使用技巧能够帮助我们解决以上问题。下面以实战的形式来具体分解 ：

断言 JavaBean 或集合类

Unitils 提供了 ReflectionAssert 用于对 objects/collections 的断言，可以断言两个对象的属性是否全部相等，例如：

//Bean 断言，比较对象中每个属性的值

User user1 = new User(1, "John" , "Doe" );

User user2 = new User(1, "John" , "Doe" );

ReflectionAssert.assertReflectionEquals (user1, user2);

也可以断言集合中的成员是否全部相等，例如：

// 集合对象断言，比较集合中每个元素的值

List<Double> myList = new ArrayList<Double>();

myList.add(1.0);

myList.add(2.0);

ReflectionAssert.assertReflectionEquals (Arrays.asList (1, 2), myList);

当然， ReflectionAssert 还有其他的一些特性，比如忽略一些空值、断言对象某些属性等等，有需要的同学可以继续深究

 

依赖 DB 数据

我们单元测试中往往会涉及到 DAO 的操作，需要 DB 中存在所依赖的数据，如果通过原生态的 jdbc 去初始化就非常繁琐， Unitils 集成了 DBUnit ，通过 xml 的方式来做数据初始化，具体实战步骤如下：

1 ）在 unitils.properties 中配置数据源

例如：

database.driverClassName= oracle.jdbc.driver.OracleDriver

database.url= jdbc : oracle:thin:@10.20.147.2:1521:TESTDB

database.userName= money

database.password= ali88

 

2 ）编写初始化数据的 XML

按照预定格式编写数据文件，例如：

< dataset >

      < billing_log_monitor

     GUID = "wb_xinmin.zhao_test121"

     classname = "ChargeByTimesTask"

     methodname = "charge"

     type = "DAY_CHARGE_THREAD_END"

     recorddate = "2009-12-06"

     description = "chargeByTimeTase,121,151"

     error_msg = "12"

     isurgent = "0"

       />

</ dataset >

3 ）单元测试中应用

应用这个特性非常简单，只需要通过 @DataSet 载入上面编写好的 XML 即可，这个注解可以加在类名上（这个类所有测试方法都会加载 XML 中的数据），也可以加在方法名上（这作用于这个方法），例如：

    @Test

    @DataSet ( "LogDAOTest.xml" )

    public void testGetLogs()

数据的载入模式是可以定制的， Unitils 提供了以下几种：

CleanInsertLoadStrategy ：先把 dataSet 中涉及到的表都清掉，再插入 dataset 中的数据

InsertLoadStrategy ：直接插入 dataset 中的数据

RefreshLoadStrategy ： DB 已存在的数据就修改，不存在的就插入

UpdateLoadStrategy ：直接通过 dataset 中的数据修改 DB 中数据，如果不存在就抛出错误

一般说来使用 RefreshLoadStrategy 比较合理，配置方法就是在 unitils.properties 中配置：

DbUnitModule.DataSet.loadStrategy.default=org.unitils.dbunit.datasetloadstrategy.impl.RefreshLoadStrategy

 

单元测试的数据清理

单元测试过后往往会留下很多垃圾数据，并且有可能会导致下次跑单元测试失败（比如唯一性属性等问题），就有清理单元测试数据的需求， unitils 提供了事务的方式来满足这个需求，也就是在单元测试过后不提交事务。配置事务的方式有两种，一种是全局化配置（ unitils.properties ），如：

DatabaseModule.Transactional.value.default=disabled|commit|rollback

其中 disabled 是没有事务， commit 是单元测试方法过后提交事务， rollback 是回滚事务。另外也可以直接在方法上加@Transactional 来更细粒度的控制，如：

    @Transactional (TransactionMode. ROLLBACK )

public void testGetLogsCount()

以上表示 testGetLogsCount 执行过后自动回滚事务

 

Mock 依赖方

Mock 的方式很多， unitils 集成了 easyMock 也提供了 mock 的功能，并且使用起来更加方便，比如要想对logService 注入一个 mock 出来的 logDAO ，只需：

    private ILogService logService ;

    @InjectInto (target = "logService" , property = "logDAO" )

然后具体调用过程中：

    logDAOMock .onceReturns(5).getLogsCount( null );

    Assert.isTrue ( new Integer(5).equals ( logService .getLogsCount( null )));

即可完成 mock 的操作，方式有很多，重在方便

 

彻底排除第三方环境对单元测试的影响

单元测试依赖第三方常常因为环境问题导致单元测试失败，当然如果能在所以单元测试中都 mock 所依赖的类是可以解决这个问题，但在大型项目中很难保证没有漏网之鱼，特别是存在多层依赖的情况。其实通过 Spring 配置并在测试目录下替换掉对第三方的依赖就可以从根本上掐断祸端。具体实施中，我们在写 spring bean 配置文件的时候，可以把对第三方依赖的接口放在一个文件中，在测试目录下我们用另外一个配置文件替换它，此文件配置的都是我们 mock 第三方接口的实现类，这样单元测试的时候只要是通过 spring 依赖的，都会自动去依赖我们的 mock ，无需在单元测试代码里再去mock 。当然，如果需要 mock 出不同行为，那就需要在我们的 mock 类里去细化，可以通过线程变量来传递信息。这种方法没有使用任何 mock 框架，原理简单，关键是来得彻底。