1. 程式人生 > >private(私有)方法單元測試無法覆蓋?那就用反射呼叫來測試private(私有)方法

private(私有)方法單元測試無法覆蓋?那就用反射呼叫來測試private(私有)方法

程式碼測試覆蓋率是衡量軟體質量的重要指標,但常規的Junit不能對private方法進行測試,那麼就會影響我們的覆蓋率。

不過有一個辦法可以在Junit框架中測試private的方法,那就是反射!

我們知道可以通過反射method.invoke來呼叫某個方法,但如果方法前面為private怎麼辦呢?我們可以使用method.setAccessible(true);來跳過安全檢查,就可以呼叫private方法了。原理就這麼簡單。

這裡,我將這個功能封裝成一個工具類,單元測試中可以直接呼叫即可:

Java程式碼 複製程式碼 收藏程式碼
  1. import java.lang.reflect.InvocationTargetException;
  2. import java.lang.reflect.Method;
  3. public class PrivateMethodTestUtils {
  4. /**
  5. *
  6. * @Description:呼叫只有一個引數的私有方法
  7. * @since 1.0.0
  8. * @Date:2013-1-23 下午2:54:00
  9. * @param methodHostInstance
  10. * @param methodName
  11. * @param arg
  12. * @return Object
  13. */
  14. public static Object invoke(Object methodHostInstance, String methodName,
  15. Object arg) {
  16. Class<?>[] parameterTypes = { arg.getClass() };
  17. Object[] args = { arg };
  18. return invoke(methodHostInstance, methodName, parameterTypes, args);
  19. }
  20. /**
  21. *
  22. * @Description:呼叫有多個引數的私有方法
  23. * @since 1.0.0
  24. * @Date:2013-1-23 下午2:54:40
  25. * @param methodHostInstance
  26. * @param methodName
  27. * @param parameterTypes
  28. * @param args
  29. * @return Object
  30. */
  31. public static Object invoke(Object methodHostInstance, String methodName,
  32. Class<?>[] parameterTypes, Object[] args) {
  33. try {
  34. Method method = methodHostInstance.getClass().getDeclaredMethod(
  35. methodName, parameterTypes);
  36. method.setAccessible(true);
  37. try {
  38. return method.invoke(methodHostInstance, args);
  39. } catch (IllegalArgumentException e) {
  40. e.printStackTrace();
  41. } catch (IllegalAccessException e) {
  42. e.printStackTrace();
  43. } catch (InvocationTargetException e) {
  44. e.printStackTrace();
  45. }
  46. method.setAccessible(false);
  47. } catch (SecurityException e) {
  48. e.printStackTrace();
  49. } catch (NoSuchMethodException e) {
  50. e.printStackTrace();
  51. }
  52. return null;
  53. }
  54. }
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;
	}
}