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
cmd_ut_category(const char * name,struct unit_test * tests,int n_ents,int argc,char * const argv[])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_OPTEE
47 U_BOOT_CMD_MKENT(optee, CONFIG_SYS_MAXARGS, 1, do_ut_optee, "", ""),
48 #endif
49 #ifdef CONFIG_UT_OVERLAY
50 U_BOOT_CMD_MKENT(overlay, CONFIG_SYS_MAXARGS, 1, do_ut_overlay, "", ""),
51 #endif
52 #ifdef CONFIG_UT_LIB
53 U_BOOT_CMD_MKENT(lib, CONFIG_SYS_MAXARGS, 1, do_ut_lib, "", ""),
54 #endif
55 #ifdef CONFIG_UT_TIME
56 U_BOOT_CMD_MKENT(time, CONFIG_SYS_MAXARGS, 1, do_ut_time, "", ""),
57 #endif
58 #if CONFIG_IS_ENABLED(UT_UNICODE) && !defined(API_BUILD)
59 U_BOOT_CMD_MKENT(unicode, CONFIG_SYS_MAXARGS, 1, do_ut_unicode, "", ""),
60 #endif
61 #ifdef CONFIG_SANDBOX
62 U_BOOT_CMD_MKENT(compression, CONFIG_SYS_MAXARGS, 1, do_ut_compression,
63 "", ""),
64 U_BOOT_CMD_MKENT(bloblist, CONFIG_SYS_MAXARGS, 1, do_ut_bloblist,
65 "", ""),
66 #endif
67 };
68
do_ut_all(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])69 static int do_ut_all(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
70 {
71 int i;
72 int retval;
73 int any_fail = 0;
74
75 for (i = 1; i < ARRAY_SIZE(cmd_ut_sub); i++) {
76 printf("----Running %s tests----\n", cmd_ut_sub[i].name);
77 retval = cmd_ut_sub[i].cmd(cmdtp, flag, 1, &cmd_ut_sub[i].name);
78 if (!any_fail)
79 any_fail = retval;
80 }
81
82 return any_fail;
83 }
84
do_ut(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])85 static int do_ut(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
86 {
87 cmd_tbl_t *cp;
88
89 if (argc < 2)
90 return CMD_RET_USAGE;
91
92 /* drop initial "ut" arg */
93 argc--;
94 argv++;
95
96 cp = find_cmd_tbl(argv[0], cmd_ut_sub, ARRAY_SIZE(cmd_ut_sub));
97
98 if (cp)
99 return cp->cmd(cmdtp, flag, argc, argv);
100
101 return CMD_RET_USAGE;
102 }
103
104 #ifdef CONFIG_SYS_LONGHELP
105 static char ut_help_text[] =
106 "all - execute all enabled tests\n"
107 #ifdef CONFIG_SANDBOX
108 "ut bloblist - Test bloblist implementation\n"
109 "ut compression - Test compressors and bootm decompression\n"
110 #endif
111 #ifdef CONFIG_UT_DM
112 "ut dm [test-name]\n"
113 #endif
114 #ifdef CONFIG_UT_ENV
115 "ut env [test-name]\n"
116 #endif
117 #ifdef CONFIG_UT_LIB
118 "ut lib [test-name] - test library functions\n"
119 #endif
120 #ifdef CONFIG_UT_OPTEE
121 "ut optee [test-name]\n"
122 #endif
123 #ifdef CONFIG_UT_OVERLAY
124 "ut overlay [test-name]\n"
125 #endif
126 #ifdef CONFIG_UT_TIME
127 "ut time - Very basic test of time functions\n"
128 #endif
129 #if defined(CONFIG_UT_UNICODE) && \
130 !defined(CONFIG_SPL_BUILD) && !defined(API_BUILD)
131 "ut unicode [test-name] - test Unicode functions\n"
132 #endif
133 ;
134 #endif /* CONFIG_SYS_LONGHELP */
135
136 U_BOOT_CMD(
137 ut, CONFIG_SYS_MAXARGS, 1, do_ut,
138 "unit tests", ut_help_text
139 );
140