test: Expand choose expr test.

trailofbits · Sep 4, 2024 · f6d12ec · f6d12ec
1 parent f8ecacf
commit f6d12ec
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/test/vast/Dialect/HighLevel/choose-expr-a.c b/test/vast/Dialect/HighLevel/choose-expr-a.c
@@ -9,5 +9,19 @@ void fn() {
     // CHECK: hl.value.yield {{%.*}} : !hl.lvalue<!hl.int>
     int z = __builtin_choose_expr(0, x, y);
     // CHECK: hl.choose_expr cond true : !hl.lvalue<!hl.int> {
-    int w = __builtin_choose_expr(1, x, y);
+    z = __builtin_choose_expr(1, x, y);
+
+    // CHECK: hl.choose_expr cond true : !hl.lvalue<!hl.int> {
+    // CHECK: hl.value.yield {{%.*}} : !hl.lvalue<!hl.int>
+    // CHECK: hl.value.yield {{%.*}} : !hl.void
+    z = __builtin_choose_expr(1, x, (void) 0);
+    // CHECK: hl.choose_expr cond false : !hl.lvalue<!hl.int> {
+    // CHECK: hl.value.yield {{%.*}} : !hl.void
+    // CHECK: hl.value.yield {{%.*}} : !hl.lvalue<!hl.int>
+    z = __builtin_choose_expr(0, (void) 0, x);
+
+    // CHECK: hl.choose_expr cond true : !hl.lvalue<!hl.int> {
+    // CHECK: hl.value.yield {{%.*}} : !hl.lvalue<!hl.int>
+    // CHECK: hl.value.yield {{%.*}} : !hl.char
+    z = __builtin_choose_expr(1, x, (char) 0);
 }