From 4089116972049123f447d72d1bd5cb5d7389c25e Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 10 Jun 2023 19:41:09 +0100
Subject: Add params to environment in function

---
 .../TypeCheckerFunctionDefinitionTests.java        | 70 ++++++++++++++++++++++
 1 file changed, 70 insertions(+)

(limited to 'src/test/java')

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
index 7ce8c7db..3eb177cc 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
@@ -104,6 +104,54 @@ public class TypeCheckerFunctionDefinitionTests {
         );
     }
 
+    @Test
+    public void canReferencePositionalArgInBody() {
+        var untyped = UntypedFunctionDefinitionNode.builder()
+            .addPositionalParam("x", Untyped.typeLevelReference("String"))
+            .withReturnType(Untyped.typeLevelReference("Unit"))
+            .addStatement(Untyped.expressionStatement(Untyped.reference("x")))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enter(P.map(
+                Map.entry("String", Types.metaType(Types.STRING)),
+                Map.entry("Unit", Types.metaType(Types.UNIT))
+            ));
+
+        var typed = TypeChecker.typeCheckNamespaceStatement(untyped, context);
+
+        assertThat(
+            typed,
+            isTypedFunctionDefinitionNode()
+                .withBody(isSequence(
+                    isTypedExpressionStatementNode(isTypedReferenceNode("x", Types.STRING))
+                ))
+        );
+    }
+
+    @Test
+    public void canReferenceNamedArgInBody() {
+        var untyped = UntypedFunctionDefinitionNode.builder()
+            .addNamedParam("x", Untyped.typeLevelReference("String"))
+            .withReturnType(Untyped.typeLevelReference("Unit"))
+            .addStatement(Untyped.expressionStatement(Untyped.reference("x")))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enter(P.map(
+                Map.entry("String", Types.metaType(Types.STRING)),
+                Map.entry("Unit", Types.metaType(Types.UNIT))
+            ));
+
+        var typed = TypeChecker.typeCheckNamespaceStatement(untyped, context);
+
+        assertThat(
+            typed,
+            isTypedFunctionDefinitionNode()
+                .withBody(isSequence(
+                    isTypedExpressionStatementNode(isTypedReferenceNode("x", Types.STRING))
+                ))
+        );
+    }
+
     @Test
     public void freeVariablesIncludeVariablesReferencedInFunctionButDefinedOutsideOfFunction() {
         var untyped = UntypedFunctionDefinitionNode.builder()
@@ -125,4 +173,26 @@ public class TypeCheckerFunctionDefinitionTests {
                 .withFreeVariables(containsExactly(equalTo("y")))
         );
     }
+
+    @Test
+    public void freeVariablesDoNotIncludeArgsReferencedInFunction() {
+        var untyped = UntypedFunctionDefinitionNode.builder()
+            .addPositionalParam("x", Untyped.typeLevelReference("String"))
+            .withReturnType(Untyped.typeLevelReference("Unit"))
+            .addStatement(Untyped.expressionStatement(Untyped.reference("x")))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enter(P.map(
+                Map.entry("String", Types.metaType(Types.STRING)),
+                Map.entry("Unit", Types.metaType(Types.UNIT))
+            ));
+
+        var typed = TypeChecker.typeCheckNamespaceStatement(untyped, context);
+
+        assertThat(
+            typed,
+            isTypedFunctionDefinitionNode()
+                .withFreeVariables(containsExactly())
+        );
+    }
 }
-- 
cgit v1.2.3