mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
Make sure include_role inherit variables from parent role (#18627)
* Make sure include_role inherit variables from parent role Setting the parent of task blocks generated by include_role after they have been produced is not sufficient - it means the tasks don't have the correct dependency chain set afterwards, and therefore, don't properly inherit variables from outer roles. In addition to manually setting the parents, pass the dep_chain when compiling the role, such that variables are correctly imported. Fixes #18540. * Add tests for include_role * Fix include_role variable inheritance for multiple parent levels
This commit is contained in:
parent
00f8945159
commit
57f4a9885e
2 changed files with 238 additions and 3 deletions
|
@ -76,10 +76,16 @@ class IncludeRole(Task):
|
||||||
actual_role = Role.load(ri, myplay, parent_role=self._parent_role, from_files=self._from_files)
|
actual_role = Role.load(ri, myplay, parent_role=self._parent_role, from_files=self._from_files)
|
||||||
actual_role._metadata.allow_duplicates = self.allow_duplicates
|
actual_role._metadata.allow_duplicates = self.allow_duplicates
|
||||||
|
|
||||||
# compile role
|
# compile role with parent roles as dependencies to ensure they inherit
|
||||||
blocks = actual_role.compile(play=myplay)
|
# variables
|
||||||
|
if not self._parent_role:
|
||||||
|
dep_chain = []
|
||||||
|
else:
|
||||||
|
dep_chain = list(self._parent_role._parents)
|
||||||
|
dep_chain.extend(self._parent_role.get_all_dependencies())
|
||||||
|
dep_chain.append(self._parent_role)
|
||||||
|
|
||||||
# set parent to ensure proper inheritance
|
blocks = actual_role.compile(play=myplay, dep_chain=dep_chain)
|
||||||
for b in blocks:
|
for b in blocks:
|
||||||
b._parent = self
|
b._parent = self
|
||||||
|
|
||||||
|
|
229
test/units/playbook/role/test_include_role.py
Normal file
229
test/units/playbook/role/test_include_role.py
Normal file
|
@ -0,0 +1,229 @@
|
||||||
|
# (c) 2016, Daniel Miranda <danielkza2@gmail.com>
|
||||||
|
#
|
||||||
|
# This file is part of Ansible
|
||||||
|
#
|
||||||
|
# Ansible is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# Ansible is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
# Make coding more python3-ish
|
||||||
|
from __future__ import (absolute_import, division, print_function)
|
||||||
|
__metaclass__ = type
|
||||||
|
|
||||||
|
from ansible.compat.tests import unittest
|
||||||
|
from ansible.compat.tests.mock import patch
|
||||||
|
|
||||||
|
from ansible.playbook import Play
|
||||||
|
from ansible.playbook.task import Task
|
||||||
|
from ansible.vars import VariableManager
|
||||||
|
|
||||||
|
from units.mock.loader import DictDataLoader
|
||||||
|
from units.mock.path import mock_unfrackpath_noop
|
||||||
|
|
||||||
|
|
||||||
|
def flatten_tasks(tasks):
|
||||||
|
for task in tasks:
|
||||||
|
if isinstance(task, Task):
|
||||||
|
yield task
|
||||||
|
else:
|
||||||
|
for t in flatten_tasks(task.block):
|
||||||
|
yield t
|
||||||
|
|
||||||
|
|
||||||
|
class TestIncludeRole(unittest.TestCase):
|
||||||
|
|
||||||
|
def setUp(self):
|
||||||
|
self.var_manager = VariableManager()
|
||||||
|
|
||||||
|
self.loader = DictDataLoader({
|
||||||
|
'/etc/ansible/roles/l1/tasks/main.yml': """
|
||||||
|
- shell: echo 'hello world from l1'
|
||||||
|
- include_role: name=l2
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l1/tasks/alt.yml': """
|
||||||
|
- shell: echo 'hello world from l1 alt'
|
||||||
|
- include_role: name=l2 tasks_from=alt defaults_from=alt
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l1/defaults/main.yml': """
|
||||||
|
test_variable: l1-main
|
||||||
|
l1_variable: l1-main
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l1/defaults/alt.yml': """
|
||||||
|
test_variable: l1-alt
|
||||||
|
l1_variable: l1-alt
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l2/tasks/main.yml': """
|
||||||
|
- shell: echo 'hello world from l2'
|
||||||
|
- include_role: name=l3
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l2/tasks/alt.yml': """
|
||||||
|
- shell: echo 'hello world from l2 alt'
|
||||||
|
- include_role: name=l3 tasks_from=alt defaults_from=alt
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l2/defaults/main.yml': """
|
||||||
|
test_variable: l2-main
|
||||||
|
l2_variable: l2-main
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l2/defaults/alt.yml': """
|
||||||
|
test_variable: l2-alt
|
||||||
|
l2_variable: l2-alt
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l3/tasks/main.yml': """
|
||||||
|
- shell: echo 'hello world from l3'
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l3/tasks/alt.yml': """
|
||||||
|
- shell: echo 'hello world from l3 alt'
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l3/defaults/main.yml': """
|
||||||
|
test_variable: l3-main
|
||||||
|
l3_variable: l3-main
|
||||||
|
""",
|
||||||
|
'/etc/ansible/roles/l3/defaults/alt.yml': """
|
||||||
|
test_variable: l3-alt
|
||||||
|
l3_variable: l3-alt
|
||||||
|
"""
|
||||||
|
})
|
||||||
|
|
||||||
|
def tearDown(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def get_tasks_vars(self, play, tasks):
|
||||||
|
for task in flatten_tasks(tasks):
|
||||||
|
role = task._role
|
||||||
|
if not role:
|
||||||
|
continue
|
||||||
|
|
||||||
|
yield (role.get_name(),
|
||||||
|
self.var_manager.get_vars(self.loader, play=play,
|
||||||
|
task=task))
|
||||||
|
|
||||||
|
@patch('ansible.playbook.role.definition.unfrackpath',
|
||||||
|
mock_unfrackpath_noop)
|
||||||
|
def test_simple(self):
|
||||||
|
|
||||||
|
"""Test one-level include with default tasks and variables"""
|
||||||
|
|
||||||
|
play = Play.load(dict(
|
||||||
|
name="test play",
|
||||||
|
hosts=['foo'],
|
||||||
|
gather_facts=False,
|
||||||
|
tasks=[
|
||||||
|
{'include_role': 'name=l3'}
|
||||||
|
]
|
||||||
|
), loader=self.loader, variable_manager=self.var_manager)
|
||||||
|
|
||||||
|
tasks = play.compile()
|
||||||
|
for role, task_vars in self.get_tasks_vars(play, tasks):
|
||||||
|
self.assertEqual(task_vars.get('l3_variable'), 'l3-main')
|
||||||
|
self.assertEqual(task_vars.get('test_variable'), 'l3-main')
|
||||||
|
|
||||||
|
@patch('ansible.playbook.role.definition.unfrackpath',
|
||||||
|
mock_unfrackpath_noop)
|
||||||
|
def test_simple_alt_files(self):
|
||||||
|
|
||||||
|
"""Test one-level include with alternative tasks and variables"""
|
||||||
|
|
||||||
|
play = Play.load(dict(
|
||||||
|
name="test play",
|
||||||
|
hosts=['foo'],
|
||||||
|
gather_facts=False,
|
||||||
|
tasks=[
|
||||||
|
{'include_role': 'name=l3 tasks_from=alt defaults_from=alt'}
|
||||||
|
]
|
||||||
|
), loader=self.loader, variable_manager=self.var_manager)
|
||||||
|
|
||||||
|
tasks = play.compile()
|
||||||
|
for role, task_vars in self.get_tasks_vars(play, tasks):
|
||||||
|
self.assertEqual(task_vars.get('l3_variable'), 'l3-alt')
|
||||||
|
self.assertEqual(task_vars.get('test_variable'), 'l3-alt')
|
||||||
|
|
||||||
|
@patch('ansible.playbook.role.definition.unfrackpath',
|
||||||
|
mock_unfrackpath_noop)
|
||||||
|
def test_nested(self):
|
||||||
|
|
||||||
|
"""
|
||||||
|
Test nested includes with default tasks and variables.
|
||||||
|
|
||||||
|
Variables from outer roles should be inherited, but overriden in inner
|
||||||
|
roles.
|
||||||
|
"""
|
||||||
|
|
||||||
|
play = Play.load(dict(
|
||||||
|
name="test play",
|
||||||
|
hosts=['foo'],
|
||||||
|
gather_facts=False,
|
||||||
|
tasks=[
|
||||||
|
{'include_role': 'name=l1'}
|
||||||
|
]
|
||||||
|
), loader=self.loader, variable_manager=self.var_manager)
|
||||||
|
|
||||||
|
tasks = play.compile()
|
||||||
|
for role, task_vars in self.get_tasks_vars(play, tasks):
|
||||||
|
# Outer-most role must not have variables from inner roles yet
|
||||||
|
if role == 'l1':
|
||||||
|
self.assertEqual(task_vars.get('l1_variable'), 'l1-main')
|
||||||
|
self.assertEqual(task_vars.get('l2_variable'), None)
|
||||||
|
self.assertEqual(task_vars.get('l3_variable'), None)
|
||||||
|
self.assertEqual(task_vars.get('test_variable'), 'l1-main')
|
||||||
|
# Middle role must have variables from outer role, but not inner
|
||||||
|
elif role == 'l2':
|
||||||
|
self.assertEqual(task_vars.get('l1_variable'), 'l1-main')
|
||||||
|
self.assertEqual(task_vars.get('l2_variable'), 'l2-main')
|
||||||
|
self.assertEqual(task_vars.get('l3_variable'), None)
|
||||||
|
self.assertEqual(task_vars.get('test_variable'), 'l2-main')
|
||||||
|
# Inner role must have variables from both outer roles
|
||||||
|
elif role == 'l3':
|
||||||
|
self.assertEqual(task_vars.get('l1_variable'), 'l1-main')
|
||||||
|
self.assertEqual(task_vars.get('l2_variable'), 'l2-main')
|
||||||
|
self.assertEqual(task_vars.get('l3_variable'), 'l3-main')
|
||||||
|
self.assertEqual(task_vars.get('test_variable'), 'l3-main')
|
||||||
|
|
||||||
|
@patch('ansible.playbook.role.definition.unfrackpath',
|
||||||
|
mock_unfrackpath_noop)
|
||||||
|
def test_nested_alt_files(self):
|
||||||
|
|
||||||
|
"""
|
||||||
|
Test nested includes with alternative tasks and variables.
|
||||||
|
|
||||||
|
Variables from outer roles should be inherited, but overriden in inner
|
||||||
|
roles.
|
||||||
|
"""
|
||||||
|
|
||||||
|
play = Play.load(dict(
|
||||||
|
name="test play",
|
||||||
|
hosts=['foo'],
|
||||||
|
gather_facts=False,
|
||||||
|
tasks=[
|
||||||
|
{'include_role': 'name=l1 tasks_from=alt defaults_from=alt'}
|
||||||
|
]
|
||||||
|
), loader=self.loader, variable_manager=self.var_manager)
|
||||||
|
|
||||||
|
tasks = play.compile()
|
||||||
|
for role, task_vars in self.get_tasks_vars(play, tasks):
|
||||||
|
# Outer-most role must not have variables from inner roles yet
|
||||||
|
if role == 'l1':
|
||||||
|
self.assertEqual(task_vars.get('l1_variable'), 'l1-alt')
|
||||||
|
self.assertEqual(task_vars.get('l2_variable'), None)
|
||||||
|
self.assertEqual(task_vars.get('l3_variable'), None)
|
||||||
|
self.assertEqual(task_vars.get('test_variable'), 'l1-alt')
|
||||||
|
# Middle role must have variables from outer role, but not inner
|
||||||
|
elif role == 'l2':
|
||||||
|
self.assertEqual(task_vars.get('l1_variable'), 'l1-alt')
|
||||||
|
self.assertEqual(task_vars.get('l2_variable'), 'l2-alt')
|
||||||
|
self.assertEqual(task_vars.get('l3_variable'), None)
|
||||||
|
self.assertEqual(task_vars.get('test_variable'), 'l2-alt')
|
||||||
|
# Inner role must have variables from both outer roles
|
||||||
|
elif role == 'l3':
|
||||||
|
self.assertEqual(task_vars.get('l1_variable'), 'l1-alt')
|
||||||
|
self.assertEqual(task_vars.get('l2_variable'), 'l2-alt')
|
||||||
|
self.assertEqual(task_vars.get('l3_variable'), 'l3-alt')
|
||||||
|
self.assertEqual(task_vars.get('test_variable'), 'l3-alt')
|
Loading…
Reference in a new issue