test: UT for core_log_set_function()

src/test/core_log/TESTS.py

@@ -46,3 +46,7 @@ class TEST5(CORE_LOG):
 
 class TEST6(CORE_LOG):
  test_case = 'test_happy_day'
+
+
+class TEST7(CORE_LOG):
+ test_case = 'test_set_custom_function'

src/test/core_log/core_log.c

@@ -2,7 +2,8 @@
 /* Copyright 2024, Intel Corporation */
 
 /*
- * core_log.c -- unit test for core_log() and core_log_va()
+ * core_log.c -- unit test for core_log(), core_log_va() and
+ * core_log_set_function()
  */
 
 #include <syslog.h>
@@ -172,11 +173,25 @@ test_level_gt_threshold(const struct test_case *tc, int argc, char *argv[])
 }
 
 static int
-test_happy_day(const struct test_case *tc, int argc, char *argv[])
+test_happy_day_helper(core_log_function *log_function)
 {
  /* Pass the message all the way to the logging function. */
  core_log_set_threshold(CORE_LOG_THRESHOLD, CORE_LOG_LEVEL_ERROR);
 
+ /*
+ * Disable the validation as custom_log_function() may be called from
+ * core_log_set_function().
+ */
+ if (log_function == CORE_LOG_USE_DEFAULT_FUNCTION) {
+ FUNC_MOCK_RCOUNTER_SET(core_log_default_function,
+ NOT_VALIDATED_CALL);
+ } else {
+ FUNC_MOCK_RCOUNTER_SET(custom_log_function,
+ NOT_VALIDATED_CALL);
+ }
+
+ core_log_set_function(log_function);
+
  reset_mocks();
 
  /* set the expectations */
@@ -195,11 +210,30 @@ test_happy_day(const struct test_case *tc, int argc, char *argv[])
  UT_ASSERTeq(RCOUNTER(last_error_msg_get), CALLED);
  UT_ASSERTeq(RCOUNTER(vsnprintf), CALLED);
  UT_ASSERTeq(RCOUNTER(__xpg_strerror_r), CALLED);
- UT_ASSERTeq(RCOUNTER(core_log_default_function), CALLED);
+ if (log_function == CORE_LOG_USE_DEFAULT_FUNCTION) {
+ UT_ASSERTeq(RCOUNTER(core_log_default_function), CALLED);
+ UT_ASSERTeq(RCOUNTER(custom_log_function), NOT_CALLED);
+ } else {
+ UT_ASSERTeq(RCOUNTER(core_log_default_function), NOT_CALLED);
+ UT_ASSERTeq(RCOUNTER(custom_log_function), CALLED);
+ }
 
  return NO_ARGS_CONSUMED;
 }
 
+static int
+test_happy_day(const struct test_case *tc, int argc, char *argv[])
+{
+ return test_happy_day_helper(CORE_LOG_USE_DEFAULT_FUNCTION);
+}
+
+/* Happy day scenario with custom logging function */
+static int
+test_set_custom_function(const struct test_case *tc, int argc, char *argv[])
+{
+ return test_happy_day_helper(custom_log_function);
+}
+
 static struct test_case test_cases[] = {
  TEST_CASE(test_CORE_LOG_LEVEL_ERROR_LAST),
  TEST_CASE(test_vsnprintf_fail),
@@ -208,6 +242,7 @@ static struct test_case test_cases[] = {
  TEST_CASE(test_strerror_r_fail),
  TEST_CASE(test_level_gt_threshold),
  TEST_CASE(test_happy_day),
+ TEST_CASE(test_set_custom_function),
 };
 
 #define NTESTS ARRAY_SIZE(test_cases)

src/test/core_log/core_log_common.c

@@ -84,6 +84,34 @@ FUNC_MOCK_RUN_DEFAULT {
 }
 FUNC_MOCK_END
 
+FUNC_MOCK_NONSTATIC(custom_log_function, void, enum core_log_level level,
+ const char *file_name, const int line_no, const char *function_name,
+ const char *message)
+ FUNC_MOCK_RUN(VALIDATED_CALL) {
+ UT_ASSERTeq(level, Log_function_.exp_level);
+ UT_ASSERTstreq(file_name, FILE_NAME);
+ UT_ASSERTeq(line_no, LINE_NO);
+ UT_ASSERTstreq(function_name, FUNC_NAME);
+ if (Common.use_last_error_msg) {
+ UT_ASSERTeq(message, LAST_ERROR_MSG_MOCK);
+ } else {
+ UT_ASSERTne(message, LAST_ERROR_MSG_MOCK);
+ }
+ return;
+ }
+FUNC_MOCK_RUN_DEFAULT {
+ _FUNC_REAL(custom_log_function)(level, file_name, line_no,
+ function_name, message);
+}
+FUNC_MOCK_END
+
+void
+custom_log_function(enum core_log_level level, const char *file_name,
+ const int line_no, const char *function_name, const char *message)
+{
+ SUPPRESS_UNUSED(level, file_name, line_no, function_name, message);
+}
+
 /* helpers */
 
 void
@@ -93,6 +121,7 @@ reset_mocks(void)
  FUNC_MOCK_RCOUNTER_SET(vsnprintf, VALIDATED_CALL);
  FUNC_MOCK_RCOUNTER_SET(__xpg_strerror_r, VALIDATED_CALL);
  FUNC_MOCK_RCOUNTER_SET(core_log_default_function, VALIDATED_CALL);
+ FUNC_MOCK_RCOUNTER_SET(custom_log_function, VALIDATED_CALL);
 }
 
 void

src/test/core_log/core_log_common.h

@@ -23,6 +23,7 @@
 #define VALIDATED_CALL 127
 #define NOT_CALLED VALIDATED_CALL
 #define CALLED (VALIDATED_CALL + 1)
+#define NOT_VALIDATED_CALL 0
 
 extern struct common_ctx {
  bool use_last_error_msg;
@@ -49,6 +50,7 @@ FUNC_MOCK_EXTERN(last_error_msg_get);
 FUNC_MOCK_EXTERN(vsnprintf);
 FUNC_MOCK_EXTERN(__xpg_strerror_r);
 FUNC_MOCK_EXTERN(core_log_default_function);
+FUNC_MOCK_EXTERN(custom_log_function);
 
 /* helpers */
 
@@ -60,3 +62,6 @@ void test_strerror_r_fail_helper(bool before_glibc_2_13);
 
 void test_log_function_call_helper(enum core_log_level level,
  bool call_log_function);
+
+void custom_log_function(enum core_log_level level, const char *file_name,
+ const int line_no, const char *function_name, const char *message);