@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));
  }
Exemple #2
0
  @Test
  public void shouldReturnCountOfReportingRightOfUser() throws SQLException {
    Long userId = 1L;
    Right right1 = new Right("template1", RightType.REPORTING);
    Right right2 = new Right("template2", RightType.REPORTING);
    Right right3 = new Right("template3", RightType.REPORTING);
    Right right4 = new Right("admin", RightType.REPORTING);
    rightMapper.insertRight(right1);
    rightMapper.insertRight(right2);
    rightMapper.insertRight(right3);
    rightMapper.insertRight(right4);

    Role role = new Role();
    role.setName("New Role");
    roleRightsMapper.insertRole(role);
    queryExecutor.executeUpdate(
        "INSERT INTO role_rights(roleId, rightName) VALUES (?,?)", role.getId(), "template1");
    queryExecutor.executeUpdate(
        "INSERT INTO role_rights(roleId, rightName) VALUES (?,?)", role.getId(), "template2");
    queryExecutor.executeUpdate(
        "INSERT INTO role_rights(roleId, rightName) VALUES (?,?)", role.getId(), "template3");
    queryExecutor.executeUpdate(
        "INSERT INTO role_rights(roleId, rightName) VALUES (?,?)", role.getId(), "admin");

    queryExecutor.executeUpdate(
        "INSERT INTO role_assignments(userId, roleId) VALUES (?,?)", userId, role.getId());

    Integer count = rightMapper.totalReportingRightsFor(userId);

    assertThat(count, is(4));
  }
 @Test
 public void shouldUpdateRole() {
   role.setRights(new HashSet<>(asList(CREATE_REQUISITION)));
   roleRightsService.updateRole(role);
   verify(roleRightsRepository).updateRole(role);
 }
 @Test
 public void shouldSaveRole() throws Exception {
   role.setRights(new HashSet<>(asList(CREATE_REQUISITION, VIEW_REQUISITION)));
   roleRightsService.saveRole(role);
   verify(roleRightsRepository).createRole(role);
 }