204 lines
6.0 KiB
Python
204 lines
6.0 KiB
Python
#
|
|
# 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 six
|
|
|
|
from heat.common import template_format
|
|
from heat.engine import stack
|
|
from heat.engine import template
|
|
from heat.tests import common
|
|
from heat.tests import utils
|
|
|
|
|
|
tmpl1 = """
|
|
heat_template_version: 2014-10-16
|
|
resources:
|
|
AResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: 'abc'
|
|
"""
|
|
|
|
tmpl2 = """
|
|
heat_template_version: 2014-10-16
|
|
resources:
|
|
AResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: 'abc'
|
|
BResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: {get_attr: [AResource, attr_A1]}
|
|
metadata:
|
|
Foo: {get_attr: [AResource, attr_A1]}
|
|
outputs:
|
|
out1:
|
|
value: {get_attr: [AResource, attr_A1]}
|
|
"""
|
|
|
|
tmpl3 = """
|
|
heat_template_version: 2014-10-16
|
|
resources:
|
|
AResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: 'abc'
|
|
BResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: {get_attr: [AResource, attr_A1]}
|
|
Doo: {get_attr: [AResource, attr_A2]}
|
|
Bar: {get_attr: [AResource, attr_A3]}
|
|
metadata:
|
|
first: {get_attr: [AResource, meta_A1]}
|
|
second: {get_attr: [AResource, meta_A2]}
|
|
third: {get_attr: [AResource, attr_A3]}
|
|
outputs:
|
|
out1:
|
|
value: {get_attr: [AResource, out_A1]}
|
|
out2:
|
|
value: {get_attr: [AResource, out_A2]}
|
|
"""
|
|
|
|
tmpl4 = """
|
|
heat_template_version: 2014-10-16
|
|
resources:
|
|
AResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: 'abc'
|
|
BResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: 'abc'
|
|
CResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: 'abc'
|
|
DResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: {get_attr: [AResource, attr_A1]}
|
|
Doo: {get_attr: [BResource, attr_B1]}
|
|
metadata:
|
|
Doo: {get_attr: [CResource, attr_C1]}
|
|
outputs:
|
|
out1:
|
|
value: [{get_attr: [AResource, attr_A1]},
|
|
{get_attr: [BResource, attr_B1]},
|
|
{get_attr: [CResource, attr_C1]}]
|
|
"""
|
|
|
|
tmpl5 = """
|
|
heat_template_version: 2014-10-16
|
|
resources:
|
|
AResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: 'abc'
|
|
BResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: {get_attr: [AResource, attr_A1]}
|
|
Doo: {get_attr: [AResource, attr_A2]}
|
|
metadata:
|
|
first: {get_attr: [AResource, meta_A1]}
|
|
CResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: {get_attr: [AResource, attr_A1]}
|
|
Doo: {get_attr: [BResource, attr_B2]}
|
|
metadata:
|
|
Doo: {get_attr: [BResource, attr_B1]}
|
|
first: {get_attr: [AResource, meta_A1]}
|
|
second: {get_attr: [BResource, meta_B2]}
|
|
outputs:
|
|
out1:
|
|
value: [{get_attr: [AResource, attr_A3]},
|
|
{get_attr: [AResource, attr_A4]},
|
|
{get_attr: [BResource, attr_B3]}]
|
|
"""
|
|
|
|
tmpl6 = """
|
|
heat_template_version: 2015-04-30
|
|
resources:
|
|
AResource:
|
|
type: ResourceWithComplexAttributesType
|
|
BResource:
|
|
type: ResourceWithPropsType
|
|
properties:
|
|
Foo: {get_attr: [AResource, list, 1]}
|
|
Doo: {get_attr: [AResource, nested_dict, dict, b]}
|
|
outputs:
|
|
out1:
|
|
value: [{get_attr: [AResource, flat_dict, key2]},
|
|
{get_attr: [AResource, nested_dict, string]},
|
|
{get_attr: [BResource, attr_B3]}]
|
|
"""
|
|
|
|
|
|
class DepAttrsTest(common.HeatTestCase):
|
|
|
|
scenarios = [
|
|
('no_attr',
|
|
dict(tmpl=tmpl1,
|
|
expected={'AResource': set()})),
|
|
('one_res_one_attr',
|
|
dict(tmpl=tmpl2,
|
|
expected={'AResource': {'attr_A1'},
|
|
'BResource': set()})),
|
|
('one_res_several_attrs',
|
|
dict(tmpl=tmpl3,
|
|
expected={'AResource': {'attr_A1', 'attr_A2', 'attr_A3',
|
|
'meta_A1', 'meta_A2', 'out_A1',
|
|
'out_A2'},
|
|
'BResource': set()})),
|
|
('several_res_one_attr',
|
|
dict(tmpl=tmpl4,
|
|
expected={'AResource': {'attr_A1'},
|
|
'BResource': {'attr_B1'},
|
|
'CResource': {'attr_C1'},
|
|
'DResource': set()})),
|
|
('several_res_several_attrs',
|
|
dict(tmpl=tmpl5,
|
|
expected={'AResource': {'attr_A1', 'attr_A2', 'meta_A1',
|
|
'attr_A3', 'attr_A4'},
|
|
'BResource': {'attr_B1', 'attr_B2', 'meta_B2',
|
|
'attr_B3'},
|
|
'CResource': set()})),
|
|
('nested_attr',
|
|
dict(tmpl=tmpl6,
|
|
expected={'AResource': set([(u'flat_dict', u'key2'),
|
|
(u'list', 1),
|
|
(u'nested_dict', u'dict', u'b'),
|
|
(u'nested_dict', u'string')]),
|
|
'BResource': set(['attr_B3'])}))
|
|
]
|
|
|
|
def setUp(self):
|
|
super(DepAttrsTest, self).setUp()
|
|
self.ctx = utils.dummy_context()
|
|
|
|
def test_dep_attrs(self):
|
|
parsed_tmpl = template_format.parse(self.tmpl)
|
|
self.stack = stack.Stack(self.ctx, 'test_stack',
|
|
template.Template(parsed_tmpl))
|
|
|
|
for res in six.itervalues(self.stack):
|
|
outputs = self.stack.outputs
|
|
resources = six.itervalues(self.stack.resources)
|
|
self.assertEqual(self.expected[res.name],
|
|
self.stack.get_dep_attrs(resources, outputs,
|
|
res.name))
|