来自 技术 2019-01-17 00:00 的文章

单元测试时静态方法注意点

Mockito 很强大, 但是它不支持静态方法.所以, 就用Powermock了. 要测试的对象是Controller中的接口, 对单元测试来说, 这个待测试的街口应该是一个白盒的. 所以, 其中的第三方(service,其他静态类)的调用都应该使用mock对象来stub起来. 下面是## 示例代码(随手写的,说明问题就行):

Controller:

@ControllerpublicclassMainController{@AutowiredprivateMainServicemainService;@GetMapping("/hello")publicStringhello(){mainService.hello();Stringresult=HelloUtil.hi();if(result.equals("hi")){return"ok";}else{return"fail";}}}

Service:

/***Createdbyiceron2017/10/13.*/@ServicepublicclassMainService{publicStringhello(){System.out.println("hello");return"hello";}}Util静态类:```javapublicclassHelloUtil{publicstaticStringhi(){return"hi";}}

我们要测试的是MainController中的hello方法.

再次注意, 是hello方法,不是hello接口. 我们的测试类:

@RunWith(PowerMockRunner.class)@PrepareForTest(HelloUtil.class)publicclassMainControllerTest{[@Mock](https://my.oschina.net/mock)privateMainServicemainService;@InjectMocksprivateMainControllermainController;[@Test](https://my.oschina.net/azibug)publicvoidhelloTest(){PowerMockito.when(mainService.hello()).thenReturn("hello");PowerMockito.mockStatic(HelloUtil.class);PowerMockito.when(HelloUtil.hi()).thenReturn("hi");HelloUtil.hi();HelloUtil.hi();Stringresult=mainController.hello();PowerMockito.verifyStatic(Mockito.times(3));HelloUtil.hi();HelloUtil.hi();HelloUtil.hi();//mainController.hello();assertEquals(result,"ok");}}

我们着重看的是上面的HelloUtil.hi()方法.

PowerMockito.verifyStatic(Mockito.times(3));

先说下这句的意思, 这句(后面简称静态校验)就是对静态方法的调用次数做了校验. 光看命名就能想通. 但是要注意的是, 这里的3指的是在静态校验之前的调用次数. 如果静态校验执行的次数不等于这个次数, 那么静态校验后面的静态方法就不能再执行了. 上面代码中, 在这句前一共执行了三次HelloUtil.hi(); 两次是主动明文执行的, 一次是在mainController.hello()里面执行的. 如果注释掉某一个, 那么就会看到日志打印:

org.mockito.exceptions.verification.TooLittleActualInvocations:cn.com.hanbinit.utils.HelloUtil.hi();Wanted3timesbutwas2times.

这里只校验静态校验之前的执行次数, 后面执行多少次都没关系. 另外, 还有一点要注意的是, 在校验语句前, 我们前面给方法mock的返回值是有效的, 在校验之后就没效果了.所以, 我们对静态方法的测试, 应该是放在verifyStatis之前进行的. 在校验之后的执行, 是为了证明我们之前的校验是ok的.

校验之后的静态方法调用, 是一定要有的. 要不然是测试不出调用次数的. 参见下面代码:

@TestpublicvoidhelloTest(){PowerMockito.when(mainService.hello()).thenReturn("hello");PowerMockito.mockStatic(HelloUtil.class);PowerMockito.when(HelloUtil.hi()).thenReturn("hi");Stringresult=HelloUtil.hi();Stringhi=HelloUtil.hi();PowerMockito.verifyStatic(Mockito.times(4));assertEquals(result,"hi");}

像上面这段代码, 测试是通过的. 改成下面这样:

@TestpublicvoidhelloTest(){PowerMockito.when(mainService.hello()).thenReturn("hello");PowerMockito.mockStatic(HelloUtil.class);PowerMockito.when(HelloUtil.hi()).thenReturn("hi");Stringresult=HelloUtil.hi();Stringhi=HelloUtil.hi();PowerMockito.verifyStatic(Mockito.times(4));HelloUtil.hi();assertEquals(result,"hi");}

就能看到日志:

org.mockito.exceptions.verification.TooLittleActualInvocations:cn.com.hanbinit.utils.HelloUtil.hi();Wanted4timesbutwas2times.

所以, 对静态方法调用次数的校验一定是verifyStatis和之后的显示调用一起来进行的. 最后的最后, 再提一句: 如果同时有多个静态方法调用. 如果都要验证调用次数, 那么应该分别像下面这样使用多次:

PowerMockito.verifyStatic(Mockito.times(4));HelloUtil.hi();