PowerMock使用-Mock私有方法

喜欢ヅ旅行 2024-03-27 09:33 31阅读 0赞

### 前言 ###

本篇文章将说明如何使用`PowerMock`对私有方法进行`Mock`。关于使用`PowerMock`需要引入哪些依赖，请参考[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));
        }
    
    }

[PowerMock_-]: https://blog.csdn.net/Sakura_Sacrifice/article/details/128773055