Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add test for creating launch templates with TagSpecifications option

Browse Source
This commit is contained in:
Don Kuntz 2019-08-19 17:58:19 -05:00
parent 743e5be4d3
commit a1aa087718
No known key found for this signature in database
GPG Key ID: EBA7AEF98657FB2E
2 changed files with 42 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
moto/ec2/responses/launch_templates.py
View File

@ -94,17 +94,25 @@ class LaunchTemplates(BaseResponse):
def create_launch_template(self): def create_launch_template(self):
name = self._get_param('LaunchTemplateName') name = self._get_param('LaunchTemplateName')
version_description = self._get_param('VersionDescription') version_description = self._get_param('VersionDescription')
tag_spec = self._get_param('TagSpecifications') tag_spec = self._parse_tag_specification("TagSpecification")
raw_template_data = self._get_dict_param('LaunchTemplateData.') raw_template_data = self._get_dict_param('LaunchTemplateData.')
parsed_template_data = parse_object(raw_template_data) parsed_template_data = parse_object(raw_template_data)
if tag_spec:
if 'TagSpecifications' not in parsed_template_data:
parsed_template_data['TagSpecifications'] = []
parsed_template_data['TagSpecifications'].extend(tag_spec)
if self.is_not_dryrun('CreateLaunchTemplate'): if self.is_not_dryrun('CreateLaunchTemplate'):
if tag_spec:
if 'TagSpecifications' not in parsed_template_data:
parsed_template_data['TagSpecifications'] = []
converted_tag_spec = []
for resource_type, tags in six.iteritems(tag_spec):
converted_tag_spec.append({
"ResourceType": resource_type,
"Tags": [{"Key": key, "Value": value} for key, value in six.iteritems(tags)],
})
parsed_template_data['TagSpecifications'].extend(converted_tag_spec)
template = self.ec2_backend.create_launch_template(name, version_description, parsed_template_data) template = self.ec2_backend.create_launch_template(name, version_description, parsed_template_data)
version = template.default_version() version = template.default_version()

29
tests/test_ec2/test_launch_templates.py
View File

@ -356,3 +356,32 @@ def test_describe_launch_templates_with_filters():
resp["LaunchTemplates"].should.have.length_of(1) resp["LaunchTemplates"].should.have.length_of(1)
resp["LaunchTemplates"][0]["LaunchTemplateName"].should.equal("no-tags") resp["LaunchTemplates"][0]["LaunchTemplateName"].should.equal("no-tags")
@mock_ec2
def test_create_launch_template_with_tag_spec():
cli = boto3.client("ec2", region_name="us-east-1")
cli.create_launch_template(
LaunchTemplateName="test-template",
LaunchTemplateData={"ImageId":"ami-abc123"},
TagSpecifications=[{
"ResourceType": "instance",
"Tags": [
{"Key": "key", "Value": "value"}
]
}],
)
resp = cli.describe_launch_template_versions(
LaunchTemplateName="test-template",
Versions=["1"])
version = resp["LaunchTemplateVersions"][0]
version["LaunchTemplateData"].should.have.key("TagSpecifications")
version["LaunchTemplateData"]["TagSpecifications"].should.have.length_of(1)
version["LaunchTemplateData"]["TagSpecifications"][0].should.equal({
"ResourceType": "instance",
"Tags": [
{"Key": "key", "Value": "value"}
]
})