use of com.sequenceiq.freeipa.entity.UserSyncStatus in project cloudbreak by hortonworks.
the class UserSyncStatusServiceTest method testGetOrCreateForStackDontExists.
@Test
public void testGetOrCreateForStackDontExists() {
Stack stack = new Stack();
when(userSyncStatusRepository.getByStack(stack)).thenReturn(Optional.empty());
when(userSyncStatusRepository.save(any())).thenAnswer(invocation -> invocation.getArgument(0, UserSyncStatus.class));
UserSyncStatus result = underTest.getOrCreateForStack(stack);
ArgumentCaptor<UserSyncStatus> captor = ArgumentCaptor.forClass(UserSyncStatus.class);
verify(userSyncStatusRepository).save(captor.capture());
UserSyncStatus saved = captor.getValue();
assertEquals(saved, result);
assertEquals(stack, saved.getStack());
assertNotNull(saved.getUmsEventGenerationIds());
}
use of com.sequenceiq.freeipa.entity.UserSyncStatus in project cloudbreak by hortonworks.
the class UserSyncForEnvServiceTest method testSyncUsers.
@Test
public void testSyncUsers() {
Stack stack1 = mock(Stack.class);
when(stack1.getEnvironmentCrn()).thenReturn(ENV_CRN);
Stack stack2 = mock(Stack.class);
when(stack2.getEnvironmentCrn()).thenReturn(ENV_CRN_2);
UserSyncRequestFilter userSyncFilter = new UserSyncRequestFilter(Set.of(), Set.of(), Optional.empty());
UserSyncOptions options = new UserSyncOptions(true, true, WorkloadCredentialsUpdateType.UPDATE_IF_CHANGED);
doAnswer(inv -> {
inv.getArgument(2, Runnable.class).run();
return null;
}).when(operationService).tryWithOperationCleanup(eq(OPERATION_ID), eq(ACCOUNT_ID), any(Runnable.class));
UmsUsersState umsUsersState1 = mock(UmsUsersState.class);
UmsUsersState umsUsersState2 = mock(UmsUsersState.class);
when(umsUsersStateProviderDispatcher.getEnvToUmsUsersStateMap(eq(ACCOUNT_ID), eq(Set.of(ENV_CRN, ENV_CRN_2)), eq(Set.of()), eq(Set.of()), any())).thenReturn(Map.of(ENV_CRN, umsUsersState1, ENV_CRN_2, umsUsersState2));
when(asyncTaskExecutor.submit(any(Callable.class))).thenAnswer(inv -> {
SyncStatusDetail result = (SyncStatusDetail) inv.getArgument(0, Callable.class).call();
Future future = mock(Future.class);
when(future.get()).thenReturn(result);
return future;
});
when(umsEventGenerationIdsProvider.getEventGenerationIds(eq(ACCOUNT_ID), any(Optional.class))).thenReturn(new UmsEventGenerationIds());
when(userSyncForStackService.synchronizeStack(stack1, umsUsersState1, options)).thenReturn(new SyncStatusDetail(ENV_CRN, SynchronizationStatus.COMPLETED, "", ImmutableMultimap.of()));
when(userSyncForStackService.synchronizeStack(stack2, umsUsersState2, options)).thenReturn(new SyncStatusDetail(ENV_CRN_2, SynchronizationStatus.COMPLETED, "", ImmutableMultimap.of()));
when(userSyncStatusService.getOrCreateForStack(stack1)).thenReturn(new UserSyncStatus());
when(userSyncStatusService.getOrCreateForStack(stack2)).thenReturn(new UserSyncStatus());
underTest.synchronizeUsers(OPERATION_ID, ACCOUNT_ID, List.of(stack1, stack2), userSyncFilter, options, System.currentTimeMillis());
verify(umsVirtualGroupCreateService).createVirtualGroups(ACCOUNT_ID, List.of(stack1, stack2));
verify(userSyncStatusService, times(2)).save(any(UserSyncStatus.class));
ArgumentCaptor<Collection> successCaptor = ArgumentCaptor.forClass(Collection.class);
ArgumentCaptor<Collection> failureCaptor = ArgumentCaptor.forClass(Collection.class);
verify(operationService).completeOperation(eq(ACCOUNT_ID), eq(OPERATION_ID), successCaptor.capture(), failureCaptor.capture());
assertTrue(failureCaptor.getValue().isEmpty());
assertTrue(successCaptor.getValue().contains(new SuccessDetails(ENV_CRN)));
assertTrue(successCaptor.getValue().contains(new SuccessDetails(ENV_CRN_2)));
}
use of com.sequenceiq.freeipa.entity.UserSyncStatus in project cloudbreak by hortonworks.
the class UserSyncPollerTest method testSyncStackWhenNotInSync.
@Test
void testSyncStackWhenNotInSync() {
UserSyncStatus userSyncStatus = setupMocks();
when(regionAwareInternalCrnGenerator.getInternalCrnForServiceAsString()).thenReturn("crn:altus:iam:us-west-1:altus:user:__internal__actor__");
when(regionAwareInternalCrnGeneratorFactory.iam()).thenReturn(regionAwareInternalCrnGenerator);
when(eventGenerationIdsChecker.isInSync(eq(userSyncStatus), any())).thenReturn(false);
when(cooldownChecker.isCooldownExpired(eq(userSyncStatus), any())).thenReturn(true);
underTest.syncAllFreeIpaStacks();
verify(userSyncService).synchronizeUsers(UserSyncTestUtils.ACCOUNT_ID, "crn:altus:iam:us-west-1:altus:user:__internal__actor__", Set.of(UserSyncTestUtils.ENVIRONMENT_CRN), Set.of(), Set.of(), WorkloadCredentialsUpdateType.UPDATE_IF_CHANGED);
}
use of com.sequenceiq.freeipa.entity.UserSyncStatus in project cloudbreak by hortonworks.
the class UserSyncPollerTest method testDontSyncStackWhenInSync.
@Test
void testDontSyncStackWhenInSync() {
UserSyncStatus userSyncStatus = setupMocks();
when(regionAwareInternalCrnGenerator.getInternalCrnForServiceAsString()).thenReturn("crn:altus:iam:us-west-1:altus:user:__internal__actor__");
when(regionAwareInternalCrnGeneratorFactory.iam()).thenReturn(regionAwareInternalCrnGenerator);
when(eventGenerationIdsChecker.isInSync(eq(userSyncStatus), any())).thenReturn(true);
underTest.syncAllFreeIpaStacks();
verify(userSyncService, times(0)).synchronizeUsers(any(), any(), any(), any(), any(), any());
}
use of com.sequenceiq.freeipa.entity.UserSyncStatus in project cloudbreak by hortonworks.
the class FreeIpaDescribeService method getResponseForStack.
private DescribeFreeIpaResponse getResponseForStack(Stack stack, Boolean includeAllInstances) {
MDCBuilder.buildMdcContext(stack);
ImageEntity image = imageService.getByStack(stack);
FreeIpa freeIpa = freeIpaService.findByStackId(stack.getId());
Optional<UserSyncStatus> userSyncStatus = userSyncStatusService.findByStack(stack);
return stackToDescribeFreeIpaResponseConverter.convert(stack, image, freeIpa, userSyncStatus, includeAllInstances);
}
Aggregations