From 804500f0aac6253cfd5262038b62667902e195ca Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 1 Oct 2023 19:51:24 +0100 Subject: Check all variants are handled --- .../compiler/typechecker/TypeCheckerWhenTests.java | 30 ++++++++++++++++++++++ 1 file changed, 30 insertions(+) (limited to 'src/test/java') diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerWhenTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerWhenTests.java index 031b20c6..ef131b4d 100644 --- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerWhenTests.java +++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerWhenTests.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.UnhandledVariantError; import org.shedlang.compiler.typechecker.errors.WhenExpressionIsNotInstanceOfSumTypeError; import org.shedlang.compiler.types.*; import org.shedlang.compiler.util.P; @@ -40,6 +41,35 @@ public class TypeCheckerWhenTests { assertThat(error.actualType(), equalTo(Types.STRING)); } + @Test + public void whenNotAllVariantsAreCoveredThenErrorIsThrown() { + var untyped = Untyped.when( + Untyped.reference("x"), + List.of( + Untyped.whenCase( + Untyped.typeLevelReference(VARIANT_TYPE_1.name()), + Untyped.block(List.of( + Untyped.blockReturn(Untyped.stringLiteral("a")) + )) + ) + ) + ); + var context = TypeCheckerContext.EMPTY + .addSumTypeInfo(SUM_TYPE, new SumTypeInfo(P.orderedSet(VARIANT_TYPE_1, VARIANT_TYPE_2))) + .enterGlobal(P.map( + Map.entry("x", SUM_TYPE), + Map.entry(VARIANT_TYPE_1.name(), Types.metaType(VARIANT_TYPE_1)), + Map.entry(VARIANT_TYPE_2.name(), Types.metaType(VARIANT_TYPE_2)) + )); + + var error = assertThrows( + UnhandledVariantError.class, + () -> TypeChecker.typeCheckExpression(untyped, context) + ); + + assertThat(error.unhandledVariant(), equalTo(VARIANT_TYPE_2)); + } + @Test public void whenAllCasesAreHandledThenExpressionTypeChecks() { var untyped = Untyped.when( -- cgit v1.2.3