utest/msh: supports autocomplete of utest cases for utest_run

Author: kurisaW

components/utilities/Kconfig

@@ -217,6 +217,12 @@ config RT_USING_UTEST
             help
                 If enable this option, the test cases will be run automatically when board boot up.
 
+        if FINSH_USING_OPTION_COMPLETION
+            config MAX_UTEST_OPTIONS
+                int "Maximum number of utest cases for auto-completion"
+                default 64
+        endif
+
         config RT_UTEST_USING_ALL_CASES
             bool "Enable all selected modules' test cases"
             default n

components/utilities/utest/utest.c

@@ -73,6 +73,36 @@ void utest_log_lv_set(rt_uint8_t lv)
     }
 }
 
+static struct msh_cmd_opt utest_testcase_run_msh_options[MAX_UTEST_OPTIONS];
+
+static void utest_build_options(void)
+{
+    rt_size_t i;
+    rt_size_t option_index = 0;
+
+    if (tc_num >= MAX_UTEST_OPTIONS - 1)
+    {
+        LOG_W("Too many test cases (%d), only first %d will have completion", tc_num, MAX_UTEST_OPTIONS - 1);
+    }
+
+    rt_memset(utest_testcase_run_msh_options, 0, sizeof(utest_testcase_run_msh_options));
+
+    rt_size_t max_cases = (tc_num < MAX_UTEST_OPTIONS - 1) ? tc_num : MAX_UTEST_OPTIONS - 1;
+    for (i = 0; i < max_cases; i++)
+    {
+        utest_testcase_run_msh_options[option_index].id = i + 1;
+        utest_testcase_run_msh_options[option_index].name = tc_table[i].name;
+        utest_testcase_run_msh_options[option_index].des = RT_NULL;
+        option_index++;
+    }
+
+    utest_testcase_run_msh_options[option_index].id = 0;
+    utest_testcase_run_msh_options[option_index].name = RT_NULL;
+    utest_testcase_run_msh_options[option_index].des = RT_NULL;
+}
+
+static void utest_build_options(void);
+
 int utest_init(void)
 {
     /* initialize the utest commands table.*/
@@ -123,6 +153,9 @@ int utest_init(void)
             LOG_E("no memory, tc_fail_list init failed!");
         }
     }
+
+    utest_build_options();
+
     return tc_num;
 }
 INIT_COMPONENT_EXPORT(utest_init);
@@ -335,6 +368,7 @@ INIT_APP_EXPORT(utest_auto_run);
 int utest_testcase_run(int argc, char** argv)
 {
     static char utest_name[UTEST_NAME_MAX_LEN];
+    int opt_id;
     rt_memset(utest_name, 0x0, sizeof(utest_name));
 
     tc_loop = 1;
@@ -343,32 +377,47 @@ int utest_testcase_run(int argc, char** argv)
     {
         utest_thread_create(RT_NULL);
     }
-    else if (argc == 2 || argc == 3 || argc == 4)
+    else if (argc >= 2)
     {
-        if (rt_strcmp(argv[1], "-thread") == 0)
+        opt_id = msh_cmd_opt_id_get(argc, argv, (void*)utest_testcase_run_msh_options);
+
+        if (opt_id >= 1 && opt_id <= tc_num)
         {
-            if (argc == 3 || argc == 4)
+            rt_size_t test_index = opt_id - 1;
+            rt_strncpy(utest_name, tc_table[test_index].name, sizeof(utest_name) -1);
+            if (argc == 3)
             {
-                rt_strncpy(utest_name, argv[2], sizeof(utest_name) -1);
-                if (argc == 4)
-                {
-                    tc_loop = atoi(argv[3]);
-                }
+                tc_loop = atoi(argv[2]);
             }
-            utest_thread_create(utest_name);
-        }
-        else if (rt_strcmp(argv[1], "-help") == 0)
-        {
-            utest_help();
+            utest_do_run(utest_name);
         }
         else
         {
-            rt_strncpy(utest_name, argv[1], sizeof(utest_name) -1);
-            if (argc == 3)
+            if (rt_strcmp(argv[1], "-thread") == 0)
             {
-                tc_loop = atoi(argv[2]);
+                if (argc == 3 || argc == 4)
+                {
+                    rt_strncpy(utest_name, argv[2], sizeof(utest_name) -1);
+                    if (argc == 4)
+                    {
+                        tc_loop = atoi(argv[3]);
+                    }
+                }
+                utest_thread_create(utest_name);
+            }
+            else if (rt_strcmp(argv[1], "-help") == 0)
+            {
+                utest_help();
+            }
+            else
+            {
+                rt_strncpy(utest_name, argv[1], sizeof(utest_name) -1);
+                if (argc == 3)
+                {
+                    tc_loop = atoi(argv[2]);
+                }
+                utest_do_run(utest_name);
             }
-            utest_do_run(utest_name);
         }
     }
     else
@@ -379,7 +428,7 @@ int utest_testcase_run(int argc, char** argv)
 
     return RT_EOK;
 }
-MSH_CMD_EXPORT_ALIAS(utest_testcase_run, utest_run, utest_run [-thread or -help] [testcase name] [loop num]);
+MSH_CMD_EXPORT_ALIAS(utest_testcase_run, utest_run, utest_run [-thread or -help] [testcase name] [loop num], optenable);
 
 utest_t utest_handle_get(void)
 {