fix comptime switch on enum with ref payload

See #43

1 files changed, 17 insertions, 0 deletions

diff --git a/test/cases/switch.zig b/test/cases/switch.zig
index ccacbd1c78..dfec42ea87 100644
--- a/test/cases/switch.zig
+++ b/test/cases/switch.zig
@@ -106,6 +106,22 @@ fn switchProngWithVarFn(a: &const SwitchProngWithVarEnum) {
     }
 }
 
+test "switch on enum using pointer capture" {
+    testSwitchEnumPtrCapture();
+    comptime testSwitchEnumPtrCapture();
+}
+
+fn testSwitchEnumPtrCapture() {
+    var value = SwitchProngWithVarEnum.One { 1234 };
+    switch (value) {
+        SwitchProngWithVarEnum.One => |*x| *x += 1,
+        else => unreachable,
+    }
+    switch (value) {
+        SwitchProngWithVarEnum.One => |x| assert(x == 1235),
+        else => unreachable,
+    }
+}
 
 test "switch with multiple expressions" {
     const x = switch (returnsFive()) {
@@ -188,3 +204,4 @@ fn testSwitchHandleAllCasesRange(x: u8) -> u8 {
         204 ... 255 => 3,
     }
 }
+