Mockito使用

前言

在Spring单元测试中,如果不想生成的对象受Spring管理的话,就可以使用Mock注解,为了减少单测与spring框架的耦合,并且尽量不使用真实对象进行模拟,可以使用@InjectMocks和@Mock注解

Mockito使用

生成@Mock、@Spy备注的mock对象,用

typescript 复制代码
@BeforeEach
	void setUp() {
		MockitoAnnotations.openMocks(this);
	}

定义一个实体类

vbnet 复制代码
@Data
public class UserQuery {

    
    private Long id;


    private String account;


    private String username;

    private Long deptId;


    private List<Long> roleIds;
    
    private String password;
    

}

定义一个service接口

scss 复制代码
public Boolean save(UserQuery userQuery) {
		UserDTO userDTO = new UserDTO();
		userDTO.setUsername(userQuery.getAccount());
		R<Boolean> result = remoteUserAdminService.isExist(userDTO);
		if (!Objects.equals(result.getCode(), CommonConstants.SUCCESS)) {
			throw new BussException("判断用户名是否存在错误");
		}
		if (result.getData()) {
			throw new BussException("用户名已存在");
		}
		userDTO.setNickName(userQuery.getUsername());
		result = remoteUserAdminService.user(userDTO);
		if (!Objects.equals(result.getCode(), CommonConstants.SUCCESS)) {
			throw new BussException("保存用户异常");
		}
		return result.getData();
	}

测试单元如下

scss 复制代码
@DisplayName("用户模块测试")
public class UserServiceImplTest {

	@InjectMocks
	private UserServiceImpl userService;

	@Mock
	private RemoteUserService remoteUserService;


	@Mock
	private RemoteUserAdminService remoteUserAdminService;

	@BeforeEach
	void setUp() {
		MockitoAnnotations.openMocks(this);
	}

	@Nested
	@DisplayName("新建用户单元测试")
	public class UserAddTest {

		@Test
		@DisplayName("保存用户,查询用户名是否存在,三方系统返回调用失败 => 返回异常")
		void save_1_test() {
			UserQuery userQuery = new UserQuery();
			userQuery.setUsername("aa");
			UserDTO userDTO = new UserDTO();
			userDTO.setUsername(userQuery.getAccount());
			R<Boolean> result = R.restResult(null, 1, "操作失败");
			Mockito.when(remoteUserAdminService.isExist(Mockito.any(UserDTO.class))).thenReturn(result);
			Assertions.assertThrows(BussException.class, () -> {
				userService.save(userQuery);
			});
			Mockito.verify(remoteUserAdminService, times(1)).isExist(Mockito.any(UserDTO.class));
		}

		@Test
		@DisplayName("保存用户,查询用户名是否存在,三方系统返回用户名已经存在 => 返回异常")
		void save_2_test() {
			UserQuery userQuery = new UserQuery();
			userQuery.setUsername("aa");			
			UserDTO userDTO = new UserDTO();
			userDTO.setUsername(userQuery.getAccount());
			R<Boolean> result = R.restResult(true, 0, "操作成功");
			Mockito.when(remoteUserService.isExist(Mockito.any(UserDTO.class))).thenReturn(result);
			Assertions.assertThrows(BussException.class, () -> {
				userService.save(userQuery);
			});
			Mockito.verify(remoteUserService, times(1)).isExist(Mockito.any(UserDTO.class));
		}

		@Test
		@DisplayName("保存用户,三方系统返回保存异常 => 返回异常")
		void save_3_test() {
			UserQuery userQuery = new UserQuery();
			userQuery.setUsername("aa");
			UserDTO userDTO = new UserDTO();
			userDTO.setUsername(userQuery.getAccount());
			R<Boolean> result = R.restResult(false, 0, "操作成功");
			Mockito.when(remoteUserService.isExist(Mockito.any(UserDTO.class))).thenReturn(result);
			R<Boolean> saveR = R.restResult(false, 1, "操作失败");
			Mockito.when(remoteUserService.user(Mockito.any(UserDTO.class))).thenReturn(saveR);
			Assertions.assertThrows(BussException.class, () -> {
				userService.save(userQuery);
			});
			Mockito.verify(remoteUserService, times(1)).isExist(Mockito.any(UserDTO.class));
			Mockito.verify(remoteUserService, times(1)).user(Mockito.any(UserDTO.class));
		}

		@Test
		@DisplayName("保存用户,三方系统返回成功 => 保存用户成功")
		void save_4_test() {
			UserQuery userQuery = new UserQuery();
			userQuery.setUsername("aa");
			UserDTO userDTO = new UserDTO();
			userDTO.setUsername(userQuery.getAccount());
			R<Boolean> result = R.restResult(false, 0, "操作成功");
			Mockito.when(remoteUserService.isExist(Mockito.any(UserDTO.class))).thenReturn(result);
			R<Boolean> saveR = R.ok(true, "操作成功");
			Mockito.when(remoteUserService.user(Mockito.any(UserDTO.class))).thenReturn(saveR);
			Boolean isSuccess = userService.save(userQuery);
			Mockito.verify(remoteUserService, times(1)).isExist(Mockito.any(UserDTO.class));
			Mockito.verify(remoteUserService, times(1)).user(Mockito.any(UserDTO.class));
			Assertions.assertTrue(isSuccess);
		}

	}

}

其中

Mockito.when

Mockito.when是模拟一个方法返回值,如

ini 复制代码
Mockito.when(remoteUserService.isExist(Mockito.any(UserDTO.class))).thenReturn(result);
			R<Boolean> saveR = R.ok(true, "操作成功");

模拟方法中调用isExist,返回result结果,Mockito.any(UserDTO.class)是传入一个UserDTO值

Mockito.verify

Mockito.verify(remoteUserService, times(1)),是验证方法中remoteUserService是不是被被用了一次

总结

在单元测试中,Mockito可以模拟出对象不受Spring框架影响,这样在没有Bean的注入和管理下,启动和验证就比较快,当然,写单元测试的时候,具体问题具体分析

相关推荐
齐 飞24 分钟前
MongoDB笔记01-概念与安装
前端·数据库·笔记·后端·mongodb
九圣残炎32 分钟前
【从零开始的LeetCode-算法】1456. 定长子串中元音的最大数目
java·算法·leetcode
wclass-zhengge34 分钟前
Netty篇(入门编程)
java·linux·服务器
LunarCod41 分钟前
WorkFlow源码剖析——Communicator之TCPServer(中)
后端·workflow·c/c++·网络框架·源码剖析·高性能高并发
Re.不晚1 小时前
Java入门15——抽象类
java·开发语言·学习·算法·intellij-idea
雷神乐乐1 小时前
Maven学习——创建Maven的Java和Web工程,并运行在Tomcat上
java·maven
码农派大星。1 小时前
Spring Boot 配置文件
java·spring boot·后端
顾北川_野1 小时前
Android 手机设备的OEM-unlock解锁 和 adb push文件
android·java
江深竹静,一苇以航1 小时前
springboot3项目整合Mybatis-plus启动项目报错:Invalid bean definition with name ‘xxxMapper‘
java·spring boot
confiself2 小时前
大模型系列——LLAMA-O1 复刻代码解读
java·开发语言