From 51f3d868199b411a4b0a010c0a26585b5718e0fb Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 21 Oct 2023 00:10:50 +0100
Subject: Support type checking field access on namespace

---
 .../typechecker/TypeCheckerFieldAccessTests.java   | 28 +++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

(limited to 'src/test/java')

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFieldAccessTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFieldAccessTests.java
index 62971de5..dfddc7ca 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFieldAccessTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFieldAccessTests.java
@@ -43,7 +43,33 @@ public class TypeCheckerFieldAccessTests {
     }
 
     @Test
-    public void whenReceiverIsNotStructTypeThenErrorIsThrown() {
+    public void canAccessFieldOnNamespaceType() {
+        var untyped = Untyped.fieldAccess(
+            Untyped.reference("Example"),
+            "name"
+        );
+        var namespaceType = Types.namespace(NamespaceName.of("Example"));
+        var context = TypeCheckerContext.EMPTY
+            .enterGlobal(P.map(Map.entry("Example", namespaceType)))
+            .addNamespaceFields(
+                namespaceType.namespaceName(),
+                P.orderedMap(Map.entry("name", Types.STRING))
+            );
+
+        var typed = TypeChecker.typeCheckExpression(untyped, context);
+
+        assertThat(
+            typed.value(),
+            isTypedFieldAccessNode(
+                isTypedReferenceNode("Example", namespaceType),
+                "name",
+                Types.STRING
+            )
+        );
+    }
+
+    @Test
+    public void whenReceiverIsTypeWithoutFieldsThenErrorIsThrown() {
         var untyped = Untyped.fieldAccess(
             Untyped.reference("user"),
             "name"
-- 
cgit v1.2.3