动态测试数据-@DataSet

之前介绍了在数据库测试时，可以通过@DateSet注解的方式将一些预定义的数据导入到目标数据库中，以实现对数据库上下文的控制。一般情况下，DataBaseRider可以很好地完成这项工作。
而在某些测试场景中，则需要某些数据是需要动态生成的，例如ID、序列号、日期、时间等等。
譬如在关于某个订单系统的测试时，系统只会处理当天的数据。而使用@ExportDataSet导出的数据，则会日期、时间等数据在导出后就不再变化了。这类数据如果是直接导入使用的话，可能会导致测试场景无法触发的问题。于是，一个很自然的需求就产生了
可否在导入数据时，能将日期字段的值替换为系统当前的日期？

DBRider的 DataSet Replacers

@DBUnit(cacheConnection = true)
public class DateReplacementsIt {
    Calendar now;
    @Before
    public void setup(){
        now = Calendar.getInstance();
    }
    @Test
    @DataSet(value = "datasets/yml/date-replacements.yml",disableConstraints = true, executorId = "rules-it")
    public void shouldReplaceDateWithNowPlaceHolder() {
        Tweet tweet = (Tweet) EntityManagerProvider.em()
.createQuery("select t from Tweet t where t.id = '1'").getSingleResult();
        assertThat(tweet).isNotNull();
        assertThat(tweet.getDate().get(Calendar.DAY_OF_MONTH)).isEqualTo(now.get(Calendar.DAY_OF_MONTH));
        assertThat(tweet.getDate().get(Calendar.HOUR_OF_DAY)).isEqualTo(now.get(Calendar.HOUR_OF_DAY));
    }

通过上述用例来验证date-replacements.yml文件导入到数据库中的内容的确是当前月份和日期。那么来看一下date-replacements.yml文件。

TWEET:
  - ID: "1"
    CONTENT: "dbunit rules!"
    DATE: "[DAY,NOW]"
    USER_ID: 1
  - ID: "2"
    CONTENT: "dbunit rules!"
    DATE: "[DAY,YESTERDAY]"
    USER_ID: 1
  - ID: "4"
    CONTENT: "dbunit rules!"
    DATE: "[DAY,YEAR_AFTER]"
    USER_ID: 1
  - ID: "6"
    CONTENT: "dbunit rules!"
    DATE: "[HOUR,PLUS_ONE]"
    USER_ID: 1
  - ID: "7"
    CONTENT: "dbunit rules!"
    USER_ID: 1
    TIMESTAMP: "[UNIX_TIMESTAMP]"

DBRider提供了[DAY,NOW] 这样的占位符，用以替换时间类的字段值。从上述示例来看，不仅支持年、月、日、小时的当前值，还可以前后进行偏移，以及时间戳。

DBRider源码分析

DBRider通过提供Repalcer接口来支持动态数据替换的功能。原生提供了日期时间、Null和时间戳这三种类型。

image.png

以最简单的时间戳来看，

package com.github.database.rider.core.replacers;

import org.dbunit.dataset.ReplacementDataSet;

/**
 * Replacer which replaces [UNIX_TIMESTAMP] placeholder with Unix timestamp (obtained through {@link System#currentTimeMillis()}
 *
 * @author njuro
 */
public class UnixTimestampReplacer implements Replacer {

    @Override
    public void addReplacements(ReplacementDataSet dataSet) {
        long timestamp = System.currentTimeMillis() / 1000L;
        dataSet.addReplacementObject("[UNIX_TIMESTAMP]", timestamp);
    }
}

读者可以参考上述案例来实现自己的CustomReplacer