Source code for heat_integrationtests.functional.test_aws_stack

#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

import hashlib
import json
import random

from six.moves.urllib import parse
from swiftclient import utils as swiftclient_utils
import yaml

from heat_integrationtests.common import test
from heat_integrationtests.functional import functional_base


[docs]class AwsStackTest(functional_base.FunctionalTestsBase): test_template = ''' HeatTemplateFormatVersion: '2012-12-12' Resources: the_nested: Type: AWS::CloudFormation::Stack Properties: TemplateURL: the.yaml Parameters: KeyName: foo Outputs: output_foo: Value: {"Fn::GetAtt": [the_nested, Outputs.Foo]} ''' nested_template = ''' HeatTemplateFormatVersion: '2012-12-12' Parameters: KeyName: Type: String Outputs: Foo: Value: bar ''' update_template = ''' HeatTemplateFormatVersion: '2012-12-12' Parameters: KeyName: Type: String Outputs: Foo: Value: foo ''' nested_with_res_template = ''' HeatTemplateFormatVersion: '2012-12-12' Parameters: KeyName: Type: String Resources: NestedResource: Type: OS::Heat::RandomString Outputs: Foo: Value: {"Fn::GetAtt": [NestedResource, value]} '''
[docs] def setUp(self): super(AwsStackTest, self).setUp() self.object_container_name = test.rand_name() self.project_id = self.identity_client.project_id self.swift_key = hashlib.sha224( str(random.getrandbits(256))).hexdigest()[:32] key_header = 'x-container-meta-temp-url-key' self.object_client.put_container(self.object_container_name, {key_header: self.swift_key}) self.addCleanup(self.object_client.delete_container, self.object_container_name)
[docs] def publish_template(self, contents, cleanup=True): oc = self.object_client # post the object oc.put_object(self.object_container_name, 'template.yaml', contents) if cleanup: self.addCleanup(oc.delete_object, self.object_container_name, 'template.yaml') path = '/v1/AUTH_%s/%s/%s' % (self.project_id, self.object_container_name, 'template.yaml') timeout = self.conf.build_timeout * 10 tempurl = swiftclient_utils.generate_temp_url(path, timeout, self.swift_key, 'GET') sw_url = parse.urlparse(oc.url) return '%s://%s%s' % (sw_url.scheme, sw_url.netloc, tempurl)
[docs] def test_nested_stack_create(self): url = self.publish_template(self.nested_template) self.template = self.test_template.replace('the.yaml', url) stack_identifier = self.stack_create(template=self.template) stack = self.client.stacks.get(stack_identifier) self.assert_resource_is_a_stack(stack_identifier, 'the_nested') self.assertEqual('bar', self._stack_output(stack, 'output_foo'))
[docs] def test_nested_stack_create_with_timeout(self): url = self.publish_template(self.nested_template) self.template = self.test_template.replace('the.yaml', url) timeout_template = yaml.safe_load(self.template) props = timeout_template['Resources']['the_nested']['Properties'] props['TimeoutInMinutes'] = '50' stack_identifier = self.stack_create( template=timeout_template) stack = self.client.stacks.get(stack_identifier) self.assert_resource_is_a_stack(stack_identifier, 'the_nested') self.assertEqual('bar', self._stack_output(stack, 'output_foo'))
[docs] def test_nested_stack_adopt_ok(self): url = self.publish_template(self.nested_with_res_template) self.template = self.test_template.replace('the.yaml', url) adopt_data = { "resources": { "the_nested": { "resource_id": "test-res-id", "resources": { "NestedResource": { "type": "OS::Heat::RandomString", "resource_id": "test-nested-res-id", "resource_data": {"value": "goopie"} } } } }, "environment": {"parameters": {}}, "template": yaml.safe_load(self.template) } stack_identifier = self.stack_adopt(adopt_data=json.dumps(adopt_data)) self.assert_resource_is_a_stack(stack_identifier, 'the_nested') stack = self.client.stacks.get(stack_identifier) self.assertEqual('goopie', self._stack_output(stack, 'output_foo'))
[docs] def test_nested_stack_adopt_fail(self): url = self.publish_template(self.nested_with_res_template) self.template = self.test_template.replace('the.yaml', url) adopt_data = { "resources": { "the_nested": { "resource_id": "test-res-id", "resources": { } } }, "environment": {"parameters": {}}, "template": yaml.safe_load(self.template) } stack_identifier = self.stack_adopt(adopt_data=json.dumps(adopt_data), wait_for_status='ADOPT_FAILED') rsrc = self.client.resources.get(stack_identifier, 'the_nested') self.assertEqual('ADOPT_FAILED', rsrc.resource_status)
[docs] def test_nested_stack_update(self): url = self.publish_template(self.nested_template) self.template = self.test_template.replace('the.yaml', url) stack_identifier = self.stack_create(template=self.template) original_nested_id = self.assert_resource_is_a_stack( stack_identifier, 'the_nested') stack = self.client.stacks.get(stack_identifier) self.assertEqual('bar', self._stack_output(stack, 'output_foo')) new_template = yaml.safe_load(self.template) props = new_template['Resources']['the_nested']['Properties'] props['TemplateURL'] = self.publish_template(self.update_template, cleanup=False) self.update_stack(stack_identifier, new_template) # Expect the physical resource name staying the same after update, # so that the nested was actually updated instead of replaced. new_nested_id = self.assert_resource_is_a_stack( stack_identifier, 'the_nested') self.assertEqual(original_nested_id, new_nested_id) updt_stack = self.client.stacks.get(stack_identifier) self.assertEqual('foo', self._stack_output(updt_stack, 'output_foo'))
[docs] def test_nested_stack_suspend_resume(self): url = self.publish_template(self.nested_template) self.template = self.test_template.replace('the.yaml', url) stack_identifier = self.stack_create(template=self.template) self.stack_suspend(stack_identifier) self.stack_resume(stack_identifier)