Compile load-store instruction to LLVM IR

1 files changed, 36 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCompilerLocalStoreTests.java b/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCompilerLocalStoreTests.java
new file mode 100644
index 00000000..2f20ec85
--- /dev/null
+++ b/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCompilerLocalStoreTests.java
@@ -0,0 +1,36 @@
+package org.shedlang.compiler.backends.llvm;
+
+import org.junit.jupiter.api.Test;
+import org.shedlang.compiler.backends.llvm.ir.ast.LlvmIr;
+import org.shedlang.compiler.backends.llvm.ir.ast.operands.LlvmOperand;
+import org.shedlang.compiler.stackir.instructions.Instructions;
+import org.zwobble.precisely.Matcher;
+
+import java.util.Map;
+
+import static org.shedlang.compiler.backends.llvm.ir.ast.LlvmIrMatchers.isLlvmOperandInt;
+import static org.zwobble.precisely.AssertThat.assertThat;
+import static org.zwobble.precisely.Matchers.*;
+
+public class LlvmCompilerLocalStoreTests {
+    @Test
+    public void localStorePopsFromStackAndUpdatesLocals() {
+        var instruction = Instructions.localStore("x");
+        var context = LlvmCompilerContext.EMPTY
+            .push(LlvmIr.operandInt(0))
+            .push(LlvmIr.operandInt(1));
+
+        var result = LlvmCompiler.compileInstruction(instruction, context);
+
+        assertThat(result.instructions(), isSequence());
+        assertThat(result.stack(), isSequence(isLlvmOperandInt(0)));
+        assertThat(result.locals().entrySet(), isSequence(isLocal("x", isLlvmOperandInt(1))));
+    }
+
+    private static Matcher<Map.Entry<String, LlvmOperand>> isLocal(String name, Matcher<LlvmOperand> operand) {
+        return allOf(
+            has("name", x -> x.getKey(), equalTo(name)),
+            has("operand", x -> x.getValue(), operand)
+        );
+    }
+}