From d28daafdd5a3df782f50f3c460038f894cc4f74b Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Wed, 5 Jul 2023 22:15:32 +0100
Subject: Add function to environment

---
 .../typechecker/TypeCheckerNamespaceTests.java     | 32 ++++++++++++++++++++++
 1 file changed, 32 insertions(+)

(limited to 'src/test/java')

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
@@ -54,6 +54,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");
-- 
cgit v1.2.3