package jnpf.dmsPermission;
|
|
import jnpf.base.UserInfo;
|
import jnpf.dmsEntity.permission.model.DmsSubjectSet;
|
import jnpf.permission.OrganizeApi;
|
import jnpf.permission.PositionApi;
|
import jnpf.permission.RoleRelationApi;
|
import jnpf.permission.entity.OrganizeEntity;
|
import jnpf.permission.entity.PositionEntity;
|
import jnpf.permission.entity.RoleRelationEntity;
|
import jnpf.permission.model.rolerelaiton.RoleRelationModel;
|
import org.junit.jupiter.api.Test;
|
import org.mockito.ArgumentCaptor;
|
|
import java.util.Arrays;
|
import java.util.Collections;
|
import java.util.LinkedHashSet;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.times;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
|
class DmsSubjectResolverTest {
|
|
@Test
|
void resolvesOrganizationAndPositionHierarchyInBatches() {
|
OrganizeApi organizeApi = mock(OrganizeApi.class);
|
PositionApi positionApi = mock(PositionApi.class);
|
RoleRelationApi roleRelationApi = mock(RoleRelationApi.class);
|
DmsSubjectResolver resolver = new DmsSubjectResolver(
|
new DmsSubjectHierarchyResolver(organizeApi, positionApi), roleRelationApi);
|
UserInfo user = new UserInfo();
|
user.setUserId("user-1");
|
user.setOrganizeIds(Arrays.asList(" org-child ", "org-second"));
|
user.setPositionIds(Collections.singletonList(" position-child "));
|
user.setRoleIds(Arrays.asList("role-direct", "role-inherited"));
|
user.setGroupIds(Collections.singletonList("platform-group-1"));
|
when(organizeApi.getListByIds(Arrays.asList("org-child", "org-second"))).thenReturn(Arrays.asList(
|
organization("org-child", "org-parent", "org-root,org-parent,org-child"),
|
organization("org-second", "0", "org-second")));
|
when(positionApi.getListByIds(Collections.singletonList("position-child"))).thenReturn(
|
Collections.singletonList(position("position-child", "position-parent",
|
"position-root,position-parent,position-child", "org-child")));
|
when(roleRelationApi.getListByObjectId(org.mockito.ArgumentMatchers.any(RoleRelationModel.class)))
|
.thenReturn(Arrays.asList(
|
role("user-1", "user", "role-direct"),
|
role("org-child", "organize", "role-1")));
|
|
DmsSubjectSet result = resolver.resolve(user);
|
|
assertEquals(set("org-child", "org-second"), result.getDirectOrgIds());
|
assertEquals(set("org-parent"), result.getImmediateParentOrgIds());
|
assertEquals(set("org-root", "org-parent", "org-child", "org-second"),
|
result.getAncestorOrgIds());
|
assertEquals(set("position-child"), result.getPositionIds());
|
assertEquals(set("position-parent"), result.getImmediateParentPositionIds());
|
assertEquals(set("position-root", "position-parent", "position-child"),
|
result.getAncestorPositionIds());
|
assertEquals(set("role-direct", "role-inherited"), result.getRoleIds());
|
assertEquals(set("role-direct"), result.getDirectUserRoleIds());
|
assertEquals(set("platform-group-1"), result.getPlatformGroupIds());
|
verify(organizeApi, times(1)).getListByIds(Arrays.asList("org-child", "org-second"));
|
verify(positionApi, times(1)).getListByIds(Collections.singletonList("position-child"));
|
ArgumentCaptor<RoleRelationModel> roleRequest = ArgumentCaptor.forClass(RoleRelationModel.class);
|
verify(roleRelationApi, times(1)).getListByObjectId(roleRequest.capture());
|
assertEquals(Collections.singletonList("user-1"), roleRequest.getValue().getObjectId());
|
assertEquals("user", roleRequest.getValue().getObjectType());
|
}
|
|
@Test
|
void preservesDirectSubjectsWhenHierarchyServicesFallBackToEmptyResults() {
|
OrganizeApi organizeApi = mock(OrganizeApi.class);
|
PositionApi positionApi = mock(PositionApi.class);
|
RoleRelationApi roleRelationApi = mock(RoleRelationApi.class);
|
DmsSubjectResolver resolver = new DmsSubjectResolver(
|
new DmsSubjectHierarchyResolver(organizeApi, positionApi), roleRelationApi);
|
UserInfo user = new UserInfo();
|
user.setUserId("user-1");
|
user.setOrganizeIds(Collections.singletonList("org-child"));
|
user.setPositionIds(Collections.singletonList("position-child"));
|
when(organizeApi.getListByIds(Collections.singletonList("org-child")))
|
.thenReturn(Collections.emptyList());
|
when(positionApi.getListByIds(Collections.singletonList("position-child")))
|
.thenReturn(Collections.emptyList());
|
|
DmsSubjectSet result = resolver.resolve(user);
|
|
assertEquals(set("org-child"), result.getDirectOrgIds());
|
assertEquals(set("org-child"), result.getAncestorOrgIds());
|
assertEquals(set("position-child"), result.getPositionIds());
|
assertEquals(set("position-child"), result.getAncestorPositionIds());
|
assertEquals(Collections.emptySet(), result.getImmediateParentOrgIds());
|
assertEquals(Collections.emptySet(), result.getImmediateParentPositionIds());
|
assertEquals(Collections.emptySet(), result.getDirectUserRoleIds());
|
}
|
|
private OrganizeEntity organization(String id, String parentId, String idTree) {
|
OrganizeEntity result = new OrganizeEntity();
|
result.setId(id);
|
result.setParentId(parentId);
|
result.setOrganizeIdTree(idTree);
|
return result;
|
}
|
|
private PositionEntity position(String id, String parentId, String idTree, String organizationId) {
|
PositionEntity result = new PositionEntity();
|
result.setId(id);
|
result.setParentId(parentId);
|
result.setPositionIdTree(idTree);
|
result.setOrganizeId(organizationId);
|
return result;
|
}
|
|
private RoleRelationEntity role(String objectId, String objectType, String roleId) {
|
RoleRelationEntity result = new RoleRelationEntity();
|
result.setObjectId(objectId);
|
result.setObjectType(objectType);
|
result.setRoleId(roleId);
|
return result;
|
}
|
|
private LinkedHashSet<String> set(String... values) {
|
return new LinkedHashSet<>(Arrays.asList(values));
|
}
|
}
|
