moveconfig: Correct pylint errors
[platform/kernel/u-boot.git] / test / test-main.c
index 4e17c9e..8fcb02e 100644 (file)
@@ -7,7 +7,6 @@
 #include <common.h>
 #include <console.h>
 #include <dm.h>
-#include <asm/state.h>
 #include <dm/root.h>
 #include <dm/test.h>
 #include <dm/uclass-internal.h>
 
 DECLARE_GLOBAL_DATA_PTR;
 
+/* This is valid when a test is running, NULL otherwise */
+static struct unit_test_state *cur_test_state;
+
+struct unit_test_state *test_get_state(void)
+{
+       return cur_test_state;
+}
+
+void test_set_state(struct unit_test_state *uts)
+{
+       cur_test_state = uts;
+}
+
 /**
  * dm_test_pre_run() - Get ready to run a driver model test
  *
@@ -33,9 +45,9 @@ static int dm_test_pre_run(struct unit_test_state *uts)
        uts->force_fail_alloc = false;
        uts->skip_post_probe = false;
        gd->dm_root = NULL;
-       if (!CONFIG_IS_ENABLED(OF_PLATDATA))
+       if (CONFIG_IS_ENABLED(UT_DM) && !CONFIG_IS_ENABLED(OF_PLATDATA))
                memset(dm_testdrv_op_count, '\0', sizeof(dm_testdrv_op_count));
-       state_reset_for_test(state_get_current());
+       arch_reset_for_test();
 
        /* Determine whether to make the live tree available */
        gd_set_of_root(of_live ? uts->of_root : NULL);
