|
@@ -1,7 +1,7 @@
|
|
|
package cn.iocoder.yudao.module.system.service.permission;
|
|
|
|
|
|
import cn.hutool.core.collection.CollUtil;
|
|
|
-import cn.hutool.core.map.MapUtil;
|
|
|
+import cn.hutool.extra.spring.SpringUtil;
|
|
|
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
|
|
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
|
|
import cn.iocoder.yudao.module.system.api.permission.dto.DeptDataPermissionRespDTO;
|
|
@@ -14,32 +14,28 @@ import cn.iocoder.yudao.module.system.dal.dataobject.user.AdminUserDO;
|
|
|
import cn.iocoder.yudao.module.system.dal.mysql.permission.RoleMenuMapper;
|
|
|
import cn.iocoder.yudao.module.system.dal.mysql.permission.UserRoleMapper;
|
|
|
import cn.iocoder.yudao.module.system.enums.permission.DataScopeEnum;
|
|
|
-import cn.iocoder.yudao.module.system.mq.producer.permission.PermissionProducer;
|
|
|
import cn.iocoder.yudao.module.system.service.dept.DeptService;
|
|
|
import cn.iocoder.yudao.module.system.service.user.AdminUserService;
|
|
|
-import com.google.common.collect.ImmutableMultimap;
|
|
|
-import com.google.common.collect.Multimap;
|
|
|
import org.junit.jupiter.api.Test;
|
|
|
+import org.mockito.MockedStatic;
|
|
|
import org.springframework.boot.test.mock.mockito.MockBean;
|
|
|
import org.springframework.context.annotation.Import;
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
|
import java.util.Collection;
|
|
|
import java.util.List;
|
|
|
-import java.util.Map;
|
|
|
import java.util.Set;
|
|
|
|
|
|
+import static cn.hutool.core.collection.ListUtil.toList;
|
|
|
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.RandomUtils.*;
|
|
|
-import static java.util.Arrays.asList;
|
|
|
+import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomLongId;
|
|
|
+import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
|
|
|
import static java.util.Collections.singleton;
|
|
|
import static java.util.Collections.singletonList;
|
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
|
import static org.mockito.ArgumentMatchers.eq;
|
|
|
-import static org.mockito.ArgumentMatchers.same;
|
|
|
-import static org.mockito.Mockito.verify;
|
|
|
-import static org.mockito.Mockito.when;
|
|
|
+import static org.mockito.Mockito.*;
|
|
|
|
|
|
@Import({PermissionServiceImpl.class})
|
|
|
public class PermissionServiceTest extends BaseDbUnitTest {
|
|
@@ -61,99 +57,140 @@ public class PermissionServiceTest extends BaseDbUnitTest {
|
|
|
@MockBean
|
|
|
private AdminUserService userService;
|
|
|
|
|
|
- @MockBean
|
|
|
- private PermissionProducer permissionProducer;
|
|
|
-
|
|
|
@Test
|
|
|
- public void testInitLocalCacheForRoleMenu() {
|
|
|
- // mock 数据
|
|
|
- RoleMenuDO roleMenuDO01 = randomPojo(RoleMenuDO.class, o -> o.setRoleId(1L).setMenuId(10L));
|
|
|
- roleMenuMapper.insert(roleMenuDO01);
|
|
|
- RoleMenuDO roleMenuDO02 = randomPojo(RoleMenuDO.class, o -> o.setRoleId(1L).setMenuId(20L));
|
|
|
- roleMenuMapper.insert(roleMenuDO02);
|
|
|
-
|
|
|
- // 调用
|
|
|
- permissionService.initLocalCacheForRoleMenu();
|
|
|
- // 断言 roleMenuCache 缓存
|
|
|
- assertEquals(1, permissionService.getRoleMenuCache().keySet().size());
|
|
|
- assertEquals(asList(10L, 20L), permissionService.getRoleMenuCache().get(1L));
|
|
|
- // 断言 menuRoleCache 缓存
|
|
|
- assertEquals(2, permissionService.getMenuRoleCache().size());
|
|
|
- assertEquals(singletonList(1L), permissionService.getMenuRoleCache().get(10L));
|
|
|
- assertEquals(singletonList(1L), permissionService.getMenuRoleCache().get(20L));
|
|
|
+ public void testHasAnyPermissions_superAdmin() {
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(PermissionServiceImpl.class)))
|
|
|
+ .thenReturn(permissionService);
|
|
|
+
|
|
|
+ // 准备参数
|
|
|
+ Long userId = 1L;
|
|
|
+ String[] roles = new String[]{"system:user:query", "system:user:create"};
|
|
|
+ // mock 用户登录的角色
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(100L));
|
|
|
+ RoleDO role = randomPojo(RoleDO.class, o -> o.setId(100L)
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
+ when(roleService.getRoleListFromCache(eq(singleton(100L)))).thenReturn(toList(role));
|
|
|
+ // mock 其它方法
|
|
|
+ when(roleService.hasAnySuperAdmin(eq(asSet(100L)))).thenReturn(true);
|
|
|
+
|
|
|
+ // 调用,并断言
|
|
|
+ assertTrue(permissionService.hasAnyPermissions(userId, roles));
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void testInitLocalCacheForUserRole() {
|
|
|
- // mock 数据
|
|
|
- UserRoleDO userRoleDO01 = randomPojo(UserRoleDO.class, o -> o.setUserId(1L).setRoleId(10L));
|
|
|
- userRoleMapper.insert(userRoleDO01);
|
|
|
- UserRoleDO roleMenuDO02 = randomPojo(UserRoleDO.class, o -> o.setUserId(1L).setRoleId(20L));
|
|
|
- userRoleMapper.insert(roleMenuDO02);
|
|
|
+ public void testHasAnyPermissions_normal() {
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(PermissionServiceImpl.class)))
|
|
|
+ .thenReturn(permissionService);
|
|
|
+
|
|
|
+ // 准备参数
|
|
|
+ Long userId = 1L;
|
|
|
+ String[] roles = new String[]{"system:user:query", "system:user:create"};
|
|
|
+ // mock 用户登录的角色
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(100L));
|
|
|
+ RoleDO role = randomPojo(RoleDO.class, o -> o.setId(100L)
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
+ when(roleService.getRoleListFromCache(eq(singleton(100L)))).thenReturn(toList(role));
|
|
|
+ // mock 菜单
|
|
|
+ Long menuId = 1000L;
|
|
|
+ when(menuService.getMenuIdListByPermissionFromCache(
|
|
|
+ eq("system:user:create"))).thenReturn(singletonList(menuId));
|
|
|
+ roleMenuMapper.insert(randomPojo(RoleMenuDO.class).setRoleId(100L).setMenuId(1000L));
|
|
|
+
|
|
|
+ // 调用,并断言
|
|
|
+ assertTrue(permissionService.hasAnyPermissions(userId, roles));
|
|
|
+ }
|
|
|
+ }
|
|
|
|
|
|
- // 调用
|
|
|
- permissionService.initLocalCacheForUserRole();
|
|
|
- // 断言 roleMenuCache 缓存
|
|
|
- assertEquals(1, permissionService.getUserRoleCache().size());
|
|
|
- assertEquals(asSet(10L, 20L), permissionService.getUserRoleCache().get(1L));
|
|
|
+ @Test
|
|
|
+ public void testHasAnyRoles() {
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(PermissionServiceImpl.class)))
|
|
|
+ .thenReturn(permissionService);
|
|
|
+
|
|
|
+ // 准备参数
|
|
|
+ Long userId = 1L;
|
|
|
+ String[] roles = new String[]{"yunai", "tudou"};
|
|
|
+ // mock 用户与角色的缓存
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(100L));
|
|
|
+ RoleDO role = randomPojo(RoleDO.class, o -> o.setId(100L).setCode("tudou")
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
+ when(roleService.getRoleListFromCache(eq(singleton(100L)))).thenReturn(toList(role));
|
|
|
+
|
|
|
+ // 调用,并断言
|
|
|
+ assertTrue(permissionService.hasAnyRoles(userId, roles));
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
+ // ========== 角色-菜单的相关方法 ==========
|
|
|
+
|
|
|
@Test
|
|
|
- public void testGetRoleMenuListFromCache_superAdmin() {
|
|
|
+ public void testAssignRoleMenu() {
|
|
|
// 准备参数
|
|
|
- Collection<Long> roleIds = singletonList(100L);
|
|
|
- Collection<Integer> menuTypes = asList(2, 3);
|
|
|
- Collection<Integer> menusStatuses = asList(0, 1);
|
|
|
- // mock 方法
|
|
|
- List<RoleDO> roleList = singletonList(randomPojo(RoleDO.class, o -> o.setId(100L)));
|
|
|
- when(roleService.getRoleListFromCache(eq(roleIds))).thenReturn(roleList);
|
|
|
- when(roleService.hasAnySuperAdmin(same(roleList))).thenReturn(true);
|
|
|
- List<MenuDO> menuList = randomPojoList(MenuDO.class);
|
|
|
- when(menuService.getMenuListFromCache(eq(menuTypes), eq(menusStatuses))).thenReturn(menuList);
|
|
|
+ Long roleId = 1L;
|
|
|
+ Set<Long> menuIds = asSet(200L, 300L);
|
|
|
+ // mock 数据
|
|
|
+ RoleMenuDO roleMenu01 = randomPojo(RoleMenuDO.class).setRoleId(1L).setMenuId(100L);
|
|
|
+ roleMenuMapper.insert(roleMenu01);
|
|
|
+ RoleMenuDO roleMenu02 = randomPojo(RoleMenuDO.class).setRoleId(1L).setMenuId(200L);
|
|
|
+ roleMenuMapper.insert(roleMenu02);
|
|
|
|
|
|
// 调用
|
|
|
- List<MenuDO> result = permissionService.getRoleMenuListFromCache(roleIds, menuTypes, menusStatuses);
|
|
|
+ permissionService.assignRoleMenu(roleId, menuIds);
|
|
|
// 断言
|
|
|
- assertSame(menuList, result);
|
|
|
+ List<RoleMenuDO> roleMenuList = roleMenuMapper.selectList();
|
|
|
+ assertEquals(2, roleMenuList.size());
|
|
|
+ assertEquals(1L, roleMenuList.get(0).getRoleId());
|
|
|
+ assertEquals(200L, roleMenuList.get(0).getMenuId());
|
|
|
+ assertEquals(1L, roleMenuList.get(1).getRoleId());
|
|
|
+ assertEquals(300L, roleMenuList.get(1).getMenuId());
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void testGetRoleMenuListFromCache_normal() {
|
|
|
+ public void testProcessRoleDeleted() {
|
|
|
// 准备参数
|
|
|
- Collection<Long> roleIds = asSet(100L, 200L);
|
|
|
- Collection<Integer> menuTypes = asList(2, 3);
|
|
|
- Collection<Integer> menusStatuses = asList(0, 1);
|
|
|
- // mock 方法
|
|
|
- Multimap<Long, Long> roleMenuCache = ImmutableMultimap.<Long, Long>builder().put(100L, 1000L)
|
|
|
- .put(200L, 2000L).put(200L, 2001L).build();
|
|
|
- permissionService.setRoleMenuCache(roleMenuCache);
|
|
|
- List<MenuDO> menuList = randomPojoList(MenuDO.class);
|
|
|
- when(menuService.getMenuListFromCache(eq(asList(1000L, 2000L, 2001L)), eq(menuTypes), eq(menusStatuses))).thenReturn(menuList);
|
|
|
+ Long roleId = randomLongId();
|
|
|
+ // mock 数据 UserRole
|
|
|
+ UserRoleDO userRoleDO01 = randomPojo(UserRoleDO.class, o -> o.setRoleId(roleId)); // 被删除
|
|
|
+ userRoleMapper.insert(userRoleDO01);
|
|
|
+ UserRoleDO userRoleDO02 = randomPojo(UserRoleDO.class); // 不被删除
|
|
|
+ userRoleMapper.insert(userRoleDO02);
|
|
|
+ // mock 数据 RoleMenu
|
|
|
+ RoleMenuDO roleMenuDO01 = randomPojo(RoleMenuDO.class, o -> o.setRoleId(roleId)); // 被删除
|
|
|
+ roleMenuMapper.insert(roleMenuDO01);
|
|
|
+ RoleMenuDO roleMenuDO02 = randomPojo(RoleMenuDO.class); // 不被删除
|
|
|
+ roleMenuMapper.insert(roleMenuDO02);
|
|
|
|
|
|
// 调用
|
|
|
- List<MenuDO> result = permissionService.getRoleMenuListFromCache(roleIds, menuTypes, menusStatuses);
|
|
|
- // 断言
|
|
|
- assertSame(menuList, result);
|
|
|
+ permissionService.processRoleDeleted(roleId);
|
|
|
+ // 断言数据 RoleMenuDO
|
|
|
+ List<RoleMenuDO> dbRoleMenus = roleMenuMapper.selectList();
|
|
|
+ assertEquals(1, dbRoleMenus.size());
|
|
|
+ assertPojoEquals(dbRoleMenus.get(0), roleMenuDO02);
|
|
|
+ // 断言数据 UserRoleDO
|
|
|
+ List<UserRoleDO> dbUserRoles = userRoleMapper.selectList();
|
|
|
+ assertEquals(1, dbUserRoles.size());
|
|
|
+ assertPojoEquals(dbUserRoles.get(0), userRoleDO02);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void testGetUserRoleIdsFromCache() {
|
|
|
+ public void testProcessMenuDeleted() {
|
|
|
// 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- Collection<Integer> roleStatuses = singleton(CommonStatusEnum.ENABLE.getStatus());
|
|
|
- // mock 方法
|
|
|
- Map<Long, Set<Long>> userRoleCache = MapUtil.<Long, Set<Long>>builder()
|
|
|
- .put(1L, asSet(10L, 20L)).build();
|
|
|
- permissionService.setUserRoleCache(userRoleCache);
|
|
|
- RoleDO roleDO01 = randomPojo(RoleDO.class, o -> o.setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleFromCache(eq(10L))).thenReturn(roleDO01);
|
|
|
- RoleDO roleDO02 = randomPojo(RoleDO.class, o -> o.setStatus(CommonStatusEnum.DISABLE.getStatus()));
|
|
|
- when(roleService.getRoleFromCache(eq(20L))).thenReturn(roleDO02);
|
|
|
+ Long menuId = randomLongId();
|
|
|
+ // mock 数据
|
|
|
+ RoleMenuDO roleMenuDO01 = randomPojo(RoleMenuDO.class, o -> o.setMenuId(menuId)); // 被删除
|
|
|
+ roleMenuMapper.insert(roleMenuDO01);
|
|
|
+ RoleMenuDO roleMenuDO02 = randomPojo(RoleMenuDO.class); // 不被删除
|
|
|
+ roleMenuMapper.insert(roleMenuDO02);
|
|
|
|
|
|
// 调用
|
|
|
- Set<Long> roleIds = permissionService.getUserRoleIdsFromCache(userId, roleStatuses);
|
|
|
- // 断言
|
|
|
- assertEquals(asSet(10L), roleIds);
|
|
|
+ permissionService.processMenuDeleted(menuId);
|
|
|
+ // 断言数据
|
|
|
+ List<RoleMenuDO> dbRoleMenus = roleMenuMapper.selectList();
|
|
|
+ assertEquals(1, dbRoleMenus.size());
|
|
|
+ assertPojoEquals(dbRoleMenus.get(0), roleMenuDO02);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
@@ -166,7 +203,7 @@ public class PermissionServiceTest extends BaseDbUnitTest {
|
|
|
when(menuService.getMenuList()).thenReturn(menuList);
|
|
|
|
|
|
// 调用
|
|
|
- Set<Long> menuIds = permissionService.getRoleMenuIds(roleId);
|
|
|
+ Set<Long> menuIds = permissionService.getRoleMenuListByRoleId(roleId);
|
|
|
// 断言
|
|
|
assertEquals(singleton(1L), menuIds);
|
|
|
}
|
|
@@ -182,34 +219,29 @@ public class PermissionServiceTest extends BaseDbUnitTest {
|
|
|
roleMenuMapper.insert(roleMenu02);
|
|
|
|
|
|
// 调用
|
|
|
- Set<Long> menuIds = permissionService.getRoleMenuIds(roleId);
|
|
|
+ Set<Long> menuIds = permissionService.getRoleMenuListByRoleId(roleId);
|
|
|
// 断言
|
|
|
assertEquals(asSet(1L, 2L), menuIds);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void testAssignRoleMenu() {
|
|
|
+ public void testGetMenuRoleIdListByMenuIdFromCache() {
|
|
|
// 准备参数
|
|
|
- Long roleId = 1L;
|
|
|
- Set<Long> menuIds = asSet(200L, 300L);
|
|
|
+ Long menuId = 1L;
|
|
|
// mock 数据
|
|
|
- RoleMenuDO roleMenu01 = randomPojo(RoleMenuDO.class).setRoleId(1L).setMenuId(100L);
|
|
|
+ RoleMenuDO roleMenu01 = randomPojo(RoleMenuDO.class).setRoleId(100L).setMenuId(1L);
|
|
|
roleMenuMapper.insert(roleMenu01);
|
|
|
- RoleMenuDO roleMenu02 = randomPojo(RoleMenuDO.class).setRoleId(1L).setMenuId(200L);
|
|
|
+ RoleMenuDO roleMenu02 = randomPojo(RoleMenuDO.class).setRoleId(200L).setMenuId(1L);
|
|
|
roleMenuMapper.insert(roleMenu02);
|
|
|
|
|
|
// 调用
|
|
|
- permissionService.assignRoleMenu(roleId, menuIds);
|
|
|
+ Set<Long> roleIds = permissionService.getMenuRoleIdListByMenuIdFromCache(menuId);
|
|
|
// 断言
|
|
|
- List<RoleMenuDO> roleMenuList = roleMenuMapper.selectList();
|
|
|
- assertEquals(2, roleMenuList.size());
|
|
|
- assertEquals(1L, roleMenuList.get(0).getRoleId());
|
|
|
- assertEquals(200L, roleMenuList.get(0).getMenuId());
|
|
|
- assertEquals(1L, roleMenuList.get(1).getRoleId());
|
|
|
- assertEquals(300L, roleMenuList.get(1).getMenuId());
|
|
|
- verify(permissionProducer).sendRoleMenuRefreshMessage();
|
|
|
+ assertEquals(asSet(100L, 200L), roleIds);
|
|
|
}
|
|
|
|
|
|
+ // ========== 用户-角色的相关方法 ==========
|
|
|
+
|
|
|
@Test
|
|
|
public void testAssignUserRole() {
|
|
|
// 准备参数
|
|
@@ -230,7 +262,24 @@ public class PermissionServiceTest extends BaseDbUnitTest {
|
|
|
assertEquals(200L, userRoleDOList.get(0).getRoleId());
|
|
|
assertEquals(1L, userRoleDOList.get(1).getUserId());
|
|
|
assertEquals(300L, userRoleDOList.get(1).getRoleId());
|
|
|
- verify(permissionProducer).sendUserRoleRefreshMessage();
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testProcessUserDeleted() {
|
|
|
+ // 准备参数
|
|
|
+ Long userId = randomLongId();
|
|
|
+ // mock 数据
|
|
|
+ UserRoleDO userRoleDO01 = randomPojo(UserRoleDO.class, o -> o.setUserId(userId)); // 被删除
|
|
|
+ userRoleMapper.insert(userRoleDO01);
|
|
|
+ UserRoleDO userRoleDO02 = randomPojo(UserRoleDO.class); // 不被删除
|
|
|
+ userRoleMapper.insert(userRoleDO02);
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ permissionService.processUserDeleted(userId);
|
|
|
+ // 断言数据
|
|
|
+ List<UserRoleDO> dbUserRoles = userRoleMapper.selectList();
|
|
|
+ assertEquals(1, dbUserRoles.size());
|
|
|
+ assertPojoEquals(dbUserRoles.get(0), userRoleDO02);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
@@ -250,296 +299,229 @@ public class PermissionServiceTest extends BaseDbUnitTest {
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void testGetUserRoleIdListByRoleIds() {
|
|
|
+ public void testGetUserRoleIdListByUserIdFromCache() {
|
|
|
// 准备参数
|
|
|
- Collection<Long> roleIds = asSet(10L, 20L);
|
|
|
+ Long userId = 1L;
|
|
|
// mock 数据
|
|
|
UserRoleDO userRoleDO01 = randomPojo(UserRoleDO.class, o -> o.setUserId(1L).setRoleId(10L));
|
|
|
userRoleMapper.insert(userRoleDO01);
|
|
|
- UserRoleDO roleMenuDO02 = randomPojo(UserRoleDO.class, o -> o.setUserId(2L).setRoleId(20L));
|
|
|
+ UserRoleDO roleMenuDO02 = randomPojo(UserRoleDO.class, o -> o.setUserId(1L).setRoleId(20L));
|
|
|
userRoleMapper.insert(roleMenuDO02);
|
|
|
|
|
|
// 调用
|
|
|
- Set<Long> result = permissionService.getUserRoleIdListByRoleIds(roleIds);
|
|
|
+ Set<Long> result = permissionService.getUserRoleIdListByUserIdFromCache(userId);
|
|
|
// 断言
|
|
|
- assertEquals(asSet(1L, 2L), result);
|
|
|
- }
|
|
|
-
|
|
|
- @Test
|
|
|
- public void testAssignRoleDataScope() {
|
|
|
- // 准备参数
|
|
|
- Long roleId = 1L;
|
|
|
- Integer dataScope = 2;
|
|
|
- Set<Long> dataScopeDeptIds = asSet(10L, 20L);
|
|
|
-
|
|
|
- // 调用
|
|
|
- permissionService.assignRoleDataScope(roleId, dataScope, dataScopeDeptIds);
|
|
|
- // 断言
|
|
|
- verify(roleService).updateRoleDataScope(eq(roleId), eq(dataScope), eq(dataScopeDeptIds));
|
|
|
- }
|
|
|
-
|
|
|
- @Test
|
|
|
- public void testProcessRoleDeleted() {
|
|
|
- // 准备参数
|
|
|
- Long roleId = randomLongId();
|
|
|
- // mock 数据 UserRole
|
|
|
- UserRoleDO userRoleDO01 = randomPojo(UserRoleDO.class, o -> o.setRoleId(roleId)); // 被删除
|
|
|
- userRoleMapper.insert(userRoleDO01);
|
|
|
- UserRoleDO userRoleDO02 = randomPojo(UserRoleDO.class); // 不被删除
|
|
|
- userRoleMapper.insert(userRoleDO02);
|
|
|
- // mock 数据 RoleMenu
|
|
|
- RoleMenuDO roleMenuDO01 = randomPojo(RoleMenuDO.class, o -> o.setRoleId(roleId)); // 被删除
|
|
|
- roleMenuMapper.insert(roleMenuDO01);
|
|
|
- RoleMenuDO roleMenuDO02 = randomPojo(RoleMenuDO.class); // 不被删除
|
|
|
- roleMenuMapper.insert(roleMenuDO02);
|
|
|
-
|
|
|
- // 调用
|
|
|
- permissionService.processRoleDeleted(roleId);
|
|
|
- // 断言数据 RoleMenuDO
|
|
|
- List<RoleMenuDO> dbRoleMenus = roleMenuMapper.selectList();
|
|
|
- assertEquals(1, dbRoleMenus.size());
|
|
|
- assertPojoEquals(dbRoleMenus.get(0), roleMenuDO02);
|
|
|
- // 断言数据 UserRoleDO
|
|
|
- List<UserRoleDO> dbUserRoles = userRoleMapper.selectList();
|
|
|
- assertEquals(1, dbUserRoles.size());
|
|
|
- assertPojoEquals(dbUserRoles.get(0), userRoleDO02);
|
|
|
- // 断言调用
|
|
|
- verify(permissionProducer).sendRoleMenuRefreshMessage();
|
|
|
- verify(permissionProducer).sendUserRoleRefreshMessage();
|
|
|
+ assertEquals(asSet(10L, 20L), result);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void testProcessMenuDeleted() {
|
|
|
+ public void testGetUserRoleIdsFromCache() {
|
|
|
// 准备参数
|
|
|
- Long menuId = randomLongId();
|
|
|
+ Long userId = 1L;
|
|
|
// mock 数据
|
|
|
- RoleMenuDO roleMenuDO01 = randomPojo(RoleMenuDO.class, o -> o.setMenuId(menuId)); // 被删除
|
|
|
- roleMenuMapper.insert(roleMenuDO01);
|
|
|
- RoleMenuDO roleMenuDO02 = randomPojo(RoleMenuDO.class); // 不被删除
|
|
|
- roleMenuMapper.insert(roleMenuDO02);
|
|
|
+ UserRoleDO userRoleDO01 = randomPojo(UserRoleDO.class, o -> o.setUserId(1L).setRoleId(10L));
|
|
|
+ userRoleMapper.insert(userRoleDO01);
|
|
|
+ UserRoleDO roleMenuDO02 = randomPojo(UserRoleDO.class, o -> o.setUserId(1L).setRoleId(20L));
|
|
|
+ userRoleMapper.insert(roleMenuDO02);
|
|
|
|
|
|
// 调用
|
|
|
- permissionService.processMenuDeleted(menuId);
|
|
|
- // 断言数据
|
|
|
- List<RoleMenuDO> dbRoleMenus = roleMenuMapper.selectList();
|
|
|
- assertEquals(1, dbRoleMenus.size());
|
|
|
- assertPojoEquals(dbRoleMenus.get(0), roleMenuDO02);
|
|
|
- // 断言调用
|
|
|
- verify(permissionProducer).sendRoleMenuRefreshMessage();
|
|
|
+ Set<Long> result = permissionService.getUserRoleIdListByUserIdFromCache(userId);
|
|
|
+ // 断言
|
|
|
+ assertEquals(asSet(10L, 20L), result);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void testProcessUserDeleted() {
|
|
|
+ public void testGetUserRoleIdListByRoleId() {
|
|
|
// 准备参数
|
|
|
- Long userId = randomLongId();
|
|
|
+ Collection<Long> roleIds = asSet(10L, 20L);
|
|
|
// mock 数据
|
|
|
- UserRoleDO userRoleDO01 = randomPojo(UserRoleDO.class, o -> o.setUserId(userId)); // 被删除
|
|
|
+ UserRoleDO userRoleDO01 = randomPojo(UserRoleDO.class, o -> o.setUserId(1L).setRoleId(10L));
|
|
|
userRoleMapper.insert(userRoleDO01);
|
|
|
- UserRoleDO userRoleDO02 = randomPojo(UserRoleDO.class); // 不被删除
|
|
|
- userRoleMapper.insert(userRoleDO02);
|
|
|
-
|
|
|
- // 调用
|
|
|
- permissionService.processUserDeleted(userId);
|
|
|
- // 断言数据
|
|
|
- List<UserRoleDO> dbUserRoles = userRoleMapper.selectList();
|
|
|
- assertEquals(1, dbUserRoles.size());
|
|
|
- assertPojoEquals(dbUserRoles.get(0), userRoleDO02);
|
|
|
- // 断言调用
|
|
|
- verify(permissionProducer).sendUserRoleRefreshMessage();
|
|
|
- }
|
|
|
-
|
|
|
- @Test
|
|
|
- public void testHasAnyPermissions_superAdmin() {
|
|
|
- // 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- String[] roles = new String[]{"system:user:query", "system:user:create"};
|
|
|
- // mock 用户与角色的缓存
|
|
|
- permissionService.setUserRoleCache(MapUtil.<Long, Set<Long>>builder().put(1L, asSet(100L)).build());
|
|
|
- RoleDO role = randomPojo(RoleDO.class, o -> o.setId(100L)
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleFromCache(eq(100L))).thenReturn(role);
|
|
|
- // mock 其它方法
|
|
|
- when(roleService.hasAnySuperAdmin(eq(asSet(100L)))).thenReturn(true);
|
|
|
+ UserRoleDO roleMenuDO02 = randomPojo(UserRoleDO.class, o -> o.setUserId(2L).setRoleId(20L));
|
|
|
+ userRoleMapper.insert(roleMenuDO02);
|
|
|
|
|
|
// 调用
|
|
|
- boolean has = permissionService.hasAnyPermissions(userId, roles);
|
|
|
+ Set<Long> result = permissionService.getUserRoleIdListByRoleId(roleIds);
|
|
|
// 断言
|
|
|
- assertTrue(has);
|
|
|
+ assertEquals(asSet(1L, 2L), result);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void testHasAnyPermissions_normal() {
|
|
|
- // 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- String[] roles = new String[]{"system:user:query", "system:user:create"};
|
|
|
- // mock 用户与角色的缓存
|
|
|
- permissionService.setUserRoleCache(MapUtil.<Long, Set<Long>>builder().put(1L, asSet(100L)).build());
|
|
|
- RoleDO role = randomPojo(RoleDO.class, o -> o.setId(100L)
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleFromCache(eq(100L))).thenReturn(role);
|
|
|
- // mock 其它方法
|
|
|
- MenuDO menu = randomPojo(MenuDO.class, o -> o.setId(1000L));
|
|
|
- when(menuService.getMenuListByPermissionFromCache(eq("system:user:create"))).thenReturn(singletonList(menu));
|
|
|
- permissionService.setMenuRoleCache(ImmutableMultimap.<Long, Long>builder().put(1000L, 100L).build());
|
|
|
-
|
|
|
-
|
|
|
- // 调用
|
|
|
- boolean has = permissionService.hasAnyPermissions(userId, roles);
|
|
|
- // 断言
|
|
|
- assertTrue(has);
|
|
|
+ public void testGetEnableUserRoleListByUserIdFromCache() {
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(PermissionServiceImpl.class)))
|
|
|
+ .thenReturn(permissionService);
|
|
|
+
|
|
|
+ // 准备参数
|
|
|
+ Long userId = 1L;
|
|
|
+ // mock 用户登录的角色
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(100L));
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(200L));
|
|
|
+ RoleDO role01 = randomPojo(RoleDO.class, o -> o.setId(100L)
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
+ RoleDO role02 = randomPojo(RoleDO.class, o -> o.setId(200L)
|
|
|
+ .setStatus(CommonStatusEnum.DISABLE.getStatus()));
|
|
|
+ when(roleService.getRoleListFromCache(eq(asSet(100L, 200L))))
|
|
|
+ .thenReturn(toList(role01, role02));
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ List<RoleDO> result = permissionService.getEnableUserRoleListByUserIdFromCache(userId);
|
|
|
+ // 断言
|
|
|
+ assertEquals(1, result.size());
|
|
|
+ assertPojoEquals(role01, result.get(0));
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
- @Test
|
|
|
- public void testHasAnyRoles_superAdmin() {
|
|
|
- // 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- String[] roles = new String[]{"yunai", "tudou"};
|
|
|
- // mock 用户与角色的缓存
|
|
|
- permissionService.setUserRoleCache(MapUtil.<Long, Set<Long>>builder().put(1L, asSet(100L)).build());
|
|
|
- RoleDO role = randomPojo(RoleDO.class, o -> o.setId(100L)
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleFromCache(eq(100L))).thenReturn(role);
|
|
|
- // mock 其它方法
|
|
|
- when(roleService.hasAnySuperAdmin(eq(asSet(100L)))).thenReturn(true);
|
|
|
-
|
|
|
- // 调用
|
|
|
- boolean has = permissionService.hasAnyRoles(userId, roles);
|
|
|
- // 断言
|
|
|
- assertTrue(has);
|
|
|
- }
|
|
|
+ // ========== 用户-部门的相关方法 ==========
|
|
|
|
|
|
@Test
|
|
|
- public void testHasAnyRoles_normal() {
|
|
|
+ public void testAssignRoleDataScope() {
|
|
|
// 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- String[] roles = new String[]{"yunai", "tudou"};
|
|
|
- // mock 用户与角色的缓存
|
|
|
- permissionService.setUserRoleCache(MapUtil.<Long, Set<Long>>builder().put(1L, asSet(100L)).build());
|
|
|
- RoleDO role = randomPojo(RoleDO.class, o -> o.setId(100L).setCode("yunai")
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleFromCache(eq(100L))).thenReturn(role);
|
|
|
- // mock 其它方法
|
|
|
- when(roleService.getRoleListFromCache(eq(asSet(100L)))).thenReturn(singletonList(role));
|
|
|
+ Long roleId = 1L;
|
|
|
+ Integer dataScope = 2;
|
|
|
+ Set<Long> dataScopeDeptIds = asSet(10L, 20L);
|
|
|
|
|
|
// 调用
|
|
|
- boolean has = permissionService.hasAnyRoles(userId, roles);
|
|
|
+ permissionService.assignRoleDataScope(roleId, dataScope, dataScopeDeptIds);
|
|
|
// 断言
|
|
|
- assertTrue(has);
|
|
|
+ verify(roleService).updateRoleDataScope(eq(roleId), eq(dataScope), eq(dataScopeDeptIds));
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void testGetDeptDataPermission_All() {
|
|
|
- // 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- // mock 用户的角色编号
|
|
|
- permissionService.setUserRoleCache(MapUtil.<Long, Set<Long>>builder().put(1L, asSet(2L)).build());
|
|
|
- // mock 获得用户的角色
|
|
|
- RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.ALL.getScope())
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(singletonList(roleDO));
|
|
|
- when(roleService.getRoleFromCache(eq(2L))).thenReturn(roleDO);
|
|
|
-
|
|
|
- // 调用
|
|
|
- DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
- // 断言
|
|
|
- assertTrue(result.getAll());
|
|
|
- assertFalse(result.getSelf());
|
|
|
- assertTrue(CollUtil.isEmpty(result.getDeptIds()));
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(PermissionServiceImpl.class)))
|
|
|
+ .thenReturn(permissionService);
|
|
|
+
|
|
|
+ // 准备参数
|
|
|
+ Long userId = 1L;
|
|
|
+ // mock 用户的角色编号
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(2L));
|
|
|
+ // mock 获得用户的角色
|
|
|
+ RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.ALL.getScope())
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
+ when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(toList(roleDO));
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
+ // 断言
|
|
|
+ assertTrue(result.getAll());
|
|
|
+ assertFalse(result.getSelf());
|
|
|
+ assertTrue(CollUtil.isEmpty(result.getDeptIds()));
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void testGetDeptDataPermission_DeptCustom() {
|
|
|
- // 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- // mock 用户的角色编号
|
|
|
- permissionService.setUserRoleCache(MapUtil.<Long, Set<Long>>builder().put(1L, asSet(2L)).build());
|
|
|
- // mock 获得用户的角色
|
|
|
- RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.DEPT_CUSTOM.getScope())
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(singletonList(roleDO));
|
|
|
- when(roleService.getRoleFromCache(eq(2L))).thenReturn(roleDO);
|
|
|
- // mock 部门的返回
|
|
|
- when(userService.getUser(eq(1L))).thenReturn(new AdminUserDO().setDeptId(3L), null, null); // 最后返回 null 的目的,看看会不会重复调用
|
|
|
-
|
|
|
- // 调用
|
|
|
- DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
- // 断言
|
|
|
- assertFalse(result.getAll());
|
|
|
- assertFalse(result.getSelf());
|
|
|
- assertEquals(roleDO.getDataScopeDeptIds().size() + 1, result.getDeptIds().size());
|
|
|
- assertTrue(CollUtil.containsAll(result.getDeptIds(), roleDO.getDataScopeDeptIds()));
|
|
|
- assertTrue(CollUtil.contains(result.getDeptIds(), 3L));
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(PermissionServiceImpl.class)))
|
|
|
+ .thenReturn(permissionService);
|
|
|
+
|
|
|
+ // 准备参数
|
|
|
+ Long userId = 1L;
|
|
|
+ // mock 用户的角色编号
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(2L));
|
|
|
+ // mock 获得用户的角色
|
|
|
+ RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.DEPT_CUSTOM.getScope())
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
+ when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(toList(roleDO));
|
|
|
+ // mock 部门的返回
|
|
|
+ when(userService.getUser(eq(1L))).thenReturn(new AdminUserDO().setDeptId(3L),
|
|
|
+ null, null); // 最后返回 null 的目的,看看会不会重复调用
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
+ // 断言
|
|
|
+ assertFalse(result.getAll());
|
|
|
+ assertFalse(result.getSelf());
|
|
|
+ assertEquals(roleDO.getDataScopeDeptIds().size() + 1, result.getDeptIds().size());
|
|
|
+ assertTrue(CollUtil.containsAll(result.getDeptIds(), roleDO.getDataScopeDeptIds()));
|
|
|
+ assertTrue(CollUtil.contains(result.getDeptIds(), 3L));
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void testGetDeptDataPermission_DeptOnly() {
|
|
|
- // 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- // mock 用户的角色编号
|
|
|
- permissionService.setUserRoleCache(MapUtil.<Long, Set<Long>>builder().put(1L, asSet(2L)).build());
|
|
|
- // mock 获得用户的角色
|
|
|
- RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.DEPT_ONLY.getScope())
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(singletonList(roleDO));
|
|
|
- when(roleService.getRoleFromCache(eq(2L))).thenReturn(roleDO);
|
|
|
- // mock 部门的返回
|
|
|
- when(userService.getUser(eq(1L))).thenReturn(new AdminUserDO().setDeptId(3L), null, null); // 最后返回 null 的目的,看看会不会重复调用
|
|
|
-
|
|
|
- // 调用
|
|
|
- DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
- // 断言
|
|
|
- assertFalse(result.getAll());
|
|
|
- assertFalse(result.getSelf());
|
|
|
- assertEquals(1, result.getDeptIds().size());
|
|
|
- assertTrue(CollUtil.contains(result.getDeptIds(), 3L));
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(PermissionServiceImpl.class)))
|
|
|
+ .thenReturn(permissionService);
|
|
|
+
|
|
|
+ // 准备参数
|
|
|
+ Long userId = 1L;
|
|
|
+ // mock 用户的角色编号
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(2L));
|
|
|
+ // mock 获得用户的角色
|
|
|
+ RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.DEPT_ONLY.getScope())
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
+ when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(toList(roleDO));
|
|
|
+ // mock 部门的返回
|
|
|
+ when(userService.getUser(eq(1L))).thenReturn(new AdminUserDO().setDeptId(3L),
|
|
|
+ null, null); // 最后返回 null 的目的,看看会不会重复调用
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
+ // 断言
|
|
|
+ assertFalse(result.getAll());
|
|
|
+ assertFalse(result.getSelf());
|
|
|
+ assertEquals(1, result.getDeptIds().size());
|
|
|
+ assertTrue(CollUtil.contains(result.getDeptIds(), 3L));
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void testGetDeptDataPermission_DeptAndChild() {
|
|
|
- // 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- // mock 用户的角色编号
|
|
|
- permissionService.setUserRoleCache(MapUtil.<Long, Set<Long>>builder().put(1L, asSet(2L)).build());
|
|
|
- // mock 获得用户的角色
|
|
|
- RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.DEPT_AND_CHILD.getScope())
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(singletonList(roleDO));
|
|
|
- when(roleService.getRoleFromCache(eq(2L))).thenReturn(roleDO);
|
|
|
- // mock 部门的返回
|
|
|
- when(userService.getUser(eq(1L))).thenReturn(new AdminUserDO().setDeptId(3L), null, null); // 最后返回 null 的目的,看看会不会重复调用
|
|
|
- // mock 方法(部门)
|
|
|
- DeptDO deptDO = randomPojo(DeptDO.class);
|
|
|
- when(deptService.getDeptListByParentIdFromCache(eq(3L), eq(true)))
|
|
|
- .thenReturn(singletonList(deptDO));
|
|
|
-
|
|
|
- // 调用
|
|
|
- DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
- // 断言
|
|
|
- assertFalse(result.getAll());
|
|
|
- assertFalse(result.getSelf());
|
|
|
- assertEquals(2, result.getDeptIds().size());
|
|
|
- assertTrue(CollUtil.contains(result.getDeptIds(), deptDO.getId()));
|
|
|
- assertTrue(CollUtil.contains(result.getDeptIds(), 3L));
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(PermissionServiceImpl.class)))
|
|
|
+ .thenReturn(permissionService);
|
|
|
+
|
|
|
+ // 准备参数
|
|
|
+ Long userId = 1L;
|
|
|
+ // mock 用户的角色编号
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(2L));
|
|
|
+ // mock 获得用户的角色
|
|
|
+ RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.DEPT_AND_CHILD.getScope())
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
+ when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(toList(roleDO));
|
|
|
+ // mock 部门的返回
|
|
|
+ when(userService.getUser(eq(1L))).thenReturn(new AdminUserDO().setDeptId(3L),
|
|
|
+ null, null); // 最后返回 null 的目的,看看会不会重复调用
|
|
|
+ // mock 方法(部门)
|
|
|
+ DeptDO deptDO = randomPojo(DeptDO.class);
|
|
|
+ when(deptService.getChildDeptIdListFromCache(eq(3L))).thenReturn(singleton(deptDO.getId()));
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
+ // 断言
|
|
|
+ assertFalse(result.getAll());
|
|
|
+ assertFalse(result.getSelf());
|
|
|
+ assertEquals(2, result.getDeptIds().size());
|
|
|
+ assertTrue(CollUtil.contains(result.getDeptIds(), deptDO.getId()));
|
|
|
+ assertTrue(CollUtil.contains(result.getDeptIds(), 3L));
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void testGetDeptDataPermission_Self() {
|
|
|
- // 准备参数
|
|
|
- Long userId = 1L;
|
|
|
- // mock 用户的角色编号
|
|
|
- permissionService.setUserRoleCache(MapUtil.<Long, Set<Long>>builder().put(1L, asSet(2L)).build());
|
|
|
- // mock 获得用户的角色
|
|
|
- RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.SELF.getScope())
|
|
|
- .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
- when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(singletonList(roleDO));
|
|
|
- when(roleService.getRoleFromCache(eq(2L))).thenReturn(roleDO);
|
|
|
-
|
|
|
- // 调用
|
|
|
- DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
- // 断言
|
|
|
- assertFalse(result.getAll());
|
|
|
- assertTrue(result.getSelf());
|
|
|
- assertTrue(CollUtil.isEmpty(result.getDeptIds()));
|
|
|
+ try (MockedStatic<SpringUtil> springUtilMockedStatic = mockStatic(SpringUtil.class)) {
|
|
|
+ springUtilMockedStatic.when(() -> SpringUtil.getBean(eq(PermissionServiceImpl.class)))
|
|
|
+ .thenReturn(permissionService);
|
|
|
+
|
|
|
+ // 准备参数
|
|
|
+ Long userId = 1L;
|
|
|
+ // mock 用户的角色编号
|
|
|
+ userRoleMapper.insert(randomPojo(UserRoleDO.class).setUserId(userId).setRoleId(2L));
|
|
|
+ // mock 获得用户的角色
|
|
|
+ RoleDO roleDO = randomPojo(RoleDO.class, o -> o.setDataScope(DataScopeEnum.SELF.getScope())
|
|
|
+ .setStatus(CommonStatusEnum.ENABLE.getStatus()));
|
|
|
+ when(roleService.getRoleListFromCache(eq(singleton(2L)))).thenReturn(toList(roleDO));
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ DeptDataPermissionRespDTO result = permissionService.getDeptDataPermission(userId);
|
|
|
+ // 断言
|
|
|
+ assertFalse(result.getAll());
|
|
|
+ assertTrue(result.getSelf());
|
|
|
+ assertTrue(CollUtil.isEmpty(result.getDeptIds()));
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
}
|