Add Patio code gen for returns

2 files changed, 28 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
index f19e88ba..790bf8d7 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
@@ -3,8 +3,10 @@ package org.shedlang.compiler.patio;
 import org.pcollections.PVector;
 import org.shedlang.compiler.patio.instructions.PatioDiscard;
 import org.shedlang.compiler.patio.instructions.PatioInstruction;
+import org.shedlang.compiler.patio.instructions.PatioReturn;
 import org.shedlang.compiler.stackir.instructions.DiscardInstruction;
 import org.shedlang.compiler.stackir.instructions.Instruction;
+import org.shedlang.compiler.stackir.instructions.ReturnInstruction;
 import org.shedlang.compiler.util.P;
 
 public class PatioCodeGen {
@@ -15,6 +17,9 @@ public class PatioCodeGen {
             case DiscardInstruction discard ->
                 P.vector(new PatioDiscard(discard.source()));
 
+            case ReturnInstruction discard ->
+                P.vector(new PatioReturn(discard.source()));
+
             default ->
                 throw new IllegalStateException("Unexpected value: " + stackIrInstruction);
         };
diff --git a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenReturnTests.java b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenReturnTests.java
new file mode 100644
index 00000000..59c5f436
--- /dev/null
+++ b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenReturnTests.java
@@ -0,0 +1,23 @@
+package org.shedlang.compiler.patio;
+
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.shedlang.compiler.stackir.instructions.Instructions;
+import org.shedlang.compiler.testing.snapshots.SnapshotResolver;
+import org.shedlang.compiler.testing.snapshots.Snapshotter;
+
+import java.io.IOException;
+
+import static org.shedlang.compiler.patio.PatioTesting.serialize;
+
+@ExtendWith({SnapshotResolver.class})
+public class PatioCodeGenReturnTests {
+    @Test
+    public void returnInstructionGeneratesReturnInstruction(Snapshotter snapshotter) throws IOException {
+        var stackIr = Instructions.return_();
+
+        var result = PatioCodeGen.generateInstruction(stackIr);
+
+        snapshotter.assertSnapshot(serialize(result));
+    }
+}