Type-check effect in function types

1 files changed, 49 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java
index 608e3caf..d1c7c2f2 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java
@@ -4,6 +4,7 @@ import org.junit.jupiter.api.Test;
 import org.shedlang.compiler.ast.untyped.Untyped;
 import org.shedlang.compiler.ast.untyped.UntypedFunctionTypeNode;
 import org.shedlang.compiler.typechecker.typesets.TypeSets;
+import org.shedlang.compiler.types.Effects;
 import org.shedlang.compiler.types.FunctionType;
 import org.shedlang.compiler.types.Types;
 import org.shedlang.compiler.util.P;
@@ -36,4 +37,52 @@ public class TypeCheckerFunctionTypeTests {
             ))
         );
     }
+
+    @Test
+    public void whenNodeHasNoEffectThenEffectIsNone() {
+        var untyped = UntypedFunctionTypeNode.builder()
+            .withReturnType(Untyped.typeLevelReference("String"))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enterGlobal(P.map(
+                Map.entry("String", Types.metaType(Types.STRING))
+            ));
+
+        var typed = TypeChecker.typeCheckTypeLevelExpression(TypeSets.META_TYPE, untyped, context);
+
+        assertThat(
+            typed,
+            isTypedFunctionTypeNode(equalTo(
+                FunctionType.builder()
+                    .withEffect(Effects.NONE)
+                    .withReturnType(Types.STRING)
+                    .build()
+            ))
+        );
+    }
+
+    @Test
+    public void effectIsTypeChecked() {
+        var untyped = UntypedFunctionTypeNode.builder()
+            .withEffect(Untyped.typeLevelReference("Io"))
+            .withReturnType(Untyped.typeLevelReference("String"))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enterGlobal(P.map(
+                Map.entry("Io", Types.effectType(Effects.IO)),
+                Map.entry("String", Types.metaType(Types.STRING))
+            ));
+
+        var typed = TypeChecker.typeCheckTypeLevelExpression(TypeSets.META_TYPE, untyped, context);
+
+        assertThat(
+            typed,
+            isTypedFunctionTypeNode(equalTo(
+                FunctionType.builder()
+                    .withEffect(Effects.IO)
+                    .withReturnType(Types.STRING)
+                    .build()
+            ))
+        );
+    }
 }