使用mockito辅助springboot service层单元测试

关注

原创最新推荐文章于 2026-09-14 12:04:33 发布 4.9k 阅读 AI 写同款· GEO 优化 ›

AI生成JUnit测试实战：TRAE插件提升Spring Boot单元测试覆盖率 单元测试是保障软件质量的关键环节，通过验证代码单元在隔离环境下的行为是否符合预期，能够有效预防回归缺陷。其核心原理在于模拟依赖、控制输入并断言输出，从而构建快速反馈的安全网。在Java生态中，JUnit与Mockito构成了单元测试的技术基石，结合JaCoCo等覆盖率工具，可以量化测试的完备性。随着AI编码助手的兴起，开发者现在可以借助像TRAE这样的智能插件，自动化生成高贴合度的测试骨架代码。这极大地提升了编写测试的效率，尤其适用于Spring Boot项目中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){
        // 读取文件
    }

}