use of com.cloudera.api.swagger.model.ApiHostTemplateList in project cloudbreak by hortonworks.
the class ClouderaManagerDecommisionerTest method testCollectDownscaleCandidatesWhenOneHostDoesNotHaveFQDN.
@Test
public void testCollectDownscaleCandidatesWhenOneHostDoesNotHaveFQDN() throws ApiException {
// GIVEN
VolumeSetAttributes volumeSetAttributes = new VolumeSetAttributes("az", false, "fstab", List.of(), 50, "vt");
Stack stack = createTestStack(volumeSetAttributes);
Cluster cluster = new Cluster();
stack.setCluster(cluster);
Set<HostGroup> hostGroups = createTestHostGroups(1, 6);
cluster.setHostGroups(hostGroups);
HostGroup downscaledHostGroup = hostGroups.iterator().next();
Optional<InstanceMetaData> hgHost2 = downscaledHostGroup.getInstanceGroup().getAllInstanceMetaData().stream().filter(instanceMetaData -> instanceMetaData.getDiscoveryFQDN().equals("hg0-host-2")).findFirst();
hgHost2.ifPresent(instanceMetaData -> instanceMetaData.setDiscoveryFQDN(null));
HostsResourceApi hostsResourceApi = mock(HostsResourceApi.class);
when(clouderaManagerApiFactory.getHostsResourceApi(any(ApiClient.class))).thenReturn(hostsResourceApi);
HostTemplatesResourceApi hostTemplatesResourceApi = mock(HostTemplatesResourceApi.class);
when(clouderaManagerApiFactory.getHostTemplatesResourceApi(any(ApiClient.class))).thenReturn(hostTemplatesResourceApi);
ApiHostTemplateList hostTemplates = createEmptyHostTemplates();
when(hostTemplatesResourceApi.readHostTemplates(stack.getName())).thenReturn(hostTemplates);
ApiHostList apiHostRefList = new ApiHostList();
List<ApiHost> apiHosts = new ArrayList<>();
hostGroups.stream().flatMap(hostGroup -> hostGroup.getInstanceGroup().getAllInstanceMetaData().stream()).map(InstanceMetaData::getDiscoveryFQDN).forEach(hostName -> {
ApiHost apiHostRef = new ApiHost();
apiHostRef.setHostname(hostName);
apiHostRef.setHostId(hostName);
apiHostRef.setHealthSummary(ApiHealthSummary.GOOD);
apiHosts.add(apiHostRef);
});
apiHostRefList.setItems(apiHosts);
when(hostsResourceApi.readHosts(any(), any(), any())).thenReturn(apiHostRefList);
Set<InstanceMetaData> downscaleCandidates = underTest.collectDownscaleCandidates(mock(ApiClient.class), stack, downscaledHostGroup, -2, downscaledHostGroup.getInstanceGroup().getAllInstanceMetaData());
assertEquals(2, downscaleCandidates.size());
assertTrue("Assert if downscaleCandidates contains hg0-host-2, because FQDN is missing", downscaleCandidates.stream().anyMatch(instanceMetaData -> "hg0-instanceid-2".equals(instanceMetaData.getInstanceId())));
assertTrue("Assert if downscaleCandidates contains hg0-host-5", downscaleCandidates.stream().anyMatch(instanceMetaData -> "hg0-host-5".equals(instanceMetaData.getDiscoveryFQDN())));
}
use of com.cloudera.api.swagger.model.ApiHostTemplateList in project cloudbreak by hortonworks.
the class ClouderaManagerDecomissionerTest method testDecommissionForNodesNowKnownByCM.
@Test
public void testDecommissionForNodesNowKnownByCM() throws ApiException {
HostTemplatesResourceApi hostTemplatesResourceApi = mock(HostTemplatesResourceApi.class);
ApiHostTemplateList apiHostTemplateList = new ApiHostTemplateList();
apiHostTemplateList.setItems(new ArrayList<>());
when(hostTemplatesResourceApi.readHostTemplates(any())).thenReturn(apiHostTemplateList);
when(clouderaManagerApiFactory.getHostTemplatesResourceApi(client)).thenReturn(hostTemplatesResourceApi);
HostGroup compute = new HostGroup();
compute.setName("compute");
HostsResourceApi hostsResourceApi = mock(HostsResourceApi.class);
when(clouderaManagerApiFactory.getHostsResourceApi(client)).thenReturn(hostsResourceApi);
ApiHostList apiHostList = new ApiHostList();
apiHostList.addItemsItem(createApiHostRef("host1.example.com"));
apiHostList.addItemsItem(createApiHostRef("host2.example.com", ApiHealthSummary.BAD));
apiHostList.addItemsItem(createApiHostRef("host5.example.com"));
when(hostsResourceApi.readHosts(any(), any(), any())).thenReturn(apiHostList);
InstanceMetaData healthy1 = createInstanceMetadata(InstanceStatus.SERVICES_HEALTHY, "host1.example.com", "compute");
InstanceMetaData bad1 = createInstanceMetadata(InstanceStatus.SERVICES_HEALTHY, "host2.example.com", "compute");
InstanceMetaData failed1 = createInstanceMetadata(InstanceStatus.ORCHESTRATION_FAILED, "host3.example.com", "compute");
InstanceMetaData failed2 = createInstanceMetadata(InstanceStatus.ORCHESTRATION_FAILED, "host4.example.com", "compute");
InstanceMetaData healthy2 = createInstanceMetadata(InstanceStatus.SERVICES_HEALTHY, "host5.example.com", "compute");
Set<InstanceMetaData> instanceMetaDataSet = Set.of(failed1, failed2, healthy1, healthy2, bad1);
Stack stack = getStack();
stack.setPlatformVariant("AWS");
Set<InstanceMetaData> removableInstances = underTest.collectDownscaleCandidates(client, stack, compute, 4, instanceMetaDataSet);
assertEquals(4, removableInstances.size());
assertTrue(removableInstances.contains(failed1));
assertTrue(removableInstances.contains(failed2));
assertTrue(removableInstances.contains(healthy2));
assertTrue(removableInstances.contains(bad1));
}
use of com.cloudera.api.swagger.model.ApiHostTemplateList in project cloudbreak by hortonworks.
the class ClouderaManagerDecommisionerTest method createHostTemplatesWithDataNodes.
private ApiHostTemplateList createHostTemplatesWithDataNodes(String name) {
ApiHostTemplateList apiHostTemplateList = new ApiHostTemplateList();
ApiHostTemplate apiHostTemplate = new ApiHostTemplate();
apiHostTemplate.setName(name);
ApiRoleConfigGroupRef roleConfigGroupRef = new ApiRoleConfigGroupRef();
roleConfigGroupRef.setRoleConfigGroupName("_DATANODE_");
apiHostTemplate.setRoleConfigGroupRefs(List.of(roleConfigGroupRef));
apiHostTemplateList.setItems(List.of(apiHostTemplate));
return apiHostTemplateList;
}
use of com.cloudera.api.swagger.model.ApiHostTemplateList in project cloudbreak by hortonworks.
the class ClouderaManagerDecommisionerTest method createEmptyHostTemplates.
private ApiHostTemplateList createEmptyHostTemplates() {
ApiHostTemplateList apiHostTemplateList = new ApiHostTemplateList();
apiHostTemplateList.setItems(List.of());
return apiHostTemplateList;
}
use of com.cloudera.api.swagger.model.ApiHostTemplateList in project cloudbreak by hortonworks.
the class ClouderaManagerDecommisionerTest method testVerifyNodesAreRemovableWithRepairAndReplication.
@Test
public void testVerifyNodesAreRemovableWithRepairAndReplication() throws ApiException {
// GIVEN
VolumeSetAttributes volumeSetAttributes = new VolumeSetAttributes("az", false, "fstab", List.of(), 50, "vt");
Stack stack = createTestStack(volumeSetAttributes);
Cluster cluster = new Cluster();
stack.setCluster(cluster);
Set<HostGroup> hostGroups = createTestHostGroups(2, 5);
cluster.setHostGroups(hostGroups);
// Multimap<Long, InstanceMetaData> hostGroupWithInstances = createTestHostGroupWithInstances();
ApiServiceConfig apiServiceConfig = createApiServiceConfigWithReplication("3", true);
ApiHostTemplateList hostTemplates = createHostTemplatesWithDataNodes(hostGroups.stream().findFirst().get().getName());
when(clouderaManagerApiFactory.getHostTemplatesResourceApi(Mockito.any(ApiClient.class))).thenReturn(hostTemplatesResourceApi);
when(hostTemplatesResourceApi.readHostTemplates(stack.getName())).thenReturn(hostTemplates);
when(resourceAttributeUtil.getTypedAttributes(stack.getDiskResources().get(0), VolumeSetAttributes.class)).thenReturn(Optional.of(volumeSetAttributes));
when(clouderaManagerApiFactory.getServicesResourceApi(Mockito.any(ApiClient.class))).thenReturn(servicesResourceApi);
when(servicesResourceApi.readServiceConfig(stack.getName(), "hdfs", "full")).thenReturn(apiServiceConfig);
when(clouderaManagerApiFactory.getHostsResourceApi(Mockito.any(ApiClient.class))).thenReturn(hostsResourceApi);
when(hostsResourceApi.readHosts(eq(null), eq(null), anyString())).thenReturn(apiHostList);
when(apiHostList.getItems()).thenReturn(List.of());
// WHEN
HostGroup firstHostGroup = hostGroups.iterator().next();
Set<InstanceMetaData> firstHostGroupInstances = firstHostGroup.getInstanceGroup().getInstanceMetaDataSet();
Set<InstanceMetaData> removableInstances = firstHostGroupInstances.stream().limit(5).collect(Collectors.toSet());
underTest.verifyNodesAreRemovable(stack, removableInstances, new ApiClient());
// THEN no exception
}
Aggregations