Skip to content

Commit

Permalink
Merge branch 'develop' into OB-5382-add-tenant-name-and-id-to-consult…
Browse files Browse the repository at this point in the history
…ant-endpoint
  • Loading branch information
tkuzynow authored Sep 5, 2023
2 parents d09d0cf + 2c3a383 commit 7627140
Show file tree
Hide file tree
Showing 6 changed files with 96 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -486,6 +486,35 @@ public void updateRole(final String userId, final UserRole role) {
this.updateRole(userId, role.getValue());
}

@Override
public void removeRoleIfPresent(final String userId, final String roleName) {
// Get realm and user resources
var realmResource = keycloakClient.getRealmResource();
UsersResource userRessource = realmResource.users();
UserResource user = userRessource.get(userId);
// Remove role
var optionalRole = findRole(user, roleName);
if (optionalRole.isPresent()) {
RoleRepresentation roleRepresentation =
realmResource.roles().get(optionalRole.get()).toRepresentation();
if (roleRepresentation != null) {
user.roles().realmLevel().remove(Collections.singletonList(roleRepresentation));
}
}
}

Optional<String> findRole(UserResource user, String roleName) {

List<RoleRepresentation> userRoles = user.roles().realmLevel().listAll();
if (userRoles != null) {
return userRoles.stream()
.filter(role -> role.getName() != null && role.getName().equals(roleName))
.map(RoleRepresentation::getName)
.findFirst();
}
return Optional.empty();
}

/**
* Assigns the role with the given name to the given user ID.
*
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,11 @@ public Consultant updateConsultant(
&& updateConsultantDTO.getIsGroupchatConsultant()) {
identityClient.updateRole(consultant.getId(), GROUP_CHAT_CONSULTANT.getValue());
}
if ((updateConsultantDTO.getIsGroupchatConsultant() != null
&& !updateConsultantDTO.getIsGroupchatConsultant())
|| isNull(updateConsultantDTO.getIsGroupchatConsultant())) {
identityClient.removeRoleIfPresent(consultant.getId(), GROUP_CHAT_CONSULTANT.getValue());
}

this.rocketChatService.updateUser(
buildUserUpdateRequestDTO(consultant.getRocketChatId(), updateConsultantDTO));
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,8 @@ KeycloakCreateUserResponseDTO createKeycloakUser(

void updateRole(final String userId, final UserRole role);

void removeRoleIfPresent(final String userId, final String roleName);

void updateRole(final String userId, final String roleName);

void updatePassword(final String userId, final String password);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -573,6 +573,39 @@ public void updateRole_Should_updateRole_When_roleUpdateIsValid() {
verify(roleScopeResource, times(1)).add(any());
}

@Test
public void removeRole_Should_removeRole_When_rolePresent() {
String validRole = "role";

UserResource userResource = mock(UserResource.class);
UsersResource usersResource = mock(UsersResource.class);
when(usersResource.get(anyString())).thenReturn(userResource);
RoleScopeResource roleScopeResource = mock(RoleScopeResource.class);
RoleRepresentation keycloakRoleMock = mock(RoleRepresentation.class);
when(keycloakRoleMock.getName()).thenReturn(validRole);
when(roleScopeResource.listAll()).thenReturn(singletonList(keycloakRoleMock));
when(roleScopeResource.listAll()).thenReturn(singletonList(keycloakRoleMock));
RoleMappingResource roleMappingResource = mock(RoleMappingResource.class);
when(roleMappingResource.realmLevel()).thenReturn(roleScopeResource);
when(userResource.roles()).thenReturn(roleMappingResource);

RoleRepresentation roleRepresentation = new EasyRandom().nextObject(RoleRepresentation.class);
roleRepresentation.setName("role");
RoleResource roleResource = mock(RoleResource.class);
when(roleResource.toRepresentation()).thenReturn(roleRepresentation);
RolesResource rolesResource = mock(RolesResource.class);
when(rolesResource.get(any())).thenReturn(roleResource);

RealmResource realmResource = mock(RealmResource.class);
when(realmResource.users()).thenReturn(usersResource);
when(realmResource.roles()).thenReturn(rolesResource);
when(keycloakClient.getRealmResource()).thenReturn(realmResource);

this.keycloakService.removeRoleIfPresent("user", validRole);

verify(roleScopeResource, times(1)).remove(any());
}

@Test
public void updateRole_Should_updateUserWithProvidedRole() {
UserRole validRole = UserRole.USER;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -95,4 +95,28 @@ public void updateConsultant_Should_callServicesCorrectly_When_givenConsultantDa
verify(this.consultantService, times(1)).saveConsultant(any());
verify(this.appointmentService, times(1)).syncConsultantData(any());
}

@Test
public void
updateConsultant_Should_callServicesCorrectly_And_RemoveGroupChatConsultantRole_When_givenConsultantDataIsValidAndGroupChatFlagIsGiven() {
Consultant consultant = new EasyRandom().nextObject(Consultant.class);
when(this.consultantService.getConsultant(any())).thenReturn(Optional.of(consultant));
UpdateAdminConsultantDTO updateConsultant =
new EasyRandom().nextObject(UpdateAdminConsultantDTO.class);
updateConsultant.setIsGroupchatConsultant(false);

this.consultantUpdateService.updateConsultant("", updateConsultant);

verify(this.keycloakService)
.removeRoleIfPresent(consultant.getId(), UserRole.GROUP_CHAT_CONSULTANT.getValue());

verify(this.keycloakService, times(1))
.updateUserData(
eq(consultant.getId()),
any(UserDTO.class),
eq(updateConsultant.getFirstname()),
eq(updateConsultant.getLastname()));
verify(this.consultantService, times(1)).saveConsultant(any());
verify(this.appointmentService, times(1)).syncConsultantData(any());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -112,6 +112,9 @@ public void updateRole(String userId, UserRole role) {}
@Override
public void updateRole(String userId, String roleName) {}

@Override
public void removeRoleIfPresent(String userId, String roleName) {}

@Override
public void updatePassword(String userId, String password) {}

Expand Down

0 comments on commit 7627140

Please sign in to comment.