use of com.amazonaws.services.ec2.model.CreateLaunchTemplateVersionResult in project cloudbreak by hortonworks.
the class AwsLaunchTemplateUpdateServiceTest method testUpdateImageWithWrongTemplateParams.
@Test
public void testUpdateImageWithWrongTemplateParams() throws IOException {
// GIVEN
String cfStackName = "cf";
CloudResource cfResource = CloudResource.builder().type(ResourceType.CLOUDFORMATION_STACK).name(cfStackName).build();
String template = FileReaderUtils.readFileFromClasspath("json/aws-cf-template.json");
String cfTemplateBody = JsonUtil.minify(String.format(template, "{\"Ref\":\"AMI\"}"));
when(cloudFormationClient.getTemplate(any())).thenReturn(new GetTemplateResult().withTemplateBody(cfTemplateBody));
Map<AutoScalingGroup, String> autoScalingGroupsResult = createAutoScalingGroupHandler();
when(autoScalingGroupHandler.getAutoScalingGroups(cloudFormationClient, autoScalingClient, cfResource.getName())).thenReturn(autoScalingGroupsResult);
when(ec2Client.createLaunchTemplateVersion(any(CreateLaunchTemplateVersionRequest.class))).thenReturn(new CreateLaunchTemplateVersionResult().withWarning(new ValidationWarning().withErrors(new ValidationError().withCode("1").withMessage("error"))));
// WHEN and THEN exception
Assert.assertThrows(CloudConnectorException.class, () -> underTest.updateFields(ac, cfResource.getName(), Map.of(LaunchTemplateField.IMAGE_ID, stack.getImage().getImageName())));
}
use of com.amazonaws.services.ec2.model.CreateLaunchTemplateVersionResult in project cloudbreak by hortonworks.
the class AwsLaunchTemplateUpdateServiceTest method testUpdateFieldsUpdatesWithMissingParams.
@Test
public void testUpdateFieldsUpdatesWithMissingParams() throws IOException {
// GIVEN
String cfStackName = "cf";
CloudResource cfResource = CloudResource.builder().type(ResourceType.CLOUDFORMATION_STACK).name(cfStackName).build();
String template = FileReaderUtils.readFileFromClasspath("json/aws-cf-template.json");
String cfTemplateBody = JsonUtil.minify(String.format(template, "{\"Ref\":\"AMI\"}"));
when(cloudFormationClient.getTemplate(any())).thenReturn(new GetTemplateResult().withTemplateBody(cfTemplateBody));
Map<AutoScalingGroup, String> autoScalingGroupsResult = createAutoScalingGroupHandler();
when(autoScalingGroupHandler.getAutoScalingGroups(cloudFormationClient, autoScalingClient, cfResource.getName())).thenReturn(autoScalingGroupsResult);
Map<LaunchTemplateField, String> updatableFieldMap = Map.of(LaunchTemplateField.IMAGE_ID, stack.getImage().getImageName(), LaunchTemplateField.DESCRIPTION, DESCRIPTION);
when(ec2Client.createLaunchTemplateVersion(any(CreateLaunchTemplateVersionRequest.class))).thenReturn(new CreateLaunchTemplateVersionResult().withLaunchTemplateVersion(new LaunchTemplateVersion().withVersionNumber(1L)));
// WHEN
underTest.updateFields(ac, cfResource.getName(), updatableFieldMap);
// THEN
Mockito.verify(ec2Client).createLaunchTemplateVersion(argumentCaptor.capture());
CreateLaunchTemplateVersionRequest request = argumentCaptor.getValue();
Assertions.assertEquals(stack.getImage().getImageName(), request.getLaunchTemplateData().getImageId());
Assertions.assertEquals(DESCRIPTION, request.getVersionDescription());
Assertions.assertNull(request.getLaunchTemplateData().getUserData());
}
Aggregations