From 1352b8d81c5bc7a36f32908ad9c15fcbf081e57e Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Tue, 6 Oct 2015 16:44:09 +0100 Subject: Simplify builder usage --- src/test/java/org/zwobble/couscous/tests/BackendEvalTests.java | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) (limited to 'src/test/java/org') diff --git a/src/test/java/org/zwobble/couscous/tests/BackendEvalTests.java b/src/test/java/org/zwobble/couscous/tests/BackendEvalTests.java index 765be9c..c9b267d 100644 --- a/src/test/java/org/zwobble/couscous/tests/BackendEvalTests.java +++ b/src/test/java/org/zwobble/couscous/tests/BackendEvalTests.java @@ -92,9 +92,8 @@ public abstract class BackendEvalTests { @Test public void canCallInstanceMethodWithNoArgumentsOnUserDefinedClass() { val classNode = ClassNode.builder("com.example.Example") - .method(MethodNode.method("main") - .statement(new ReturnNode(literal(42))) - .build()) + .method("main", method -> method + .statement(new ReturnNode(literal(42)))) .build(); val result = evalExpression(asList(classNode), methodCall(constructorCall(classNode.getName(), asList()), "main", asList(), IntegerValue.REF)); @@ -105,10 +104,9 @@ public abstract class BackendEvalTests { public void canCallInstanceMethodWithArgumentsOnUserDefinedClass() { val argument = formalArg(var(42, "x", IntegerValue.REF)); val classNode = ClassNode.builder("com.example.Example") - .method(MethodNode.method("main") + .method("main", method -> method .argument(argument) - .statement(new ReturnNode(reference(argument))) - .build()) + .statement(new ReturnNode(reference(argument)))) .build(); val result = evalExpression( -- cgit v1.2.3