Add switch instruction to stack IR

2 files changed, 51 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java b/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
index b0143a9b..e2294588 100644
--- a/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
+++ b/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
@@ -328,6 +328,46 @@ public class InstructionSerializerTests {
     }
 
     @Test
+    public void canSerializeSwitch() {
+        var instruction = Instructions.switch_(List.of(
+            Instructions.switchCase(
+                Types.struct(NamespaceName.of("Example", "Project"), "A"),
+                List.of(
+                    Instructions.return_()
+                )
+            ),
+            Instructions.switchCase(
+                Types.struct(NamespaceName.of("Example", "Project"), "B"),
+                List.of(
+                    Instructions.discard()
+                )
+            )
+        ));
+
+        var string = serialize(instruction);
+
+        assertThat(string, equalTo("""
+            (
+                switch
+                (
+                    (
+                        "Example/Project.A"
+                        (
+                            (return)
+                        )
+                    )
+                    (
+                        "Example/Project.B"
+                        (
+                            (discard)
+                        )
+                    )
+                )
+            )"""
+        ));
+    }
+
+    @Test
     public void canSerializeTupleCreate() {
         var instruction = Instructions.tupleCreate(2);
 
diff --git a/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java b/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java
index 3ac0ac62..4971575c 100644
--- a/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java
+++ b/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java
@@ -134,6 +134,17 @@ public class Instructions {
         return new StringAddInstruction(Sources.NULL);
     }
 
+    public static Instruction switch_(List<SwitchInstruction.Case> cases) {
+        return new SwitchInstruction(cases, Sources.NULL);
+    }
+
+    public static SwitchInstruction.Case switchCase(
+        StructType type,
+        List<Instruction> body
+    ) {
+        return new SwitchInstruction.Case(type, body, Sources.NULL);
+    }
+
     public static Instruction tupleCreate(int elementCount) {
         return new TupleCreateInstruction(elementCount, Sources.NULL);
     }