使用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；