codegen: fix switch expressions for enums with payloads

1 files changed, 12 insertions, 4 deletions

diff --git a/test/self_hosted.zig b/test/self_hosted.zig
index 048b820ec9..645726c657 100644
--- a/test/self_hosted.zig
+++ b/test/self_hosted.zig
@@ -51,16 +51,24 @@ error SecondError;
 
 #attribute("test")
 fn constant_enum_with_payload() {
-    should_be_empty(AnEnumWithPayload.Empty);
-    should_be_not_empty(AnEnumWithPayload.Full(13));
+    var empty = AnEnumWithPayload.Empty;
+    var full = AnEnumWithPayload.Full(13);
+    should_be_empty(empty);
+    should_be_not_empty(full);
 }
 
 fn should_be_empty(x: AnEnumWithPayload) {
-    if (x != AnEnumWithPayload.Empty) unreachable{}
+    switch (x) {
+        AnEnumWithPayload.Empty => {},
+        else => unreachable{},
+    }
 }
 
 fn should_be_not_empty(x: AnEnumWithPayload) {
-    if (x == AnEnumWithPayload.Empty) unreachable{}
+    switch (x) {
+        AnEnumWithPayload.Empty => unreachable{},
+        else => {},
+    }
 }
 
 enum AnEnumWithPayload {