@@ -49,17 +61,25 @@ static int dm_test_post_run(struct unit_test_state *uts)
 {
        int id;
 
-       for (id = 0; id < UCLASS_COUNT; id++) {
-               struct uclass *uc;
-
-               /*
-                * If the uclass doesn't exist we don't want to create it. So
-                * check that here before we call uclass_find_device().
-                */
-               uc = uclass_find(id);
-               if (!uc)
-                       continue;
-               ut_assertok(uclass_destroy(uc));
+       /*
+        * With of-platdata-inst the uclasses are created at build time. If we
+        * destroy them we cannot get them back since uclass_add() is not
+        * supported. So skip this.
+        */
+       if (!CONFIG_IS_ENABLED(OF_PLATDATA_INST)) {
+               for (id = 0; id < UCLASS_COUNT; id++) {
+                       struct uclass *uc;
+
+                       /*
+                        * If the uclass doesn't exist we don't want to create
+                        * it. So check that here before we call
+                        * uclass_find_device().
+                        */
+                       uc = uclass_find(id);
+                       if (!uc)
+                               continue;
+                       ut_assertok(uclass_destroy(uc));
+               }
        }
 
        return 0;
@@ -86,7 +106,7 @@ static int do_autoprobe(struct unit_test_state *uts)
  * This skips long/slow tests where there is not much value in running a flat
  * DT test in addition to a live DT test.
  *
- * @return true to run the given test on the flat device tree
+ * Return: true to run the given test on the flat device tree
  */
 static bool ut_test_run_on_flattree(struct unit_test *test)
 {
@@ -99,11 +119,100 @@ static bool ut_test_run_on_flattree(struct unit_test *test)
 }
 
 /**
+ * test_matches() - Check if a test should be run
+ *
+ * This checks if the a test should be run. In the normal case of running all
+ * tests, @select_name is NULL.
+ *
+ * @prefix: String prefix for the tests. Any tests that have this prefix will be
+ *     printed without the prefix, so that it is easier to see the unique part
+ *     of the test name. If NULL, any suite name (xxx_test) is considered to be
+ *     a prefix.
+ * @test_name: Name of current test
+ * @select_name: Name of test to run (or NULL for all)
+ * Return: true to run this test, false to skip it
+ */
+static bool test_matches(const char *prefix, const char *test_name,
+                        const char *select_name)
+{
+       size_t len;
+
+       if (!select_name)
+               return true;
+
+       /* Allow glob expansion in the test name */
+       len = select_name[strlen(select_name) - 1] == '*' ? strlen(select_name) : 0;
+       if (len-- == 1)
+               return true;
+
+       if (!strncmp(test_name, select_name, len))
+               return true;
+
+       if (prefix) {
+               /* All tests have this prefix */
+               if (!strncmp(test_name, prefix, strlen(prefix)))
+                       test_name += strlen(prefix);
+       } else {
+               const char *p = strstr(test_name, "_test_");
+
+               /* convert xxx_test_yyy to yyy, i.e. remove the suite name */
+               if (p)
+                       test_name = p + strlen("_test_");
+       }
+
+       if (!strncmp(test_name, select_name, len))
+               return true;
+
+       return false;
+}
+
+/**
+ * ut_list_has_dm_tests() - Check if a list of tests has driver model ones
+ *
+ * @tests: List of tests to run
+ * @count: Number of tests to ru
+ * Return: true if any of the tests have the UT_TESTF_DM flag
+ */
+static bool ut_list_has_dm_tests(struct unit_test *tests, int count)
+{
+       struct unit_test *test;
+
+       for (test = tests; test < tests + count; test++) {
+               if (test->flags & UT_TESTF_DM)
+                       return true;
+       }
+
+       return false;
+}
+
+/**
+ * dm_test_restore() Put things back to normal so sandbox works as expected
+ *
+ * @of_root: Value to set for of_root
+ * Return: 0 if OK, -ve on error
+ */
+static int dm_test_restore(struct device_node *of_root)
+{
+       int ret;
+
+       gd_set_of_root(of_root);
+       gd->dm_root = NULL;
+       ret = dm_init(CONFIG_IS_ENABLED(OF_LIVE));
+       if (ret)
+               return ret;
+       dm_scan_plat(false);
+       if (!CONFIG_IS_ENABLED(OF_PLATDATA))
+               dm_scan_fdt(false);
+
+       return 0;
+}
+
+/**
  * test_pre_run() - Handle any preparation needed to run a test
  *
  * @uts: Test state
  * @test: Test to prepare for
- * @return 0 if OK, -EAGAIN to skip this test since some required feature is not
+ * Return: 0 if OK, -EAGAIN to skip this test since some required feature is not
  *     available, other -ve on error (meaning that testing cannot likely
  *     continue)
  */
@@ -144,7 +253,7 @@ static int test_pre_run(struct unit_test_state *uts, struct unit_test *test)
  *
  * @uts: Test state
  * @test: Test to clean up after
- * @return 0 if OK, -ve on error (meaning that testing cannot likely continue)
+ * Return: 0 if OK, -ve on error (meaning that testing cannot likely continue)
  */
 static int test_post_run(struct unit_test_state *uts, struct unit_test *test)
 {
@@ -166,7 +275,7 @@ static int test_post_run(struct unit_test_state *uts, struct unit_test *test)
  *     incremented by the number of failures (0, one hopes)
  * @test_name: Test to run
  * @name: Name of test, possibly skipping a prefix that should not be displayed
- * @return 0 if all tests passed, -EAGAIN if the test should be skipped, -1 if
+ * Return: 0 if all tests passed, -EAGAIN if the test should be skipped, -1 if
  *     any failed
  */
 static int ut_run_test(struct unit_test_state *uts, struct unit_test *test,
@@ -180,6 +289,9 @@ static int ut_run_test(struct unit_test_state *uts, struct unit_test *test,
                note = " (flat tree)";
        printf("Test: %s: %s%s\n", test_name, fname, note);
 
+       /* Allow access to test state from drivers */
+       test_set_state(uts);
+
        ret = test_pre_run(uts, test);
        if (ret == -EAGAIN)
                return -EAGAIN;
@@ -192,11 +304,30 @@ static int ut_run_test(struct unit_test_state *uts, struct unit_test *test,
        if (ret)
                return ret;
 
+       test_set_state( NULL);
+
        return 0;
 }
 
-int ut_run_test_live_flat(struct unit_test_state *uts, struct unit_test *test,
-                         const char *name)
+/**
+ * ut_run_test_live_flat() - Run a test with both live and flat tree
+ *
+ * This calls ut_run_test() with livetree enabled, which is the standard setup
+ * for runnig tests. Then, for driver model test, it calls it again with
+ * livetree disabled. This allows checking of flattree being used when OF_LIVE
+ * is enabled, as is the case in U-Boot proper before relocation, as well as in
+ * SPL.
+ *
+ * @uts: Test state to update. The caller should ensure that this is zeroed for
+ *     the first call to this function. On exit, @uts->fail_count is
+ *     incremented by the number of failures (0, one hopes)
+ * @test: Test to run
+ * @name: Name of test, possibly skipping a prefix that should not be displayed
+ * Return: 0 if all tests passed, -EAGAIN if the test should be skipped, -1 if
+ *     any failed
+ */
+static int ut_run_test_live_flat(struct unit_test_state *uts,
+                                struct unit_test *test, const char *name)
 {
        int runs;
 
@@ -224,24 +355,39 @@ int ut_run_test_live_flat(struct unit_test_state *uts, struct unit_test *test,
        return 0;
 }
 
-int ut_run_tests(struct unit_test_state *uts, const char *prefix,
-                struct unit_test *tests, int count, const char *select_name)
+/**
+ * ut_run_tests() - Run a set of tests
+ *
+ * This runs the tests, handling any preparation and clean-up needed. It prints
+ * the name of each test before running it.
+ *
+ * @uts: Test state to update. The caller should ensure that this is zeroed for
+ *     the first call to this function. On exit, @uts->fail_count is
+ *     incremented by the number of failures (0, one hopes)
+ * @prefix: String prefix for the tests. Any tests that have this prefix will be
+ *     printed without the prefix, so that it is easier to see the unique part
+ *     of the test name. If NULL, no prefix processing is done
+ * @tests: List of tests to run
+ * @count: Number of tests to run
+ * @select_name: Name of a single test to run (from the list provided). If NULL
+ *     then all tests are run
+ * Return: 0 if all tests passed, -ENOENT if test @select_name was not found,
+ *     -EBADF if any failed
+ */
+static int ut_run_tests(struct unit_test_state *uts, const char *prefix,
+                       struct unit_test *tests, int count,
+                       const char *select_name)
 {
        struct unit_test *test;
-       int prefix_len = prefix ? strlen(prefix) : 0;
        int found = 0;
 
        for (test = tests; test < tests + count; test++) {
                const char *test_name = test->name;
                int ret;
 
-               /* Remove the prefix */
-               if (prefix && !strncmp(test_name, prefix, prefix_len))
-                       test_name += prefix_len;
-
-               if (select_name && strcmp(select_name, test_name))
+               if (!test_matches(prefix, test_name, select_name))
                        continue;
-               ret = ut_run_test_live_flat(uts, test, test_name);
+               ret = ut_run_test_live_flat(uts, test, select_name);
                found++;
                if (ret == -EAGAIN)
                        continue;
@@ -258,11 +404,27 @@ int ut_run_list(const char *category, const char *prefix,
                struct unit_test *tests, int count, const char *select_name)
 {
        struct unit_test_state uts = { .fail_count = 0 };
+       bool has_dm_tests = false;
        int ret;
 
+       if (!CONFIG_IS_ENABLED(OF_PLATDATA) &&
+           ut_list_has_dm_tests(tests, count)) {
+               has_dm_tests = true;
+               /*
+                * If we have no device tree, or it only has a root node, then
+                * these * tests clearly aren't going to work...
+                */
+               if (!gd->fdt_blob || fdt_next_node(gd->fdt_blob, 0, NULL) < 0) {
+                       puts("Please run with test device tree:\n"
+                            "    ./u-boot -d arch/sandbox/dts/test.dtb\n");
+                       return CMD_RET_FAILURE;
+               }
+       }
+
        if (!select_name)
                printf("Running %d %s tests\n", count, category);
 
+       uts.of_root = gd_of_root();
        ret = ut_run_tests(&uts, prefix, tests, count, select_name);
 
        if (ret == -ENOENT)
@@ -270,5 +432,9 @@ int ut_run_list(const char *category, const char *prefix,
        else
                printf("Failures: %d\n", uts.fail_count);
 
+       /* Best efforts only...ignore errors */
+       if (has_dm_tests)
+               dm_test_restore(uts.of_root);
+
        return ret;
 }