mockito的用法

well，說來慚愧，之前一直知道有這么個東西，但總是看不進去。剛好趁着這次迭代間隙有些閑暇，認真看了下，大概明白是怎么回事了。

首先，mock是個概念，這個詞的本意就是“虛假的”、“模仿的”。在測試的時候，很多情況下都無法獲取真正的對象，如Servlet對象，但測試又需要這個對象，怎么辦？

當然是弄個假的給糊弄一下啦。

其次，mock的實現有很多，本文只關心Mockito，其他的請自行百度哈。

 

看到所有的資料都說Mockito有兩方面的作用：1. 驗證方法調用； 2. 指定特定條件下某個方法的返回值，或者執行特定的動作。

廢話少說，直接上代碼驗證一下。

1. 先上pom文件：

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <groupId>win.larryzeal</groupId>
    <artifactId>mockito-study</artifactId>
    <version>1.0-SNAPSHOT</version>
    <packaging>jar</packaging>
    <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <maven.compiler.source>1.8</maven.compiler.source>
        <maven.compiler.target>1.8</maven.compiler.target>
    </properties>
    
    <dependencies>
        <!--2.x not good 呵呵-->
        <dependency>
            <groupId>org.mockito</groupId>
            <artifactId>mockito-all</artifactId>
            <version>1.10.19</version>
        </dependency>
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.12</version>
        </dependency>
    </dependencies>
</project>

 

2. 再提供一個接口 Processor，不提供實現。

package win.larryzeal.mockito.study;

public interface Processor {
    void process();
    
    int get();
    
    void cb(int a, int b,  Callback cb);
}

package win.larryzeal.mockito.study;

public interface Callback {
    int onSuccess();
    int onFail();
}

 

3. 然后看一下怎么驗證調用 - 需要特別指出的是，mock的對象相當於加了一層代理，可以認為是實現。如下：

package win.larryzeal.mockito.study;

import org.junit.Test;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

//1. 驗證方法調用
public class MockTest {

    @Test
    public void r1() {

        Processor mock = Mockito.mock(Processor.class);
        mock.process();

        //注意，這里相當於驗證之前的調用！
        Mockito
                .verify(mock) //Alias to verify(mock, times(1))  驗證是否調用一次
                .process();

    }

    @Test
    public void r2() {

        Processor mock = Mockito.mock(Processor.class);
        mock.process();

        //注意，這里相當於驗證之前的調用！
        Mockito
                .verify(mock, Mockito.times(3)) //Alias to verify(mock, times(1))  驗證是否調用n次
                .process();

    }


}

上面的驗證很簡單，就是驗證指定mock的特定方法的調用 - 可以是調用次數，也可以是超時等，還可以是復合的驗證條件，如Mockito.timeout(1000).times(3)。

 

4. 再來看看怎么定制特定方法的返回結果 - 很有用噢，特別是針對遠程調用、接口調用等情況。

package win.larryzeal.mockito.study;

import org.junit.Test;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

//2. 指定某個方法的返回值
public class MockTest {

    @Test
    public void r3() {

        Processor mock = Mockito.mock(Processor.class);

        //注意，這里相當於設定mock里某個方法的返回值！在測試遇到嵌套方法，且嵌套方法不便於調用時，非常有用。
        Mockito
                .when(mock.get()) 
                .thenReturn(0xff); //這里是可變參數，多次調用的情況下，依次返回相應的數值；如果超出次數，返回最后一個值。

        System.out.println(mock.get());
    }

}

嗯嗯，when用於指定調用的方法，thenXxx用於指定返回結果 或者 拋出異常。

 

5. 再來看看回調的情況 - 傳入的參數是條件，根據條件來調用特定的回調。

package win.larryzeal.mockito.study;

import org.junit.Test;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;


//2. 指定某個方法的返回值，或者是執行特定的動作
public class MockTest {

    @Test
    public void r4() {
        Processor mock = Mockito.mock(Processor.class);

        //這里，是設定在特定條件下調用特定的回調！ - 這個特定的條件是不同的參數，可以是具體的，也可以是Mockito.anyXxx。
        Mockito
                .doAnswer((InvocationOnMock invocation) -> {
                    Callback cb = invocation.getArgumentAt(2, Callback.class);
                    cb.onFail();
                    return null; //嗯？這里的返回值是？
                })
                .when(mock)
                .cb(Mockito.eq(1), Mockito.eq(-1), Mockito.any(Callback.class)); //TODO 如果用matcher，那所有的參數都需要是matcher

        //實際的調用
        mock.cb(1, -1, new Callback() {
            @Override
            public int onSuccess() {
                System.out.println("ok");
                return 200;
            }

            @Override
            public int onFail() {
                System.out.println("fail");
                return 500;
            }
        });
        mock.cb(1, 1, null);
    }

}

前面設定條件，然后實際調用下，就是這么簡單！

ps：暫時不明白那個Invocation的返回值是干嘛用的，稍后再說。

 

6. 最后看看 spy，Mockito 除了mock()方法，還提供了spy，簡單的說就是可以不指定方法的返回值，而是返回類型的默認值（零值）。如下：

package win.larryzeal.mockito.study;

import org.junit.Test;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

public class MockTest {


    @Test
    public void r5() {

        //spy，除非設定，返回返回默認值
        Processor spy = Mockito.spy(Processor.class);

        //多次調用的情況下，依次返回相應的數值；如果超出次數，返回最后一個值。
        //Mockito.when(spy.get()).thenReturn(3, 5, 9); //不指定的情況，返回0

        System.out.println(spy.get());
        System.out.println(spy.get());
        System.out.println(spy.get());
        System.out.println(spy.get());

        //同樣可以校驗執行次數
//        Mockito.verify(spy).get(); //一次
        Mockito.verify(spy, Mockito.times(4)).get();
    }

}

 

ok，至此基本就算大功告成了，趕快去試試吧？