import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;

import org.antlr.v4.runtime.tree.ParseTree;
import org.junit.jupiter.api.Test;

import de.hsrm.compiler.Klang.ContextAnalysis;

public class FieldAssignmentTest {

    @Test
    void variableNotDefined() {
        ParseTree tree = Helper.prepareParser("struct test { a: int; } function foo(): int { str.a = 1; return 1; } foo();");
        var funcs = Helper.getFuncs(tree);
        var structs = Helper.getStructs(tree);
        var enums = Helper.getEnums(tree);
        ContextAnalysis ctxAnal = new ContextAnalysis(funcs, structs, enums);
        
        Exception e = assertThrows(RuntimeException.class, () -> ctxAnal.visit(tree));
        assertEquals("Error in line 1:46 Variable or parameter with name \"str\" not defined.", e.getMessage());
    }

    @Test
    void fieldAssignmentOnNonStruct() {
        ParseTree tree = Helper.prepareParser("struct test { a: int; } function foo(): int { let x: int = 0; x.a = 0; return 1; } foo();");
        var funcs = Helper.getFuncs(tree);
        var structs = Helper.getStructs(tree);
        var enums = Helper.getEnums(tree);
        ContextAnalysis ctxAnal = new ContextAnalysis(funcs, structs, enums);
        
        Exception e = assertThrows(RuntimeException.class, () -> ctxAnal.visit(tree));
        assertEquals("Error in line 1:62 Variable or parameter must reference a struct but references int.", e.getMessage());
    }
}