From e450be55c05a6bb1eda4e49595b50cf776e6e6d6 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Tue, 7 Nov 2023 16:55:48 +0000 Subject: Disallow returns in effect handlers --- .../typechecker/TypeCheckerFunctionReturnTests.java | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'src/test/java') diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionReturnTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionReturnTests.java index da130bcc..650d1531 100644 --- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionReturnTests.java +++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionReturnTests.java @@ -2,6 +2,7 @@ package org.shedlang.compiler.typechecker; import org.junit.jupiter.api.Test; import org.shedlang.compiler.ast.untyped.Untyped; +import org.shedlang.compiler.typechecker.errors.CannotReturnFromEffectHandlerFieldError; import org.shedlang.compiler.typechecker.errors.CannotReturnOutsideOfFunctionError; import org.shedlang.compiler.typechecker.errors.UnexpectedTypeError; import org.shedlang.compiler.typechecker.typesets.TypeSets; @@ -52,6 +53,18 @@ public class TypeCheckerFunctionReturnTests { ); } + @Test + public void whenInEffectHandlerThenReturnIsForbidden() { + var untyped = Untyped.functionReturn(Untyped.stringLiteral("hello")); + var context = TypeCheckerContext.EMPTY + .enterFunction(P.map(), Effects.NONE, ReturnTypeState.effectHandler(Types.UNIT)); + + assertThrows( + CannotReturnFromEffectHandlerFieldError.class, + () -> TypeChecker.typeCheckFunctionStatement(untyped, context) + ); + } + @Test public void canReturnSubtypeOfReturnType() { var untyped = Untyped.functionReturn(Untyped.stringLiteral("hello")); -- cgit v1.2.3