Examples with UmsEventGenerationIds com.sequenceiq.freeipa.service.freeipa.user.model.UmsEventGenerationIds used on opensource projects

Example 11 with UmsEventGenerationIds

use of com.sequenceiq.freeipa.service.freeipa.user.model.UmsEventGenerationIds in project cloudbreak by hortonworks.

the class UserSyncForEnvServiceTest method testSyncUsersDoesntTimeout.

@Test
public void testSyncUsersDoesntTimeout() {
    ReflectionTestUtils.setField(underTest, "operationTimeout", 0L);
    Stack stack1 = mock(Stack.class);
    when(stack1.getEnvironmentCrn()).thenReturn(ENV_CRN);
    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);
    when(umsUsersStateProviderDispatcher.getEnvToUmsUsersStateMap(eq(ACCOUNT_ID), eq(Set.of(ENV_CRN)), eq(Set.of()), eq(Set.of()), any())).thenReturn(Map.of(ENV_CRN, umsUsersState1));
    when(userSyncForStackService.synchronizeStack(stack1, umsUsersState1, options)).thenReturn(new SyncStatusDetail(ENV_CRN, SynchronizationStatus.COMPLETED, "", ImmutableMultimap.of()));
    Future<SyncStatusDetail> future = mock(Future.class);
    when(asyncTaskExecutor.submit(any(Callable.class))).thenAnswer(inv -> {
        SyncStatusDetail result = (SyncStatusDetail) inv.getArgument(0, Callable.class).call();
        when(future.get(anyLong(), eq(TimeUnit.MILLISECONDS))).thenReturn(result);
        return future;
    });
    when(umsEventGenerationIdsProvider.getEventGenerationIds(eq(ACCOUNT_ID), any(Optional.class))).thenReturn(new UmsEventGenerationIds());
    when(userSyncStatusService.getOrCreateForStack(stack1)).thenReturn(new UserSyncStatus());
    when(entitlementService.isUserSyncThreadTimeoutEnabled(ACCOUNT_ID)).thenReturn(Boolean.TRUE);
    underTest.synchronizeUsers(OPERATION_ID, ACCOUNT_ID, List.of(stack1), userSyncFilter, options, System.currentTimeMillis());
    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());
    verify(future, never()).cancel(true);
    verify(future, never()).cancel(false);
    assertTrue(failureCaptor.getValue().isEmpty());
    assertTrue(successCaptor.getValue().contains(new SuccessDetails(ENV_CRN)));
}

Example 12 with UmsEventGenerationIds

use of com.sequenceiq.freeipa.service.freeipa.user.model.UmsEventGenerationIds in project cloudbreak by hortonworks.

the class EnvironmentUserSyncStateCalculatorTest method internalCalculateEnvironmentUserSyncStateLastSyncCompletedSuccessNotInSync.

@Test
void internalCalculateEnvironmentUserSyncStateLastSyncCompletedSuccessNotInSync() {
    UserSyncStatus userSyncStatus = new UserSyncStatus();
    Operation lastSync = new Operation();
    lastSync.setOperationId(UUID.randomUUID().toString());
    lastSync.setStatus(OperationState.COMPLETED);
    lastSync.setSuccessList(List.of(new SuccessDetails(ENVIRONMENT_CRN)));
    userSyncStatus.setLastStartedFullSync(lastSync);
    UmsEventGenerationIds current = new UmsEventGenerationIds();
    when(umsEventGenerationIdsProvider.getEventGenerationIds(eq(ACCOUNT_ID), any())).thenReturn(current);
    when(eventGenerationIdsChecker.isInSync(userSyncStatus, current)).thenReturn(false);
    EnvironmentUserSyncState result = underTest.internalCalculateEnvironmentUserSyncState(ACCOUNT_ID, ENVIRONMENT_CRN, Optional.of(userSyncStatus));
    assertEquals(UserSyncState.STALE, result.getState());
    assertEquals(lastSync.getOperationId(), result.getLastUserSyncOperationId());
}

Example 13 with UmsEventGenerationIds

use of com.sequenceiq.freeipa.service.freeipa.user.model.UmsEventGenerationIds in project cloudbreak by hortonworks.

the class UmsEventGenerationIdsProviderTest method testGetEventGenerationIds.

@Test
void testGetEventGenerationIds() {
    GetEventGenerationIdsResponse response = createGetEventGenerationIdsResponse();
    when(grpcUmsClient.getEventGenerationIds(any(), any(), any())).thenReturn(response);
    UmsEventGenerationIds umsEventGenerationIds = underTest.getEventGenerationIds(ACCOUNT_ID, Optional.of(UUID.randomUUID().toString()));
    for (EventMapping eventMapping : EventMapping.values()) {
        assertEquals(eventMapping.getConverter().apply(response.getEventGenerationIds()), umsEventGenerationIds.getEventGenerationIds().get(eventMapping.getEventName()));
    }
}

Example 14 with UmsEventGenerationIds

use of com.sequenceiq.freeipa.service.freeipa.user.model.UmsEventGenerationIds in project cloudbreak by hortonworks.

the class UserSyncStatusToUserSyncStatusResponseConverterTest method convert.

@Test
void convert() {
    String requestedId = UUID.randomUUID().toString();
    Operation requested = new Operation();
    requested.setOperationId(requestedId);
    String successfulId = UUID.randomUUID().toString();
    Operation successful = new Operation();
    successful.setOperationId(successfulId);
    Map<String, String> eventGenerationIdMap = Map.of("key1", "value1", "key2", "value2");
    UmsEventGenerationIds eventGenerationIds = new UmsEventGenerationIds();
    eventGenerationIds.setEventGenerationIds(eventGenerationIdMap);
    UserSyncStatus status = new UserSyncStatus();
    status.setLastStartedFullSync(requested);
    status.setLastSuccessfulFullSync(successful);
    status.setUmsEventGenerationIds(new Json(eventGenerationIds));
    UserSyncStatusResponse response = underTest.convert(status);
    assertThat(response).returns(requestedId, UserSyncStatusResponse::getLastStartedUserSyncId).returns(successfulId, UserSyncStatusResponse::getLastSuccessfulUserSyncId).returns(eventGenerationIdMap, UserSyncStatusResponse::getEventGenerationIds);
}

Example 15 with UmsEventGenerationIds

use of com.sequenceiq.freeipa.service.freeipa.user.model.UmsEventGenerationIds in project cloudbreak by hortonworks.

the class UserSyncStatusService method createNewUserSyncStatusForStack.

private UserSyncStatus createNewUserSyncStatusForStack(Stack stack) {
    UserSyncStatus userSyncStatus = new UserSyncStatus();
    userSyncStatus.setStack(stack);
    userSyncStatus.setUmsEventGenerationIds(new Json(new UmsEventGenerationIds()));
    return userSyncStatusRepository.save(userSyncStatus);
}