Support capture of vars in effect handler methods

1 files changed, 28 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
index eeb17968..696876b9 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
@@ -3,6 +3,7 @@ package org.shedlang.compiler.patio;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.shedlang.compiler.ast.typed.FreeVariables;
+import org.shedlang.compiler.patio.ast.types.PatioTypes;
 import org.shedlang.compiler.patio.instructions.Patio;
 import org.shedlang.compiler.sources.Sources;
 import org.shedlang.compiler.stackir.instructions.DefineFunctionInstruction;
@@ -91,6 +92,33 @@ public class PatioCodeGenDefineFunctionTests {
     }
 
     @Test
+    public void pointersToFreeMutableVariablesAreStoredInAndLoadedFromEnvironment(Snapshotter snapshotter) throws IOException {
+        var instruction = new DefineFunctionInstruction(
+            "f",
+            List.of(),
+            List.of(),
+            FreeVariables.builder()
+                .plus("a", Sources.NULL)
+                .plus("b", Sources.NULL)
+                .build(),
+            List.of(
+                Instructions.localLoad("a"),
+                Instructions.return_()
+            ),
+            Sources.NULL
+        );
+        var context = PatioCodeGenContext.EMPTY
+            .declareStackIrVal("a", PatioTypes.DATA_POINTER).context()
+            .initStackIrLocal("a")
+            .declareStackIrVal("b", PatioTypes.DATA_POINTER).context()
+            .initStackIrLocal("b");
+
+        var result = PatioCodeGen.generateInstruction(instruction, context);
+
+        snapshotter.assertSnapshot(serialize(result));
+    }
+
+    @Test
     public void positionalAndNamedParamsAreDefined(Snapshotter snapshotter) throws IOException {
         var instruction = new DefineFunctionInstruction(
             "f",