Skip to content

Commit

Permalink
check for cycles in nested macros
Browse files Browse the repository at this point in the history
  • Loading branch information
bernt-matthias committed Jan 14, 2022
1 parent 372439c commit d89538e
Show file tree
Hide file tree
Showing 2 changed files with 41 additions and 4 deletions.
18 changes: 14 additions & 4 deletions lib/galaxy/util/xml_macros.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def _expand_tokens_str(s, tokens):
return s


def _expand_macros(elements, macros, tokens):
def _expand_macros(elements, macros, tokens, visited=None):
if not macros and not tokens:
return

Expand All @@ -164,12 +164,21 @@ def _expand_macros(elements, macros, tokens):
expand_el = element.find('.//expand')
if expand_el is None:
break
_expand_macro(expand_el, macros, tokens)
if visited is None:
v = set()
else:
v = visited
_expand_macro(expand_el, macros, tokens, v)


def _expand_macro(expand_el, macros, tokens):
def _expand_macro(expand_el, macros, tokens, visited):
macro_name = expand_el.get('macro')
assert macro_name is not None, "Attempted to expand macro with no 'macro' attribute defined."

# check for cycles in the nested macro expansion
assert macro_name not in visited, f"Cycle in nested macros {visited}"
visited.add(macro_name)

assert macro_name in macros, f"No macro named {macro_name} found, known macros are {', '.join(macros.keys())}."
macro_def = macros[macro_name]
expanded_elements = deepcopy(macro_def.element)
Expand All @@ -180,8 +189,9 @@ def _expand_macro(expand_el, macros, tokens):
_expand_tokens(expanded_elements, macro_tokens)

# Recursively expand contained macros.
_expand_macros(expanded_elements, macros, tokens)
_expand_macros(expanded_elements, macros, tokens, visited)
_xml_replace(expand_el, expanded_elements)
visited.remove(macro_name)


def _expand_yield_statements(macro_def, expand_el):
Expand Down
27 changes: 27 additions & 0 deletions test/unit/tool_util/test_tool_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -644,3 +644,30 @@ def test_loader_specify_nested_macro_by_token():
<A/>
<B/>
</tool>'''


def test_loader_circular_macros():
"""
check the cycles in nested macros are detected
"""
with TestToolDirectory() as tool_dir:
tool_dir.write('''
<tool>
<macros>
<xml name="a">
<A>
<expand macro="b"/>
</A>
</xml>
<xml name="b">
<B>
<expand macro="a"/>
</B>
</xml>
</macros>
<expand macro="a"/>
</tool>''')
try:
tool_dir.load()
except AssertionError as a:
assert str(a) == "Cycle in nested macros {'a', 'b'}"

0 comments on commit d89538e

Please sign in to comment.