log: Test case for log levels

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

Author: db7

include/dice/log.h

@@ -37,6 +37,10 @@ DICE_WEAK caslock_t log_lock;
     #define LOG_SUFFIX "\n"
 #endif
 
+#define LOG_PASTE(LEVEL)  LOG_LEVEL##_##LEVEL
+#define LOG_EXPAND(LEVEL) LOG_PASTE(LEVEL)
+#define LOG_LEVEL_ LOG_EXPAND(LOG_LEVEL)
+
 #define LOG_MAX_LEN 1024
 #define log_printf(fmt, ...)                                                   \
     do {                                                                       \
@@ -48,10 +52,8 @@ DICE_WEAK caslock_t log_lock;
         }                                                                      \
     } while (0)
 
-#define LOG_PASTE(LEVEL)  LOG_LEVEL##_##LEVEL
-#define LOG_EXPAND(LEVEL) LOG_PASTE(LEVEL)
 
-#if LOG_EXPAND(LOG_LEVEL) >= LOG_LEVEL_DEBUG
+#if LOG_LEVEL_ >= LOG_LEVEL_DEBUG
     #define log_debug(fmt, ...)                                                \
         do {                                                                   \
             LOG_LOCK_ACQUIRE;                                                  \
@@ -64,7 +66,7 @@ DICE_WEAK caslock_t log_lock;
     #define log_debug log_none
 #endif
 
-#if LOG_EXPAND(LOG_LEVEL) >= LOG_LEVEL_INFO
+#if LOG_LEVEL_ >= LOG_LEVEL_INFO
     #define log_info(fmt, ...)                                                 \
         do {                                                                   \
             LOG_LOCK_ACQUIRE;                                                  \

test/CMakeLists.txt

@@ -52,3 +52,4 @@ add_custom_target(expand-tests)
 add_subdirectory(interpose)
 add_subdirectory(tsan)
 add_subdirectory(order)
+add_subdirectory(log)

test/log/CMakeLists.txt

@@ -0,0 +1,17 @@
+# Copyright (C) Huawei Technologies Co., Ltd. 2025. All rights reserved.
+# SPDX-License-Identifier: MIT
+
+add_executable(log-fatal-test log_levels.c)
+target_link_libraries(log-fatal-test PRIVATE dice.h)
+target_compile_options(log-fatal-test PRIVATE -DLOG_LEVEL=FATAL)
+add_test(NAME log-fatal-test COMMAND log-fatal-test)
+
+add_executable(log-info-test log_levels.c)
+target_link_libraries(log-info-test PRIVATE dice.h)
+target_compile_options(log-info-test PRIVATE -DLOG_LEVEL=INFO)
+add_test(NAME log-info-test COMMAND log-info-test)
+
+add_executable(log-debug-test log_levels.c)
+target_link_libraries(log-debug-test PRIVATE dice.h)
+target_compile_options(log-debug-test PRIVATE -DLOG_LEVEL=DEBUG)
+add_test(NAME log-debug-test COMMAND log-debug-test)

test/log/log_levels.c

@@ -0,0 +1,85 @@
+/*
+ * Copyright (C) Huawei Technologies Co., Ltd. 2025. All rights reserved.
+ * SPDX-License-Identifier: MIT
+ */
+
+#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];
+
+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);
+    nest++;
+    assert(*tail);
+    assert(tail < head);
+    if (strncmp((char*) buf, *tail, count) != 0) {
+        log_printf("exp: %s\n", *tail);
+        log_printf("buf: %s\n", (char*)buf);
+        abort();
+    }
+    tail++;
+    nest--;
+    return count;
+}
+
+static void expect(char *e) {
+    assert(head < strings + MAX_EXP);
+    *head = e;
+    head++;
+}
+static bool empty(void) {
+    return head == tail;
+}
+
+
+int
+main()
+{
+    // this should always work
+    expect("print");
+    log_printf("print");
+    assert(empty());
+
+    // this should always work, but we remove the abort to about actually
+    // aborting
+    printf("level >= fatal\n");
+    expect("dice: ");
+    expect("fatal");
+    expect("\n");
+#define abort()
+    log_fatal("fatal");
+#undef abort
+    assert(empty());
+
+#if LOG_LEVEL_ >= LOG_LEVEL_INFO
+    printf("level >= info\n");
+    expect("dice: ");
+    expect("info");
+    expect("\n");
+    log_info("info");
+    assert(empty());
+#endif
+
+#if LOG_LEVEL_ >= LOG_LEVEL_DEBUG
+    printf("level >= debug \n");
+    expect("dice: ");
+    expect("debug");
+    expect("\n");
+    log_debug("debug");
+    assert(empty());
+#endif
+
+
+    return 0;
+}