From 405fc623d679c0d40a8e4fc031dd4e5e32d5f35f Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Tue, 24 Oct 2023 21:21:29 +0100 Subject: Type-check effect in function types --- .../typechecker/TypeCheckerFunctionTypeTests.java | 49 ++++++++++++++++++++++ 1 file changed, 49 insertions(+) (limited to 'src/test/java') 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() + )) + ); + } } -- cgit v1.2.3