At present the 'ut' command handles its flags in a strange way, in that
they must come after the subcommand.

So, we must use 'ut bloblist -r2' to run the bloblist tests twice. This
is an artefact of the way tests were run, through subcommands.

It is now possible to correct this, by doing flag-processing before
running the suite.

Update the code to handle this, so that 'ut -r2 bloblist' works. Update
the 'test_suite' test to check the new arguments.

Add a sanity-check for -I while we are here.

Signed-off-by: Simon Glass <s...@chromium.org>
---

(no changes since v1)

 test/cmd_ut.c               | 93 +++++++++++++++++++------------------
 test/py/tests/test_suite.py | 20 ++++++--
 2 files changed, 66 insertions(+), 47 deletions(-)

diff --git a/test/cmd_ut.c b/test/cmd_ut.c
index bc2da617e51..6c0b39ff60c 100644
--- a/test/cmd_ut.c
+++ b/test/cmd_ut.c
@@ -28,11 +28,11 @@ struct suite {
        const char *help;
 };
 
-static int do_ut_all(struct unit_test_state *uts, struct cmd_tbl *cmdtp,
-                    int flag, int argc, char *const argv[]);
+static int do_ut_all(struct unit_test_state *uts, const char *select_name,
+                    int runs_per_test, bool force_run,
+                    const char *test_insert);
 
-static int do_ut_info(struct cmd_tbl *cmdtp, int flag, int argc,
-                     char *const argv[]);
+static int do_ut_info(bool show_suites);
 
 /* declare linker-list symbols for the start and end of a suite */
 #define SUITE_DECL(_name) \
@@ -119,40 +119,18 @@ static bool has_tests(struct suite *ste)
 
 /** run_suite() - Run a suite of tests */
 static int run_suite(struct unit_test_state *uts, struct suite *ste,
-                    struct cmd_tbl *cmdtp, int flag, int argc,
-                    char *const argv[])
+                    const char *select_name, int runs_per_test, bool force_run,
+                    const char *test_insert)
 {
        int n_ents = ste->end - ste->start;
-       const char *test_insert = NULL;
-       int runs_per_text = 1;
-       bool force_run = false;
        char prefix[30];
        int ret;
 
        /* use a standard prefix */
        snprintf(prefix, sizeof(prefix), "%s_test_", ste->name);
 
-       while (argc > 1 && *argv[1] == '-') {
-               const char *str = argv[1];
-
-               switch (str[1]) {
-               case 'r':
-                       runs_per_text = dectoul(str + 2, NULL);
-                       break;
-               case 'f':
-                       force_run = true;
-                       break;
-               case 'I':
-                       test_insert = str + 2;
-                       break;
-               }
-               argv++;
-               argc--;
-       }
-
        ret = ut_run_list(uts, ste->name, prefix, ste->start, n_ents,
-                         cmd_arg1(argc, argv), runs_per_text, force_run,
-                         test_insert);
+                         select_name, runs_per_test, force_run, test_insert);
 
        return ret;
 }
@@ -187,8 +165,8 @@ static void update_stats(struct unit_test_state *uts, const 
struct suite *ste)
        }
 }
 
