Type check named params in function type

1 files changed, 28 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java
index 2b212b4b..6b48a604 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java
@@ -113,4 +113,32 @@ public class TypeCheckerFunctionTypeTests {
             ))
         );
     }
+
+    @Test
+    public void namedParamsAreTypeChecked() {
+        var untyped = UntypedFunctionTypeNode.builder()
+            .addNamedParam("a", Untyped.typeLevelReference("Int"))
+            .addNamedParam("b", Untyped.typeLevelReference("Bool"))
+            .withReturnType(Untyped.typeLevelReference("String"))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enterGlobal(P.map(
+                Map.entry("Bool", Types.metaType(Types.BOOL)),
+                Map.entry("Int", Types.metaType(Types.INT)),
+                Map.entry("String", Types.metaType(Types.STRING))
+            ));
+
+        var typed = TypeChecker.typeCheckTypeLevelExpression(TypeSets.META_TYPE, untyped, context);
+
+        assertThat(
+            typed,
+            isTypedFunctionTypeNode(equalTo(
+                FunctionType.builder()
+                    .addNamedParam("a", Types.INT)
+                    .addNamedParam("b", Types.BOOL)
+                    .withReturnType(Types.STRING)
+                    .build()
+            ))
+        );
+    }
 }