單元測試時靜態方法注意點

Mockito 很強大, 但是它不支援靜態方法.所以, 就用Powermock了. 要測試的物件是Controller中的介面, 對單元測試來說, 這個待測試的街口應該是一個白盒的. 所以, 其中的第三方(service,其他靜態類)的呼叫都應該使用mock物件來stub起來. 下面是## 示例程式碼(隨手寫的,說明問題就行):

Controller:  

@Controllerpublic class MainController {
    @Autowired    private MainService mainService;
    @GetMapping("/hello")
    public String hello(){
        mainService.hello();
        String result = HelloUtil.hi();
        if(result.equals("hi")){
            return "ok";
        }else {
            return "fail";
        }
    }
}

Service:

/**
 * Created by icer on 2017/10/13.
 */@Service public class MainService {
    public String hello(){
        System.out.println("hello");
        return "hello";
    }
}
Util靜態類:  
```javapublic class HelloUtil {
    public static String hi(){
        return "hi";
    }
}

我們要測試的是MainController中的hello方法.

再次注意, 是hello方法,不是hello介面.  我們的測試類:

@RunWith(PowerMockRunner.class)@PrepareForTest(HelloUtil.class)public class MainControllerTest {
    [@Mock]()    private MainService mainService;
    @InjectMocks    private MainController mainController;
    [@Test]()    public void helloTest(){
        PowerMockito.when(mainService.hello()).thenReturn("hello");
        PowerMockito.mockStatic(HelloUtil.class);
        PowerMockito.when(HelloUtil.hi()).thenReturn("hi");
        HelloUtil.hi();
        HelloUtil.hi();
        String result = 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();Wanted 3 times but was 2 times.

這裡只校驗靜態校驗之前的執行次數, 後面執行多少次都沒關係. 另外, 還有一點要注意的是, 在校驗語句前, 我們前面給方法mock的返回值是有效的, 在校驗之後就沒效果了. 
所以, 我們對靜態方法的測試, 應該是放在verifyStatis之前進行的.  在校驗之後的執行, 是為了證明我們之前的校驗是ok的.

校驗之後的靜態方法呼叫, 是一定要有的. 要不然是測試不出呼叫次數的. 參見下面程式碼:

@Test    public void helloTest(){
        PowerMockito.when(mainService.hello()).thenReturn("hello");
        PowerMockito.mockStatic(HelloUtil.class);
        PowerMockito.when(HelloUtil.hi()).thenReturn("hi");
        String result = HelloUtil.hi();
        String hi = HelloUtil.hi();
        PowerMockito.verifyStatic(Mockito.times(4));
        assertEquals(result, "hi");
    }

像上面這段程式碼, 測試是透過的.  改成下面這樣:

@Test    public void helloTest(){
        PowerMockito.when(mainService.hello()).thenReturn("hello");
        PowerMockito.mockStatic(HelloUtil.class);
        PowerMockito.when(HelloUtil.hi()).thenReturn("hi");
        String result = HelloUtil.hi();
        String hi = HelloUtil.hi();
        PowerMockito.verifyStatic(Mockito.times(4));
        HelloUtil.hi();
        assertEquals(result, "hi");
    }

就能看到日誌:

org.mockito.exceptions.verification.TooLittleActualInvocations: cn.com.hanbinit.utils.HelloUtil.hi();Wanted 4 times but was 2 times.

所以, 對靜態方法呼叫次數的校驗一定是verifyStatis和之後的顯示呼叫一起來進行的. 最後的最後, 再提一句:  如果同時有多個靜態方法呼叫. 如果都要驗證呼叫次數, 那麼應該分別像下面這樣使用多次:

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