use of com.sequenceiq.common.api.telemetry.model.Telemetry in project cloudbreak by hortonworks.
the class TelemetryDecoratorTest method testDecorateWithDatabus.
@Test
public void testDecorateWithDatabus() {
// GIVEN
Map<String, SaltPillarProperties> servicePillar = new HashMap<>();
FluentConfigView fluentConfigView = new FluentConfigView.Builder().build();
DatabusConfigView dataConfigView = new DatabusConfigView.Builder().withEnabled(true).withAccessKeyId("myAccessKeyId").withAccessKeySecret("mySecret".toCharArray()).withEndpoint("databusEndpoint").build();
mockConfigServiceResults(dataConfigView, fluentConfigView, new MeteringConfigView.Builder().build());
// WHEN
Map<String, SaltPillarProperties> result = underTest.decoratePillar(servicePillar, createStack(), new Telemetry());
// THEN
Map<String, Object> results = createMapFromFluentPillars(result, "databus");
assertEquals(results.get("accessKeyId"), "myAccessKeyId");
assertEquals(results.get("enabled"), true);
}
use of com.sequenceiq.common.api.telemetry.model.Telemetry in project cloudbreak by hortonworks.
the class TelemetryServiceTest method testCreateTelemetryConfigs.
@Test
public void testCreateTelemetryConfigs() {
// GIVEN
given(stackService.getById(STACK_ID)).willReturn(stack("{}"));
given(componentConfigProviderService.getTelemetry(STACK_ID)).willReturn(new Telemetry());
given(telemetryDecorator.decoratePillar(anyMap(), any(), any(), any())).willReturn(new HashMap<>());
// WHEN
underTest.createTelemetryConfigs(STACK_ID, Set.of(TelemetryComponentType.CDP_TELEMETRY));
// THEN
verify(telemetryDecorator, times(1)).decoratePillar(anyMap(), any(), any(), isNotNull());
}
use of com.sequenceiq.common.api.telemetry.model.Telemetry in project cloudbreak by hortonworks.
the class TelemetryServiceTest method testCreateTelemetryConfigsWithIOException.
@Test
public void testCreateTelemetryConfigsWithIOException() {
// GIVEN
given(stackService.getById(STACK_ID)).willReturn(stack("wrongJson"));
given(componentConfigProviderService.getTelemetry(STACK_ID)).willReturn(new Telemetry());
given(telemetryDecorator.decoratePillar(anyMap(), any(), any(), any())).willReturn(new HashMap<>());
// WHEN
underTest.createTelemetryConfigs(STACK_ID, Set.of(TelemetryComponentType.CDP_TELEMETRY));
// THEN
verify(telemetryDecorator, times(1)).decoratePillar(anyMap(), any(), any(), isNull());
}
use of com.sequenceiq.common.api.telemetry.model.Telemetry in project cloudbreak by hortonworks.
the class MachineUserRemoveHandler method cleanupMachineUser.
private void cleanupMachineUser(Long stackId) {
Stack stack = stackService.getStackById(stackId);
Telemetry telemetry = stack.getTelemetry();
if (telemetry != null && (telemetry.isClusterLogsCollectionEnabled() || StringUtils.isNotBlank(stack.getDatabusCredential()))) {
ThreadBasedUserCrnProvider.doAsInternalActor(regionAwareInternalCrnGeneratorFactory.iam().getInternalCrnForServiceAsString(), () -> altusMachineUserService.cleanupMachineUser(stack, telemetry));
} else {
LOGGER.info("Machine user cleanup is not needed.");
}
}
use of com.sequenceiq.common.api.telemetry.model.Telemetry in project cloudbreak by hortonworks.
the class DiagnosticsCollectionValidatorTest method testValidateWithValidCloudStorage.
@Test
void testValidateWithValidCloudStorage() {
BaseDiagnosticsCollectionRequest request = new BaseDiagnosticsCollectionRequest();
request.setDestination(DiagnosticsDestination.CLOUD_STORAGE);
Telemetry telemetry = new Telemetry();
Logging logging = new Logging();
logging.setS3(new S3CloudStorageV1Parameters());
telemetry.setLogging(logging);
underTest.validate(request, createStack(), telemetry);
}
Aggregations