Type-check function body

1 files changed, 21 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
index 88027a69..889698ac 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java
@@ -35,4 +35,25 @@ public class TypeCheckerFunctionDefinitionTests {
                 .withBody(isSequence())
         );
     }
+
+    @Test
+    public void bodyIsTypeChecked() {
+        var untyped = UntypedFunctionDefinitionNode.builder()
+            .withName("f")
+            .withReturnType(Untyped.typeLevelReference("Unit"))
+            .addStatement(Untyped.expressionStatement(Untyped.stringLiteral("hello")))
+            .build();
+        var context = TypeCheckerContext.EMPTY
+            .enter(P.map(Map.entry("Unit", Types.metaType(Types.UNIT))));
+
+        var typed = TypeChecker.typeCheckNamespaceStatement(untyped, context);
+
+        assertThat(
+            typed,
+            isTypedFunctionDefinitionNode()
+                .withBody(isSequence(
+                    isTypedExpressionStatementNode(isTypedStringLiteralNode("hello"))
+                ))
+        );
+    }
 }