From 9f7c92ba7dd23debfc37cf6f7f88b0dd8c926b05 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 26 Nov 2023 19:01:54 +0000
Subject: Test exception handler is discarded after try

---
 .../backends/llvm/LlvmPatioCodeGenTests.java       |  6 +++
 .../shedlang/compiler/patio/PatioExampleTests.java | 59 ++++++++++++++++++++++
 2 files changed, 65 insertions(+)

(limited to 'src/test/java')

diff --git a/src/test/java/org/shedlang/compiler/backends/llvm/LlvmPatioCodeGenTests.java b/src/test/java/org/shedlang/compiler/backends/llvm/LlvmPatioCodeGenTests.java
index a1af75c4..65db8040 100644
--- a/src/test/java/org/shedlang/compiler/backends/llvm/LlvmPatioCodeGenTests.java
+++ b/src/test/java/org/shedlang/compiler/backends/llvm/LlvmPatioCodeGenTests.java
@@ -90,4 +90,10 @@ public class LlvmPatioCodeGenTests extends PatioExampleTests {
     public void multipleCatchClausesCanBeAttachedToSameTry() {
 
     }
+
+    @Disabled("WIP")
+    @Test
+    public void exceptionHandlerIsDiscardedAfterTry() {
+
+    }
 }
diff --git a/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java b/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java
index 3d590111..e95fc3fc 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java
@@ -455,6 +455,65 @@ public abstract class PatioExampleTests {
         assertThat(result, isSilentExit(1));
     }
 
+    @Test
+    public void exceptionHandlerIsDiscardedAfterTry() throws IOException, InterruptedException {
+        var patioFunction = PatioFunctionDefinition.builder()
+            .withName(FUNCTION_NAME)
+            .withReturnType(PatioTypes.INT)
+            .addLocal(Patio.val("r1", PatioTypes.INT))
+            .addLocal(Patio.val("r2", PatioTypes.INT))
+            .addLocal(Patio.val("e1", PatioTypes.INT))
+            .addLocal(Patio.val("e2", PatioTypes.INT))
+            .addBodyInstruction(Patio.try_(
+                List.of(Patio.localRef("r1")),
+                List.of(
+                    Patio.catchClause(
+                        "Error",
+                        Patio.localRef("e1"),
+                        Patio.block(
+                            List.of(),
+                            List.of(Patio.int_(1))
+                        )
+                    )
+                ),
+                Patio.block(
+                    List.of(
+                        Patio.try_(
+                            List.of(Patio.localRef("r2")),
+                            List.of(
+                                Patio.catchClause(
+                                    "Error",
+                                    Patio.localRef("e2"),
+                                    Patio.block(
+                                        List.of(),
+                                        List.of(Patio.int_(2))
+                                    )
+                                )
+                            ),
+                            Patio.block(
+                                List.of(),
+                                List.of(Patio.int_(0))
+                            )
+                        ),
+                        Patio.throw_("Error", Patio.int_(0))
+                    ),
+                    List.of(Patio.localRef("r2"))
+                )
+            ))
+            .addBodyInstruction(Patio.return_(
+                Patio.localRef("r1")
+            ))
+            .build();
+        var patio = Patio.compilationUnit(List.of(
+            Patio.exceptionTypeDefinition("Error", PatioTypes.INT),
+            patioFunction
+        ));
+
+        var result = run(patio);
+
+        assertThat(result, isSilentExit(1));
+    }
+
     @Test
     public void vars() throws IOException, InterruptedException {
         var patioFunction = PatioFunctionDefinition.builder()
-- 
cgit v1.2.3