Add var-load and var-store to Stack IR

2 files changed, 26 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 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);
+    }
 }