Tom Rini | 83d290c | 2018-05-06 17:58:06 -0400 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0 |
Joe Hershberger | c617ede | 2015-05-20 14:27:28 -0500 | [diff] [blame] | 2 | /* |
| 3 | * (C) Copyright 2015 |
| 4 | * Joe Hershberger, National Instruments, joe.hershberger@ni.com |
Joe Hershberger | c617ede | 2015-05-20 14:27:28 -0500 | [diff] [blame] | 5 | */ |
| 6 | |
| 7 | #include <common.h> |
| 8 | #include <command.h> |
| 9 | #include <test/suites.h> |
Simon Glass | 4d869c1 | 2017-11-25 11:57:29 -0700 | [diff] [blame] | 10 | #include <test/test.h> |
Joe Hershberger | c617ede | 2015-05-20 14:27:28 -0500 | [diff] [blame] | 11 | |
| 12 | static int do_ut_all(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]); |
| 13 | |
Philippe Reynes | 4ad4edf | 2019-12-17 19:07:04 +0100 | [diff] [blame] | 14 | int cmd_ut_category(const char *name, const char *prefix, |
| 15 | struct unit_test *tests, int n_ents, |
Simon Glass | 4d869c1 | 2017-11-25 11:57:29 -0700 | [diff] [blame] | 16 | int argc, char * const argv[]) |
| 17 | { |
| 18 | struct unit_test_state uts = { .fail_count = 0 }; |
| 19 | struct unit_test *test; |
Philippe Reynes | 4ad4edf | 2019-12-17 19:07:04 +0100 | [diff] [blame] | 20 | int prefix_len = prefix ? strlen(prefix) : 0; |
Simon Glass | 4d869c1 | 2017-11-25 11:57:29 -0700 | [diff] [blame] | 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++) { |
Philippe Reynes | 4ad4edf | 2019-12-17 19:07:04 +0100 | [diff] [blame] | 26 | const char *test_name = test->name; |
| 27 | |
| 28 | /* Remove the prefix */ |
Philippe Reynes | 3f05f08 | 2020-01-09 17:34:02 +0100 | [diff] [blame] | 29 | if (prefix && !strncmp(test_name, prefix, prefix_len)) |
Philippe Reynes | 4ad4edf | 2019-12-17 19:07:04 +0100 | [diff] [blame] | 30 | test_name += prefix_len; |
| 31 | |
| 32 | if (argc > 1 && strcmp(argv[1], test_name)) |
Simon Glass | 4d869c1 | 2017-11-25 11:57:29 -0700 | [diff] [blame] | 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 | |
Joe Hershberger | c617ede | 2015-05-20 14:27:28 -0500 | [diff] [blame] | 46 | static cmd_tbl_t cmd_ut_sub[] = { |
| 47 | U_BOOT_CMD_MKENT(all, CONFIG_SYS_MAXARGS, 1, do_ut_all, "", ""), |
Joe Hershberger | 40441e0 | 2015-05-20 14:27:29 -0500 | [diff] [blame] | 48 | #if defined(CONFIG_UT_DM) |
| 49 | U_BOOT_CMD_MKENT(dm, CONFIG_SYS_MAXARGS, 1, do_ut_dm, "", ""), |
| 50 | #endif |
Joe Hershberger | 421f86f | 2015-05-20 14:27:36 -0500 | [diff] [blame] | 51 | #if defined(CONFIG_UT_ENV) |
| 52 | U_BOOT_CMD_MKENT(env, CONFIG_SYS_MAXARGS, 1, do_ut_env, "", ""), |
| 53 | #endif |
Heiko Stuebner | 96383bd | 2019-10-23 16:46:41 +0200 | [diff] [blame] | 54 | #ifdef CONFIG_UT_OPTEE |
| 55 | U_BOOT_CMD_MKENT(optee, CONFIG_SYS_MAXARGS, 1, do_ut_optee, "", ""), |
| 56 | #endif |
Maxime Ripard | f2a9942 | 2016-07-05 10:26:46 +0200 | [diff] [blame] | 57 | #ifdef CONFIG_UT_OVERLAY |
| 58 | U_BOOT_CMD_MKENT(overlay, CONFIG_SYS_MAXARGS, 1, do_ut_overlay, "", ""), |
| 59 | #endif |
Heinrich Schuchardt | 2dd0111 | 2019-01-30 07:53:31 +0100 | [diff] [blame] | 60 | #ifdef CONFIG_UT_LIB |
| 61 | U_BOOT_CMD_MKENT(lib, CONFIG_SYS_MAXARGS, 1, do_ut_lib, "", ""), |
| 62 | #endif |
Joe Hershberger | c812f72 | 2015-05-20 14:27:30 -0500 | [diff] [blame] | 63 | #ifdef CONFIG_UT_TIME |
| 64 | U_BOOT_CMD_MKENT(time, CONFIG_SYS_MAXARGS, 1, do_ut_time, "", ""), |
| 65 | #endif |
Heinrich Schuchardt | f11a164 | 2018-08-31 21:31:28 +0200 | [diff] [blame] | 66 | #if CONFIG_IS_ENABLED(UT_UNICODE) && !defined(API_BUILD) |
| 67 | U_BOOT_CMD_MKENT(unicode, CONFIG_SYS_MAXARGS, 1, do_ut_unicode, "", ""), |
| 68 | #endif |
Simon Glass | 0aac10f | 2017-11-25 11:57:33 -0700 | [diff] [blame] | 69 | #ifdef CONFIG_SANDBOX |
| 70 | U_BOOT_CMD_MKENT(compression, CONFIG_SYS_MAXARGS, 1, do_ut_compression, |
| 71 | "", ""), |
Simon Glass | 919e7a8 | 2018-11-15 18:43:53 -0700 | [diff] [blame] | 72 | U_BOOT_CMD_MKENT(bloblist, CONFIG_SYS_MAXARGS, 1, do_ut_bloblist, |
| 73 | "", ""), |
Simon Glass | 0aac10f | 2017-11-25 11:57:33 -0700 | [diff] [blame] | 74 | #endif |
Joe Hershberger | c617ede | 2015-05-20 14:27:28 -0500 | [diff] [blame] | 75 | }; |
| 76 | |
| 77 | static int do_ut_all(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) |
| 78 | { |
| 79 | int i; |
| 80 | int retval; |
| 81 | int any_fail = 0; |
| 82 | |
| 83 | for (i = 1; i < ARRAY_SIZE(cmd_ut_sub); i++) { |
| 84 | printf("----Running %s tests----\n", cmd_ut_sub[i].name); |
| 85 | retval = cmd_ut_sub[i].cmd(cmdtp, flag, 1, &cmd_ut_sub[i].name); |
| 86 | if (!any_fail) |
| 87 | any_fail = retval; |
| 88 | } |
| 89 | |
| 90 | return any_fail; |
| 91 | } |
| 92 | |
| 93 | static int do_ut(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) |
| 94 | { |
| 95 | cmd_tbl_t *cp; |
| 96 | |
| 97 | if (argc < 2) |
| 98 | return CMD_RET_USAGE; |
| 99 | |
| 100 | /* drop initial "ut" arg */ |
| 101 | argc--; |
| 102 | argv++; |
| 103 | |
| 104 | cp = find_cmd_tbl(argv[0], cmd_ut_sub, ARRAY_SIZE(cmd_ut_sub)); |
| 105 | |
| 106 | if (cp) |
| 107 | return cp->cmd(cmdtp, flag, argc, argv); |
| 108 | |
| 109 | return CMD_RET_USAGE; |
| 110 | } |
| 111 | |
| 112 | #ifdef CONFIG_SYS_LONGHELP |
| 113 | static char ut_help_text[] = |
| 114 | "all - execute all enabled tests\n" |
Heinrich Schuchardt | f11a164 | 2018-08-31 21:31:28 +0200 | [diff] [blame] | 115 | #ifdef CONFIG_SANDBOX |
Simon Glass | 919e7a8 | 2018-11-15 18:43:53 -0700 | [diff] [blame] | 116 | "ut bloblist - Test bloblist implementation\n" |
Heinrich Schuchardt | f11a164 | 2018-08-31 21:31:28 +0200 | [diff] [blame] | 117 | "ut compression - Test compressors and bootm decompression\n" |
| 118 | #endif |
Joe Hershberger | 40441e0 | 2015-05-20 14:27:29 -0500 | [diff] [blame] | 119 | #ifdef CONFIG_UT_DM |
| 120 | "ut dm [test-name]\n" |
| 121 | #endif |
Joe Hershberger | 421f86f | 2015-05-20 14:27:36 -0500 | [diff] [blame] | 122 | #ifdef CONFIG_UT_ENV |
| 123 | "ut env [test-name]\n" |
| 124 | #endif |
Heinrich Schuchardt | 2dd0111 | 2019-01-30 07:53:31 +0100 | [diff] [blame] | 125 | #ifdef CONFIG_UT_LIB |
| 126 | "ut lib [test-name] - test library functions\n" |
| 127 | #endif |
Heiko Stuebner | 96383bd | 2019-10-23 16:46:41 +0200 | [diff] [blame] | 128 | #ifdef CONFIG_UT_OPTEE |
| 129 | "ut optee [test-name]\n" |
| 130 | #endif |
Maxime Ripard | f2a9942 | 2016-07-05 10:26:46 +0200 | [diff] [blame] | 131 | #ifdef CONFIG_UT_OVERLAY |
| 132 | "ut overlay [test-name]\n" |
| 133 | #endif |
Joe Hershberger | c812f72 | 2015-05-20 14:27:30 -0500 | [diff] [blame] | 134 | #ifdef CONFIG_UT_TIME |
| 135 | "ut time - Very basic test of time functions\n" |
| 136 | #endif |
Heinrich Schuchardt | f11a164 | 2018-08-31 21:31:28 +0200 | [diff] [blame] | 137 | #if defined(CONFIG_UT_UNICODE) && \ |
| 138 | !defined(CONFIG_SPL_BUILD) && !defined(API_BUILD) |
| 139 | "ut unicode [test-name] - test Unicode functions\n" |
Simon Glass | 0aac10f | 2017-11-25 11:57:33 -0700 | [diff] [blame] | 140 | #endif |
Joe Hershberger | c617ede | 2015-05-20 14:27:28 -0500 | [diff] [blame] | 141 | ; |
Heinrich Schuchardt | f11a164 | 2018-08-31 21:31:28 +0200 | [diff] [blame] | 142 | #endif /* CONFIG_SYS_LONGHELP */ |
Joe Hershberger | c617ede | 2015-05-20 14:27:28 -0500 | [diff] [blame] | 143 | |
| 144 | U_BOOT_CMD( |
| 145 | ut, CONFIG_SYS_MAXARGS, 1, do_ut, |
| 146 | "unit tests", ut_help_text |
| 147 | ); |