1 // SPDX-License-Identifier: GPL-2.0 2 /* 3 * (C) Copyright 2015 4 * Joe Hershberger, National Instruments, joe.hershberger@ni.com 5 */ 6 7 #include <common.h> 8 #include <command.h> 9 #include <test/suites.h> 10 #include <test/test.h> 11 12 static int do_ut_all(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]); 13 14 int cmd_ut_category(const char *name, struct unit_test *tests, int n_ents, 15 int argc, char * const argv[]) 16 { 17 struct unit_test_state uts = { .fail_count = 0 }; 18 struct unit_test *test; 19 20 if (argc == 1) 21 printf("Running %d %s tests\n", n_ents, name); 22 23 for (test = tests; test < tests + n_ents; test++) { 24 if (argc > 1 && strcmp(argv[1], test->name)) 25 continue; 26 printf("Test: %s\n", test->name); 27 28 uts.start = mallinfo(); 29 30 test->func(&uts); 31 } 32 33 printf("Failures: %d\n", uts.fail_count); 34 35 return uts.fail_count ? CMD_RET_FAILURE : 0; 36 } 37 38 static cmd_tbl_t cmd_ut_sub[] = { 39 U_BOOT_CMD_MKENT(all, CONFIG_SYS_MAXARGS, 1, do_ut_all, "", ""), 40 #if defined(CONFIG_UT_DM) 41 U_BOOT_CMD_MKENT(dm, CONFIG_SYS_MAXARGS, 1, do_ut_dm, "", ""), 42 #endif 43 #if defined(CONFIG_UT_ENV) 44 U_BOOT_CMD_MKENT(env, CONFIG_SYS_MAXARGS, 1, do_ut_env, "", ""), 45 #endif 46 #ifdef CONFIG_UT_OVERLAY 47 U_BOOT_CMD_MKENT(overlay, CONFIG_SYS_MAXARGS, 1, do_ut_overlay, "", ""), 48 #endif 49 #ifdef CONFIG_UT_LIB 50 U_BOOT_CMD_MKENT(lib, CONFIG_SYS_MAXARGS, 1, do_ut_lib, "", ""), 51 #endif 52 #ifdef CONFIG_UT_TIME 53 U_BOOT_CMD_MKENT(time, CONFIG_SYS_MAXARGS, 1, do_ut_time, "", ""), 54 #endif 55 #if CONFIG_IS_ENABLED(UT_UNICODE) && !defined(API_BUILD) 56 U_BOOT_CMD_MKENT(unicode, CONFIG_SYS_MAXARGS, 1, do_ut_unicode, "", ""), 57 #endif 58 #ifdef CONFIG_SANDBOX 59 U_BOOT_CMD_MKENT(compression, CONFIG_SYS_MAXARGS, 1, do_ut_compression, 60 "", ""), 61 U_BOOT_CMD_MKENT(bloblist, CONFIG_SYS_MAXARGS, 1, do_ut_bloblist, 62 "", ""), 63 #endif 64 }; 65 66 static int do_ut_all(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) 67 { 68 int i; 69 int retval; 70 int any_fail = 0; 71 72 for (i = 1; i < ARRAY_SIZE(cmd_ut_sub); i++) { 73 printf("----Running %s tests----\n", cmd_ut_sub[i].name); 74 retval = cmd_ut_sub[i].cmd(cmdtp, flag, 1, &cmd_ut_sub[i].name); 75 if (!any_fail) 76 any_fail = retval; 77 } 78 79 return any_fail; 80 } 81 82 static int do_ut(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) 83 { 84 cmd_tbl_t *cp; 85 86 if (argc < 2) 87 return CMD_RET_USAGE; 88 89 /* drop initial "ut" arg */ 90 argc--; 91 argv++; 92 93 cp = find_cmd_tbl(argv[0], cmd_ut_sub, ARRAY_SIZE(cmd_ut_sub)); 94 95 if (cp) 96 return cp->cmd(cmdtp, flag, argc, argv); 97 98 return CMD_RET_USAGE; 99 } 100 101 #ifdef CONFIG_SYS_LONGHELP 102 static char ut_help_text[] = 103 "all - execute all enabled tests\n" 104 #ifdef CONFIG_SANDBOX 105 "ut bloblist - Test bloblist implementation\n" 106 "ut compression - Test compressors and bootm decompression\n" 107 #endif 108 #ifdef CONFIG_UT_DM 109 "ut dm [test-name]\n" 110 #endif 111 #ifdef CONFIG_UT_ENV 112 "ut env [test-name]\n" 113 #endif 114 #ifdef CONFIG_UT_LIB 115 "ut lib [test-name] - test library functions\n" 116 #endif 117 #ifdef CONFIG_UT_OVERLAY 118 "ut overlay [test-name]\n" 119 #endif 120 #ifdef CONFIG_UT_TIME 121 "ut time - Very basic test of time functions\n" 122 #endif 123 #if defined(CONFIG_UT_UNICODE) && \ 124 !defined(CONFIG_SPL_BUILD) && !defined(API_BUILD) 125 "ut unicode [test-name] - test Unicode functions\n" 126 #endif 127 ; 128 #endif /* CONFIG_SYS_LONGHELP */ 129 130 U_BOOT_CMD( 131 ut, CONFIG_SYS_MAXARGS, 1, do_ut, 132 "unit tests", ut_help_text 133 ); 134