使用mockito輔助springboot service層單元測試

目的：

        測試一個springboot service層的一段插入MySQL代碼性能；

要求：

        1. 單元測試，不走web調用；

        2. 資料來自第三方服務，資料量較大，傳回時間10s+，希望mock掉此過程；

方案：

        junit+mockito

代碼：

?下面的類是要測的service，startCalculation()函數new thread新開線程（錯誤做法）裡的“批量插入data”

是要測試的代碼，dataScienceRpc.callMethod()是插入資料的來源，也是要通過mock回避掉的過程：

@Service
public class ExampleServiceImpl implements ExampleService {

    // 其它變量&函數
    @Autowired
    private DataScienceRpc dataScienceRpc;

    @Override
    public boolean startCalculation(long id) {

        // 其它操作
        Runnable myRunnable = () -> {
            // 其它操作
            JSONArray config = JSONUtils.toJSONArray(str);
            JSONArray data = dataScienceRpc.callMethod(id, config);
            // 批量插入data
        };

        Thread thread = new Thread(myRunnable);
        thread.start();

    }

}
           

?下面是springboot裡pom.xml對junit+mockito的配置（springboot自帶mockito）：

<dependency>
    <groupId>junit</groupId>
    <artifactId>junit</artifactId>
    <version>4.12</version>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-test</artifactId>
    <version>2.0.6.RELEASE</version>
    <!--<scope>test</scope>-->
</dependency>
           

 ?下面是單元測試代碼：

import com.alibaba.fastjson.JSONArray;
import com.example.DataScienceRpc;
import com.example.ExampleService;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import java.io.*;
import java.util.ArrayList;

import static org.mockito.Mockito.when;

/**
 * 必須和springboot啟動類放在同一命名空間下
 */

// 使用這個跑單測會加載整個容器
@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest
public class ExampleTest {

    @MockBean
    private DataScienceRpc dataScienceRpc;

    @Autowired
    ExampleService exampleService;


    @Before
    public void setup(){
        String dataPath = "/Desktop/data.txt";
        String data = readToBuffer(dataPath);
        JSONArray jsonArr = JSONArray.parseArray(data);

        when(dataScienceRpc.callMethod(Mockito.anyLong(), Mockito.any(JSONArray.class)))
                .thenReturn(jsonArr);

    }

    @Test
    public void test(){

        exampleService.startCalculation(133);
    }

    public String readToBuffer(String filePath){
        // 讀取檔案
    }

}
           

說明（坑）：

        1.  測試類必須和springboot啟動類放在同一命名空間下；

        2. @RunWith(SpringJUnit4ClassRunner.class)，@SpringBootTest注解，加載整個容器；

        3. @MockBean，将mock出spring bean，若與spring容器無關，可用@Mock；

        4. Mockito.when方法mock函數傳回值，不會進入該函數内部；若使用Mockito.anyLong()等方法模糊比對函數的任意此類型參數，注意當運作時函數參數類型與指定Mockito類型不符，該mock将不會成功；

        此博文不錯：https://www.cnblogs.com/ceshi2016/p/9550823.html

        5. 此情況下startCalculation()函數執行完後立即傳回，new出的thread（異步執行）沒執行完，JVM就會退出，導緻要測試的代碼執行不完；故不建議此種寫法，異步執行的過程應該抽象出一個函數，一是保持上層函數簡潔，可讀性強，二是友善單元測試；