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(
"""
