前言

本篇文章將說明如何使用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 {@Testpublic 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 {@Testpublic void mockPrivate() throws Exception {MockPrivateMethod mockPrivateMethod = PowerMockito.mock(MockPrivateMethod.class);PowerMockito.when(mockPrivateMethod, "returnTrue").thenReturn(false);assertThat(Whitebox.invokeMethod(mockPrivateMethod, "returnTrue"),is(false));}}

總結

以上是生活随笔為你收集整理的PowerMock使用-Mock私有方法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。