PowerMock使用-Mock静态私有方法

樱花祭的约定

已于 2023-01-28 21:00:12 修改

阅读量2.5k

点赞数 1

分类专栏：单元测试文章标签： java 单元测试 PowerMock Mock静态私有方法

于 2023-01-27 23:05:18 首次发布

本文链接：https://blog.csdn.net/Sakura_Sacrifice/article/details/128773737

版权

单元测试专栏收录该内容

12 篇文章 2 订阅

订阅专栏

前言

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

正文

被测试类如下所示。

public class MockStaticPrivateMethod {

    public static boolean isTrue() {
        return returnTrue();
    }

    private static boolean returnTrue() {
        return true;
    }

}

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

@RunWith(PowerMockRunner.class)
@PrepareForTest(MockStaticPrivateMethod.class)
public class PowerMockTest {

    @Test
    public void mockStaticPrivate() throws Exception {
        PowerMockito.mockStatic(MockStaticPrivateMethod.class);
        PowerMockito.when(MockStaticPrivateMethod.class, "returnTrue")
                .thenReturn(false);
        PowerMockito.when(MockStaticPrivateMethod.isTrue()).thenCallRealMethod();
        assertThat(MockStaticPrivateMethod.isTrue(), is(false));
    }

}

同样测试代码对业务代码造成了入侵，可以使用Whitebox来方便的调用静态私有方法，如下所示。

public class MockStaticPrivateMethod {

    private static boolean returnTrue() {
        return true;
    }

}

@RunWith(PowerMockRunner.class)
@PrepareForTest(MockStaticPrivateMethod.class)
public class PowerMockTest {

    @Test
    public void mockStaticPrivate() throws Exception {
        PowerMockito.mockStatic(MockStaticPrivateMethod.class);
        PowerMockito.when(MockStaticPrivateMethod.class, "returnTrue")
                .thenReturn(false);
        assertThat(Whitebox.invokeMethod(MockStaticPrivateMethod.class, "returnTrue"),
                is(false));
    }

}