From 2e9b821350221fc2defa89e9c4ecefc1c273855b Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 14 Jul 2023 22:02:10 +0100
Subject: Handle params when defining functions in LLVM backend

---
 .../llvm/LlvmCodeGenDefineFunctionTests.java       | 35 ++++++++++++++++------
 1 file changed, 26 insertions(+), 9 deletions(-)

(limited to 'src/test/java')

diff --git a/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCodeGenDefineFunctionTests.java b/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCodeGenDefineFunctionTests.java
index 1baf847c..885306b3 100644
--- a/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCodeGenDefineFunctionTests.java
+++ b/src/test/java/org/shedlang/compiler/backends/llvm/LlvmCodeGenDefineFunctionTests.java
@@ -3,24 +3,15 @@ package org.shedlang.compiler.backends.llvm;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.shedlang.compiler.ast.typed.FreeVariables;
-import org.shedlang.compiler.backends.llvm.ir.LlvmIrSerializer;
 import org.shedlang.compiler.backends.llvm.ir.ast.LlvmIr;
-import org.shedlang.compiler.backends.llvm.ir.ast.operands.LlvmOperand;
 import org.shedlang.compiler.sources.Sources;
 import org.shedlang.compiler.stackir.instructions.DefineFunctionInstruction;
 import org.shedlang.compiler.stackir.instructions.Instructions;
 import org.shedlang.compiler.testing.snapshots.SnapshotResolver;
 import org.shedlang.compiler.testing.snapshots.Snapshotter;
-import org.zwobble.precisely.Matcher;
 
 import java.io.IOException;
 import java.util.List;
-import java.util.Map;
-
-import static org.shedlang.compiler.backends.llvm.ir.ast.LlvmIrMatchers.isLlvmVariableLocal;
-import static org.shedlang.compiler.util.Iterables.forEachInterspersed;
-import static org.zwobble.precisely.AssertThat.assertThat;
-import static org.zwobble.precisely.Matchers.*;
 
 @ExtendWith({SnapshotResolver.class})
 public class LlvmCodeGenDefineFunctionTests {
@@ -65,4 +56,30 @@ public class LlvmCodeGenDefineFunctionTests {
 
         snapshotter.assertSnapshot(LlvmCodeGenTesting.serializeContext(result));
     }
+
+    @Test
+    public void positionalAndNamedParamsAreDefined(Snapshotter snapshotter) throws IOException {
+        var instruction = new DefineFunctionInstruction(
+            "f",
+            List.of(Instructions.param("param1"), Instructions.param("param2")),
+            List.of(Instructions.param("paramA"), Instructions.param("paramB")),
+            FreeVariables.of("a", "b"),
+            List.of(
+                Instructions.localLoad("a"),
+                Instructions.localLoad("param1"),
+                Instructions.intAdd(),
+                Instructions.localLoad("paramA"),
+                Instructions.intAdd(),
+                Instructions.return_()
+            ),
+            Sources.NULL
+        );
+        var context = LlvmCodeGenContext.EMPTY
+            .addShedLocal("a", LlvmIr.operandInt(1))
+            .addShedLocal("b", LlvmIr.operandInt(2));
+
+        var result = LlvmCodeGen.generateInstruction(instruction, context);
+
+        snapshotter.assertSnapshot(LlvmCodeGenTesting.serializeContext(result));
+    }
 }
-- 
cgit v1.2.3