From 76fc235449f03fd76ab9a5bd96d9e8f2aec025c3 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 14 Jul 2023 21:27:24 +0100
Subject: Add if instruction

---
 .../stackir/InstructionSerializerTests.java        | 29 ++++++++++++++++++++++
 .../stackir/instructions/Instructions.java         |  7 ++++++
 2 files changed, 36 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 3ce690b2..72056a3e 100644
--- a/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
+++ b/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
@@ -75,6 +75,35 @@ public class InstructionSerializerTests {
         assertThat(string, equalTo("(field-load \"Example/Lib\" \"x\")"));
     }
 
+    @Test
+    public void canSerializeIf() {
+        var instruction = Instructions.if_(
+            List.of(
+                Instructions.pushUnit()
+            ),
+            List.of(
+                Instructions.localLoad("x"),
+                Instructions.discard(),
+                Instructions.pushBool(true)
+            )
+        );
+
+        var string = serialize(instruction);
+
+        assertThat(string, equalTo("""
+            (
+                if
+                (
+                    (push-unit)
+                )
+                (
+                    (local-load "x")
+                    (discard)
+                    (push-bool true)
+                )
+            )"""));
+    }
+
     @Test
     public void canSerializeIntAdd() {
         var instruction = Instructions.intAdd();
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 df8b193b..131207b8 100644
--- a/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java
+++ b/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java
@@ -22,6 +22,13 @@ public class Instructions {
         return new FieldLoadInstruction(namespaceName, fieldName, Sources.NULL);
     }
 
+    public static Instruction if_(
+        List<Instruction> trueBody,
+        List<Instruction> falseBody
+    ) {
+        return new IfInstruction(trueBody, falseBody, Sources.NULL);
+    }
+
     public static Instruction intAdd() {
         return new IntAddInstruction(Sources.NULL);
     }
-- 
cgit v1.2.3