Should execute 'resolve' before check the values of 'for_each'

Change-Id: I75f58885a91d11e52ed3737a806b5920774d586c
Closes-Bug: #1466748
This commit is contained in:
lvdongbing 2015-06-29 15:23:43 +08:00
parent e4a72cd92c
commit a7c4b2332f
2 changed files with 15 additions and 8 deletions

View File

@ -301,7 +301,7 @@ class Repeat(function.Function):
self.fn_name)
try:
for_each = self.args['for_each']
for_each = function.resolve(self.args['for_each'])
template = self.args['template']
except (KeyError, TypeError):
example = ('''repeat:
@ -335,9 +335,8 @@ class Repeat(function.Function):
for (k, v) in template.items())
def result(self):
for_each = function.resolve(self._for_each)
keys = list(six.iterkeys(for_each))
lists = [for_each[key] for key in keys]
keys = list(six.iterkeys(self._for_each))
lists = [self._for_each[key] for key in keys]
template = function.resolve(self._template)
return [self._do_replacement(keys, items, template)
for items in itertools.product(*lists)]

View File

@ -593,13 +593,21 @@ class HOTemplateTest(common.HeatTestCase):
def test_repeat(self):
"""Test repeat function."""
snippet = {'repeat': {'template': 'this is %var%',
'for_each': {'%var%': ['a', 'b', 'c']}}}
hot_tpl = template_format.parse('''
heat_template_version: 2015-04-30
parameters:
param:
type: comma_delimited_list
default: 'a,b,c'
''')
snippet = {'repeat': {'template': 'this is var%',
'for_each': {'var%': {'get_param': 'param'}}}}
snippet_resolved = ['this is a', 'this is b', 'this is c']
tmpl = template.Template(hot_kilo_tpl_empty)
tmpl = template.Template(hot_tpl)
stack = parser.Stack(utils.dummy_context(), 'test_stack', tmpl)
self.assertEqual(snippet_resolved, self.resolve(snippet, tmpl))
self.assertEqual(snippet_resolved, self.resolve(snippet, tmpl, stack))
def test_repeat_dict_template(self):
"""Test repeat function with a dictionary as a template."""