快捷導(dǎo)航

SpringBoot結(jié)合mockito測試實戰(zhàn)

更新時間：2022年11月21日 09:32:05 作者：程序員小董

與集成測試將系統(tǒng)作為一個整體測試不同，單元測試更應(yīng)該專注于某個類。所以當被測試類與外部類有依賴的時候，尤其是與數(shù)據(jù)庫相關(guān)的這種費時且有狀態(tài)的類，很難做單元測試。但好在可以通過“Mockito”這種仿真框架來模擬這些比較費時的類，從而專注于測試某個類內(nèi)部的邏輯

題目起的很像那么回事，但是這篇博客本身寫的很簡單。

我不想寫說一堆概念，然后闡釋各種概念是什么意思。我喜歡的是直接從例子出發(fā)。

package com.example.demo.controller;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.example.demo.entity.Person;
import com.example.demo.mapper.PersonMapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import java.util.List;
import java.util.Random;
@RequestMapping({"/user"})
@RestController
@Slf4j
public class UserController {
    @Autowired
    PersonMapper personMapper;
    @RequestMapping("/getUserById")
    public Person getUserById(Long id){
        Person person =personMapper.selectById(id);
        if ("張三".equals(person.getName())){
            return person;
        }
        if ("李四".equals(person.getName())){
            person.setAge(17);
            return person;
        }
        return person;
    }
}
//具體的PersonMapper如下
package com.example.demo.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.example.demo.entity.Person;
import org.springframework.stereotype.Component;
@Component(value = "personMapper")
public interface PersonMapper extends BaseMapper<Person> {
}

首先就是上面的例子，getUserById。

這個方法里面的邏輯可以分為兩部分，一個是通過mapper從db里面拿數(shù)據(jù)，還有就是很多額外完全沒有邏輯的代碼，不用管它的具體語義，反正就是寫些業(yè)務(wù)邏輯。

我現(xiàn)在想測試的目的就是先假定mapper本身是OK的，然后驗證那些無意義的額外邏輯也是正確的。

ok，看測試代碼。

package com.example.demo.controller;
import com.example.demo.entity.Person;
import com.example.demo.mapper.PersonMapper;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
import org.springframework.boot.test.context.SpringBootTest;
import static org.mockito.Mockito.when;
@SpringBootTest
@ExtendWith(MockitoExtension.class)
public class MockitoTest {
    @InjectMocks
    private  UserController userController;
    @Mock
    private PersonMapper personMapper;
    @Test
    public void testGet(){
        Person person = new Person();
        person.setName("張三");
        Long id = 15L;
        when(personMapper.selectById(id)).thenReturn(person);
        Person result = userController.getUserById(id);
        Assertions.assertEquals(person,result);
    }
}

運行之后：

那pom文件呢？

我用的是

	<parent>
	    <groupId>org.springframework.boot</groupId>
	    <artifactId>spring-boot-starter-parent</artifactId>
	    <version>2.7.5</version>
	    <relativePath/> <!-- lookup parent from repository -->
	</parent>
...
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
            <exclusions>
                <exclusion>
                    <groupId>org.junit.vintage</groupId>
                    <artifactId>junit-vintage-engine</artifactId>
                </exclusion>
            </exclusions>
        </dependency>

沒有別的依賴了。

ok，測試代碼已經(jīng)能跑了，現(xiàn)在就來解釋代碼。

代碼解釋

@InjectMocks

這個注解就是說我要測試被這個注解標注的類，他里面每個方法都會真正的被執(zhí)行。

換句話說，被這個注解標注的類，是我們真正想測試的類。

InjectMocks 被這個注解標注的類，是我們真正想測試的類。

@Mock

被這個注解標注的是我們想忽略的或者說，默認就認為是ok的類。

when