From 9a79e487595ac1fdbf4e9c69963440bd979aa107 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 3 Jul 2023 12:11:20 +0100
Subject: Define val type

---
 .../compiler/ast/typed/TypedNodeMatchers.java      |  4 +--
 .../compiler/typechecker/TypeCheckerValTests.java  | 30 ++++++++++++++++++++++
 2 files changed, 32 insertions(+), 2 deletions(-)

(limited to 'src/test/java')

diff --git a/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java b/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java
index 48c67ad1..ccda182a 100644
--- a/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java
+++ b/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java
@@ -125,8 +125,8 @@ public class TypedNodeMatchers {
     }
 
     public static Matcher<TypedFunctionStatementNode> isTypedValNode(
-        Matcher<TypedTargetNode> target,
-        Matcher<TypedExpressionNode> expression
+        Matcher<? super TypedTargetNode> target,
+        Matcher<? super TypedExpressionNode> expression
     ) {
         return instanceOf(
             TypedValNode.class,
diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerValTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerValTests.java
index ec1e3c8a..8ebe4f79 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerValTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerValTests.java
@@ -2,9 +2,16 @@ package org.shedlang.compiler.typechecker;
 
 import org.junit.jupiter.api.Test;
 import org.shedlang.compiler.ast.untyped.Untyped;
+import org.shedlang.compiler.ast.untyped.UntypedFunctionDefinitionNode;
+import org.shedlang.compiler.types.Types;
+import org.shedlang.compiler.util.P;
+
+import java.util.Map;
 
 import static org.shedlang.compiler.ast.typed.TypedNodeMatchers.*;
 import static org.zwobble.precisely.AssertThat.assertThat;
+import static org.zwobble.precisely.Matchers.anything;
+import static org.zwobble.precisely.Matchers.isSequence;
 
 public class TypeCheckerValTests {
     @Test
@@ -18,4 +25,27 @@ public class TypeCheckerValTests {
             isTypedStringLiteralNode("hello")
         ));
     }
+
+    @Test
+    public void valCanBeReferencedAfterDefinition() {
+        var untyped = UntypedFunctionDefinitionNode.builder()
+            .addStatement(Untyped.val(Untyped.targetVariable("x"), Untyped.stringLiteral("hello")))
+            .addStatement(Untyped.expressionStatement(Untyped.reference("x")))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enterBlock(P.map(
+                Map.entry("Unit", Types.metaType(Types.UNIT))
+            ));
+
+        var typed = TypeChecker.typeCheckNamespaceStatement(untyped, context);
+
+        assertThat(
+            typed,
+            isTypedFunctionDefinitionNode()
+                .withBody(isSequence(
+                    isTypedValNode(anything(), anything()),
+                    isTypedExpressionStatementNode(isTypedReferenceNode("x", Types.STRING))
+                ))
+        );
+    }
 }
-- 
cgit v1.2.3