diff --git a/components/utilities/utest/utest.c b/components/utilities/utest/utest.c index 0ee0b94bf5..a0a42d7438 100644 --- a/components/utilities/utest/utest.c +++ b/components/utilities/utest/utest.c @@ -8,9 +8,10 @@ * 2018-11-19 MurphyZhao the first version */ -#include "utest.h" #include -#include +#include +#include "utest.h" +#include #undef DBG_SECTION_NAME #undef DBG_LEVEL @@ -31,6 +32,7 @@ #error "RT_CONSOLEBUF_SIZE is less than 256!" #endif +static rt_uint8_t utest_log_lv = UTEST_LOG_ALL; static utest_tc_export_t tc_table = RT_NULL; static rt_size_t tc_num; static struct utest local_utest = {UTEST_PASSED, 0, 0}; @@ -39,6 +41,14 @@ static struct utest local_utest = {UTEST_PASSED, 0, 0}; #pragma section="UtestTcTab" #endif +void utest_log_lv_set(rt_uint8_t lv) +{ + if (lv == UTEST_LOG_ALL || lv == UTEST_LOG_ASSERT) + { + utest_log_lv = lv; + } +} + int utest_init(void) { /* initialize the utest commands table.*/ @@ -200,7 +210,10 @@ void utest_assert(int value, const char *file, int line, const char *func, const } else { - LOG_D("[ OK ] [ unit ] (%s:%d) is passed", func, line); + if (utest_log_lv == UTEST_LOG_ALL) + { + LOG_D("[ OK ] [ unit ] (%s:%d) is passed", func, line); + } local_utest.error = UTEST_PASSED; local_utest.passed_num ++; } diff --git a/components/utilities/utest/utest_log.h b/components/utilities/utest/utest_log.h index c954046b76..9f38d3f9c8 100644 --- a/components/utilities/utest/utest_log.h +++ b/components/utilities/utest/utest_log.h @@ -11,6 +11,8 @@ #ifndef __UTEST_LOG_H__ #define __UTEST_LOG_H__ +#include + #define UTEST_DEBUG #undef DBG_SECTION_NAME @@ -28,4 +30,9 @@ #define DBG_COLOR #include +#define UTEST_LOG_ALL (1u) +#define UTEST_LOG_ASSERT (2u) + +void utest_log_lv_set(rt_uint8_t lv); + #endif /* __UTEST_LOG_H__ */