天天看點

PowerMock - 無所不能的PowerMock,mock私有方法,靜态方法,測試私有方法,final類

1、為什麼要用mock

我的一本書的解釋:

(1)建立所需的DB資料可能需要很長時間,如:調用别的接口,模拟很多資料

(2)調用第三方API接口,測試很慢,

(3)編寫滿足所有外部依賴的測試可能很複雜,複雜到不值得編寫,Mock模拟内部或外部依賴可以幫助我們解決這些問題

另一本TDD書的解釋:

(1)對象的結果不确定,如每擷取目前時間,得到的結果都不一樣,無法符合我們的預期;

(2)實作這個接口的對象不存在;

(3)對象速度緩慢

對于TDD還有一個更重要原因:通過模拟可以隔離目前方法使用的的所有依賴,讓我們更加專注于單個單元,忽略其調用的代碼的内部工作原理

一本部落格的幹貨:

(1)Mock可以用來解除測試對象對外部服務的依賴(比如資料庫,第三方接口等),使得測試用例可以獨立運作。不管是傳統的單體應用,還是現在流行的微服務,這點都特别重要,因為任何外部依賴的存在都會極大的限制測試用例的可遷移性和穩定性。

(2)Mock的第二個好處是替換外部服務調用,提升測試用例的運作速度。任何外部服務調用至少是跨程序級别的消耗,甚至是跨系統、跨網絡的消耗,而Mock可以把消耗降低到程序内。比如原來一次秒級的網絡請求,通過Mock可以降至毫秒級,整整3個數量級的差别。

(3)Mock的第三個好處是提升測試效率。這裡說的測試效率有兩層含義。第一層含義是機關時間運作的測試用例數,這是運作速度提升帶來的直接好處。而第二層含義是一個測試人員機關時間建立的測試用例數。

以單體應用為例,随着業務複雜度的上升,為了運作一個測試用例可能需要準備很多測試資料,與此同時還要盡量保證多個測試用例之間的測試資料互不幹擾。為了做到這一點,測試人員往往需要花費大量的時間來維護一套可運作的測試資料。有了Mock之後,由于去除了測試用例之間共享的資料庫依賴,測試人員就可以針對每一個或者每一組測試用例設計一套獨立的測試資料,進而很容易的做到不同測試用例之間的資料隔離性。而對于微服務,由于一個微服務可能級聯依賴很多其他的微服務,運作一個測試用例甚至需要跨系統準備一套測試資料,如果沒有Mock,基本上可以說是不可能的。是以,不管是單體應用還是微服務,有了Mock之後,QE就可以省去大量的準備測試資料的時間,專注于測試用例本身,自然也就提升了單人的測試效率。

現如今比較流行的Mock工具如jMock 、EasyMock 、Mockito等都有一個共同的缺點:不能mock靜态、final、私有方法等。而PowerMock能夠完美的彌補以上三個Mock工具的不足

2、實戰

好了,我們用PoweMockito架構,直接上代碼:如何mock私有方法,靜态方法,測試私有方法,final類

依賴

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito2</artifactId>
    <version>1.7.4</version>
    <scope>test</scope>
</dependency>

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>1.7.4</version>
    <scope>test</scope>
</dependency>      

powermock可能與mockito有版本沖突,我們可以講mockito版本改成2.8.47

<properties>
    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
    <java.version>1.8</java.version>
    <mockito.version>2.8.47</mockito.version>
</properties>      

service層代碼:一個簡單的儲存,傳回user,這裡就不調用持久層了

@Override
public User save(User user) {
    return user;
}      
package com.example.demo.controller;

import com.example.demo.entity.User;
import com.example.demo.service.UserService;

public final class UserController {

    private final UserService service;

    public UserController(UserService service) {
        this.service = service;
    }

    public User saveUser(User user) {
        User save = service.save(user);
        return save;
    }

    public String returnName(){
       return getStaticName("ljw1");
    }

    public static String getStaticName(String name) {
        return "A_" + name;
    }
    public  String getPrivateName(String name) {

        if (publicCheck()){
            return "public 被mock 了";
        }
        if (check(name)){
            return "private 被mock 了";
        }
        return "A_" + name;
    }

    public boolean publicCheck() {
        return false;
    }

    private boolean check(String name) {
        return false;
    }

    private String say(String content) {
        return "ljw say " + content;
    }

}      
package com.example.demo.demo1;

import com.example.demo.controller.UserController;
import com.example.demo.entity.User;
import com.example.demo.service.impl.UserServiceImpl;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import org.powermock.reflect.Whitebox;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import static org.junit.Assert.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.powermock.api.mockito.PowerMockito.when;

@RunWith(PowerMockRunner.class)
@PrepareForTest(UserController.class)
public class LastMock {
    @Mock
    private UserServiceImpl serviceImpl;

    @InjectMocks
    private UserController controller;

    /**
     * mock service的儲存方法
     */
    @Test
    public void mockSave() {
        User user1 = new User();
        User user2 = new User();
        user1.setId("1");
        user2.setId("2");
        Mockito.when(serviceImpl.save(user1)).thenReturn(user2); //當調用service的save()時,mock讓他傳回user2
        User saveUser = controller.saveUser(user1); //調用
        Mockito.verify(serviceImpl,Mockito.times(1)).save(user1);//verify驗證mock次數
        assertEquals(user2, saveUser);//斷言是否mock傳回的是user2
    }

    /**
     * mock spy public方法
     * @throws Exception xx
     */
    @Test
    public void spy_public_method() throws Exception {
        UserController spy = PowerMockito.spy(controller); //監視controller的publicCheck方法,讓他傳回true
        Mockito.when(spy.publicCheck()).thenReturn(true);
        String name = spy.getPrivateName("ljw");//執行該方法
        assertEquals("public 被mock 了", name);//驗證
    }

    /**
     * mock私有方法
     * @throws Exception xx
     */
    @Test
    public void spy_private_method() throws Exception {
        UserController spy = PowerMockito.spy(controller);
        PowerMockito.when(spy, "check", any()).thenReturn(true);//私有方法mockito不行了,需要用無所不能的PowerMock監視spy
        String name = spy.getPrivateName("ljw");
        assertEquals("private 被mock 了", name);
    }

    /**
     * mock 靜态方法
     */
    @Test
    public void mockStaticMethod() {
        PowerMockito.mockStatic(UserController.class);//mock靜态方法
        when(UserController.getStaticName(any())).thenReturn("hi");
        String staticName = UserController.getStaticName("ljw");//執行
        assertEquals("hi", staticName);//驗證
    }

    @Test
    public void mockStaticMethod_2() {
        PowerMockito.mockStatic(UserController.class);
        when(UserController.getStaticName(any())).thenReturn("hi");
        String staticName = controller.returnName();//通過returnName()調用,看能否被mock
        assertEquals("hi", staticName);
    }

    /**
     * 測試私有方法一
     * @throws InvocationTargetException xx
     * @throws IllegalAccessException xx
     */
    @Test
    public void testPrivateMethod() throws InvocationTargetException, IllegalAccessException {
        Method method = PowerMockito.method(UserController.class, "say", String.class);
        Object say = method.invoke(controller, "hi");
        assertEquals("ljw say hi", say);
    }

    /**
     * 測試私有方法二
     * @throws Exception xx
     */
    @Test
    public void testPrivateMethod_2() throws Exception {
        Object say = Whitebox.invokeMethod(controller, "say", "hi");
        assertEquals("ljw say hi", say);
    }

}      

繼續閱讀