-static int do_ut_all(struct unit_test_state *uts, struct cmd_tbl *cmdtp,
-                    int flag, int argc, char *const argv[])
+static int do_ut_all(struct unit_test_state *uts, const char *select_name,
+                    int runs_per_test, bool force_run, const char *test_insert)
 {
        int i;
        int retval;
@@ -196,11 +174,11 @@ static int do_ut_all(struct unit_test_state *uts, struct 
cmd_tbl *cmdtp,
 
        for (i = 0; i < ARRAY_SIZE(suites); i++) {
                struct suite *ste = &suites[i];
-               char *const argv[] = {(char *)ste->name, NULL};
 
                if (has_tests(ste)) {
                        printf("----Running %s tests----\n", ste->name);
-                       retval = run_suite(uts, ste, cmdtp, flag, 1, argv);
+                       retval = run_suite(uts, ste, select_name, runs_per_test,
+                                          force_run, test_insert);
                        if (!any_fail)
                                any_fail = retval;
                        update_stats(uts, ste);
@@ -210,11 +188,9 @@ static int do_ut_all(struct unit_test_state *uts, struct 
cmd_tbl *cmdtp,
        return any_fail;
 }
 
-static int do_ut_info(struct cmd_tbl *cmdtp, int flag, int argc,
-                     char *const argv[])
+static int do_ut_info(bool show_suites)
 {
        int suite_count, i;
-       const char *flags;
 
        for (suite_count = 0, i = 0; i < ARRAY_SIZE(suites); i++) {
                struct suite *ste = &suites[i];
@@ -226,8 +202,7 @@ static int do_ut_info(struct cmd_tbl *cmdtp, int flag, int 
argc,
        printf("Test suites: %d\n", suite_count);
        printf("Total tests: %d\n", (int)UNIT_TEST_ALL_COUNT());
 
-       flags = cmd_arg1(argc, argv);
-       if (flags && !strcmp("-s", flags)) {
+       if (show_suites) {
                int i, total;
 
                puts("\nTests  Suite         Purpose");
@@ -267,24 +242,53 @@ static struct suite *find_suite(const char *name)
 
 static int do_ut(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
 {
+       const char *test_insert = NULL, *select_name;
        struct unit_test_state uts;
+       bool show_suites = false;
+       bool force_run = false;
+       int runs_per_text = 1;
        struct suite *ste;
        char *name;
        int ret;
 
-       if (argc < 2)
-               return CMD_RET_USAGE;
-
        /* drop initial "ut" arg */
        argc--;
        argv++;
 
+       while (argc > 0 && *argv[0] == '-') {
+               const char *str = argv[0];
+
+               switch (str[1]) {
+               case 'r':
+                       runs_per_text = dectoul(str + 2, NULL);
+                       break;
+               case 'f':
+                       force_run = true;
+                       break;
+               case 'I':
+                       test_insert = str + 2;
+                       if (!strchr(test_insert, ':'))
+                               return CMD_RET_USAGE;
+                       break;
+               case 's':
+                       show_suites = true;
+                       break;
+               }
+               argv++;
+               argc--;
+       }
+
+       if (argc < 1)
+               return CMD_RET_USAGE;
+
        ut_init_state(&uts);
        name = argv[0];
+       select_name = cmd_arg1(argc, argv);
        if (!strcmp(name, "all")) {
-               ret = do_ut_all(&uts, cmdtp, flag, argc, argv);
+               ret = do_ut_all(&uts, select_name, runs_per_text, force_run,
+                               test_insert);
        } else if (!strcmp(name, "info")) {
-               ret = do_ut_info(cmdtp, flag, argc, argv);
+               ret = do_ut_info(show_suites);
        } else {
                int any_fail = 0;
                const char *p;
@@ -300,7 +304,8 @@ static int do_ut(struct cmd_tbl *cmdtp, int flag, int argc, 
char *const argv[])
                                return CMD_RET_FAILURE;
                        }
 
-                       ret = run_suite(&uts, ste, cmdtp, flag, argc, argv);
+                       ret = run_suite(&uts, ste, select_name, runs_per_text,
+                                       force_run, test_insert);
                        if (!any_fail)
                                any_fail = ret;
                        update_stats(&uts, ste);
diff --git a/test/py/tests/test_suite.py b/test/py/tests/test_suite.py
index d0025a7ba30..9ddc883394b 100644
--- a/test/py/tests/test_suite.py
+++ b/test/py/tests/test_suite.py
@@ -135,7 +135,7 @@ def xtest_suite(u_boot_console, u_boot_config):
 
        - The number of suites matches that reported by the 'ut info'
        - Where available, the number of tests is each suite matches that
-         reported by 'ut info -s'
+         reported by 'ut -s info'
        - The total number of tests adds up to the total that are actually run
          with 'ut all'
        - All suites are run with 'ut all'
@@ -167,7 +167,7 @@ def xtest_suite(u_boot_console, u_boot_config):
 
     # Run 'ut info' and compare with the log results
     with cons.log.section('Check suite test-counts'):
-        output = cons.run_command('ut info -s')
+        output = cons.run_command('ut -s info')
 
         suite_count, total_test_count, test_count = process_ut_info(cons,
                                                                     output)
@@ -191,4 +191,18 @@ def xtest_suite(u_boot_console, u_boot_config):
     # Run three suites
     with cons.log.section('Check multiple suites'):
         output = cons.run_command('ut bloblist,setexpr,mem')
-    assert 'Suites run: 3' in output
+        assert 'Suites run: 3' in output
+
+    # Run a particular test
+    with cons.log.section('Check single test'):
+        output = cons.run_command('ut bloblist reloc')
+        assert 'Test: reloc: bloblist.c' in output
+
+    # Run tests multiple times
+    with cons.log.section('Check multiple runs'):
+        output = cons.run_command('ut -r2 bloblist')
+        lines = output.splitlines()
+        run = len([line for line in lines if 'Test:' in line])
+        count = re.search(r'Tests run: (\d*)', lines[-1]).group(1)
+
+        assert run == 2 * int(count)
-- 
2.43.0

Reply via email to