PowerMock使用-Mock公共方法

￡神魔★判官ぃ 2024-03-27 09:33 58阅读 0赞

### 前言 ###

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

### 正文 ###

如下是一个有一个公共方法的待`Mock`类。

public class MockPublicMethod {
        
    
        public boolean isTrue() {
        
            return true;
        }
    
    }

如下是使用`PowerMock`对`MockPublicMethod`进行打桩，让`MockPublicMethod#isTrue`方法返回**false**。

public class PowerMockTest {
        
    
        @Test
        public void mockPublic() {
        
            MockPublicMethod mockPublicMethod = PowerMockito
                    .mock(MockPublicMethod.class);
            PowerMockito.when(mockPublicMethod.isTrue()).thenReturn(false);
            assertThat(mockPublicMethod.isTrue(), is(false));
        }
    
    }

`Mock`公共方法时需要使用`PowerMockito.mock(方法所在类.class)`获取`Mock`出来的对象，这里称之为**mock**实例，**mock**实例的方法均为假方法，不对**mock**实例进行任何操作的情况下，调用**mock**实例的方法会返回（如果有返回值的话）返回值类型的默认值（零值，比如`String`返回**null**，`Integer`返回0）。如果想要调用**mock**实例的方法时使其执行真实方法，那么打桩时需要使用`thenCallRealMethod()`，如下所示。

public class MockPublicMethod {
        
    
        public boolean isTrue() {
        
            return true;
        }
    
    }
    
    public class PowerMockTest {
        
    
        @Test
        public void mockPublic() {
        
            MockPublicMethod mockPublicMethod = PowerMockito
                    .mock(MockPublicMethod.class);
            PowerMockito.when(mockPublicMethod.isTrue()).thenCallRealMethod();
            assertThat(mockPublicMethod.isTrue(), is(true));
        }
    
    }

同时可以使用`whenNew()`来实现在程序中**new**一个对象时得到一个**mock**实例。如下所示。

public class MockPublicMethod {
        
    
        public boolean isTrue() {
        
            return true;
        }
    
    }
    
    public class TestObj {
        
    
        public boolean isTrue() {
        
            MockPublicMethod mockPublicMethod = new MockPublicMethod();
            return mockPublicMethod.isTrue();
        }
    
    }
    
    @RunWith(PowerMockRunner.class)
    @PrepareForTest(TestObj.class)
    public class PowerMockTest {
        
    
        @Test
        public void mockWhenNew() throws Exception {
        
            MockPublicMethod mockPublicMethod = PowerMockito.mock(MockPublicMethod.class);
            PowerMockito.when(mockPublicMethod.isTrue()).thenReturn(false);
            PowerMockito.whenNew(MockPublicMethod.class).withAnyArguments()
                    .thenReturn(mockPublicMethod);
            TestObj testObj = new TestObj();
            assertThat(testObj.isTrue(), is(false));
        }
    
    }

**特别注意**：在使用`whenNew()`时，`@PrepareForTest`注解中一定得是被测试类的`Class`对象。上面例子中，在被测试类`TestObj`的方法中**new**了一个`Mock`类的对象，所以在`@PrepareForTest`注解中加入了`TestObj.class`，此时`whenNew()`才会在`TestObj`中生效。

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