如何对Java单例模式进行mock

Posted 2020-08-11

最近刚开始转行到Java编程，项目使用的是Junit4框架。其中遇到了采用单例模式（饿汉）实现的Java代码，那么如何对该类中的方法进行模拟呢？因为该模式下所有的共有方法都是通过获取该类的静态私有实例来调用类中的共有方法，这就需要对该类的静态私有实例对象来模拟。经过研究发现，可以使用Whitebox类来帮助我们模拟该静态私有实例对象。

Java源代码：

public class HelloWorld {
    private static HelloWorld instance = new HelloWorld();

    private HelloWorld() {}

    public static HelloWorld getInstance() {return instance;}

    public void say() {
        System.out.println("Hello World!");
    }
}

public class HelloWorld {
    private static HelloWorld instance = new HelloWorld();

    private HelloWorld() {}

    public static HelloWorld getInstance() {return instance;}

    public void say() {
        System.out.println("Hello World!");
    }
}

测试代码：

public class HelloWorldAppTest {
    private HelloWorldApp helloWorldApp;

    @Before
    public void setUp() throws Exception {
        helloWorldApp = new HelloWorldApp();
    }

    @Test
    public void sayHelloWorld() throws Exception {
        HelloWorld instanceMock = PowerMockito.mock(HelloWorld.class);
        Whitebox.setInternalState(HelloWorld.class, "instance", instanceMock);

        Mockito.doNothing().when(instanceMock).say();

        helloWorldApp.sayHelloWorld();
        Mockito.verify(instanceMock).say();
    }
}

要注意的是必须使用"org.powermock.reflect.whitebox"，关于该类的详细用法可参考http://static.javadoc.io/org.powermock/powermock-reflect/1.6.4/org/powermock/reflect/Whitebox.html

不能使用“org.mockito.internal.util.reflection.Whitebox”，这个类不支持对私有字段进行模拟。