From b47264c037aae9b9ed288af457d81ae7ea91726a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8A=8B=E9=81=93=E6=BA=90=E7=A0=81?=
<zhijiantianya@gmail.com>
Date: Sat, 30 Apr 2022 15:14:58 +0000
Subject: [PATCH] =?UTF-8?q?=E5=9B=9E=E9=80=80=20'Pull=20Request=20!144=20:?=
=?UTF-8?q?=20=E6=B3=A8=E9=87=8A=E6=8E=89MenuServiceTest=E7=B1=BB=E6=B5=8B?=
=?UTF-8?q?=E8=AF=95=E4=BB=A3=E7=A0=81=EF=BC=8C=E4=BF=AE=E5=A4=8Dmaster?=
=?UTF-8?q?=E5=88=86=E6=94=AF=E7=9A=84=E6=BA=90=E7=A0=81=E7=BC=96=E8=AF=91?=
=?UTF-8?q?=E6=8A=A5=E9=94=99=E9=97=AE=E9=A2=98=EF=BC=8C=E6=96=B9=E4=BE=BF?=
=?UTF-8?q?=E5=85=B6=E4=BB=96=E7=AB=A5=E9=9E=8B=E4=B8=8B=E8=BD=BD=E5=90=8E?=
=?UTF-8?q?=E7=9B=B4=E6=8E=A5=E8=BF=90=E8=A1=8C=E3=80=82'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
.../service/permission/MenuServiceTest.java | 786 +++++++++---------
1 file changed, 393 insertions(+), 393 deletions(-)
diff --git a/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/permission/MenuServiceTest.java b/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/permission/MenuServiceTest.java
index eefef0527..5d206409b 100644
--- a/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/permission/MenuServiceTest.java
+++ b/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/permission/MenuServiceTest.java
@@ -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);
+ });
+ }
+
+}