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