Handle definition of struct when checking dependencies

1 files changed, 34 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 7663a0aa..34179cb2 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerNamespaceTests.java
@@ -364,6 +364,40 @@ public class TypeCheckerNamespaceTests {
     }
 
     @Test
+    public void valCanReferenceFunctionDefinedLaterThatUsesStructDefinedEvenLater() {
+        var untyped = Untyped.namespace(NamespaceName.of("Example", "Project"), List.of(
+            Untyped.val("none", Untyped.call(
+                Untyped.reference("createNone"),
+                List.of()
+            )),
+            UntypedFunctionDefinitionNode.builder()
+                .withName("createNone")
+                .withReturnType(Untyped.typeLevelReference("None"))
+                .addStatement(Untyped.functionReturn(Untyped.call(
+                    Untyped.reference("None"),
+                    List.of()
+                )))
+                .build(),
+            UntypedStructDefinitionNode.builder()
+                .withName("None")
+                .build()
+        ));
+        var context = TypeCheckerContext.EMPTY;
+
+        var result = TypeChecker.typeCheckNamespace(untyped, context);
+
+        assertThat(
+            result.value(),
+            isTypedNamespaceNode()
+                .withBody(isSequence(
+                    isTypedValNode("none", anything()),
+                    isTypedFunctionDefinitionNode(),
+                    isTypedStructDefinitionNode()
+                ))
+        );
+    }
+
+    @Test
     public void functionBodyCanAccessFieldOnTypeDefinedLater() {
         var untyped = Untyped.namespace(NamespaceName.of("Example", "Project"), List.of(
             UntypedFunctionDefinitionNode.builder()