aboutsummaryrefslogtreecommitdiff
path: root/src/test/java/org/zwobble/couscous/tests/interpreter/InterpreterMethodTests.java
blob: c1a0584ffc286b0cce21c981118d9a5500727db5 (plain)
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package org.zwobble.couscous.tests.interpreter;

import org.junit.Test;
import org.zwobble.couscous.ast.*;
import org.zwobble.couscous.types.ScalarType;
import org.zwobble.couscous.types.Type;
import org.zwobble.couscous.interpreter.Interpreter;
import org.zwobble.couscous.interpreter.JavaProject;
import org.zwobble.couscous.interpreter.Project;
import org.zwobble.couscous.interpreter.errors.NoSuchMethod;
import org.zwobble.couscous.interpreter.errors.UnboundVariable;
import org.zwobble.couscous.interpreter.errors.UnexpectedValueType;
import org.zwobble.couscous.interpreter.errors.VariableNotInScope;
import org.zwobble.couscous.interpreter.values.InterpreterValue;
import org.zwobble.couscous.interpreter.values.InterpreterValues;
import org.zwobble.couscous.tests.BackendMethodTests;
import org.zwobble.couscous.tests.MethodRunner;
import org.zwobble.couscous.types.Types;
import org.zwobble.couscous.values.PrimitiveValue;

import java.util.List;
import java.util.stream.Collectors;

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.LiteralNode.literal;
import static org.zwobble.couscous.ast.LocalVariableDeclarationNode.localVariableDeclaration;
import static org.zwobble.couscous.ast.MethodNode.staticMethod;
import static org.zwobble.couscous.ast.MethodSignature.signature;
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.tests.util.ExtraAsserts.assertThrows;
import static org.zwobble.couscous.util.ExtraLists.list;
import static org.zwobble.couscous.values.PrimitiveValues.value;

public class InterpreterMethodTests extends BackendMethodTests {
    @Test
    public void errorIfWrongNumberOfArgumentsArePassed() {
        MethodNode.Builder method = staticMethod("hello");
        
        NoSuchMethod exception = assertThrows(NoSuchMethod.class,
            () -> runMethod(method, value("hello, world!")));
        
        assertEquals(signature("hello", list(Types.STRING), Types.VOID), exception.getSignature());
    }
    
    @Test
    public void errorIfTryingToAssignValueOfWrongTypeToVariable() {
        FormalArgumentNode arg = formalArg(var(ANY_ID, "x", Types.STRING));
        MethodNode.Builder method = staticMethod("hello")
            .argument(arg)
            .statement(expressionStatement(assign(reference(arg), literal(0))));

        UnexpectedValueType exception = assertThrows(UnexpectedValueType.class,
            () -> runMethod(method, value("")));
        
        assertEquals(new UnexpectedValueType(Types.STRING, Types.INT), exception);
    }
    
    @Test
    public void errorIfTryingToAssignToVariableNotInScope() {
        LocalVariableDeclarationNode localVariableDeclaration = localVariableDeclaration(
            ANY_ID, "x", Types.STRING, literal(""));
        MethodNode.Builder method = staticMethod("hello")
            .statement(expressionStatement(assign(reference(localVariableDeclaration), LiteralNode.literal("[updated value]"))));

        VariableNotInScope exception = assertThrows(VariableNotInScope.class,
            () -> runMethod(method));
        
        assertEquals(new VariableNotInScope(ANY_ID), exception);
    }
    
    @Test
    public void errorIfTryingToGetValueOfVariableNotInScope() {
        LocalVariableDeclarationNode localVariableDeclaration = localVariableDeclaration(
            ANY_ID, "x", Types.STRING, literal(""));
        MethodNode.Builder method = staticMethod("hello")
            .statement(returns(reference(localVariableDeclaration)));

        VariableNotInScope exception = assertThrows(VariableNotInScope.class,
            () -> runMethod(method));
        
        assertEquals(new VariableNotInScope(ANY_ID), exception);
    }
    
    @Test
    public void errorIfTryingToGetValueOfUnboundVariable() {
        LocalVariableDeclarationNode localVariableDeclaration = localVariableDeclaration(
            ANY_ID, "x", Types.STRING, literal(""));
        MethodNode.Builder method = staticMethod("hello")
            .statement(returns(reference(localVariableDeclaration)))
            .statement(localVariableDeclaration);

        UnboundVariable exception = assertThrows(UnboundVariable.class,
            () -> runMethod(method));
        
        assertEquals(new UnboundVariable(ANY_ID), exception);
    }

    @Override
    protected MethodRunner buildMethodRunner() {
        return new MethodRunner() {
            @Override
            public PrimitiveValue runMethod(
                List<TypeNode> classNodes,
                ScalarType className,
                String methodName,
                List<PrimitiveValue> arguments,
                Type returnType) {
                
                Project project = JavaProject.of(classNodes);
                Interpreter interpreter = new Interpreter(project);
                List<InterpreterValue> argumentValues = arguments.stream()
                    .map(InterpreterValues::value)
                    .collect(Collectors.toList());
                
                return interpreter.run(className, methodName, argumentValues, returnType)
                    .toPrimitiveValue().get();
            }
        };
    }
}