From ffe44975d9c6ad0cd9c1e870bf4beb669ca69b62 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sat, 2 Mar 2024 14:54:29 +0000 Subject: Implement type substitution for effect unions --- .../compiler/types/TypeSubstitutionTests.java | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'src/test/java') diff --git a/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java b/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java index 8bca5f3f..37341a7a 100644 --- a/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java +++ b/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java @@ -41,6 +41,26 @@ public class TypeSubstitutionTests { assertThat(result, equalTo(Effects.IO)); } + @Test + public void substitutingEffectUnionUpdatesMembersOfUnion() { + var effect1 = Effects.userDefined(NAMESPACE_NAME, "Effect1"); + var effect2 = Effects.userDefined(NAMESPACE_NAME, "Effect2"); + var typeSubstitutionMap = TypeSubstitutionMap.of( + Map.entry(EffectParam.forStructType(STRUCT_TYPE, "E1"), effect1), + Map.entry(EffectParam.forStructType(STRUCT_TYPE, "E2"), effect2) + ); + + var result = TypeSubstitution.substituteEffect( + Effects.union( + EffectParam.forStructType(STRUCT_TYPE, "E1"), + EffectParam.forStructType(STRUCT_TYPE, "E2") + ), + typeSubstitutionMap + ); + + assertThat(result, equalTo(Effects.union(effect1, effect2))); + } + @Test public void substitutingEffectIoDoesNothing() { var typeSubstitutionMap = TypeSubstitutionMap.of( -- cgit v1.2.3