From c4bd013b53d3d00db54844573c8254bfbb504cde Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 30 Jul 2023 17:08:57 +0100
Subject: Add var-load and var-store to Stack IR

---
 .../compiler/stackir/InstructionSerializerTests.java   | 18 ++++++++++++++++++
 .../compiler/stackir/instructions/Instructions.java    |  8 ++++++++
 2 files changed, 26 insertions(+)

(limited to 'src/test/java/org/shedlang/compiler/stackir')

diff --git a/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java b/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
index e4893c31..aff45481 100644
--- a/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
+++ b/src/test/java/org/shedlang/compiler/stackir/InstructionSerializerTests.java
@@ -244,6 +244,24 @@ public class InstructionSerializerTests {
         assertThat(string, equalTo("(string-add)"));
     }
 
+    @Test
+    public void canSerializeVarLoad() {
+        var instruction = Instructions.varLoad("x");
+
+        var string = serialize(instruction);
+
+        assertThat(string, equalTo("(var-load \"x\")"));
+    }
+
+    @Test
+    public void canSerializeVarStore() {
+        var instruction = Instructions.varStore("x");
+
+        var string = serialize(instruction);
+
+        assertThat(string, equalTo("(var-store \"x\")"));
+    }
+
     private String serialize(Instruction instruction) {
         var builder = new StringBuilder();
         InstructionSerializer.serializeInstruction(instruction, builder);
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 131207b8..18526f4a 100644
--- a/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java
+++ b/src/test/java/org/shedlang/compiler/stackir/instructions/Instructions.java
@@ -88,4 +88,12 @@ public class Instructions {
     public static Instruction stringAdd() {
         return new StringAddInstruction(Sources.NULL);
     }
+
+    public static Instruction varLoad(String name) {
+        return new VarLoadInstruction(name, Sources.NULL);
+    }
+
+    public static Instruction varStore(String name) {
+        return new VarStoreInstruction(name, Sources.NULL);
+    }
 }
-- 
cgit v1.2.3