快捷導(dǎo)航

SpringBoot結(jié)合mockito測(cè)試實(shí)戰(zhàn)

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

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

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

我不想寫說(shuō)一堆概念，然后闡釋各種概念是什么意思。我喜歡的是直接從例子出發(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。

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

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

ok，看測(cè)試代碼。

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);
    }
}

運(yùn)行之后：

那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>

沒(méi)有別的依賴了。

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

代碼解釋

@InjectMocks

這個(gè)注解就是說(shuō)我要測(cè)試被這個(gè)注解標(biāo)注的類，他里面每個(gè)方法都會(huì)真正的被執(zhí)行。

換句話說(shuō)，被這個(gè)注解標(biāo)注的類，是我們真正想測(cè)試的類。

InjectMocks 被這個(gè)注解標(biāo)注的類，是我們真正想測(cè)試的類。

@Mock

被這個(gè)注解標(biāo)注的是我們想忽略的或者說(shuō)，默認(rèn)就認(rèn)為是ok的類。

when

personMapper本身就被@Mock標(biāo)注了，那么其實(shí)里面的每個(gè)方法都不會(huì)真正的被執(zhí)行，在UserController的Person person =personMapper.selectById(id);里面selectById就直接返回一個(gè)null。

但是加上下面的代碼

when(personMapper.selectById(id)).thenReturn(person);

功能我就不解釋了。

到此這篇關(guān)于SpringBoot結(jié)合mockito測(cè)試實(shí)戰(zhàn)的文章就介紹到這了,更多相關(guān)SpringBoot mockito內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: