Mockito mockStatic, mockConstruction

之前项目中都是使用PowerMock的mockStatic和whenNew来mock 静态方法和构造函数。但是在升级jdk17的过程中发现PowerMock已停止维护且不支持jdk17,我们转而将Mockito升级,使用Mockito中提供的Mockito.mockStatic, Mockito.mockConstruction。

Mockito.mockStatic

首先来看一下较为简单的mockStatic方法

第一步先声明一个MockedStatic对象

MockedStatic mockedStatic = Mockito.mockStatic(StringUtil.class);

第二步声明对应的行为

//1 官方推荐
mockedStatic.when(() -> StringUtil.someMethod()).thenReturn(someObject);

//2
Mockito.when(StringUtil.someMethod()).thenReturn(someObject);

这里1和2两种方式都有效,但是方式1是官方文档中的做法,方式2也许在未来某个时段会失效。

第三步,关闭mockedStatic

mockedStatic.close()

这里一定要注意,这是Mockito和PowerMock区别最大的一个地方,对于Mockito来说,一旦声明了一个MockedStatic,它会一直留在当前的Thread中并且会对当前Thread中所有调用的代码产生影响,这个影响不仅仅局限于你的测试代码,甚至会对测试框架(TestNG,Junit等)产生影响,所以我们一定要保证在测试代码结束后对MockedStatic进行关闭。如果你使用jdk1.7及以上版本,可以通过try(resource){}块来声明一个MockedStatic,或者通过try{}finally{}方式来确保MockedStatic及时关闭。

Mockito.mockConstruction

MockedConstruction mockedConstruction = Mockito.mockConstruction(FileOutputStream.class, ((fileOutputStream, context) -> {
            Mockito.doNothing().when(fileOutputStream).write(any());
        }));

和MockedStatic类似,在使用完成之后需要对MockedConstruction进行close

mockedConstruction.close();

本文来自网络,不代表协通编程立场,如若转载,请注明出处:https://www.net2asp.com/e445a2f340.html