PowerMock使用-Mock私有方法-Toy模板网

这篇具有很好参考价值的文章主要介绍了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来方便的调用私有方法，如下所示。文章来源地址https://www.toymoban.com/news/detail-504624.html

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));
    }

}