def test_perform_ast_include_errors(file_name, fake_open):
files = {
'missing.inc': dedent(
"""
run just_a_command
commands-from-file really_missing.inc
"""
).lstrip(),
'simple_loop': dedent(
"""
# a simple include loop
commands-from-file simple_loop
"""
).lstrip(),
'nested_loop1': dedent(
"""
# a nested include loop file 1
commands-from-file nested_loop2
"""
).lstrip(),
'nested_loop2': dedent(
"""
# a nested include loop file 2
commands-from-file nested_loop1
"""
).lstrip(),
'script_nested_loop1': dedent(
"""
# a nested loop via script file 1
commands-from-shell \
echo commands-from-file script_nested_loop2
"""
).lstrip(),
'script_nested_loop2': dedent(
"""
# a nested lopp via script file 2
commands-from-file script_nested_loop1
"""
).lstrip(),
}
asts_contexts = (
(
lines_to_ast(
StringIO('commands-from-file simple_missing.inc'),
file_name
),
context(file_name, 1),
),
(
lines_to_ast(StringIO('commands-from-file missing.inc')),
context('missing.inc', 2),
),
(
lines_to_ast(StringIO('commands-from-file simple_loop')),
context('simple_loop', 2),
),
(
lines_to_ast(StringIO('commands-from-file nested_loop1')),
context('nested_loop2', 2),
),
(
lines_to_ast(StringIO('commands-from-file script_nested_loop1')),
context('script_nested_loop2', 2),
),
)
for ast, ctx in asts_contexts:
with pytest.raises(VirtSyntaxError) as err:
with fake_open(files):
t = tuple(perform_ast_includes(ast))
print(t)
assert ctx == err.value.context
def test_perform_ast_includes(file_name, fake_open):
files = {
'file1.inc': dedent(
"""
run /file1/arg1
write /file1/arg2:long\\
input
"""
).lstrip(),
'incdir/file2.inc': dedent(
"""
# file2 comment
run /file2/script
commands-from-file file3.inc
commands-from-shell \\
echo "# shell ctx: $VIRT_COMMAND_FILE ($VIRT_COMMAND_LINE)" \\
echo "root-password password:foo"
write /last/file2:command
"""
).lstrip(),
'incdir/file3.inc': dedent(
"""
# file3 comment
run file3_command
"""
).lstrip(),
'file4': dedent(
"""
# will be included from script
"""
).lstrip(),
}
ast = lines_to_ast(StringIO(dedent(
"""
# The main AST
commands-from-file file1.inc
commands-from-file incdir/file2.inc
commands-from-shell \\
echo "# shell comment" \\
echo commands-from-file file4
selinux-relabel
"""
).lstrip()), context_file=file_name)
expected = (
comment(' The main AST', context(file_name, 1)),
# commands-from-file file1.inc
command('run', '/file1/arg1', context('file1.inc', 1)),
command('write', '/file1/arg2:long\ninput', context('file1.inc', 2)),
# commands-from-file file2.inc
comment(' file2 comment', context('incdir/file2.inc', 1)),
command('run', '/file2/script', context('incdir/file2.inc', 2)),
# commands-from-file file3.inc (in file2)
comment(' file3 comment', context('incdir/file3.inc', 1)),
command('run', 'file3_command', context('incdir/file3.inc', 2)),
# commands-from-shell (in file2)
comment(
" shell ctx: incdir/file2.inc (4)",
nested_context(
'script<incdir/file2.inc (4)>', 1,
context('incdir/file2.inc', 4)
)
),
command(
'root-password', 'password:foo',
nested_context(
'script<incdir/file2.inc (4)>', 2,
context('incdir/file2.inc', 4)
)
),
# back to file2
command(
'write', '/last/file2:command',
context('incdir/file2.inc', 7)
),
# commands-from-shell (in main ast)
comment(
" shell comment",
nested_context(
'script<{0} (4)>'.format(file_name), 1,
context(file_name, 4)
),
),
# commands-from-file file4
comment(' will be included from script', context('file4', 1)),
# back to main
command('selinux-relabel', None, context(file_name, 7)),
)
with fake_open(files):
result = tuple(perform_ast_includes(ast))
assert expected == result