def test_preprocessor_assignment_expression(patch, magic, tree):
patch.many(Preprocessor, ['fake_tree', 'replace_expression'])
block = magic()
Preprocessor.assignment_expression(block, tree)
Preprocessor.fake_tree.assert_called_with(block)
parent = block.rules.assignment.assignment_fragment
args = (Preprocessor.fake_tree(), parent, tree.inline_expression)
Preprocessor.replace_expression.assert_called_with(*args)
def test_preprocessor_service_arguments(patch, magic, tree):
patch.many(Preprocessor, ['fake_tree', 'replace_expression'])
argument = magic()
tree.find_data.return_value = [argument]
Preprocessor.service_arguments('block', tree)
Preprocessor.fake_tree.assert_called_with('block')
tree.find_data.assert_called_with('arguments')
argument.node.assert_called_with('values.inline_expression')
args = (Preprocessor.fake_tree(), argument, argument.node())
Preprocessor.replace_expression.assert_called_with(*args)
def test_preprocessor_fake_tree(patch):
patch.init(FakeTree)
result = Preprocessor.fake_tree('block')
FakeTree.__init__.assert_called_with('block')
assert isinstance(result, FakeTree)