|
|
@@ -1,128 +1,259 @@
|
|
|
package cn.iocoder.dashboard.modules.system.service.user;
|
|
|
|
|
|
import cn.hutool.core.util.RandomUtil;
|
|
|
-import cn.iocoder.dashboard.BaseSpringBootUnitTest;
|
|
|
+import cn.iocoder.dashboard.BaseDbUnitTest;
|
|
|
import cn.iocoder.dashboard.common.enums.CommonStatusEnum;
|
|
|
+import cn.iocoder.dashboard.modules.infra.service.file.InfFileService;
|
|
|
+import cn.iocoder.dashboard.modules.system.controller.user.vo.profile.SysUserProfileUpdatePasswordReqVO;
|
|
|
+import cn.iocoder.dashboard.modules.system.controller.user.vo.profile.SysUserProfileUpdateReqVO;
|
|
|
import cn.iocoder.dashboard.modules.system.controller.user.vo.user.SysUserCreateReqVO;
|
|
|
import cn.iocoder.dashboard.modules.system.controller.user.vo.user.SysUserImportExcelVO;
|
|
|
import cn.iocoder.dashboard.modules.system.controller.user.vo.user.SysUserImportRespVO;
|
|
|
import cn.iocoder.dashboard.modules.system.controller.user.vo.user.SysUserUpdateReqVO;
|
|
|
import cn.iocoder.dashboard.modules.system.dal.dataobject.dept.SysDeptDO;
|
|
|
-import cn.iocoder.dashboard.modules.system.dal.mysql.dept.SysDeptMapper;
|
|
|
+import cn.iocoder.dashboard.modules.system.dal.dataobject.dept.SysPostDO;
|
|
|
+import cn.iocoder.dashboard.modules.system.dal.dataobject.user.SysUserDO;
|
|
|
+import cn.iocoder.dashboard.modules.system.dal.mysql.user.SysUserMapper;
|
|
|
+import cn.iocoder.dashboard.modules.system.enums.common.SysSexEnum;
|
|
|
import cn.iocoder.dashboard.modules.system.service.dept.SysDeptService;
|
|
|
-import cn.iocoder.dashboard.util.RandomUtils;
|
|
|
+import cn.iocoder.dashboard.modules.system.service.dept.SysPostService;
|
|
|
+import cn.iocoder.dashboard.modules.system.service.permission.SysPermissionService;
|
|
|
+import cn.iocoder.dashboard.util.collection.ArrayUtils;
|
|
|
+import cn.iocoder.dashboard.util.collection.CollectionUtils;
|
|
|
import org.junit.jupiter.api.Test;
|
|
|
-import org.springframework.beans.factory.annotation.Autowired;
|
|
|
+import org.mockito.stubbing.Answer;
|
|
|
+import org.springframework.boot.test.mock.mockito.MockBean;
|
|
|
import org.springframework.context.annotation.Import;
|
|
|
import org.springframework.security.crypto.password.PasswordEncoder;
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
|
+import java.io.ByteArrayInputStream;
|
|
|
import java.util.ArrayList;
|
|
|
-import java.util.Collections;
|
|
|
import java.util.List;
|
|
|
+import java.util.function.Consumer;
|
|
|
|
|
|
-import static cn.iocoder.dashboard.util.RandomUtils.randomPojo;
|
|
|
-import static cn.iocoder.dashboard.util.RandomUtils.randomString;
|
|
|
-import static org.junit.jupiter.api.Assertions.*;
|
|
|
+import static cn.hutool.core.util.RandomUtil.randomBytes;
|
|
|
+import static cn.hutool.core.util.RandomUtil.randomEle;
|
|
|
+import static cn.iocoder.dashboard.util.AssertUtils.assertPojoEquals;
|
|
|
+import static cn.iocoder.dashboard.util.RandomUtils.*;
|
|
|
+import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
|
+import static org.junit.jupiter.api.Assertions.assertNull;
|
|
|
+import static org.mockito.ArgumentMatchers.*;
|
|
|
+import static org.mockito.Mockito.*;
|
|
|
|
|
|
/**
|
|
|
- * @Author zxl
|
|
|
- * @Date 2021/3/6 14:28
|
|
|
- * @Desc
|
|
|
+ * {@link SysUserService} 的单元测试类
|
|
|
+ *
|
|
|
+ * @author zxl
|
|
|
*/
|
|
|
-@Import(PasswordEncoder.class)
|
|
|
-public class SysUserServiceImplTest extends BaseSpringBootUnitTest {
|
|
|
+@Import(SysUserServiceImpl.class)
|
|
|
+public class SysUserServiceImplTest extends BaseDbUnitTest {
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private SysUserServiceImpl userService;
|
|
|
|
|
|
- @Autowired
|
|
|
- private SysUserService userService;
|
|
|
- @Autowired
|
|
|
- private PasswordEncoder passwordEncoder;
|
|
|
- @Autowired
|
|
|
- private SysDeptService deptService;
|
|
|
@Resource
|
|
|
- private SysDeptMapper deptMapper;
|
|
|
+ private SysUserMapper userMapper;
|
|
|
|
|
|
+ @MockBean
|
|
|
+ private SysDeptService deptService;
|
|
|
+ @MockBean
|
|
|
+ private SysPostService postService;
|
|
|
+ @MockBean
|
|
|
+ private SysPermissionService permissionService;
|
|
|
+ @MockBean
|
|
|
+ private PasswordEncoder passwordEncoder;
|
|
|
+ @MockBean
|
|
|
+ private InfFileService fileService;
|
|
|
|
|
|
@Test
|
|
|
- public void test_creatUser(){
|
|
|
+ public void testCreatUser_success() {
|
|
|
// 准备参数
|
|
|
- SysUserCreateReqVO reqVO = randomPojo(SysUserCreateReqVO.class,o->{
|
|
|
- o.setDeptId(null);
|
|
|
- o.setPostIds(Collections.emptySet());
|
|
|
- o.setSex(1);
|
|
|
- o.setPassword(randomString());
|
|
|
+ SysUserCreateReqVO reqVO = randomPojo(SysUserCreateReqVO.class, o -> {
|
|
|
+ o.setSex(RandomUtil.randomEle(SysSexEnum.values()).getSex());
|
|
|
+ o.setMobile(randomString());
|
|
|
});
|
|
|
- // 调用方法
|
|
|
- Long userId = userService.createUser(reqVO);
|
|
|
- // 校验数据是否存在
|
|
|
- assertNotNull(userService.getUser(userId));
|
|
|
+ // mock deptService 的方法
|
|
|
+ SysDeptDO dept = randomPojo(SysDeptDO.class, o -> {
|
|
|
+ o.setId(reqVO.getDeptId());
|
|
|
+ o.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
|
+ });
|
|
|
+ when(deptService.getDept(eq(dept.getId()))).thenReturn(dept);
|
|
|
+ // mock postService 的方法
|
|
|
+ List<SysPostDO> posts = CollectionUtils.convertList(reqVO.getPostIds(), postId ->
|
|
|
+ randomPojo(SysPostDO.class, o -> {
|
|
|
+ o.setId(postId);
|
|
|
+ o.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
|
+ }));
|
|
|
+ when(postService.getPosts(eq(reqVO.getPostIds()), isNull())).thenReturn(posts);
|
|
|
+ // mock passwordEncoder 的方法
|
|
|
+ when(passwordEncoder.encode(eq(reqVO.getPassword()))).thenReturn("yudaoyuanma");
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ Long userId = userService.createUser(reqVO);
|
|
|
+ // 断言
|
|
|
+ SysUserDO user = userMapper.selectById(userId);
|
|
|
+ assertPojoEquals(reqVO, user, "password");
|
|
|
+ assertEquals("yudaoyuanma", user.getPassword());
|
|
|
+ assertEquals(CommonStatusEnum.ENABLE.getStatus(), user.getStatus());
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void test_updateUser(){
|
|
|
+ public void testUpdateUser_success() {
|
|
|
+ // mock 数据
|
|
|
+ SysUserDO dbUser = randomSysUserDO();
|
|
|
+ userMapper.insert(dbUser);
|
|
|
// 准备参数
|
|
|
- SysUserCreateReqVO reqVO = randomPojo(SysUserCreateReqVO.class,o->{
|
|
|
- o.setDeptId(null);
|
|
|
- o.setPostIds(Collections.emptySet());
|
|
|
- o.setSex(1);
|
|
|
- o.setPassword(randomString());
|
|
|
+ SysUserUpdateReqVO reqVO = randomPojo(SysUserUpdateReqVO.class, o -> {
|
|
|
+ o.setId(dbUser.getId());
|
|
|
+ o.setSex(RandomUtil.randomEle(SysSexEnum.values()).getSex());
|
|
|
+ o.setMobile(randomString());
|
|
|
});
|
|
|
- // 先插入一条
|
|
|
- Long userId = userService.createUser(reqVO);
|
|
|
- // 准备更新参数:更新手机号
|
|
|
- SysUserUpdateReqVO updateVo = new SysUserUpdateReqVO();
|
|
|
- updateVo.setId(userId);
|
|
|
- updateVo.setMobile(RandomUtil.randomNumbers(11));
|
|
|
- // 调用方法、
|
|
|
- userService.updateUser(updateVo);
|
|
|
- // 校验结果
|
|
|
- assertEquals(userService.getUser(userId).getMobile(),updateVo.getMobile());
|
|
|
+ // mock deptService 的方法
|
|
|
+ SysDeptDO dept = randomPojo(SysDeptDO.class, o -> {
|
|
|
+ o.setId(reqVO.getDeptId());
|
|
|
+ o.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
|
+ });
|
|
|
+ when(deptService.getDept(eq(dept.getId()))).thenReturn(dept);
|
|
|
+ // mock postService 的方法
|
|
|
+ List<SysPostDO> posts = CollectionUtils.convertList(reqVO.getPostIds(), postId ->
|
|
|
+ randomPojo(SysPostDO.class, o -> {
|
|
|
+ o.setId(postId);
|
|
|
+ o.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
|
+ }));
|
|
|
+ when(postService.getPosts(eq(reqVO.getPostIds()), isNull())).thenReturn(posts);
|
|
|
|
|
|
+ // 调用
|
|
|
+ userService.updateUser(reqVO);
|
|
|
+ // 断言
|
|
|
+ SysUserDO user = userMapper.selectById(reqVO.getId());
|
|
|
+ assertPojoEquals(reqVO, user);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void test_deleteUser(){
|
|
|
+ public void testUpdateUserProfile_success() {
|
|
|
+ // mock 数据
|
|
|
+ SysUserDO dbUser = randomSysUserDO();
|
|
|
+ userMapper.insert(dbUser);
|
|
|
// 准备参数
|
|
|
- SysUserCreateReqVO reqVO = randomPojo(SysUserCreateReqVO.class,o->{
|
|
|
- o.setDeptId(null);
|
|
|
- o.setPostIds(Collections.emptySet());
|
|
|
- o.setSex(1);
|
|
|
- o.setPassword(randomString());
|
|
|
+ Long userId = dbUser.getId();
|
|
|
+ SysUserProfileUpdateReqVO reqVO = randomPojo(SysUserProfileUpdateReqVO.class, o -> {
|
|
|
+ o.setMobile(randomString());
|
|
|
+ o.setSex(RandomUtil.randomEle(SysSexEnum.values()).getSex());
|
|
|
});
|
|
|
- // 先插入一条
|
|
|
- Long userId = userService.createUser(reqVO);
|
|
|
- // 调用数据
|
|
|
- userService.deleteUser(userId);
|
|
|
- // 校验结果
|
|
|
- assertNull(userService.getUser(userId));
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ userService.updateUserProfile(userId, reqVO);
|
|
|
+ // 断言
|
|
|
+ SysUserDO user = userMapper.selectById(userId);
|
|
|
+ assertPojoEquals(reqVO, user);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void test_updateUserPassword(){
|
|
|
+ public void testUpdateUserPassword_success() {
|
|
|
+ // mock 数据
|
|
|
+ SysUserDO dbUser = randomSysUserDO(o -> o.setPassword("encode:yudao"));
|
|
|
+ userMapper.insert(dbUser);
|
|
|
// 准备参数
|
|
|
- SysUserCreateReqVO reqVO = randomPojo(SysUserCreateReqVO.class,o->{
|
|
|
- o.setDeptId(null);
|
|
|
- o.setPostIds(Collections.emptySet());
|
|
|
- o.setSex(1);
|
|
|
- o.setPassword("123");
|
|
|
+ Long userId = dbUser.getId();
|
|
|
+ SysUserProfileUpdatePasswordReqVO reqVO = randomPojo(SysUserProfileUpdatePasswordReqVO.class, o -> {
|
|
|
+ o.setOldPassword("yudao");
|
|
|
+ o.setNewPassword("yuanma");
|
|
|
});
|
|
|
- // 先插入一条
|
|
|
- Long userId = userService.createUser(reqVO);
|
|
|
- String newPassword = RandomUtils.randomString();
|
|
|
+ // mock 方法
|
|
|
+ when(passwordEncoder.encode(anyString())).then(
|
|
|
+ (Answer<String>) invocationOnMock -> "encode:" + invocationOnMock.getArgument(0));
|
|
|
+ when(passwordEncoder.matches(eq(reqVO.getOldPassword()), eq(dbUser.getPassword()))).thenReturn(true);
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ userService.updateUserPassword(userId, reqVO);
|
|
|
+ // 断言
|
|
|
+ SysUserDO user = userMapper.selectById(userId);
|
|
|
+ assertEquals("encode:yuanma", user.getPassword());
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testUpdateUserAvatar_success() {
|
|
|
+ // mock 数据
|
|
|
+ SysUserDO dbUser = randomSysUserDO();
|
|
|
+ userMapper.insert(dbUser);
|
|
|
+ // 准备参数
|
|
|
+ Long userId = dbUser.getId();
|
|
|
+ byte[] avatarFileBytes = randomBytes(10);
|
|
|
+ ByteArrayInputStream avatarFile = new ByteArrayInputStream(avatarFileBytes);
|
|
|
+ // mock 方法
|
|
|
+ String avatar = randomString();
|
|
|
+ when(fileService.createFile(anyString(), eq(avatarFileBytes))).thenReturn(avatar);
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ userService.updateUserAvatar(userId, avatarFile);
|
|
|
+ // 断言
|
|
|
+ SysUserDO user = userMapper.selectById(userId);
|
|
|
+ assertEquals(avatar, user.getAvatar());
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testUpdateUserPassword02_success() {
|
|
|
+ // mock 数据
|
|
|
+ SysUserDO dbUser = randomSysUserDO();
|
|
|
+ userMapper.insert(dbUser);
|
|
|
+ // 准备参数
|
|
|
+ Long userId = dbUser.getId();
|
|
|
+ String password = "yudao";
|
|
|
+ // mock 方法
|
|
|
+ when(passwordEncoder.encode(anyString())).then(
|
|
|
+ (Answer<String>) invocationOnMock -> "encode:" + invocationOnMock.getArgument(0));
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ userService.updateUserPassword(userId, password);
|
|
|
+ // 断言
|
|
|
+ SysUserDO user = userMapper.selectById(userId);
|
|
|
+ assertEquals("encode:" + password, user.getPassword());
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testUpdateUserStatus() {
|
|
|
+ // mock 数据
|
|
|
+ SysUserDO dbUser = randomSysUserDO();
|
|
|
+ userMapper.insert(dbUser);
|
|
|
+ // 准备参数
|
|
|
+ Long userId = dbUser.getId();
|
|
|
+ Integer status = randomCommonStatus();
|
|
|
+
|
|
|
// 调用
|
|
|
- userService.updateUserPassword(userId,newPassword);
|
|
|
+ userService.updateUserStatus(userId, status);
|
|
|
+ // 断言
|
|
|
+ SysUserDO user = userMapper.selectById(userId);
|
|
|
+ assertEquals(status, user.getStatus());
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testDeleteUser(){
|
|
|
+ // mock 数据
|
|
|
+ SysUserDO dbUser = randomSysUserDO();
|
|
|
+ userMapper.insert(dbUser);
|
|
|
+ // 准备参数
|
|
|
+ Long userId = dbUser.getId();
|
|
|
+
|
|
|
+ // 调用数据
|
|
|
+ userService.deleteUser(userId);
|
|
|
// 校验结果
|
|
|
- assertNotEquals(passwordEncoder.encode(newPassword),userService.getUser(userId).getPassword());
|
|
|
+ assertNull(userService.getUser(userId));
|
|
|
+ // 校验调用次数
|
|
|
+ verify(permissionService, times(1)).processUserDeleted(eq(userId));
|
|
|
}
|
|
|
|
|
|
+
|
|
|
+
|
|
|
@Test
|
|
|
public void test_importUsers(){
|
|
|
SysDeptDO dept = randomPojo(SysDeptDO.class, o -> { // 等会查询到
|
|
|
o.setName("开发部");
|
|
|
- o.setSort("1");
|
|
|
+ o.setSort(1);
|
|
|
o.setStatus(CommonStatusEnum.ENABLE.getStatus());
|
|
|
});
|
|
|
- int depId = deptMapper.insert(dept);
|
|
|
+// int depId = deptMapper.insert(dept);
|
|
|
+ int depId = 0;
|
|
|
// 准备参数
|
|
|
List<SysUserImportExcelVO> list = new ArrayList<>();
|
|
|
list.add(randomPojo(SysUserImportExcelVO.class, o->{
|
|
|
@@ -153,18 +284,15 @@ public class SysUserServiceImplTest extends BaseSpringBootUnitTest {
|
|
|
assertEquals(respVOUpdate.getUpdateUsernames().size(),3);
|
|
|
}
|
|
|
|
|
|
- public SysUserCreateReqVO randomUserVO(){
|
|
|
-
|
|
|
- SysUserCreateReqVO userVO = new SysUserCreateReqVO();
|
|
|
- userVO.setUsername(RandomUtils.randomString());
|
|
|
- userVO.setNickname(RandomUtils.randomString());
|
|
|
- userVO.setMobile(RandomUtil.randomNumbers(11));
|
|
|
- userVO.setEmail(RandomUtils.randomString()+"@ruoyi.com");
|
|
|
- userVO.setDeptId(null);
|
|
|
- userVO.setPostIds(Collections.emptySet());
|
|
|
-
|
|
|
- userVO.setPassword(RandomUtils.randomString());
|
|
|
- return userVO;
|
|
|
+ // ========== 随机对象 ==========
|
|
|
|
|
|
+ @SafeVarargs
|
|
|
+ private static SysUserDO randomSysUserDO(Consumer<SysUserDO>... consumers) {
|
|
|
+ Consumer<SysUserDO> consumer = (o) -> {
|
|
|
+ o.setStatus(randomEle(CommonStatusEnum.values()).getStatus()); // 保证 status 的范围
|
|
|
+ o.setSex(randomEle(SysSexEnum.values()).getSex()); // 保证 sex 的范围
|
|
|
+ };
|
|
|
+ return randomPojo(SysUserDO.class, ArrayUtils.append(consumer, consumers));
|
|
|
}
|
|
|
+
|
|
|
}
|
YunaiV