PowerMock使用-Mock私有方法

前言

本篇文章将说明如何使用PowerMock对私有方法进行Mock。关于使用PowerMock需要引入哪些依赖，请参考PowerMock使用-依赖准备。

正文

被测试类如下所示。

public class MockPrivateMethod {    public boolean isTrue() {        return returnTrue();    }    private boolean returnTrue() {        return true;    }}

被测试类中有一个公共方法isTrue()，在isTrue()方法中会调用MockPrivateMethod的私有方法returnTrue()。测试类如下所示。

@RunWith(PowerMockRunner.class)@PrepareForTest(MockPrivateMethod.class)public class PowerMockTest {    @Test    public void mockPrivate() throws Exception {        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来方便的调用私有方法，如下所示。

public class MockPrivateMethod {    private boolean returnTrue() {        return true;    }}@RunWith(PowerMockRunner.class)@PrepareForTest(MockPrivateMethod.class)public class PowerMockTest {    @Test    public void mockPrivate() throws Exception {        MockPrivateMethod mockPrivateMethod = PowerMockito.mock(MockPrivateMethod.class);        PowerMockito.when(mockPrivateMethod, "returnTrue").thenReturn(false);        assertThat(Whitebox.invokeMethod(mockPrivateMethod, "returnTrue"),                is(false));    }}

来源地址：https://blog.csdn.net/Sakura_Sacrifice/article/details/128773618