Type check positional params in function types

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 d1c7c2f2..2b212b4b 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionTypeTests.java
@@ -85,4 +85,32 @@ public class TypeCheckerFunctionTypeTests {
             ))
         );
     }
+
+    @Test
+    public void positionalParamsAreTypeChecked() {
+        var untyped = UntypedFunctionTypeNode.builder()
+            .addPositionalParam(Untyped.typeLevelReference("Int"))
+            .addPositionalParam(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()
+                    .addPositionalParam(Types.INT)
+                    .addPositionalParam(Types.BOOL)
+                    .withReturnType(Types.STRING)
+                    .build()
+            ))
+        );
+    }
 }