Use LlvmVariableLocal for labels

1 files changed, 8 insertions, 4 deletions

diff --git a/src/test/java/org/shedlang/compiler/backends/llvm/ir/ast/LlvmIr.java b/src/test/java/org/shedlang/compiler/backends/llvm/ir/ast/LlvmIr.java
index bbf72a3e..2ed63e9c 100644
--- a/src/test/java/org/shedlang/compiler/backends/llvm/ir/ast/LlvmIr.java
+++ b/src/test/java/org/shedlang/compiler/backends/llvm/ir/ast/LlvmIr.java
@@ -46,11 +46,15 @@ public class LlvmIr {
     }
 
     public static LlvmBr br(LlvmVariableLocal condition, String ifTrue, String ifFalse) {
-        return new LlvmBr(condition, ifTrue, ifFalse);
+        return new LlvmBr(
+            condition,
+            new LlvmVariableLocal(ifTrue),
+            new LlvmVariableLocal(ifFalse)
+        );
     }
 
     public static LlvmBrUnconditional brUnconditional(String target) {
-        return new LlvmBrUnconditional(target);
+        return new LlvmBrUnconditional(new LlvmVariableLocal(target));
     }
 
     public static LlvmCall call(
@@ -196,11 +200,11 @@ public class LlvmIr {
         String defaultDestLabel,
         List<LlvmSwitch.Case> cases
     ) {
-        return new LlvmSwitch(valueType, value, defaultDestLabel, cases);
+        return new LlvmSwitch(valueType, value, new LlvmVariableLocal(defaultDestLabel), cases);
     }
 
     public static LlvmSwitch.Case switchCase(int value, String destLabel) {
-        return new LlvmSwitch.Case(value, destLabel);
+        return new LlvmSwitch.Case(value, new LlvmVariableLocal(destLabel));
     }
 
     public static LlvmTrunc trunc(