Support accessing fields on effects

1 files changed, 41 insertions, 3 deletions

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;
@@ -72,6 +70,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(
             Untyped.reference("user"),