Handle ifs with multiple conditional branches

1 files changed, 47 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/stackir/StackCompilerIfTests.java b/src/test/java/org/shedlang/compiler/stackir/StackCompilerIfTests.java
index 49c47208..a07e9aa6 100644
--- a/src/test/java/org/shedlang/compiler/stackir/StackCompilerIfTests.java
+++ b/src/test/java/org/shedlang/compiler/stackir/StackCompilerIfTests.java
@@ -43,4 +43,51 @@ public class StackCompilerIfTests {
             )
         ));
     }
+
+    @Test
+    public void ifWithMultipleConditionalBranchesIsCompiledToNestedIfInstructions() {
+        var node = Typed.if_(
+            List.of(
+                Typed.conditionalBranch(
+                    Typed.reference("c"),
+                    Typed.block(List.of(
+                        Typed.blockReturn(Typed.intLiteral(42))
+                    ))
+                ),
+                Typed.conditionalBranch(
+                    Typed.reference("d"),
+                    Typed.block(List.of(
+                        Typed.blockReturn(Typed.intLiteral(1))
+                    ))
+                )
+            ),
+            Typed.block(List.of(
+                Typed.functionReturn(Typed.stringLiteral("x"))
+            )),
+            Types.INT
+        );
+
+        var instructions = StackCompiler.compileExpression(node);
+
+        assertThat(instructions, isSequence(
+            isLocalLoadInstruction("c"),
+            isIfInstruction(
+                isSequence(
+                    isPushIntInstruction(42)
+                ),
+                isSequence(
+                    isLocalLoadInstruction("d"),
+                    isIfInstruction(
+                        isSequence(
+                            isPushIntInstruction(1)
+                        ),
+                        isSequence(
+                            isPushStringInstruction("x"),
+                            isReturnInstruction()
+                        )
+                    )
+                )
+            )
+        ));
+    }
 }