diff options
Diffstat (limited to 'src/test/java')
| -rw-r--r-- | src/test/java/org/zwobble/couscous/tests/BackendEvalTests.java | 10 |
1 files changed, 4 insertions, 6 deletions
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( |