From 298f03e167d4cf3ef5b5088f355c8d8e4ee21467 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 12 Jan 2024 17:51:59 +0000
Subject: Handle type level params when substituting in function type

---
 .../compiler/types/TypeSubstitutionTests.java      | 44 ++++++++++++++++++++++
 1 file changed, 44 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 064012f9..6eb7105e 100644
--- a/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java
+++ b/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java
@@ -63,6 +63,30 @@ public class TypeSubstitutionTests {
         ));
     }
 
+    @Test
+    public void substitutingFunctionPreservesTypeLevelParams() {
+        var typeSubstitutionMap = TypeSubstitutionMap.of(
+            Map.entry(TypeParam.forStructType(STRUCT_TYPE, "T"), Types.INT)
+        );
+
+        var result = TypeSubstitution.substituteType(
+            FunctionType.builder()
+                .withTypeLevelParams(List.of(
+                    TypeParam.forFunction(NAMESPACE_NAME, "f", "T")
+                ))
+                .build(),
+            typeSubstitutionMap
+        );
+
+        assertThat(result, equalTo(
+            FunctionType.builder()
+                .withTypeLevelParams(List.of(
+                    TypeParam.forFunction(NAMESPACE_NAME, "f", "T")
+                ))
+                .build())
+        );
+    }
+
     @Test
     public void substitutingFunctionSubstitutesPositionalArgs() {
         var typeSubstitutionMap = TypeSubstitutionMap.of(
@@ -99,6 +123,26 @@ public class TypeSubstitutionTests {
             .build()));
     }
 
+    @Test
+    public void substitutingFunctionPreservesEffect() {
+        var typeSubstitutionMap = TypeSubstitutionMap.of(
+            Map.entry(TypeParam.forStructType(STRUCT_TYPE, "T"), Types.INT)
+        );
+
+        var result = TypeSubstitution.substituteType(
+            FunctionType.builder()
+                .withEffect(Effects.IO)
+                .build(),
+            typeSubstitutionMap
+        );
+
+        assertThat(result, equalTo(
+            FunctionType.builder()
+                .withEffect(Effects.IO)
+                .build())
+        );
+    }
+
     @Test
     public void substitutingFunctionSubstitutesReturnType() {
         var typeSubstitutionMap = TypeSubstitutionMap.of(
-- 
cgit v1.2.3