之前介绍了在数据库测试时，可以通过@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
