代码测试覆盖率是衡量软件质量的重要指标,但常规的Junit不能对private方法进行测试,那么就会影响我们的覆盖率。
不过有一个办法可以在Junit框架中测试private的方法,那就是反射!
我们知道可以通过反射method.invoke来调用某个方法,但如果方法前面为private怎么办呢?我们可以使用method.setAccessible(true);来跳过安全检查,就可以调用private方法了。原理就这么简单。
这里,我将这个功能封装成一个工具类,单元测试中可以直接调用即可:
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
public class PrivateMethodTestUtils {
/**
*
* @Description:调用只有一个参数的私有方法
* @since 1.0.0
* @Date:2013-1-23 下午2:54:00
* @param methodHostInstance
* @param methodName
* @param arg
* @return Object
*/
public static Object invoke(Object methodHostInstance, String methodName,
Object arg) {
Class<?>[] parameterTypes = { arg.getClass() };
Object[] args = { arg };
return invoke(methodHostInstance, methodName, parameterTypes, args);
}
/**
*
* @Description:调用有多个参数的私有方法
* @since 1.0.0
* @Date:2013-1-23 下午2:54:40
* @param methodHostInstance
* @param methodName
* @param parameterTypes
* @param args
* @return Object
*/
public static Object invoke(Object methodHostInstance, String methodName,
Class<?>[] parameterTypes, Object[] args) {
try {
Method method = methodHostInstance.getClass().getDeclaredMethod(
methodName, parameterTypes);
method.setAccessible(true);
try {
return method.invoke(methodHostInstance, args);
} catch (IllegalArgumentException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
} catch (InvocationTargetException e) {
e.printStackTrace();
}
method.setAccessible(false);
} catch (SecurityException e) {
e.printStackTrace();
} catch (NoSuchMethodException e) {
e.printStackTrace();
}
return null;
}
}
假如有如下待测试类:
public class UserManager{
public void addUser(User user){
……
encryptPwd(user.getPwd);
……
}
private String encryptPwd(String pwd){
String encryptPwd=encrypt(pwd);
return encryptPwd;
}
}
下面测试private的encryptPwd方法:
public class UserManagerTest{
@SpringBeanByType
private UserManager manager;
@Test
public void test密码加密(){
String pwd="123456";
Object result= PrivateMethodTestUtils.invoke(manager, "encryptPwd", pwd);
assertEquals("werjadf234f314dfdfg", result);
}
}
如果private方法 有多个参数,则将参数类型及参数对象封装成两个数组:
Class<?>[] parameterTypes = { arg.getClass() };
Object[] args = { arg };
然后调用PrivateMethodTestUtils.invoke(methodHostInstance, methodName,parameterTypes, args)
