Add function to environment

1 files changed, 32 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java
index 3ae36edb..e4063606 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java
@@ -55,6 +55,38 @@ public class TypeCheckerNamespaceTests {
     }
 
     @Test
+    public void functionCanCallFunctionDefinedEarlier() {
+        var untyped = Untyped.namespace(List.of(
+            UntypedFunctionDefinitionNode.builder()
+                .withName("f")
+                .withReturnType(Untyped.typeLevelReference("Unit"))
+                .build(),
+            UntypedFunctionDefinitionNode.builder()
+                .withName("g")
+                .withReturnType(Untyped.typeLevelReference("Unit"))
+                .addStatement(Untyped.expressionStatement(Untyped.call(
+                    Untyped.reference("f"),
+                    List.of()
+                )))
+                .build()
+        ));
+        var context = TypeCheckerContext.EMPTY
+            .enterNamespaceWithDefinedVariables(P.map(Map.entry("Unit", Types.metaType(Types.UNIT))));
+
+        var result = TypeChecker.typeCheckNamespace(untyped, context);
+
+        assertThat(
+            result.value(),
+            isTypedNamespaceNode()
+                .withBody(isSequence(
+                    isTypedFunctionDefinitionNode().withName("f"),
+                    isTypedFunctionDefinitionNode().withName("g")
+                )
+            )
+        );
+    }
+
+    @Test
     public void typesOfFieldsAreStored() {
         var namespaceName = NamespaceName.of("Example");
         var untyped = Untyped.namespace(