From c3784202feea6d9e16044007dbb992dab3628553 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 27 Aug 2023 20:51:49 +0100
Subject: Add loop instructions to Stack IR

---
 .../stackir/InstructionSerializerTests.java        | 39 ++++++++++++++++++++++
 .../stackir/instructions/Instructions.java         | 12 +++++++
 2 files changed, 51 insertions(+)

(limited to 'src/test/java')

diff --git a/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java b/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
index aff45481..407621fb 100644
--- a/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
+++ b/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
@@ -158,6 +158,45 @@ public class InstructionSerializerTests {
         assertThat(string, equalTo("(local-store \"x\")"));
     }
 
+    @Test
+    public void canSerializeLoop() {
+        var instruction = Instructions.loop(
+            List.of(
+                Instructions.localLoad("x"),
+                Instructions.discard()
+            )
+        );
+
+        var string = serialize(instruction);
+
+        assertThat(string, equalTo("""
+            (
+                loop
+                (
+                    (local-load "x")
+                    (discard)
+                )
+            )"""));
+    }
+
+    @Test
+    public void canSerializeLoopBreak() {
+        var instruction = Instructions.loopBreak();
+
+        var string = serialize(instruction);
+
+        assertThat(string, equalTo("(break)"));
+    }
+
+    @Test
+    public void canSerializeLoopContinue() {
+        var instruction = Instructions.loopContinue();
+
+        var string = serialize(instruction);
+
+        assertThat(string, equalTo("(continue)"));
+    }
+
     @Test
     public void canSerializeNamespaceFieldStore() {
         var instruction = Instructions.namespaceFieldStore(
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 18526f4a..4bc3dc50 100644
--- a/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java
+++ b/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java
@@ -53,6 +53,18 @@ public class Instructions {
         return new LocalStoreInstruction(name, Sources.NULL);
     }
 
+    public static Instruction loop(List<Instruction> body) {
+        return new LoopInstruction(body, Sources.NULL);
+    }
+
+    public static Instruction loopBreak() {
+        return new LoopBreakInstruction(Sources.NULL);
+    }
+
+    public static Instruction loopContinue() {
+        return new LoopContinueInstruction(Sources.NULL);
+    }
+
     public static Instruction namespaceFieldStore(NamespaceName namespaceName, String fieldName) {
         return new NamespaceFieldStoreInstruction(namespaceName, fieldName, Sources.NULL);
     }
-- 
cgit v1.2.3