单元测试工具类ReflectionTestUtils
前言在写单元测试时,是否经常遇到要测试的目标类中有很多私有注入的变量呢?然而我们经常并没有报漏此私有属性的getter and setter,因为这些私有属性的注入可能是通过配置文件,或者其他渠道,是在程序启动时根据环境不同或者其他因素来决定的,所以当我们测试的时候,就需要通过为该属性设置不同的值而来测试各种分支case,但是前面已经说了一般这种变量都是私有变量,如果说专门为了测试而将修饰符改为public/protected 那未免过于暴力....
于是乎,大家常见的做法可能就是写一堆反射去修改,如果只有一处地方还好,但是如果需要测试此种场景的过多,那是不是写都写烦了,所以ReflectionTestUtils解放你的双手。
用法
首先搞一个service来辅助:
public class TestServiceImpl {
private String configId = "11111";
public String getConfigId() {
return configId;
}
}新建一个测试类:
class TestServiceImplTest {
private TestServiceImpl testService = new TestServiceImpl();
@Test
void getConfigId() {
System.out.println("修改前:" + testService.getConfigId());
ReflectionTestUtils.setField(testService,"configId","2222");
System.out.println("修改后:" + testService.getConfigId());
}
}
输出:
修改前:11111
22:35:40.628 DEBUG org.springframework.test.util.ReflectionTestUtils - Setting field 'configId' of type on target object or target class to value
修改后:2222通过输出以及代码,可以看到非常的简单,一行代码即可完成私有属性的更改,是不是十分的方便!
原理
这里就简单贴一下代码,看看:
public static void setField(@Nullable Object targetObject, @Nullable Class<?> targetClass, @Nullable String name, @Nullable Object value, @Nullable Class<?> type) {
Assert.isTrue(targetObject != null || targetClass != null, "Either targetObject or targetClass for the field must be specified");
if (targetObject != null && springAopPresent) {
targetObject = AopTestUtils.getUltimateTargetObject(targetObject);
}
if (targetClass == null) {
targetClass = targetObject.getClass();
}
Field field = ReflectionUtils.findField(targetClass, name, type);
if (field == null) {
throw new IllegalArgumentException(String.format("Could not find field '%s' of type [%s] on %s or target class [%s]", name, type, safeToString(targetObject), targetClass));
} else {
if (logger.isDebugEnabled()) {
logger.debug(String.format("Setting field '%s' of type [%s] on %s or target class [%s] to value [%s]", name, type, safeToString(targetObject), targetClass, value));
}
ReflectionUtils.makeAccessible(field);
ReflectionUtils.setField(field, targetObject, value);
}
}
了解下
页:
[1]