def test_compiler_chained_mutations(patch, magic, tree): patch.object(Objects, 'mutation_fragment') mutation = magic() tree.find_data.return_value = [mutation] result = JSONCompiler(story=None).chained_mutations(tree) Objects.mutation_fragment.assert_called_with(mutation.mutation_fragment) assert result == [Objects.mutation_fragment()]
def test_compiler_mutation_block_nested(patch, compiler, lines, tree): patch.many(Objects, ['expression', 'mutation_fragment']) patch.object(JSONCompiler, 'chained_mutations', return_value=['chained']) tree.path = None compiler.mutation_block(tree, None) JSONCompiler.chained_mutations.assert_called_with(tree.nested_block) args = [Objects.expression(), Objects.mutation_fragment(), 'chained', 'chained'] kwargs = {'args': args, 'parent': None} lines.append.assert_called_with('mutation', tree.position(), **kwargs)
def test_compiler_mutation_block_from_service(patch, compiler, lines, tree): patch.many(Objects, ['path', 'mutation_fragment']) patch.object(JSONCompiler, 'chained_mutations', return_value=['chained']) tree.nested_block = None tree.data = 'mutation_block' compiler.mutation_block(tree, None) Objects.path.assert_called_with(tree.path) Objects.mutation_fragment.assert_called_with(tree.mutation_fragment) JSONCompiler.chained_mutations.assert_called_with(tree) args = [Objects.path(), Objects.mutation_fragment(), 'chained'] kwargs = {'args': args, 'parent': None} lines.append.assert_called_with('mutation', tree.position(), **kwargs)
def test_compiler_mutation_block_from_service(patch, compiler, lines, tree): patch.many(Objects, ["path", "mutation_fragment"]) patch.object(JSONCompiler, "chained_mutations", return_value=["chained"]) tree.nested_block = None tree.data = "mutation_block" compiler.mutation_block(tree, None) Objects.path.assert_called_with(tree.path) Objects.mutation_fragment.assert_called_with(tree.mutation_fragment) JSONCompiler.chained_mutations.assert_called_with(tree) args = [Objects.path(), Objects.mutation_fragment(), "chained"] kwargs = {"args": args, "parent": None} lines.append.assert_called_with("mutation", tree.position(), **kwargs)
def test_compiler_mutation_block_nested(patch, compiler, lines, tree): patch.many(Objects, ["expression", "mutation_fragment"]) patch.object(JSONCompiler, "chained_mutations", return_value=["chained"]) tree.path = None compiler.mutation_block(tree, None) JSONCompiler.chained_mutations.assert_called_with(tree.nested_block) args = [ Objects.expression(), Objects.mutation_fragment(), "chained", "chained", ] kwargs = {"args": args, "parent": None} lines.append.assert_called_with("mutation", tree.position(), **kwargs)
def test_compiler_mutation_block(patch, compiler, lines, tree): patch.many(Objects, ['primary_expression', 'mutation_fragment']) patch.object(JSONCompiler, 'chained_mutations', return_value=['chained']) tree.path = None tree.nested_block = None compiler.mutation_block(tree, None) expr = tree.mutation.primary_expression Objects.primary_expression.assert_called_with(expr) Objects.mutation_fragment.assert_called_with( tree.mutation.mutation_fragment) JSONCompiler.chained_mutations.assert_called_with(tree.mutation) args = [ Objects.primary_expression(), Objects.mutation_fragment(), 'chained' ] kwargs = {'args': args, 'parent': None} lines.append.assert_called_with('mutation', tree.line(), **kwargs)
def test_compiler_extract_values_mutation(patch, tree): patch.many(Objects, ['values', 'mutation_fragment']) result = JSONCompiler(story=None).extract_values(tree) Objects.values.assert_called_with(tree.expression.values) Objects.mutation_fragment.assert_called_with(tree.expression.mutation) assert result == [Objects.values(), Objects.mutation_fragment()]