SpringBoot结合mockito测试实战

时间:2022-11-22 程序员小董人气:0

题目起的很像那么回事，但是这篇博客本身写的很简单。

我不想写说一堆概念，然后阐释各种概念是什么意思。我喜欢的是直接从例子出发。

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里面拿数据，还有就是很多额外完全没有逻辑的代码，不用管它的具体语义，反正就是写些业务逻辑。

我现在想测试的目的就是先假定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，测试代码已经能跑了，现在就来解释代码。

代码解释

@InjectMocks

这个注解就是说我要测试被这个注解标注的类，他里面每个方法都会真正的被执行。

换句话说，被这个注解标注的类，是我们真正想测试的类。

InjectMocks 被这个注解标注的类，是我们真正想测试的类。

@Mock

被这个注解标注的是我们想忽略的或者说，默认就认为是ok的类。

when

personMapper本身就被@Mock标注了，那么其实里面的每个方法都不会真正的被执行，在UserController的Person person =personMapper.selectById(id);里面selectById就直接返回一个null。

但是加上下面的代码

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

功能我就不解释了。

加载全部内容