From 74764070e84869fe8e2480350a76cb8fc58c1aeb Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Thu, 20 Jul 2023 18:28:32 +0100
Subject: Add if instruction to Patio

---
 .../compiler/patio/PatioSerializerTests.java       | 27 ++++++++++++++++++++++
 .../compiler/patio/instructions/Patio.java         | 13 +++++++++++
 2 files changed, 40 insertions(+)

(limited to 'src/test/java')

diff --git a/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java b/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java
index 864095c2..67227c7b 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioSerializerTests.java
@@ -127,6 +127,33 @@ public class PatioSerializerTests {
         assertThat(string, equalTo("(function-pointer-get \"target\" \"f\")"));
     }
 
+    @Test
+    public void canSerializeIf() {
+        var instruction = Patio.if_(
+            "condition",
+            List.of(
+                Patio.intGet("x", 1)
+            ),
+            List.of(
+                Patio.intGet("y", 2)
+            )
+        );
+
+        var string = PatioTesting.serializeInstruction(instruction);
+
+        assertThat(string, equalTo("""
+            (
+                if
+                "condition"
+                (
+                    (int-get "x" 1)
+                )
+                (
+                    (int-get "y" 2)
+                )
+            )"""));
+    }
+
     @Test
     public void canSerializeIntAdd() {
         var instruction = Patio.intAdd("target", "left", "right");
diff --git a/src/test/java/org/shedlang/compiler/patio/instructions/Patio.java b/src/test/java/org/shedlang/compiler/patio/instructions/Patio.java
index 48b458c7..0f434cd5 100644
--- a/src/test/java/org/shedlang/compiler/patio/instructions/Patio.java
+++ b/src/test/java/org/shedlang/compiler/patio/instructions/Patio.java
@@ -52,6 +52,19 @@ public class Patio {
         return new PatioFunctionPointerGet(new PatioLocalRef(target), name, Sources.NULL);
     }
 
+    public static PatioInstruction if_(
+        String condition,
+        List<PatioInstruction> trueBody,
+        List<PatioInstruction> falseBody
+    ) {
+        return new PatioIf(
+            new PatioLocalRef(condition),
+            trueBody,
+            falseBody,
+            Sources.NULL
+        );
+    }
+
     public static PatioInstruction intAdd(String target, String left, String right) {
         return new PatioIntAdd(
             new PatioLocalRef(target),
-- 
cgit v1.2.3