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的注入和管理下,启动和验证就比较快,当然,写单元测试的时候,具体问题具体分析

相关推荐
z202305084 分钟前
linux之调度管理(13)- wake affine 唤醒特性
java·开发语言
AI人H哥会Java5 分钟前
【JAVA】Java高级:Java网络编程——TCP/IP与UDP协议基础
java·开发语言
萧萧玉树22 分钟前
分布式在线评测系统
前端·c++·后端·负载均衡
桃园码工40 分钟前
第一章:Go 语言概述 2.安装和配置 Go 开发环境 --Go 语言轻松入门
开发语言·后端·golang
robin_suli1 小时前
Java多线程八股(三)一>多线程环境使用哈希表和ArrayList
java·开发语言·多线程·哈希表
NiNg_1_2341 小时前
Java中的多线程
java·开发语言
丁总学Java1 小时前
nohup java -jar supporterSys.jar --spring.profiles.active=prod &
java·spring·jar
呆呆小雅1 小时前
C# 结构体
android·java·c#
谢尔登1 小时前
使用 Maven 创建 jar / war 项目
java·maven·jar
理想不理想v1 小时前
前端开发工程师需要学什么?
java·前端·vue.js·webpack·node.js