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 <console.h>
10 #include <test/suites.h>
11 #include <test/test.h>
12 #include <test/ut.h>
13
14 static int do_ut_all(struct cmd_tbl *cmdtp, int flag, int argc,
15 char *const argv[]);
16
17 static int do_ut_info(struct cmd_tbl *cmdtp, int flag, int argc,
18 char *const argv[]);
19
cmd_ut_category(const char * name,const char * prefix,struct unit_test * tests,int n_ents,int argc,char * const argv[])20 int cmd_ut_category(const char *name, const char *prefix,
21 struct unit_test *tests, int n_ents,
22 int argc, char *const argv[])
23 {
24 const char *test_insert = NULL;
25 int runs_per_text = 1;
26 bool force_run = false;
27 int ret;
28
29 while (argc > 1 && *argv[1] == '-') {
30 const char *str = argv[1];
31
32 switch (str[1]) {
33 case 'r':
34 runs_per_text = dectoul(str + 2, NULL);
35 break;
36 case 'f':
37 force_run = true;
38 break;
39 case 'I':
40 test_insert = str + 2;
41 break;
42 }
43 argv++;
44 argc--;
45 }
46
47 ret = ut_run_list(name, prefix, tests, n_ents,
48 argc > 1 ? argv[1] : NULL, runs_per_text, force_run,
49 test_insert);
50
51 return ret ? CMD_RET_FAILURE : 0;
52 }
53
54 static struct cmd_tbl cmd_ut_sub[] = {
55 U_BOOT_CMD_MKENT(all, CONFIG_SYS_MAXARGS, 1, do_ut_all, "", ""),
56 U_BOOT_CMD_MKENT(info, 1, 1, do_ut_info, "", ""),
57 #ifdef CONFIG_BOOTSTD
58 U_BOOT_CMD_MKENT(bootstd, CONFIG_SYS_MAXARGS, 1, do_ut_bootstd,
59 "", ""),
60 #endif
61 U_BOOT_CMD_MKENT(common, CONFIG_SYS_MAXARGS, 1, do_ut_common, "", ""),
62 #if defined(CONFIG_UT_DM)
63 U_BOOT_CMD_MKENT(dm, CONFIG_SYS_MAXARGS, 1, do_ut_dm, "", ""),
64 #endif
65 #if defined(CONFIG_UT_ENV)
66 U_BOOT_CMD_MKENT(env, CONFIG_SYS_MAXARGS, 1, do_ut_env, "", ""),
67 #endif
68 U_BOOT_CMD_MKENT(exit, CONFIG_SYS_MAXARGS, 1, do_ut_exit, "", ""),
69 #ifdef CONFIG_CMD_FDT
70 U_BOOT_CMD_MKENT(fdt, CONFIG_SYS_MAXARGS, 1, do_ut_fdt, "", ""),
71 #endif
72 #ifdef CONFIG_CONSOLE_TRUETYPE
73 U_BOOT_CMD_MKENT(font, CONFIG_SYS_MAXARGS, 1, do_ut_font, "", ""),
74 #endif
75 #ifdef CONFIG_UT_OPTEE
76 U_BOOT_CMD_MKENT(optee, CONFIG_SYS_MAXARGS, 1, do_ut_optee, "", ""),
77 #endif
78 #ifdef CONFIG_UT_OVERLAY
79 U_BOOT_CMD_MKENT(overlay, CONFIG_SYS_MAXARGS, 1, do_ut_overlay, "", ""),
80 #endif
81 #ifdef CONFIG_UT_LIB
82 U_BOOT_CMD_MKENT(lib, CONFIG_SYS_MAXARGS, 1, do_ut_lib, "", ""),
83 #endif
84 #ifdef CONFIG_UT_LOG
85 U_BOOT_CMD_MKENT(log, CONFIG_SYS_MAXARGS, 1, do_ut_log, "", ""),
86 #endif
87 U_BOOT_CMD_MKENT(mem, CONFIG_SYS_MAXARGS, 1, do_ut_mem, "", ""),
88 #if defined(CONFIG_SANDBOX) && defined(CONFIG_CMD_SETEXPR)
89 U_BOOT_CMD_MKENT(setexpr, CONFIG_SYS_MAXARGS, 1, do_ut_setexpr, "",
90 ""),
91 #endif
92 U_BOOT_CMD_MKENT(print, CONFIG_SYS_MAXARGS, 1, do_ut_print, "", ""),
93 #ifdef CONFIG_UT_TIME
94 U_BOOT_CMD_MKENT(time, CONFIG_SYS_MAXARGS, 1, do_ut_time, "", ""),
95 #endif
96 #if CONFIG_IS_ENABLED(UT_UNICODE) && !defined(API_BUILD)
97 U_BOOT_CMD_MKENT(unicode, CONFIG_SYS_MAXARGS, 1, do_ut_unicode, "", ""),
98 #endif
99 #ifdef CONFIG_SANDBOX
100 U_BOOT_CMD_MKENT(compression, CONFIG_SYS_MAXARGS, 1, do_ut_compression,
101 "", ""),
102 U_BOOT_CMD_MKENT(bloblist, CONFIG_SYS_MAXARGS, 1, do_ut_bloblist,
103 "", ""),
104 U_BOOT_CMD_MKENT(bootm, CONFIG_SYS_MAXARGS, 1, do_ut_bootm, "", ""),
105 #endif
106 U_BOOT_CMD_MKENT(str, CONFIG_SYS_MAXARGS, 1, do_ut_str, "", ""),
107 #ifdef CONFIG_CMD_ADDRMAP
108 U_BOOT_CMD_MKENT(addrmap, CONFIG_SYS_MAXARGS, 1, do_ut_addrmap, "", ""),
109 #endif
110 #ifdef CONFIG_CMD_LOADM
111 U_BOOT_CMD_MKENT(loadm, CONFIG_SYS_MAXARGS, 1, do_ut_loadm, "", ""),
112 #endif
113 #ifdef CONFIG_CMD_PCI_MPS
114 U_BOOT_CMD_MKENT(pci_mps, CONFIG_SYS_MAXARGS, 1, do_ut_pci_mps, "", ""),
115 #endif
116 #ifdef CONFIG_CMD_SEAMA
117 U_BOOT_CMD_MKENT(seama, CONFIG_SYS_MAXARGS, 1, do_ut_seama, "", ""),
118 #endif
119 };
120
do_ut_all(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])121 static int do_ut_all(struct cmd_tbl *cmdtp, int flag, int argc,
122 char *const argv[])
123 {
124 int i;
125 int retval;
126 int any_fail = 0;
127
128 for (i = 1; i < ARRAY_SIZE(cmd_ut_sub); i++) {
129 printf("----Running %s tests----\n", cmd_ut_sub[i].name);
130 retval = cmd_ut_sub[i].cmd(cmdtp, flag, 1, &cmd_ut_sub[i].name);
131 if (!any_fail)
132 any_fail = retval;
133 }
134
135 return any_fail;
136 }
137
do_ut_info(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])138 static int do_ut_info(struct cmd_tbl *cmdtp, int flag, int argc,
139 char *const argv[])
140 {
141 printf("Test suites: %d\n", (int)ARRAY_SIZE(cmd_ut_sub));
142 printf("Total tests: %d\n", (int)UNIT_TEST_ALL_COUNT());
143
144 return 0;
145 }
146
do_ut(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])147 static int do_ut(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
148 {
149 struct cmd_tbl *cp;
150
151 if (argc < 2)
152 return CMD_RET_USAGE;
153
154 /* drop initial "ut" arg */
155 argc--;
156 argv++;
157
158 cp = find_cmd_tbl(argv[0], cmd_ut_sub, ARRAY_SIZE(cmd_ut_sub));
159
160 if (cp)
161 return cp->cmd(cmdtp, flag, argc, argv);
162
163 return CMD_RET_USAGE;
164 }
165
166 #ifdef CONFIG_SYS_LONGHELP
167 static char ut_help_text[] =
168 "[-r] [-f] [<suite>] - run unit tests\n"
169 " -r<runs> Number of times to run each test\n"
170 " -f Force 'manual' tests to run as well\n"
171 " <suite> Test suite to run, or all\n"
172 "\n"
173 "\nOptions for <suite>:"
174 "\nall - execute all enabled tests"
175 "\ninfo - show info about tests"
176 #ifdef CONFIG_CMD_ADDRMAP
177 "\naddrmap - very basic test of addrmap command"
178 #endif
179 #ifdef CONFIG_SANDBOX
180 "\nbloblist - bloblist implementation"
181 #endif
182 #ifdef CONFIG_BOOTSTD
183 "\nbootstd - standard boot implementation"
184 #endif
185 #ifdef CONFIG_SANDBOX
186 "\ncompression - compressors and bootm decompression"
187 #endif
188 #ifdef CONFIG_UT_DM
189 "\ndm - driver model"
190 #endif
191 #ifdef CONFIG_UT_ENV
192 "\nenv - environment"
193 #endif
194 #ifdef CONFIG_CMD_FDT
195 "\nfdt - fdt command"
196 #endif
197 #ifdef CONFIG_CONSOLE_TRUETYPE
198 "\nut font - font command"
199 #endif
200 #ifdef CONFIG_CMD_LOADM
201 "\nloadm - loadm command parameters and loading memory blob"
202 #endif
203 #ifdef CONFIG_UT_LIB
204 "\nlib - library functions"
205 #endif
206 #ifdef CONFIG_UT_LOG
207 "\nlog - logging functions"
208 #endif
209 "\nmem - memory-related commands"
210 #ifdef CONFIG_UT_OPTEE
211 "\noptee - test OP-TEE"
212 #endif
213 #ifdef CONFIG_UT_OVERLAY
214 "\noverlay - device tree overlays"
215 #endif
216 #ifdef CONFIG_CMD_PCI_MPS
217 "\npci_mps - PCI Express Maximum Payload Size"
218 #endif
219 "\nprint - printing things to the console"
220 "\nsetexpr - setexpr command"
221 #ifdef CONFIG_SANDBOX
222 "\nstr - basic test of string functions"
223 #endif
224 #ifdef CONFIG_CMD_SEAMA
225 "\nseama - seama command parameters loading and decoding"
226 #endif
227 #ifdef CONFIG_UT_TIME
228 "\ntime - very basic test of time functions"
229 #endif
230 #if defined(CONFIG_UT_UNICODE) && \
231 !defined(CONFIG_SPL_BUILD) && !defined(API_BUILD)
232 "\nunicode - Unicode functions"
233 #endif
234 ;
235 #endif /* CONFIG_SYS_LONGHELP */
236
237 U_BOOT_CMD(
238 ut, CONFIG_SYS_MAXARGS, 1, do_ut,
239 "unit tests", ut_help_text
240 );
241