Add Patio for calling local function with args

1 files changed, 31 insertions, 1 deletions

diff --git a/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java b/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java
index 9fc97792..21811753 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java
@@ -683,7 +683,7 @@ public abstract class PatioExampleTests {
     }
 
     @Test
-    public void canCallFunctionDefinedInModule() throws IOException, InterruptedException {
+    public void canCallFunctionWithoutArgsDefinedInModule() throws IOException, InterruptedException {
         var calledFunction = PatioFunctionDefinition.builder()
             .withName("get")
             .withReturnType(PatioTypes.INT)
@@ -712,6 +712,36 @@ public abstract class PatioExampleTests {
     }
 
     @Test
+    public void canCallFunctionWithArgsDefinedInModule() throws IOException, InterruptedException {
+        var calledFunction = PatioFunctionDefinition.builder()
+            .withName("identity")
+            .addLocal(PatioLocal.param("x", PatioTypes.INT))
+            .withReturnType(PatioTypes.INT)
+            .addBodyInstruction(Patio.return_(Patio.localRef("x")))
+            .build();
+
+        var patioFunction = PatioFunctionDefinition.builder()
+            .withName(FUNCTION_NAME)
+            .withReturnType(PatioTypes.VOID)
+            .addLocal(Patio.val("x", PatioTypes.INT))
+            .addBodyInstruction(Patio.call(
+                Optional.of(Patio.localRef("x")),
+                calledFunction.name(),
+                List.of(Patio.intConst(42))
+            ))
+            .addBodyInstruction(emitInt(Patio.localRef("x")))
+            .addBodyInstruction(Patio.returnVoid())
+            .build();
+        var patio = Patio.compilationUnit(List.of(calledFunction, patioFunction));
+
+        var result = run(patio);
+
+        assertThat(result, isSequence(
+            isInt(42)
+        ));
+    }
+
+    @Test
     public void canCallPointerToFunctionDefinedInModule() throws IOException, InterruptedException {
         var calledFunction = PatioFunctionDefinition.builder()
             .withName("get")