Type check function definition params

1 files changed, 28 insertions, 1 deletions

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
index 889698ac..1215389b 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
@@ -37,9 +37,36 @@ public class TypeCheckerFunctionDefinitionTests {
     }
 
     @Test
+    public void paramsAreTypeChecked() {
+        var untyped = UntypedFunctionDefinitionNode.builder()
+            .addPositionalParam("x", Untyped.typeLevelReference("String"))
+            .addNamedParam("y", Untyped.typeLevelReference("Int"))
+            .withReturnType(Untyped.typeLevelReference("Unit"))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enter(P.map(
+                Map.entry("Int", Types.metaType(Types.INT)),
+                Map.entry("String", Types.metaType(Types.STRING)),
+                Map.entry("Unit", Types.metaType(Types.UNIT))
+            ));
+
+        var typed = TypeChecker.typeCheckNamespaceStatement(untyped, context);
+
+        assertThat(
+            typed,
+            isTypedFunctionDefinitionNode()
+                .withParams(
+                    isTypedParamsNode(
+                        isSequence(isTypedParamNode("x", isTypedTypeLevelReferenceNode("String", Types.STRING))),
+                        isSequence(isTypedParamNode("y", isTypedTypeLevelReferenceNode("Int", Types.INT)))
+                    )
+                )
+        );
+    }
+
+    @Test
     public void bodyIsTypeChecked() {
         var untyped = UntypedFunctionDefinitionNode.builder()
-            .withName("f")
             .withReturnType(Untyped.typeLevelReference("Unit"))
             .addStatement(Untyped.expressionStatement(Untyped.stringLiteral("hello")))
             .build();