def test_nested_context(file_name, some_context): ctx = nested_context(file=file_name, line=68, context=some_context) assert ctx.file == file_name assert ctx.line == 68 assert ctx.context == some_context ctx = nested_context(file_name, 68, some_context) assert ctx.file == file_name assert ctx.line == 68 assert ctx.context == some_context
def test_add_context_nested(file_name, some_context): lines = ( 'line1', 'line2', 'line3', ) ctx = ( nested_context(file_name, cnt, some_context) for cnt in _it.count(1) ) expected = ( line_in_context(context=next(ctx), line='line1'), line_in_context(context=next(ctx), line='line2'), line_in_context(context=next(ctx), line='line3'), ) result = tuple( add_context(file=file_name, lines=lines, nest_in=some_context) ) assert expected == result
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
def test_shell_cmd_to_ast_errors(some_context): with pytest.raises(VirtSyntaxError) as err: tuple(shell_cmd_to_ast('false\n', some_context)) assert some_context == err.value.context with pytest.raises(VirtSyntaxError) as err: tuple(shell_cmd_to_ast('false\ntrue\n', some_context)) assert some_context == err.value.context @pytest.mark.parametrize( ('given', 'inc_ctx', 'expected'), [ ('/abs/inc', None, '/abs/inc'), ('/abs/inc', context('foo', 1), '/abs/inc'), ('/abs/inc', nested_context('bar', 1, context('foo', 1)), '/abs/inc'), ('inc', context('parent', 1), 'inc'), ('inc', context('dir/parent', 1), 'dir/inc'), ('inc', nested_context('bar', 1, context('f', 1)), 'inc'), ('inc', nested_context('bar', 1, context('hdir/f', 1)), 'hdir/inc'), ] ) def test_find_included_file(given, inc_ctx, expected): result = find_included_file(given, inc_ctx) assert expected == result def test_perform_ast_includes(file_name, fake_open): files = { 'file1.inc': dedent( """