1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
package org.zwobble.couscous.tests;
import org.junit.Test;
import org.zwobble.couscous.ast.ClassNode;
import org.zwobble.couscous.ast.FormalArgumentNode;
import org.zwobble.couscous.ast.LiteralNode;
import org.zwobble.couscous.ast.LocalVariableDeclarationNode;
import org.zwobble.couscous.ast.MethodNode;
import org.zwobble.couscous.values.PrimitiveValue;
import org.zwobble.couscous.values.PrimitiveValues;
import org.zwobble.couscous.values.StringValue;
import static java.util.Arrays.asList;
import static org.junit.Assert.assertEquals;
import static org.zwobble.couscous.ast.AssignmentNode.assign;
import static org.zwobble.couscous.ast.ExpressionStatementNode.expressionStatement;
import static org.zwobble.couscous.ast.FormalArgumentNode.formalArg;
import static org.zwobble.couscous.ast.IfStatementNode.ifStatement;
import static org.zwobble.couscous.ast.LiteralNode.literal;
import static org.zwobble.couscous.ast.LocalVariableDeclarationNode.localVariableDeclaration;
import static org.zwobble.couscous.ast.MethodNode.staticMethod;
import static org.zwobble.couscous.ast.ReturnNode.returns;
import static org.zwobble.couscous.ast.VariableDeclaration.var;
import static org.zwobble.couscous.ast.VariableReferenceNode.reference;
import static org.zwobble.couscous.tests.TestIds.ANY_ID;
import static org.zwobble.couscous.values.PrimitiveValues.value;
public abstract class BackendMethodTests {
@Test
public void methodWithNoStatementsReturnsUnit() {
MethodNode.Builder method = staticMethod("hello");
final java.lang.Object result = runMethod(method);
assertEquals(PrimitiveValues.UNIT, result);
}
@Test
public void canReturnLiteralValue() {
MethodNode.Builder method = staticMethod("hello")
.statement(returns(literal("hello, world!")));
final java.lang.Object result = runMethod(method);
assertEquals(value("hello, world!"), result);
}
@Test
public void canPassValueToMethod() {
FormalArgumentNode arg = formalArg(var(ANY_ID, "x", StringValue.REF));
MethodNode.Builder method = staticMethod("hello")
.argument(arg)
.statement(returns(reference(arg)));
final java.lang.Object result = runMethod(method, value("hello, world!"));
assertEquals(value("hello, world!"), result);
}
@Test
public void canReassignValueToArgument() {
FormalArgumentNode arg = formalArg(var(ANY_ID, "x", StringValue.REF));
MethodNode.Builder method = staticMethod("hello")
.argument(arg)
.statement(expressionStatement(assign(
reference(arg),
literal("[updated value]"))))
.statement(returns(reference(arg)));
final java.lang.Object result = runMethod(method, value("[initial value]"));
assertEquals(value("[updated value]"), result);
}
@Test
public void canDeclareVariable() {
LocalVariableDeclarationNode localVariableDeclaration = localVariableDeclaration(ANY_ID, "x", StringValue.REF, LiteralNode.literal("[initial value]"));
MethodNode.Builder method = staticMethod("hello")
.statement(localVariableDeclaration)
.statement(returns(reference(localVariableDeclaration)));
final java.lang.Object result = runMethod(method);
assertEquals(value("[initial value]"), result);
}
@Test
public void canDeclareVariableAndThenAssignValues() {
LocalVariableDeclarationNode localVariableDeclaration = localVariableDeclaration(ANY_ID, "x", StringValue.REF, LiteralNode.literal("[initial value]"));
MethodNode.Builder method = staticMethod("hello")
.statement(localVariableDeclaration)
.statement(expressionStatement(assign(
reference(localVariableDeclaration),
literal("[updated value]"))))
.statement(returns(reference(localVariableDeclaration)));
final java.lang.Object result = runMethod(method);
assertEquals(value("[updated value]"), result);
}
@Test
public void whenConditionIsTrueIfStatementExecutesTrueBranch() {
MethodNode.Builder method = staticMethod("hello")
.statement(ifStatement(
literal(true),
asList(returns(literal("[true]"))),
asList(returns(literal("[false]")))));
assertEquals(value("[true]"), runMethod(method));
}
protected PrimitiveValue runMethod(MethodNode.Builder methodBuilder, PrimitiveValue... arguments) {
MethodNode method = methodBuilder.build();
String className = "com.example.Program";
ClassNode classNode = ClassNode.builder(className).method(method).build();
MethodRunner runner = buildMethodRunner();
return runner.runMethod(asList(classNode), classNode.getName(), method.getName(), asList(arguments));
}
protected abstract MethodRunner buildMethodRunner();
}
|
