PowerMock使用-Mock私有方法

前言

本篇文章将说明如何使用

PowerMock

对私有方法进行

Mock

。关于使用

PowerMock

需要引入哪些依赖，请参考PowerMock使用-依赖准备。

正文

被测试类如下所示。

publicclassMockPrivateMethod{publicbooleanisTrue(){returnreturnTrue();}privatebooleanreturnTrue(){returntrue;}}

被测试类中有一个公共方法

isTrue()

，在

isTrue()

方法中会调用

MockPrivateMethod

的私有方法

returnTrue()

。测试类如下所示。

@RunWith(PowerMockRunner.class)@PrepareForTest(MockPrivateMethod.class)publicclassPowerMockTest{@TestpublicvoidmockPrivate()throwsException{MockPrivateMethod mockPrivateMethod =PowerMockito.mock(MockPrivateMethod.class);PowerMockito.when(mockPrivateMethod,"returnTrue").thenReturn(false);PowerMockito.when(mockPrivateMethod.isTrue()).thenCallRealMethod();assertThat(mockPrivateMethod.isTrue(),is(false));}}

Mock

私有方法打桩时，需要使用

PowerMockito.when(mock实例, "私有方法名").thenReturn(期望返回值)

的形式设置mock实例的私有方法的返回值，如果私有方法有参数，还需要在私有方法名后面添加参数占位符，比如

PowerMockito.when(mock实例, "私有方法名", anyInt()).thenReturn(期望返回值)

。上面例子中进行断言时，调用私有方法采取了调用公共方法来间接调用私有方法的形式，单元测试代码对业务代码造成了入侵，因此如果仅仅只是为了验证一个私有方法，可以使用

Whitebox

来方便的调用私有方法，如下所示。

publicclassMockPrivateMethod{privatebooleanreturnTrue(){returntrue;}}@RunWith(PowerMockRunner.class)@PrepareForTest(MockPrivateMethod.class)publicclassPowerMockTest{@TestpublicvoidmockPrivate()throwsException{MockPrivateMethod mockPrivateMethod =PowerMockito.mock(MockPrivateMethod.class);PowerMockito.when(mockPrivateMethod,"returnTrue").thenReturn(false);assertThat(Whitebox.invokeMethod(mockPrivateMethod,"returnTrue"),is(false));}}