在java写单测过程中经常使用Mockito来为类中的属性和方法赋值,以简单高效的测试方法中逻辑的正确性。但遇到父类中存在引入变量时,初始化子类就会抛出空指针异常,导致整个单测报错,具体实例如下
public class A {
@Value{"${test}"}
protect String c;
public void method() {
System.out.println(c);
}
}
public class B extends A {
@Resource
D d;
public void method2() {
d.method3();
}
}
当需要写单测测试method2方法时需要初始化B类,但此时B类初始化也要初始化A类,A类中c是空的,直接报错。今天查了很多方法也不能实现,结合其中一个博客做了改进终于实现了,这里记录一下,以备后用。
首先,需要考虑的是为父类中的变量赋值,这是一切的基础。废话不说,直接上代码
public class FieldUtils {
// fieldName: 字段名, handle:类名,这里可以直接写子类, target: 赋的变量
public static void setSuperField(String fieldName, Object handle, Object target) {
Field field = ReflectionUtils.findField(handle.getClass().getSuperclass(), fieldName);
ReflectionUtils.makeAccessible(field);
ReflectionUtils.setField(field, handle, target);
}
}
这段代码可以直接为父类中的变量赋值,此时要做的是把子类初始化出来,这里不能直接new,需要使用到了Mockite的方法
B b = Mockito.mock(B.class, Mockito.CALLS_REAL_METHODS);
这样就可以顺利实现为父类中变量赋值了