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