Type check imports

1 files changed, 54 insertions, 4 deletions

diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerNamespaceTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerNamespaceTests.java
index cfe52c3..c4b31b8 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerNamespaceTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerNamespaceTests.java
@@ -3,12 +3,11 @@ package org.zwobble.hobgoblin.compiler.typechecker;
 import org.junit.jupiter.api.Test;
 import org.zwobble.hobgoblin.compiler.ast.typed.TypedNamespaceNode;
 import org.zwobble.hobgoblin.compiler.ast.typed.TypedStructDefinitionNode;
-import org.zwobble.hobgoblin.compiler.ast.untyped.UntypedNamespaceNode;
-import org.zwobble.hobgoblin.compiler.ast.untyped.UntypedStructDefinitionNode;
-import org.zwobble.hobgoblin.compiler.types.NamespaceName;
-import org.zwobble.hobgoblin.compiler.types.StructType;
+import org.zwobble.hobgoblin.compiler.ast.untyped.*;
+import org.zwobble.hobgoblin.compiler.types.*;
 
 import java.util.List;
+import java.util.Optional;
 
 import static org.zwobble.hobgoblin.compiler.types.TypeMatchers.isMetaType;
 import static org.zwobble.precisely.AssertThat.assertThat;
@@ -47,4 +46,55 @@ public class TypeCheckerNamespaceTests {
             isMetaType(new StructType(NamespaceName.of("a", "b"), "X"))
         );
     }
+
+    @Test
+    public void importsAreTypeChecked() {
+        var untyped = UntypedNamespaceNode.arbitrary()
+            .withNamespaceName(NamespaceName.of("a", "b"))
+            .withImports(List.of(
+                UntypedImportNode.arbitrary()
+                    .withAncestorDepth(Optional.empty())
+                    .withNamespaceName(NamespaceName.of("a", "c"))
+                    .withImportedNames(List.of("Y"))
+                    .build()
+            ))
+            .withBody(List.of(
+                UntypedStructDefinitionNode.arbitrary()
+                    .withName("X")
+                    .withFields(List.of(
+                        UntypedStructFieldDefinitionNode.arbitrary()
+                            .withName("y")
+                            .withType(UntypedTypeLevelReferenceNode.arbitrary().withName("Y").build())
+                            .build()
+                    ))
+                    .build()
+            ))
+            .build();
+        var context = TypeCheckerContextArb.globalContext();
+        context.defineNamespace(NamespaceName.of("a", "c"), List.of(
+            new Field("Y", new TypeLevelValueType(new SimpleNativeType(NamespaceName.of("a", "c"), "Y"))
+        )));
+
+        var typed = TypeChecker.typeCheckNamespace(untyped, context);
+
+        assertThat(typed, instanceOf(
+            TypedNamespaceNode.class,
+            has("body", TypedNamespaceNode::body, isSequence(
+                instanceOf(
+                    TypedStructDefinitionNode.class,
+                    has(
+                        "type",
+                        TypedStructDefinitionNode::type,
+                        equalTo(new StructType(NamespaceName.of("a", "b"), "X"))
+                    )
+                )
+            ))
+        ));
+        assertThat(
+            context.toTypesInfo().fieldsOf(new StructType(NamespaceName.of("a", "b"), "X")),
+            isSequence(
+                equalTo(new Field("y", new SimpleNativeType(NamespaceName.of("a", "c"), "Y")))
+            )
+        );
+    }
 }