回退 'Pull Request !144 : 注释掉MenuServiceTest类测试代码,修复master分支的源码编译报错问题,方便其他童鞋下载后直接运行。'

This commit is contained in:
芋道源码 2022-04-30 15:14:58 +00:00 committed by Gitee
parent e651d9c59d
commit b47264c037
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 393 additions and 393 deletions

View File

@ -1,393 +1,393 @@
//package cn.iocoder.yudao.module.system.service.permission;
//
//import cn.hutool.core.bean.BeanUtil;
//import cn.hutool.core.lang.Assert;
//import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
//import cn.iocoder.yudao.framework.common.util.object.ObjectUtils;
//import cn.iocoder.yudao.framework.common.util.spring.SpringAopUtils;
//import cn.iocoder.yudao.module.system.controller.admin.permission.vo.menu.MenuCreateReqVO;
//import cn.iocoder.yudao.module.system.controller.admin.permission.vo.menu.MenuListReqVO;
//import cn.iocoder.yudao.module.system.controller.admin.permission.vo.menu.MenuUpdateReqVO;
//import cn.iocoder.yudao.module.system.dal.dataobject.permission.MenuDO;
//import cn.iocoder.yudao.module.system.dal.mysql.permission.MenuMapper;
//import cn.iocoder.yudao.module.system.enums.permission.MenuTypeEnum;
//import cn.iocoder.yudao.module.system.mq.producer.permission.MenuProducer;
//import cn.iocoder.yudao.module.system.service.tenant.TenantService;
//import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
//import com.google.common.collect.Multimap;
//import org.junit.jupiter.api.Test;
//import org.springframework.boot.test.mock.mockito.MockBean;
//import org.springframework.context.annotation.Import;
//
//import javax.annotation.Resource;
//import java.util.*;
//
//import static cn.iocoder.yudao.framework.common.util.collection.SetUtils.asSet;
//import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertPojoEquals;
//import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServiceException;
//import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
//import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.*;
//import static org.junit.jupiter.api.Assertions.*;
//import static org.mockito.ArgumentMatchers.argThat;
//import static org.mockito.Mockito.doNothing;
//import static org.mockito.Mockito.verify;
//
//// TODO @芋艿单测的代码质量可以提升下
//@Import(MenuServiceImpl.class)
//public class MenuServiceTest extends BaseDbUnitTest {
//
// @Resource
// private MenuServiceImpl menuService;
//
// @Resource
// private MenuMapper menuMapper;
//
// @MockBean
// private PermissionService permissionService;
// @MockBean
// private MenuProducer menuProducer;
// @MockBean
// private TenantService tenantService;
//
// @Test
// public void testInitLocalCache_success() throws Exception {
// MenuDO menuDO1 = createMenuDO(MenuTypeEnum.MENU, "xxxx", 0L);
// menuMapper.insert(menuDO1);
// MenuDO menuDO2 = createMenuDO(MenuTypeEnum.MENU, "xxxx", 0L);
// menuMapper.insert(menuDO2);
//
// // 调用
// menuService.initLocalCache();
//
// // 获取代理对象
// MenuServiceImpl target = (MenuServiceImpl) SpringAopUtils.getTarget(menuService);
//
// Map<Long, MenuDO> menuCache =
// (Map<Long, MenuDO>) BeanUtil.getFieldValue(target, "menuCache");
// Assert.isTrue(menuCache.size() == 2);
// assertPojoEquals(menuDO1, menuCache.get(menuDO1.getId()));
// assertPojoEquals(menuDO2, menuCache.get(menuDO2.getId()));
//
// Multimap<String, MenuDO> permissionMenuCache =
// (Multimap<String, MenuDO>) BeanUtil.getFieldValue(target, "permissionMenuCache");
// Assert.isTrue(permissionMenuCache.size() == 2);
// assertPojoEquals(menuDO1, permissionMenuCache.get(menuDO1.getPermission()));
// assertPojoEquals(menuDO2, permissionMenuCache.get(menuDO2.getPermission()));
//
// Date maxUpdateTime = (Date) BeanUtil.getFieldValue(target, "maxUpdateTime");
// assertEquals(ObjectUtils.max(menuDO1.getUpdateTime(), menuDO2.getUpdateTime()), maxUpdateTime);
// }
//
// @Test
// public void testCreateMenu_success() {
// //构造父目录
// MenuDO menuDO = createMenuDO(MenuTypeEnum.MENU, "parent", 0L);
// menuMapper.insert(menuDO);
// Long parentId = menuDO.getId();
//
// //调用
// MenuCreateReqVO vo = randomPojo(MenuCreateReqVO.class, o -> {
// o.setParentId(parentId);
// o.setName("testSonName");
// o.setType(MenuTypeEnum.MENU.getType());
// o.setStatus(randomCommonStatus());
// });
// Long menuId = menuService.createMenu(vo);
//
// //断言
// assertNotNull(menuId);
// // 校验记录的属性是否正确
// MenuDO ret = menuMapper.selectById(menuId);
// assertPojoEquals(vo, ret);
// // 校验调用
// verify(menuProducer).sendMenuRefreshMessage();
// }
//
// @Test
// public void testUpdateMenu_success() {
// //构造父子目录
// MenuDO sonMenuDO = initParentAndSonMenuDO();
// Long sonId = sonMenuDO.getId();
// Long parentId = sonMenuDO.getParentId();
//
// //调用
// MenuUpdateReqVO vo = randomPojo(MenuUpdateReqVO.class, o -> {
// o.setId(sonId);
// o.setParentId(parentId);
// o.setType(MenuTypeEnum.MENU.getType());
// o.setStatus(randomCommonStatus());
// o.setName("pppppp"); //修改名字
// });
// menuService.updateMenu(vo);
//
// //断言
// // 校验记录的属性是否正确
// MenuDO ret = menuMapper.selectById(sonId);
// assertPojoEquals(vo, ret);
// // 校验调用
// verify(menuProducer).sendMenuRefreshMessage();
// }
//
// @Test
// public void testUpdateMenu_sonIdNotExist() {
// Long sonId = 99999L;
// Long parentId = 10000L;
//
// //调用
// MenuUpdateReqVO vo = randomPojo(MenuUpdateReqVO.class, o -> {
// o.setId(sonId);
// o.setParentId(parentId);
// o.setType(MenuTypeEnum.MENU.getType());
// o.setStatus(randomCommonStatus());
// });
// //断言
// assertServiceException(() -> menuService.updateMenu(vo), MENU_NOT_EXISTS);
// }
//
// @Test
// public void testDeleteMenu_success() {
// MenuDO sonMenuDO = initParentAndSonMenuDO();
// Long sonId = sonMenuDO.getId();
//
// // 调用
// menuService.deleteMenu(sonId);
//
// // 断言
// MenuDO menuDO = menuMapper.selectById(sonId);
// assertNull(menuDO);
// verify(permissionService).processMenuDeleted(sonId);
// verify(menuProducer).sendMenuRefreshMessage();
// }
//
// @Test
// public void testDeleteMenu_menuNotExist() {
// Long sonId = 99999L;
//
// assertServiceException(() -> menuService.deleteMenu(sonId), MENU_NOT_EXISTS);
// }
//
// @Test
// public void testDeleteMenu_existChildren() {
// MenuDO sonMenu = initParentAndSonMenuDO();
// Long parentId = sonMenu.getParentId();
//
// assertServiceException(() -> menuService.deleteMenu(parentId), MENU_EXISTS_CHILDREN);
// }
//
// @Test
// public void testGetMenus() {
// // mock 数据
// MenuDO menu100 = randomPojo(MenuDO.class, o -> o.setId(100L).setStatus(CommonStatusEnum.ENABLE.getStatus()));
// menuMapper.insert(menu100);
// MenuDO menu101 = randomPojo(MenuDO.class, o -> o.setId(101L).setStatus(CommonStatusEnum.DISABLE.getStatus()));
// menuMapper.insert(menu101);
// // 准备参数
// MenuListReqVO reqVO = new MenuListReqVO().setStatus(CommonStatusEnum.ENABLE.getStatus());
//
// // 调用
// List<MenuDO> result = menuService.getMenus(reqVO);
// // 断言
// assertEquals(1, result.size());
// assertPojoEquals(menu100, result.get(0));
// }
//
// @Test
// public void testTenantMenus() {
// // mock 数据
// MenuDO menu100 = randomPojo(MenuDO.class, o -> o.setId(100L).setStatus(CommonStatusEnum.ENABLE.getStatus()));
// menuMapper.insert(menu100);
// MenuDO menu101 = randomPojo(MenuDO.class, o -> o.setId(101L).setStatus(CommonStatusEnum.DISABLE.getStatus()));
// menuMapper.insert(menu101);
// MenuDO menu102 = randomPojo(MenuDO.class, o -> o.setId(102L).setStatus(CommonStatusEnum.ENABLE.getStatus()));
// menuMapper.insert(menu102);
// // mock 过滤菜单
// // mock 账户额度充足
// Set<Long> menuIds = asSet(100L, 101L);
// doNothing().when(tenantService).handleTenantMenu(argThat(handler -> {
// handler.handle(menuIds);
// return true;
// }));
// // 准备参数
// MenuListReqVO reqVO = new MenuListReqVO().setStatus(CommonStatusEnum.ENABLE.getStatus());
//
// // 调用
// List<MenuDO> result = menuService.getTenantMenus(reqVO);
// // 断言
// assertEquals(1, result.size());
// assertPojoEquals(menu100, result.get(0));
// }
//
// @Test
// public void testGetMenusReqVo_success() {
// Map<Long, MenuDO> idMenuMap = new HashMap<>();
// // 用于验证可以模糊搜索名称包含"name"状态为1的menu
// MenuDO menu = createMenuDO(MenuTypeEnum.MENU, "name2", 0L, 1);
// menuMapper.insert(menu);
// idMenuMap.put(menu.getId(), menu);
//
// menu = createMenuDO(MenuTypeEnum.MENU, "11name111", 0L, 1);
// menuMapper.insert(menu);
// idMenuMap.put(menu.getId(), menu);
//
// menu = createMenuDO(MenuTypeEnum.MENU, "name", 0L, 1);
// menuMapper.insert(menu);
// idMenuMap.put(menu.getId(), menu);
//
// // 以下是不符合搜索条件的的menu
// menu = createMenuDO(MenuTypeEnum.MENU, "xxxxxx", 0L, 1);
// menuMapper.insert(menu);
// menu = createMenuDO(MenuTypeEnum.MENU, "name", 0L, 2);
// menuMapper.insert(menu);
//
// // 调用
// MenuListReqVO reqVO = new MenuListReqVO();
// reqVO.setStatus(1);
// reqVO.setName("name");
// List<MenuDO> menuDOS = menuService.getMenus(reqVO);
//
// // 断言
// assertEquals(menuDOS.size(), idMenuMap.size());
// menuDOS.forEach(m -> assertPojoEquals(idMenuMap.get(m.getId()), m));
// }
//
// @Test
// public void testListMenusFromCache_success() throws Exception {
// Map<Long, MenuDO> mockCacheMap = new HashMap<>();
// // 获取代理对象
// MenuServiceImpl target = (MenuServiceImpl) SpringAopUtils.getTarget(menuService);
// BeanUtil.setFieldValue(target, "menuCache", mockCacheMap);
//
// Map<Long, MenuDO> idMenuMap = new HashMap<>();
// // 用于验证搜索类型为MENU,状态为1的menu
// MenuDO menuDO = createMenuDO(1L, MenuTypeEnum.MENU, "name", 0L, 1);
// mockCacheMap.put(menuDO.getId(), menuDO);
// idMenuMap.put(menuDO.getId(), menuDO);
//
// menuDO = createMenuDO(2L, MenuTypeEnum.MENU, "name", 0L, 1);
// mockCacheMap.put(menuDO.getId(), menuDO);
// idMenuMap.put(menuDO.getId(), menuDO);
//
// // 以下是不符合搜索条件的menu
// menuDO = createMenuDO(3L, MenuTypeEnum.BUTTON, "name", 0L, 1);
// mockCacheMap.put(menuDO.getId(), menuDO);
// menuDO = createMenuDO(4L, MenuTypeEnum.MENU, "name", 0L, 2);
// mockCacheMap.put(menuDO.getId(), menuDO);
//
// List<MenuDO> menuDOS = menuService.getMenuListFromCache(Collections.singletonList(MenuTypeEnum.MENU.getType()),
// Collections.singletonList(CommonStatusEnum.DISABLE.getStatus()));
// assertEquals(menuDOS.size(), idMenuMap.size());
// menuDOS.forEach(m -> assertPojoEquals(idMenuMap.get(m.getId()), m));
// }
//
// @Test
// public void testListMenusFromCache2_success() throws Exception {
// Map<Long, MenuDO> mockCacheMap = new HashMap<>();
// // 获取代理对象
// MenuServiceImpl target = (MenuServiceImpl) SpringAopUtils.getTarget(menuService);
// BeanUtil.setFieldValue(target, "menuCache", mockCacheMap);
//
// Map<Long, MenuDO> idMenuMap = new HashMap<>();
// // 验证搜索id为1, 类型为MENU, 状态为1 的menu
// MenuDO menuDO = createMenuDO(1L, MenuTypeEnum.MENU, "name", 0L, 1);
// mockCacheMap.put(menuDO.getId(), menuDO);
// idMenuMap.put(menuDO.getId(), menuDO);
//
// // 以下是不符合搜索条件的menu
// menuDO = createMenuDO(2L, MenuTypeEnum.MENU, "name", 0L, 1);
// mockCacheMap.put(menuDO.getId(), menuDO);
// menuDO = createMenuDO(3L, MenuTypeEnum.BUTTON, "name", 0L, 1);
// mockCacheMap.put(menuDO.getId(), menuDO);
// menuDO = createMenuDO(4L, MenuTypeEnum.MENU, "name", 0L, 2);
// mockCacheMap.put(menuDO.getId(), menuDO);
//
// List<MenuDO> menuDOS = menuService.getMenuListFromCache(Collections.singletonList(1L),
// Collections.singletonList(MenuTypeEnum.MENU.getType()), Collections.singletonList(1));
// assertEquals(menuDOS.size(), idMenuMap.size());
// menuDOS.forEach(menu -> assertPojoEquals(idMenuMap.get(menu.getId()), menu));
// }
//
// @Test
// public void testCheckParentResource_success() {
// MenuDO menuDO = createMenuDO(MenuTypeEnum.MENU, "parent", 0L);
// menuMapper.insert(menuDO);
// Long parentId = menuDO.getId();
//
// menuService.checkParentResource(parentId, null);
// }
//
// @Test
// public void testCheckParentResource_canNotSetSelfToBeParent() {
// assertServiceException(() -> menuService.checkParentResource(1L, 1L), MENU_PARENT_ERROR);
// }
//
// @Test
// public void testCheckParentResource_parentNotExist() {
// assertServiceException(() -> menuService.checkParentResource(randomLongId(), null), MENU_PARENT_NOT_EXISTS);
// }
//
// @Test
// public void testCheckParentResource_parentTypeError() {
// MenuDO menuDO = createMenuDO(MenuTypeEnum.BUTTON, "parent", 0L);
// menuMapper.insert(menuDO);
// Long parentId = menuDO.getId();
//
// assertServiceException(() -> menuService.checkParentResource(parentId, null), MENU_PARENT_NOT_DIR_OR_MENU);
// }
//
// @Test
// public void testCheckResource_success() {
// MenuDO sonMenu = initParentAndSonMenuDO();
// Long parentId = sonMenu.getParentId();
//
// Long otherSonMenuId = randomLongId();
// String otherSonMenuName = randomString();
//
// menuService.checkResource(parentId, otherSonMenuName, otherSonMenuId);
// }
//
// @Test
// public void testCheckResource_sonMenuNameDuplicate(){
// MenuDO sonMenu=initParentAndSonMenuDO();
// Long parentId=sonMenu.getParentId();
//
// Long otherSonMenuId=randomLongId();
// String otherSonMenuName=sonMenu.getName(); //相同名称
//
// assertServiceException(() -> menuService.checkResource(parentId, otherSonMenuName, otherSonMenuId), MENU_NAME_DUPLICATE);
// }
//
// /**
// * 构造父子目录返回子目录
// *
// * @return
// */
// private MenuDO initParentAndSonMenuDO() {
// //构造父子目录
// MenuDO menuDO = createMenuDO(MenuTypeEnum.MENU, "parent", 0L);
// menuMapper.insert(menuDO);
// Long parentId = menuDO.getId();
//
// MenuDO sonMenuDO = createMenuDO(MenuTypeEnum.MENU, "testSonName", parentId);
// menuMapper.insert(sonMenuDO);
// return sonMenuDO;
// }
//
// private MenuDO createMenuDO(MenuTypeEnum typeEnum, String menuName, Long parentId) {
// return createMenuDO(typeEnum, menuName, parentId, randomCommonStatus());
// }
//
// private MenuDO createMenuDO(MenuTypeEnum typeEnum, String menuName, Long parentId, Integer status) {
// return createMenuDO(null, typeEnum, menuName, parentId, status);
// }
//
// private MenuDO createMenuDO(Long id, MenuTypeEnum typeEnum, String menuName, Long parentId, Integer status) {
// return randomPojo(MenuDO.class, o -> {
// o.setId(id);
// o.setParentId(parentId);
// o.setType(typeEnum.getType());
// o.setStatus(status);
// o.setName(menuName);
// });
// }
//
//}
package cn.iocoder.yudao.module.system.service.permission;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.lang.Assert;
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
import cn.iocoder.yudao.framework.common.util.object.ObjectUtils;
import cn.iocoder.yudao.framework.common.util.spring.SpringAopUtils;
import cn.iocoder.yudao.module.system.controller.admin.permission.vo.menu.MenuCreateReqVO;
import cn.iocoder.yudao.module.system.controller.admin.permission.vo.menu.MenuListReqVO;
import cn.iocoder.yudao.module.system.controller.admin.permission.vo.menu.MenuUpdateReqVO;
import cn.iocoder.yudao.module.system.dal.dataobject.permission.MenuDO;
import cn.iocoder.yudao.module.system.dal.mysql.permission.MenuMapper;
import cn.iocoder.yudao.module.system.enums.permission.MenuTypeEnum;
import cn.iocoder.yudao.module.system.mq.producer.permission.MenuProducer;
import cn.iocoder.yudao.module.system.service.tenant.TenantService;
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
import com.google.common.collect.Multimap;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.context.annotation.Import;
import javax.annotation.Resource;
import java.util.*;
import static cn.iocoder.yudao.framework.common.util.collection.SetUtils.asSet;
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertPojoEquals;
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServiceException;
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.*;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.argThat;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.verify;
// TODO @芋艿单测的代码质量可以提升下
@Import(MenuServiceImpl.class)
public class MenuServiceTest extends BaseDbUnitTest {
@Resource
private MenuServiceImpl menuService;
@Resource
private MenuMapper menuMapper;
@MockBean
private PermissionService permissionService;
@MockBean
private MenuProducer menuProducer;
@MockBean
private TenantService tenantService;
@Test
public void testInitLocalCache_success() throws Exception {
MenuDO menuDO1 = createMenuDO(MenuTypeEnum.MENU, "xxxx", 0L);
menuMapper.insert(menuDO1);
MenuDO menuDO2 = createMenuDO(MenuTypeEnum.MENU, "xxxx", 0L);
menuMapper.insert(menuDO2);
// 调用
menuService.initLocalCache();
// 获取代理对象
MenuServiceImpl target = (MenuServiceImpl) SpringAopUtils.getTarget(menuService);
Map<Long, MenuDO> menuCache =
(Map<Long, MenuDO>) BeanUtil.getFieldValue(target, "menuCache");
Assert.isTrue(menuCache.size() == 2);
assertPojoEquals(menuDO1, menuCache.get(menuDO1.getId()));
assertPojoEquals(menuDO2, menuCache.get(menuDO2.getId()));
Multimap<String, MenuDO> permissionMenuCache =
(Multimap<String, MenuDO>) BeanUtil.getFieldValue(target, "permissionMenuCache");
Assert.isTrue(permissionMenuCache.size() == 2);
assertPojoEquals(menuDO1, permissionMenuCache.get(menuDO1.getPermission()));
assertPojoEquals(menuDO2, permissionMenuCache.get(menuDO2.getPermission()));
Date maxUpdateTime = (Date) BeanUtil.getFieldValue(target, "maxUpdateTime");
assertEquals(ObjectUtils.max(menuDO1.getUpdateTime(), menuDO2.getUpdateTime()), maxUpdateTime);
}
@Test
public void testCreateMenu_success() {
//构造父目录
MenuDO menuDO = createMenuDO(MenuTypeEnum.MENU, "parent", 0L);
menuMapper.insert(menuDO);
Long parentId = menuDO.getId();
//调用
MenuCreateReqVO vo = randomPojo(MenuCreateReqVO.class, o -> {
o.setParentId(parentId);
o.setName("testSonName");
o.setType(MenuTypeEnum.MENU.getType());
o.setStatus(randomCommonStatus());
});
Long menuId = menuService.createMenu(vo);
//断言
assertNotNull(menuId);
// 校验记录的属性是否正确
MenuDO ret = menuMapper.selectById(menuId);
assertPojoEquals(vo, ret);
// 校验调用
verify(menuProducer).sendMenuRefreshMessage();
}
@Test
public void testUpdateMenu_success() {
//构造父子目录
MenuDO sonMenuDO = initParentAndSonMenuDO();
Long sonId = sonMenuDO.getId();
Long parentId = sonMenuDO.getParentId();
//调用
MenuUpdateReqVO vo = randomPojo(MenuUpdateReqVO.class, o -> {
o.setId(sonId);
o.setParentId(parentId);
o.setType(MenuTypeEnum.MENU.getType());
o.setStatus(randomCommonStatus());
o.setName("pppppp"); //修改名字
});
menuService.updateMenu(vo);
//断言
// 校验记录的属性是否正确
MenuDO ret = menuMapper.selectById(sonId);
assertPojoEquals(vo, ret);
// 校验调用
verify(menuProducer).sendMenuRefreshMessage();
}
@Test
public void testUpdateMenu_sonIdNotExist() {
Long sonId = 99999L;
Long parentId = 10000L;
//调用
MenuUpdateReqVO vo = randomPojo(MenuUpdateReqVO.class, o -> {
o.setId(sonId);
o.setParentId(parentId);
o.setType(MenuTypeEnum.MENU.getType());
o.setStatus(randomCommonStatus());
});
//断言
assertServiceException(() -> menuService.updateMenu(vo), MENU_NOT_EXISTS);
}
@Test
public void testDeleteMenu_success() {
MenuDO sonMenuDO = initParentAndSonMenuDO();
Long sonId = sonMenuDO.getId();
// 调用
menuService.deleteMenu(sonId);
// 断言
MenuDO menuDO = menuMapper.selectById(sonId);
assertNull(menuDO);
verify(permissionService).processMenuDeleted(sonId);
verify(menuProducer).sendMenuRefreshMessage();
}
@Test
public void testDeleteMenu_menuNotExist() {
Long sonId = 99999L;
assertServiceException(() -> menuService.deleteMenu(sonId), MENU_NOT_EXISTS);
}
@Test
public void testDeleteMenu_existChildren() {
MenuDO sonMenu = initParentAndSonMenuDO();
Long parentId = sonMenu.getParentId();
assertServiceException(() -> menuService.deleteMenu(parentId), MENU_EXISTS_CHILDREN);
}
@Test
public void testGetMenus() {
// mock 数据
MenuDO menu100 = randomPojo(MenuDO.class, o -> o.setId(100L).setStatus(CommonStatusEnum.ENABLE.getStatus()));
menuMapper.insert(menu100);
MenuDO menu101 = randomPojo(MenuDO.class, o -> o.setId(101L).setStatus(CommonStatusEnum.DISABLE.getStatus()));
menuMapper.insert(menu101);
// 准备参数
MenuListReqVO reqVO = new MenuListReqVO().setStatus(CommonStatusEnum.ENABLE.getStatus());
// 调用
List<MenuDO> result = menuService.getMenus(reqVO);
// 断言
assertEquals(1, result.size());
assertPojoEquals(menu100, result.get(0));
}
@Test
public void testTenantMenus() {
// mock 数据
MenuDO menu100 = randomPojo(MenuDO.class, o -> o.setId(100L).setStatus(CommonStatusEnum.ENABLE.getStatus()));
menuMapper.insert(menu100);
MenuDO menu101 = randomPojo(MenuDO.class, o -> o.setId(101L).setStatus(CommonStatusEnum.DISABLE.getStatus()));
menuMapper.insert(menu101);
MenuDO menu102 = randomPojo(MenuDO.class, o -> o.setId(102L).setStatus(CommonStatusEnum.ENABLE.getStatus()));
menuMapper.insert(menu102);
// mock 过滤菜单
// mock 账户额度充足
Set<Long> menuIds = asSet(100L, 101L);
doNothing().when(tenantService).handleTenantMenu(argThat(handler -> {
handler.handle(menuIds);
return true;
}));
// 准备参数
MenuListReqVO reqVO = new MenuListReqVO().setStatus(CommonStatusEnum.ENABLE.getStatus());
// 调用
List<MenuDO> result = menuService.getTenantMenus(reqVO);
// 断言
assertEquals(1, result.size());
assertPojoEquals(menu100, result.get(0));
}
@Test
public void testGetMenusReqVo_success() {
Map<Long, MenuDO> idMenuMap = new HashMap<>();
// 用于验证可以模糊搜索名称包含"name"状态为1的menu
MenuDO menu = createMenuDO(MenuTypeEnum.MENU, "name2", 0L, 1);
menuMapper.insert(menu);
idMenuMap.put(menu.getId(), menu);
menu = createMenuDO(MenuTypeEnum.MENU, "11name111", 0L, 1);
menuMapper.insert(menu);
idMenuMap.put(menu.getId(), menu);
menu = createMenuDO(MenuTypeEnum.MENU, "name", 0L, 1);
menuMapper.insert(menu);
idMenuMap.put(menu.getId(), menu);
// 以下是不符合搜索条件的的menu
menu = createMenuDO(MenuTypeEnum.MENU, "xxxxxx", 0L, 1);
menuMapper.insert(menu);
menu = createMenuDO(MenuTypeEnum.MENU, "name", 0L, 2);
menuMapper.insert(menu);
// 调用
MenuListReqVO reqVO = new MenuListReqVO();
reqVO.setStatus(1);
reqVO.setName("name");
List<MenuDO> menuDOS = menuService.getMenus(reqVO);
// 断言
assertEquals(menuDOS.size(), idMenuMap.size());
menuDOS.forEach(m -> assertPojoEquals(idMenuMap.get(m.getId()), m));
}
@Test
public void testListMenusFromCache_success() throws Exception {
Map<Long, MenuDO> mockCacheMap = new HashMap<>();
// 获取代理对象
MenuServiceImpl target = (MenuServiceImpl) SpringAopUtils.getTarget(menuService);
BeanUtil.setFieldValue(target, "menuCache", mockCacheMap);
Map<Long, MenuDO> idMenuMap = new HashMap<>();
// 用于验证搜索类型为MENU,状态为1的menu
MenuDO menuDO = createMenuDO(1L, MenuTypeEnum.MENU, "name", 0L, 1);
mockCacheMap.put(menuDO.getId(), menuDO);
idMenuMap.put(menuDO.getId(), menuDO);
menuDO = createMenuDO(2L, MenuTypeEnum.MENU, "name", 0L, 1);
mockCacheMap.put(menuDO.getId(), menuDO);
idMenuMap.put(menuDO.getId(), menuDO);
// 以下是不符合搜索条件的menu
menuDO = createMenuDO(3L, MenuTypeEnum.BUTTON, "name", 0L, 1);
mockCacheMap.put(menuDO.getId(), menuDO);
menuDO = createMenuDO(4L, MenuTypeEnum.MENU, "name", 0L, 2);
mockCacheMap.put(menuDO.getId(), menuDO);
List<MenuDO> menuDOS = menuService.getMenuListFromCache(Collections.singletonList(MenuTypeEnum.MENU.getType()),
Collections.singletonList(CommonStatusEnum.DISABLE.getStatus()));
assertEquals(menuDOS.size(), idMenuMap.size());
menuDOS.forEach(m -> assertPojoEquals(idMenuMap.get(m.getId()), m));
}
@Test
public void testListMenusFromCache2_success() throws Exception {
Map<Long, MenuDO> mockCacheMap = new HashMap<>();
// 获取代理对象
MenuServiceImpl target = (MenuServiceImpl) SpringAopUtils.getTarget(menuService);
BeanUtil.setFieldValue(target, "menuCache", mockCacheMap);
Map<Long, MenuDO> idMenuMap = new HashMap<>();
// 验证搜索id为1, 类型为MENU, 状态为1 的menu
MenuDO menuDO = createMenuDO(1L, MenuTypeEnum.MENU, "name", 0L, 1);
mockCacheMap.put(menuDO.getId(), menuDO);
idMenuMap.put(menuDO.getId(), menuDO);
// 以下是不符合搜索条件的menu
menuDO = createMenuDO(2L, MenuTypeEnum.MENU, "name", 0L, 1);
mockCacheMap.put(menuDO.getId(), menuDO);
menuDO = createMenuDO(3L, MenuTypeEnum.BUTTON, "name", 0L, 1);
mockCacheMap.put(menuDO.getId(), menuDO);
menuDO = createMenuDO(4L, MenuTypeEnum.MENU, "name", 0L, 2);
mockCacheMap.put(menuDO.getId(), menuDO);
List<MenuDO> menuDOS = menuService.getMenuListFromCache(Collections.singletonList(1L),
Collections.singletonList(MenuTypeEnum.MENU.getType()), Collections.singletonList(1));
assertEquals(menuDOS.size(), idMenuMap.size());
menuDOS.forEach(menu -> assertPojoEquals(idMenuMap.get(menu.getId()), menu));
}
@Test
public void testCheckParentResource_success() {
MenuDO menuDO = createMenuDO(MenuTypeEnum.MENU, "parent", 0L);
menuMapper.insert(menuDO);
Long parentId = menuDO.getId();
menuService.checkParentResource(parentId, null);
}
@Test
public void testCheckParentResource_canNotSetSelfToBeParent() {
assertServiceException(() -> menuService.checkParentResource(1L, 1L), MENU_PARENT_ERROR);
}
@Test
public void testCheckParentResource_parentNotExist() {
assertServiceException(() -> menuService.checkParentResource(randomLongId(), null), MENU_PARENT_NOT_EXISTS);
}
@Test
public void testCheckParentResource_parentTypeError() {
MenuDO menuDO = createMenuDO(MenuTypeEnum.BUTTON, "parent", 0L);
menuMapper.insert(menuDO);
Long parentId = menuDO.getId();
assertServiceException(() -> menuService.checkParentResource(parentId, null), MENU_PARENT_NOT_DIR_OR_MENU);
}
@Test
public void testCheckResource_success() {
MenuDO sonMenu = initParentAndSonMenuDO();
Long parentId = sonMenu.getParentId();
Long otherSonMenuId = randomLongId();
String otherSonMenuName = randomString();
menuService.checkResource(parentId, otherSonMenuName, otherSonMenuId);
}
@Test
public void testCheckResource_sonMenuNameDuplicate(){
MenuDO sonMenu=initParentAndSonMenuDO();
Long parentId=sonMenu.getParentId();
Long otherSonMenuId=randomLongId();
String otherSonMenuName=sonMenu.getName(); //相同名称
assertServiceException(() -> menuService.checkResource(parentId, otherSonMenuName, otherSonMenuId), MENU_NAME_DUPLICATE);
}
/**
* 构造父子目录返回子目录
*
* @return
*/
private MenuDO initParentAndSonMenuDO() {
//构造父子目录
MenuDO menuDO = createMenuDO(MenuTypeEnum.MENU, "parent", 0L);
menuMapper.insert(menuDO);
Long parentId = menuDO.getId();
MenuDO sonMenuDO = createMenuDO(MenuTypeEnum.MENU, "testSonName", parentId);
menuMapper.insert(sonMenuDO);
return sonMenuDO;
}
private MenuDO createMenuDO(MenuTypeEnum typeEnum, String menuName, Long parentId) {
return createMenuDO(typeEnum, menuName, parentId, randomCommonStatus());
}
private MenuDO createMenuDO(MenuTypeEnum typeEnum, String menuName, Long parentId, Integer status) {
return createMenuDO(null, typeEnum, menuName, parentId, status);
}
private MenuDO createMenuDO(Long id, MenuTypeEnum typeEnum, String menuName, Long parentId, Integer status) {
return randomPojo(MenuDO.class, o -> {
o.setId(id);
o.setParentId(parentId);
o.setType(typeEnum.getType());
o.setStatus(status);
o.setName(menuName);
});
}
}