From aa88533bb9809fd9cc597d0b6fc7676b35f1ede2 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 5 Nov 2023 19:57:34 +0000
Subject: Support accessing fields on effects

---
 .../typechecker/TypeCheckerFieldAccessTests.java   | 44 ++++++++++++++++++++--
 1 file changed, 41 insertions(+), 3 deletions(-)

(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 ffd9b7af..8b5c6371 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFieldAccessTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFieldAccessTests.java
@@ -4,9 +4,7 @@ import org.junit.jupiter.api.Test;
 import org.shedlang.compiler.ast.untyped.Untyped;
 import org.shedlang.compiler.typechecker.errors.MissingStructTypeInfoError;
 import org.shedlang.compiler.typechecker.errors.NoSuchFieldError;
-import org.shedlang.compiler.types.NamespaceName;
-import org.shedlang.compiler.types.StructTypeInfo;
-import org.shedlang.compiler.types.Types;
+import org.shedlang.compiler.types.*;
 import org.shedlang.compiler.util.P;
 
 import java.util.Map;
@@ -71,6 +69,46 @@ public class TypeCheckerFieldAccessTests {
         );
     }
 
+    @Test
+    public void canAccessFieldOnEffect() {
+        var untyped = Untyped.fieldAccess(
+            Untyped.reference("Effect"),
+            "mark"
+        );
+        var effect = Effects.userDefined(NamespaceName.of("Example"), "Effect");
+        var context = TypeCheckerContext.EMPTY
+            .enterGlobal(P.map(Map.entry("Effect", Types.effectType(effect))))
+            .addUserDefinedEffectInfo(
+                effect,
+                UserDefinedEffectInfo.builder()
+                    .addField(
+                        "mark",
+                        FunctionType.builder()
+                            .addPositionalParam(Types.BOOL)
+                            .addNamedParam("x", Types.INT)
+                            .withReturnType(Types.STRING)
+                            .build()
+                    )
+                    .build()
+            );
+
+        var typed = typeCheckExpression(untyped, context);
+
+        assertThat(
+            typed.value(),
+            isTypedFieldAccessNode(
+                isTypedReferenceNode("Effect", Types.effectType(effect)),
+                "mark",
+                FunctionType.builder()
+                    .addPositionalParam(Types.BOOL)
+                    .addNamedParam("x", Types.INT)
+                    .withEffect(effect)
+                    .withReturnType(Types.STRING)
+                    .build()
+            )
+        );
+    }
+
     @Test
     public void whenReceiverIsTypeWithoutFieldsThenErrorIsThrown() {
         var untyped = Untyped.fieldAccess(
-- 
cgit v1.2.3