原文: https://howtodoinjava.com/spring-boot2/testing/spring-boot-mockito-junit-example/
学习使用 JUnit 和 Mockito 测试框架为 Spring 应用的服务层编写单元测试。 本教程演示 SpringBoot 测试服务层示例。
1. Maven 依赖
spring-boot-starter-test
依赖项包括创建和执行测试所需的所有依赖项。
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
</dependency>
如果不使用 spring boot,则包括以下依赖项。
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
<version>4.12</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.mockito</groupId>
<artifactId>mockito-core</artifactId>
<version>2.15.0</version>
</dependency>
<dependency>
<groupId>org.mockito</groupId>
<artifactId>mockito-inline</artifactId>
<version>2.15.0</version>
</dependency>
2. 起步
在此示例中,我们主要对两个类EmployeeManager
和EmployeeDao
进行单元测试。 顾名思义,管理器类代表服务层,而 dao 类正在与数据库进行交互。
EmployeeManager
类依赖于EmployeeDao
和委托方法调用来获取 pis 最终返回到控制器类的数据。
为了测试EmployeeManager
中的方法,我们可以通过以下两种方式创建 JUnit 测试类 TestEmployeeManager
:
2.1 MockitoJUnitRunner
类
MockitoJUnitRunner
类自动初始化所有带有@Mock
和@InjectMock
注解的对象。
@RunWith(MockitoJUnitRunner.class)
public class TestEmployeeManager {
@InjectMock
EmployeeManager manager;
@Mock
EmployeeDao dao;
//tests
}
2.2 MockitoAnnotations.initMock()
方法
如果我们不使用MockitoJUnitRunner
类方法,则可以使用静态方法MockitoAnnotations.initMock()
。 在初始化 junit 测试时,该方法还会初始化模拟对象。
public class TestEmployeeManager {
@InjectMock
EmployeeManager manager;
@Mock
EmployeeDao dao;
@Before
public void init() {
MockitoAnnotations.initMock(this);
}
//tests
}
2.3 @Mock
vs @InjectMock
@Mock
注解会为其注解的类创建一个模拟实现。@InjectMock
还将创建模拟实现,另外将标有注解@Mock
的从属模拟注入其中。
在上面的示例中,我们用@InjectMock
注解了EmployeeManager
类,因此模仿者将为EmployeeManager
类创建模仿对象,并将EmployeeDao
的模仿依赖项注入其中。
3. 使用 Mockito 进行 JUnit 测试
让我们看几个例子,这些例子编写了 junit 测试,以使用通过嘲讽创建的模拟对象对服务层方法进行单元测试。
很少有示例方法是getAllEmployees()
,该方法将返回EmployeeVO
对象的列表,getEmployeeById(int id)
以给定 id 返回雇员。 和createEmployee()
将添加一个雇员对象并返回void
。
3.1 服务层测试
package com.howtodoinjava.demo;
import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import java.util.ArrayList;
import java.util.List;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMock;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.mockito.junit.MockitoJUnitRunner;
import com.howtodoinjava.demo.dao.EmployeeDao;
import com.howtodoinjava.demo.model.EmployeeVO;
import com.howtodoinjava.demo.service.EmployeeManager;
public class TestEmployeeManager {
@InjectMock
EmployeeManager manager;
@Mock
EmployeeDao dao;
@Before
public void init() {
MockitoAnnotations.initMock(this);
}
@Test
public void getAllEmployeesTest()
{
List<EmployeeVO> list = new ArrayList<EmployeeVO>();
EmployeeVO empOne = new EmployeeVO(1, "John", "John", "howtodoinjava@gmail.com");
EmployeeVO empTwo = new EmployeeVO(2, "Alex", "kolenchiski", "alexk@yahoo.com");
EmployeeVO empThree = new EmployeeVO(3, "Steve", "Waugh", "swaugh@gmail.com");
list.add(empOne);
list.add(empTwo);
list.add(empThree);
when(dao.getEmployeeList()).thenReturn(list);
//test
List<EmployeeVO> empList = manager.getEmployeeList();
assertEquals(3, empList.size());
verify(dao, times(1)).getEmployeeList();
}
@Test
public void getEmployeeByIdTest()
{
when(dao.getEmployeeById(1)).thenReturn(new EmployeeVO(1,"Lokesh","Gupta","user@email.com"));
EmployeeVO emp = manager.getEmployeeById(1);
assertEquals("Lokesh", emp.getFirstName());
assertEquals("Gupta", emp.getLastName());
assertEquals("user@email.com", emp.getEmail());
}
@Test
public void createEmployeeTest()
{
EmployeeVO emp = new EmployeeVO(1,"Lokesh","Gupta","user@email.com");
manager.addEmployee(emp);
verify(dao, times(1)).addEmployee(emp);
}
}
单元测试结果将是这样。
Spring boot mockito junit 示例
3.2 服务层类
import java.util.List;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.howtodoinjava.demo.dao.EmployeeDao;
import com.howtodoinjava.demo.model.EmployeeVO;
@Service
public class EmployeeManager
{
@Autowired
EmployeeDao dao;
public List<EmployeeVO> getEmployeeList() {
return dao.getEmployeeList();
}
public EmployeeVO getEmployeeById(int id) {
return dao.getEmployeeById(id);
}
public void addEmployee(EmployeeVO employee) {
dao.addEmployee(employee);
}
}
3.3 DAO 层类
package com.howtodoinjava.demo.dao;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.springframework.stereotype.Repository;
import com.howtodoinjava.demo.model.EmployeeVO;
@Repository
public class EmployeeDao {
private Map<Integer, EmployeeVO> DB = new HashMap<>();
public List<EmployeeVO> getEmployeeList()
{
List<EmployeeVO> list = new ArrayList<>();
if(list.isEmpty()) {
list.addAll(DB.values());
}
return list;
}
public EmployeeVO getEmployeeById(int id) {
return DB.get(id);
}
public void addEmployee(EmployeeVO employee) {
employee.setEmployeeId(DB.keySet().size() + 1);
DB.put(employee.getEmployeeId(), employee);
}
public void updateEmployee(EmployeeVO employee) {
DB.put(employee.getEmployeeId(), employee);
}
public void deleteEmployee(int id) {
DB.remove(id);
}
}
4. Spring Boot Mockito 示例 – 总结
在此 mockito 教程中,我们学习了使用 JUnit 和 Mockito 对 Spring Boot 应用中的服务层进行单元测试。 我们学习了如何设置测试类并编写 JUnit 测试。
我们还了解了@Mock
和@InjectMock
注解之间的区别。
学习愉快!