Examples with InstanceGroup com.google.api.services.compute.model.InstanceGroup used on opensource projects

Example 1 with InstanceGroup

use of com.google.api.services.compute.model.InstanceGroup in project cloudbreak by hortonworks.

the class GcpInstanceResourceBuilderTest method addToInstanceGroupFailsAuth.

@Test
public void addToInstanceGroupFailsAuth() throws Exception {
    // GIVEN
    Group group = newGroupWithParams(ImmutableMap.of());
    List<CloudResource> buildableResources = builder.create(context, group.getInstances().get(0), privateId, authenticatedContext, group, image);
    List<CloudResource> resourcesWithGroup = buildableResources.stream().map(b -> CloudResource.builder().cloudResource(b).group(group.getName()).build()).collect(Collectors.toList());
    context.addComputeResources(0L, buildableResources);
    // WHEN
    when(compute.instances()).thenReturn(instances);
    when(instances.insert(anyString(), anyString(), any(Instance.class))).thenReturn(insert);
    when(insert.setPrettyPrint(anyBoolean())).thenReturn(insert);
    when(insert.execute()).thenReturn(operation);
    Operation addOperation = new Operation();
    addOperation.setName("operation");
    addOperation.setHttpErrorStatusCode(401);
    addOperation.setHttpErrorMessage("Not Authorized");
    addOperation.setError(new Operation.Error());
    CloudResource instanceGroup = CloudResource.builder().type(ResourceType.GCP_INSTANCE_GROUP).status(CommonStatus.CREATED).name(group.getName()).group(group.getName()).build();
    context.addGroupResources(group.getName(), Collections.singletonList(instanceGroup));
    when(compute.instanceGroups()).thenReturn(instanceGroups);
    when(instanceGroups.addInstances(anyString(), anyString(), anyString(), any())).thenReturn(addInstances);
    InstanceGroups.List list = mock(InstanceGroups.List.class);
    when(instanceGroups.list(anyString(), anyString())).thenReturn(list);
    InstanceGroupList instanceGroupList = new InstanceGroupList();
    instanceGroupList.setItems(singletonList(new InstanceGroup().setName(group.getName())));
    when(list.execute()).thenReturn(instanceGroupList);
    when(addInstances.execute()).thenReturn(addOperation);
    Assert.assertThrows("Not Authorized", GcpResourceException.class, () -> builder.build(context, group.getInstances().get(0), privateId, authenticatedContext, group, resourcesWithGroup, cloudStack));
    // THEN
    verify(compute).instances();
    verify(instances).insert(anyString(), anyString(), instanceArg.capture());
    assertNull(instanceArg.getValue().getHostname());
}

Example 2 with InstanceGroup

use of com.google.api.services.compute.model.InstanceGroup in project cloudbreak by hortonworks.

the class GcpInstanceResourceBuilderTest method addInstanceGroupFromUpscale.

@Test
public void addInstanceGroupFromUpscale() throws Exception {
    // GIVEN
    Group group = newGroupWithParams(ImmutableMap.of());
    List<CloudResource> buildableResources = builder.create(context, group.getInstances().get(0), privateId, authenticatedContext, group, image);
    context.addComputeResources(0L, buildableResources);
    // WHEN
    when(compute.instances()).thenReturn(instances);
    when(instances.insert(anyString(), anyString(), any(Instance.class))).thenReturn(insert);
    when(insert.setPrettyPrint(anyBoolean())).thenReturn(insert);
    when(insert.execute()).thenReturn(operation);
    Operation addOperation = new Operation();
    addOperation.setName("operation");
    CloudResource instanceGroup = CloudResource.builder().type(ResourceType.GCP_INSTANCE_GROUP).status(CommonStatus.CREATED).name(group.getName()).group(group.getName()).build();
    CloudResource instanceGroup2 = CloudResource.builder().type(ResourceType.GCP_INSTANCE_GROUP).status(CommonStatus.CREATED).name("gateway").build();
    CloudResource instanceGroup3 = CloudResource.builder().type(ResourceType.GCP_INSTANCE_GROUP).status(CommonStatus.CREATED).name("idbroker").build();
    CloudResource instanceGroup4 = CloudResource.builder().type(ResourceType.GCP_INSTANCE_GROUP).status(CommonStatus.CREATED).name("free-master0").build();
    context.addGroupResources(group.getName(), List.of(instanceGroup4, instanceGroup2, instanceGroup, instanceGroup3));
    when(compute.instanceGroups()).thenReturn(instanceGroups);
    ArgumentCaptor<String> groupName = ArgumentCaptor.forClass(String.class);
    when(instanceGroups.addInstances(anyString(), anyString(), groupName.capture(), any())).thenReturn(addInstances);
    InstanceGroups.List list = mock(InstanceGroups.List.class);
    when(instanceGroups.list(anyString(), anyString())).thenReturn(list);
    InstanceGroupList instanceGroupList = new InstanceGroupList();
    instanceGroupList.setItems(singletonList(new InstanceGroup().setName(group.getName())));
    when(list.execute()).thenReturn(instanceGroupList);
    when(addInstances.execute()).thenReturn(addOperation);
    builder.build(context, group.getInstances().get(0), privateId, authenticatedContext, group, buildableResources, cloudStack);
    // THEN
    verify(compute).instances();
    assertEquals("master", groupName.getValue());
    verify(instances).insert(anyString(), anyString(), instanceArg.capture());
    assertNull(instanceArg.getValue().getHostname());
}

Example 3 with InstanceGroup

use of com.google.api.services.compute.model.InstanceGroup in project cloudbreak by hortonworks.

the class GcpInstanceGroupResourceBuilder method build.

@Override
public CloudResource build(GcpContext context, AuthenticatedContext auth, Group group, Network network, Security security, CloudResource resource) throws Exception {
    LOGGER.info("Building GCP instancegroup {} for project {}", group.getName(), context.getProjectId());
    Insert insert = context.getCompute().instanceGroups().insert(context.getProjectId(), context.getLocation().getAvailabilityZone().value(), new InstanceGroup().setName(resource.getName()));
    return executeOperationalRequest(resource, insert);
}