@Test
public void shouldGetAllRolesMapByRightType() throws Exception {
List<Role> allRoles = new ArrayList<>();
Role role1 = new Role();
role1.setId(1L);
Role role2 = new Role();
role2.setId(2L);
Role role3 = new Role();
role3.setId(3L);
Role role4 = new Role();
role4.setId(4L);
allRoles.add(role1);
allRoles.add(role2);
allRoles.add(role3);
allRoles.add(role4);
when(roleRightsRepository.getAllRoles()).thenReturn(allRoles);
when(roleRightsRepository.getRightTypeForRoleId(1L)).thenReturn(RightType.REQUISITION);
when(roleRightsRepository.getRightTypeForRoleId(2L)).thenReturn(RightType.ADMIN);
when(roleRightsRepository.getRightTypeForRoleId(3L)).thenReturn(RightType.ALLOCATION);
when(roleRightsRepository.getRightTypeForRoleId(4L)).thenReturn(RightType.ALLOCATION);
Map<String, List<Role>> allRolesMap = roleRightsService.getAllRolesMap();
assertThat(allRolesMap.size(), is(3));
assertThat(allRolesMap.get(RightType.ADMIN.name()).size(), is(1));
assertThat(allRolesMap.get(RightType.REQUISITION.name()).size(), is(1));
assertThat(allRolesMap.get(RightType.ALLOCATION.name()).size(), is(2));
}
@SuppressWarnings("unchecked")
private Role parseRole(Element element) throws RESTLightClientException {
Role role = new Role();
role.setResourceURI(element.getChildText(RESOURCE_URI_ELEMENT));
role.setId(element.getChildText(ROLE_ID_ELEMENT));
role.setName(element.getChildText(ROLE_NAME_ELEMENT));
role.setDescription(element.getChildText(ROLE_DESCRIPTION_ELEMENT));
role.setSessionTimeout(Integer.parseInt(element.getChildText(ROLE_SESSION_TIMEOUT_ELEMENT)));
Element subRoles = element.getChild(ROLE_ROLES_ELEMENT);
if (subRoles != null) {
for (Element subRole : (List<Element>) subRoles.getChildren(ROLE_ROLE_ELEMENT)) {
role.getRoles().add(subRole.getValue());
}
}
Element privileges = element.getChild(ROLE_PRIVILEGES_ELEMENT);
if (privileges != null) {
for (Element privilege : (List<Element>) privileges.getChildren(ROLE_PRIVILEGE_ELEMENT)) {
role.getPrivileges().add(privilege.getValue());
}
}
role.setUserManaged(
element.getChildText(ROLE_USER_MANAGED_ELEMENT).equals("true") ? true : false);
return role;
}
public void setRoleIdList(List<String> roleIdList) {
roleList = Lists.newArrayList();
for (String roleId : roleIdList) {
Role role = new Role();
role.setId(roleId);
roleList.add(role);
}
}
protected RoleList defaultRoleList() {
// Having an empty role list is not valid, they should always have one role
Role role = new Role();
role.setName("derpRole");
role.setId("9");
role.setDescription("Derp description");
RoleList roleList = new RoleList();
roleList.getRole().add(role);
return roleList;
}
private static AuthenticateResponse getServiceResponse() {
AuthenticateResponse rsp = new AuthenticateResponse();
Token token = new Token();
token.setId("tokenid");
GregorianCalendar cal = new GregorianCalendar(2013, 11, 12);
token.setExpires(new XMLGregorianCalendarImpl(cal));
TenantForAuthenticateResponse tenantForAuthenticateResponse =
new TenantForAuthenticateResponse();
tenantForAuthenticateResponse.setId("tenantId");
tenantForAuthenticateResponse.setName("tenantName");
token.setTenant(tenantForAuthenticateResponse);
rsp.setToken(token);
ServiceCatalog catalog = new ServiceCatalog();
List<ServiceForCatalog> serviceCatalogList = new ArrayList<>();
ServiceForCatalog serviceForCatalog = new ServiceForCatalog();
serviceForCatalog.setName("catName");
serviceForCatalog.setType("type");
serviceCatalogList.add(serviceForCatalog);
catalog.getService().addAll(serviceCatalogList);
rsp.setServiceCatalog(catalog);
UserForAuthenticateResponse user = new UserForAuthenticateResponse();
user.setId("userId");
user.setName("userName");
RoleList roles = new RoleList();
Role role = new Role();
role.setDescription("role description");
role.setId("roleId");
role.setName("roleName");
role.setServiceId("serviceId");
role.setTenantId("roleTenantId");
roles.getRole().add(role);
user.setRoles(roles);
rsp.setUser(user);
return rsp;
}
/**
* Create a new package based on an existing template.
*
* @param packageId
* @param aPackage
* @return PackageId
*/
public PackageId createPackageFromTemplate(PackageId packageId, Package aPackage) {
String path =
template
.urlFor(UrlTemplate.TEMPLATE_PATH)
.replace("{packageId}", packageId.getId())
.build();
List<Role> roles = aPackage.getRoles();
aPackage.setRoles(Collections.<Role>emptyList());
String packageJson = Serialization.toJson(aPackage);
PackageId newPackageId = null;
try {
String response = client.post(path, packageJson);
newPackageId = Serialization.fromJson(response, PackageId.class);
} catch (RequestException e) {
throw new EslServerException("Could not create a new package", e);
} catch (Exception e) {
throw new EslException("Could not create a new package", e);
}
Package createdPackage = getApiPackage(newPackageId.getId());
for (Role role : roles) {
String roleUid = findRoleUidByName(createdPackage.getRoles(), role.getName());
if (roleUid == null) {
continue;
}
role.setId(roleUid);
updateRole(newPackageId, role);
}
return newPackageId;
}
