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_TIME 50 U_BOOT_CMD_MKENT(time, CONFIG_SYS_MAXARGS, 1, do_ut_time, "", ""), 51 #endif 52 #ifdef CONFIG_SANDBOX 53 U_BOOT_CMD_MKENT(compression, CONFIG_SYS_MAXARGS, 1, do_ut_compression, 54 "", ""), 55 #endif 56 }; 57 58 static int do_ut_all(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) 59 { 60 int i; 61 int retval; 62 int any_fail = 0; 63 64 for (i = 1; i < ARRAY_SIZE(cmd_ut_sub); i++) { 65 printf("----Running %s tests----\n", cmd_ut_sub[i].name); 66 retval = cmd_ut_sub[i].cmd(cmdtp, flag, 1, &cmd_ut_sub[i].name); 67 if (!any_fail) 68 any_fail = retval; 69 } 70 71 return any_fail; 72 } 73 74 static int do_ut(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) 75 { 76 cmd_tbl_t *cp; 77 78 if (argc < 2) 79 return CMD_RET_USAGE; 80 81 /* drop initial "ut" arg */ 82 argc--; 83 argv++; 84 85 cp = find_cmd_tbl(argv[0], cmd_ut_sub, ARRAY_SIZE(cmd_ut_sub)); 86 87 if (cp) 88 return cp->cmd(cmdtp, flag, argc, argv); 89 90 return CMD_RET_USAGE; 91 } 92 93 #ifdef CONFIG_SYS_LONGHELP 94 static char ut_help_text[] = 95 "all - execute all enabled tests\n" 96 #ifdef CONFIG_UT_DM 97 "ut dm [test-name]\n" 98 #endif 99 #ifdef CONFIG_UT_ENV 100 "ut env [test-name]\n" 101 #endif 102 #ifdef CONFIG_UT_OVERLAY 103 "ut overlay [test-name]\n" 104 #endif 105 #ifdef CONFIG_UT_TIME 106 "ut time - Very basic test of time functions\n" 107 #endif 108 #ifdef CONFIG_SANDBOX 109 "ut compression - Test compressors and bootm decompression\n" 110 #endif 111 ; 112 #endif 113 114 U_BOOT_CMD( 115 ut, CONFIG_SYS_MAXARGS, 1, do_ut, 116 "unit tests", ut_help_text 117 ); 118