private(私有)方法單元測試無法覆蓋?那就用反射呼叫來測試private(私有)方法
阿新 • • 發佈:2019-02-06
程式碼測試覆蓋率是衡量軟體質量的重要指標,但常規的Junit不能對private方法進行測試,那麼就會影響我們的覆蓋率。
不過有一個辦法可以在Junit框架中測試private的方法,那就是反射!
我們知道可以通過反射method.invoke來呼叫某個方法,但如果方法前面為private怎麼辦呢?我們可以使用method.setAccessible(true);來跳過安全檢查,就可以呼叫private方法了。原理就這麼簡單。
這裡,我將這個功能封裝成一個工具類,單元測試中可以直接呼叫即可:
Java程式碼- 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;
- }
- }
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;
}
}