fix: Fix include_role with loop

haidaraM · Nov 9, 2021 · b607512 · b607512
1 parent 3fa8bea
commit b607512
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/tests/fixtures/include_role_with_loop.yml b/tests/fixtures/include_role_with_loop.yml
@@ -0,0 +1,12 @@
+---
+- hosts: all
+  tags:
+    - play2
+  tasks:
+    - name: Include with loop
+      include_role:
+        name: '{{ item }}'
+      loop:
+        - fake_role
+        - display_some_facts
+
diff --git a/tests/test_parser.py b/tests/test_parser.py
@@ -40,6 +40,20 @@ def test_include_role_parsing(grapher_cli: PlaybookGrapherCLI, display: Display)
     assert len(include_role_2.tasks) == 3
 
 
+@pytest.mark.parametrize('grapher_cli', [["include_role_with_loop.yml"]], indirect=True)
+def test_include_role_with_loop_parsing(grapher_cli: PlaybookGrapherCLI, display: Display):
+    """
+    Test parsing of include_role with loop
+    :param grapher_cli:
+    :param display:
+    :return:
+    """
+    parser = PlaybookParser(grapher_cli.options.playbook_filename, display=display, include_role_tasks=True)
+    playbook_node = parser.parse()
+    assert len(playbook_node.plays) == 1
+    play_node = playbook_node.plays[0].destination
+
+
 @pytest.mark.parametrize('grapher_cli', [["with_block.yml"]], indirect=True)
 def test_block_parsing(grapher_cli: PlaybookGrapherCLI, display: Display):
     """