update

liujie

2026-08-14 92a75a7b157e6fd6423df17cfd32015017bdb8c9

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package com.tourguide.modules.system.service;
 
import com.tourguide.modules.system.entity.SystemMenu;
import com.tourguide.modules.system.entity.SystemRole;
import com.tourguide.modules.system.entity.SystemRoleMenu;
import com.tourguide.modules.system.mapper.SystemMenuMapper;
import com.tourguide.modules.system.mapper.SystemRoleMapper;
import com.tourguide.modules.system.mapper.SystemRoleMenuMapper;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
 
import java.util.Arrays;
import java.util.List;
 
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
 
class AdminPermissionServiceTest {
    private SystemRoleMapper roleMapper;
    private SystemMenuMapper menuMapper;
    private SystemRoleMenuMapper roleMenuMapper;
    private AdminPermissionService permissionService;
 
    @BeforeEach
    void setUp() {
        roleMapper = mock(SystemRoleMapper.class);
        menuMapper = mock(SystemMenuMapper.class);
        roleMenuMapper = mock(SystemRoleMenuMapper.class);
        permissionService = new AdminPermissionService(roleMapper, menuMapper, roleMenuMapper);
    }
 
    @Test
    void customRoleShouldOnlyReturnItsEnabledMenuCodes() {
        SystemRole role = role(4L, "CUSTOM_GUIDE");
        when(roleMapper.selectOne(any())).thenReturn(role);
        when(roleMenuMapper.selectList(any())).thenReturn(Arrays.asList(relation(10L), relation(11L)));
        when(menuMapper.selectList(any())).thenReturn(Arrays.asList(menu("GUIDES"), menu("GUIDE_LIST")));
 
        List<String> menuCodes = permissionService.getMenuCodes("CUSTOM_GUIDE");
 
        assertThat(menuCodes).containsExactly("GUIDES", "GUIDE_LIST");
    }
 
    @Test
    void superAdminShouldReceiveAllEnabledMenuCodesWithoutRoleRelations() {
        when(roleMapper.selectOne(any())).thenReturn(role(1L, "ADMIN"));
        when(menuMapper.selectList(any())).thenReturn(Arrays.asList(menu("DASHBOARD"), menu("EMPLOYEE_LIST")));
 
        List<String> menuCodes = permissionService.getMenuCodes("ADMIN");
 
        assertThat(menuCodes).containsExactly("DASHBOARD", "EMPLOYEE_LIST");
        verify(roleMenuMapper, never()).selectList(any());
    }
 
    private SystemRole role(Long id, String roleCode) {
        SystemRole role = new SystemRole();
        role.setId(id);
        role.setRoleCode(roleCode);
        role.setStatus(1);
        return role;
    }
 
    private SystemRoleMenu relation(Long menuId) {
        SystemRoleMenu relation = new SystemRoleMenu();
        relation.setMenuId(menuId);
        return relation;
    }
 
    private SystemMenu menu(String menuCode) {
        SystemMenu menu = new SystemMenu();
        menu.setMenuCode(menuCode);
        return menu;
    }
}