private(私有)方法单元测试无法覆盖?那就用反射调用来测试private(私有)方法

February 13, 2015

继续阅读

代码测试覆盖率是衡量软件质量的重要指标,但常规的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)