test: Add ensure() macro

Signed-off-by: Diogo Behrens <[email protected]>

Author: db7

test/CMakeLists.txt

@@ -2,7 +2,7 @@
 # SPDX-License-Identifier: MIT
 
 add_subdirectory(checkers)
-
+include_directories(include)
 add_compile_options(-DVATOMIC_BUILTINS)
 set(TSAN_mp on)
 set(TSAN_simple-tsan on)

test/include/dice/ensure.h

@@ -0,0 +1,20 @@
+/*
+ * Copyright (C) Huawei Technologies Co., Ltd. 2025. All rights reserved.
+ * SPDX-License-Identifier: MIT
+ */
+#ifndef DICE_ENSURE_H
+#define DICE_ENSURE_H
+
+#ifdef NDEBUG
+    #include <dice/log.h>
+    #define ensure(COND)                                                       \
+        {                                                                      \
+            if (!(COND)) {                                                     \
+                log_fatal("error: %s", #COND);                                 \
+            }                                                                  \
+        }
+#else
+    #include <assert.h>
+    #define ensure assert
+#endif
+#endif /* DICE_ENSURE_H */

test/log/log_levels.c

@@ -3,42 +3,48 @@
  * SPDX-License-Identifier: MIT
  */
 
+#include <stdbool.h>
+#include <string.h>
+
+#include <dice/ensure.h>
 #include <dice/interpose.h>
 #include <dice/log.h>
-#include <string.h>
-#include <assert.h>
-#include <stdbool.h>
 
 #define MAX_EXP 16
 static char *strings[MAX_EXP] = {0};
-static char **head = &strings[0];
-static char **tail = &strings[0];
+static char **head            = &strings[0];
+static char **tail            = &strings[0];
 
-INTERPOSE(ssize_t, write, int fd, const void *buf, size_t count) {
+INTERPOSE(ssize_t, write, int fd, const void *buf, size_t count)
+{
     static int nest = 0;
     if (nest == 1) {
         return REAL(write, fd, buf, count);
     }
-    assert(nest == 0);
+    ensure(nest == 0);
     nest++;
-    assert(*tail);
-    assert(tail < head);
-    if (strncmp((char*) buf, *tail, count) != 0) {
+    ensure(*tail);
+    ensure(tail < head);
+    if (strncmp((char *)buf, *tail, count) != 0) {
         log_printf("exp: %s\n", *tail);
-        log_printf("buf: %s\n", (char*)buf);
+        log_printf("buf: %s\n", (char *)buf);
         abort();
     }
     tail++;
     nest--;
     return count;
 }
 
-static void expect(char *e) {
-    assert(head < strings + MAX_EXP);
+static void
+expect(char *e)
+{
+    ensure(head < strings + MAX_EXP);
     *head = e;
     head++;
 }
-static bool empty(void) {
+static bool
+empty(void)
+{
     return head == tail;
 }
 
@@ -49,7 +55,7 @@ main()
     // this should always work
     expect("print");
     log_printf("print");
-    assert(empty());
+    ensure(empty());
 
     // this should always work, but we remove the abort to about actually
     // aborting
@@ -60,15 +66,15 @@ main()
 #define abort()
     log_fatal("fatal");
 #undef abort
-    assert(empty());
+    ensure(empty());
 
 #if LOG_LEVEL_ >= LOG_LEVEL_INFO
     printf("level >= info\n");
     expect("dice: ");
     expect("info");
     expect("\n");
     log_info("info");
-    assert(empty());
+    ensure(empty());
 #endif
 
 #if LOG_LEVEL_ >= LOG_LEVEL_DEBUG
@@ -77,9 +83,8 @@ main()
     expect("debug");
     expect("\n");
     log_debug("debug");
-    assert(empty());
+    ensure(empty());
 #endif
 
-
     return 0;
 }