def test_compiler_chained_mutations(patch, magic, tree):
patch.object(Objects, 'mutation')
mutation = magic()
tree.find_data.return_value = [mutation]
result = Compiler.chained_mutations(tree)
Objects.mutation.assert_called_with(mutation.mutation_fragment)
assert result == [Objects.mutation()]
def test_compiler_indented_chain(patch, compiler, lines, tree):
patch.object(Compiler, 'chained_mutations')
lines.last.return_value = '1'
lines.lines = {'1': {'method': 'mutation', 'args': ['args']}}
compiler.indented_chain(tree, '0')
Compiler.chained_mutations.assert_called_with(tree)
assert lines.lines['1']['args'] == ['args'] + Compiler.chained_mutations()