blob: 88ef267458e57d556a32fa56114ae2f73cfdeecf [file] [log] [blame]
Simon Glass2e7d35d2014-02-26 15:59:21 -07001/*
2 * Copyright (c) 2013 Google, Inc
3 *
4 * SPDX-License-Identifier: GPL-2.0+
5 */
6
7#include <common.h>
Joe Hershberger40441e02015-05-20 14:27:29 -05008#include <command.h>
Simon Glass24b852a2015-11-08 23:47:45 -07009#include <console.h>
Simon Glass2e7d35d2014-02-26 15:59:21 -070010#include <dm.h>
11#include <errno.h>
Simon Glass756ac0b2014-10-04 11:29:50 -060012#include <malloc.h>
Simon Glass3884c982015-11-08 23:47:44 -070013#include <asm/state.h>
Simon Glass2e7d35d2014-02-26 15:59:21 -070014#include <dm/test.h>
15#include <dm/root.h>
16#include <dm/uclass-internal.h>
Joe Hershbergere721b882015-05-20 14:27:27 -050017#include <test/ut.h>
Simon Glass2e7d35d2014-02-26 15:59:21 -070018
19DECLARE_GLOBAL_DATA_PTR;
20
Joe Hershbergere721b882015-05-20 14:27:27 -050021struct unit_test_state global_dm_test_state;
22static struct dm_test_state _global_priv_dm_test_state;
Simon Glass2e7d35d2014-02-26 15:59:21 -070023
24/* Get ready for testing */
Simon Glassc166c472017-05-18 20:09:16 -060025static int dm_test_init(struct unit_test_state *uts, bool of_live)
Simon Glass2e7d35d2014-02-26 15:59:21 -070026{
Joe Hershbergere721b882015-05-20 14:27:27 -050027 struct dm_test_state *dms = uts->priv;
28
Simon Glass2e7d35d2014-02-26 15:59:21 -070029 memset(dms, '\0', sizeof(*dms));
30 gd->dm_root = NULL;
31 memset(dm_testdrv_op_count, '\0', sizeof(dm_testdrv_op_count));
Simon Glass34b744b2017-05-18 20:09:13 -060032 state_reset_for_test(state_get_current());
Simon Glass2e7d35d2014-02-26 15:59:21 -070033
Simon Glassc166c472017-05-18 20:09:16 -060034#ifdef CONFIG_OF_LIVE
35 /* Determine whether to make the live tree available */
36 gd->of_root = of_live ? uts->of_root : NULL;
37#endif
38 ut_assertok(dm_init(of_live));
Simon Glass2e7d35d2014-02-26 15:59:21 -070039 dms->root = dm_root();
40
41 return 0;
42}
43
44/* Ensure all the test devices are probed */
Joe Hershbergere721b882015-05-20 14:27:27 -050045static int do_autoprobe(struct unit_test_state *uts)
Simon Glass2e7d35d2014-02-26 15:59:21 -070046{
Heiko Schocher54c5d082014-05-22 12:43:05 +020047 struct udevice *dev;
Simon Glass2e7d35d2014-02-26 15:59:21 -070048 int ret;
49
50 /* Scanning the uclass is enough to probe all the devices */
51 for (ret = uclass_first_device(UCLASS_TEST, &dev);
52 dev;
53 ret = uclass_next_device(&dev))
54 ;
55
56 return ret;
57}
58
Joe Hershbergere721b882015-05-20 14:27:27 -050059static int dm_test_destroy(struct unit_test_state *uts)
Simon Glass2e7d35d2014-02-26 15:59:21 -070060{
61 int id;
62
63 for (id = 0; id < UCLASS_COUNT; id++) {
64 struct uclass *uc;
65
66 /*
67 * If the uclass doesn't exist we don't want to create it. So
68 * check that here before we call uclass_find_device()/
69 */
70 uc = uclass_find(id);
71 if (!uc)
72 continue;
73 ut_assertok(uclass_destroy(uc));
74 }
75
76 return 0;
77}
78
Simon Glassc166c472017-05-18 20:09:16 -060079static int dm_do_test(struct unit_test_state *uts, struct unit_test *test,
80 bool of_live)
Simon Glassf86db102017-05-18 20:09:14 -060081{
82 struct sandbox_state *state = state_get_current();
Simon Glass801587b2017-05-18 20:09:15 -060083 const char *fname = strrchr(test->file, '/') + 1;
Simon Glassf86db102017-05-18 20:09:14 -060084
Simon Glassc166c472017-05-18 20:09:16 -060085 printf("Test: %s: %s%s\n", test->name, fname,
86 !of_live ? " (flat tree)" : "");
87 ut_assertok(dm_test_init(uts, of_live));
Simon Glassf86db102017-05-18 20:09:14 -060088
89 uts->start = mallinfo();
90 if (test->flags & DM_TESTF_SCAN_PDATA)
91 ut_assertok(dm_scan_platdata(false));
92 if (test->flags & DM_TESTF_PROBE_TEST)
93 ut_assertok(do_autoprobe(uts));
94 if (test->flags & DM_TESTF_SCAN_FDT)
95 ut_assertok(dm_scan_fdt(gd->fdt_blob, false));
96
97 /*
98 * Silence the console and rely on console reocrding to get
99 * our output.
100 */
101 console_record_reset();
102 if (!state->show_test_output)
103 gd->flags |= GD_FLG_SILENT;
104 test->func(uts);
105 gd->flags &= ~GD_FLG_SILENT;
106 state_set_skip_delays(false);
107
108 ut_assertok(dm_test_destroy(uts));
109
110 return 0;
111}
112
Joe Hershberger40441e02015-05-20 14:27:29 -0500113static int dm_test_main(const char *test_name)
Simon Glass2e7d35d2014-02-26 15:59:21 -0700114{
Joe Hershbergere721b882015-05-20 14:27:27 -0500115 struct unit_test *tests = ll_entry_start(struct unit_test, dm_test);
116 const int n_ents = ll_entry_count(struct unit_test, dm_test);
117 struct unit_test_state *uts = &global_dm_test_state;
Joe Hershbergere721b882015-05-20 14:27:27 -0500118 struct unit_test *test;
Simon Glassc02790c2015-07-06 12:54:23 -0600119 int run_count;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700120
Simon Glassc166c472017-05-18 20:09:16 -0600121 uts->priv = &_global_priv_dm_test_state;
Stephen Warren26e1bec2016-01-27 23:57:46 -0700122 uts->fail_count = 0;
123
Simon Glass2e7d35d2014-02-26 15:59:21 -0700124 /*
125 * If we have no device tree, or it only has a root node, then these
126 * tests clearly aren't going to work...
127 */
128 if (!gd->fdt_blob || fdt_next_node(gd->fdt_blob, 0, NULL) < 0) {
129 puts("Please run with test device tree:\n"
Przemyslaw Marczakf64000c2015-05-13 13:38:34 +0200130 " ./u-boot -d arch/sandbox/dts/test.dtb\n");
Simon Glass2e7d35d2014-02-26 15:59:21 -0700131 ut_assert(gd->fdt_blob);
132 }
133
Simon Glass57f54d52015-03-25 12:23:04 -0600134 if (!test_name)
135 printf("Running %d driver model tests\n", n_ents);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700136
Simon Glassc02790c2015-07-06 12:54:23 -0600137 run_count = 0;
Simon Glassc166c472017-05-18 20:09:16 -0600138#ifdef CONFIG_OF_LIVE
139 uts->of_root = gd->of_root;
140#endif
Simon Glass2e7d35d2014-02-26 15:59:21 -0700141 for (test = tests; test < tests + n_ents; test++) {
Simon Glassc02790c2015-07-06 12:54:23 -0600142 const char *name = test->name;
143
144 /* All tests have this prefix */
145 if (!strncmp(name, "dm_test_", 8))
146 name += 8;
147 if (test_name && strcmp(test_name, name))
Simon Glass57f54d52015-03-25 12:23:04 -0600148 continue;
Simon Glassc166c472017-05-18 20:09:16 -0600149 ut_assertok(dm_do_test(uts, test, false));
Simon Glassc02790c2015-07-06 12:54:23 -0600150 run_count++;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700151 }
152
Simon Glassc02790c2015-07-06 12:54:23 -0600153 if (test_name && !run_count)
154 printf("Test '%s' not found\n", test_name);
155 else
156 printf("Failures: %d\n", uts->fail_count);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700157
Joe Hershbergerb6227d32015-05-20 14:27:35 -0500158 gd->dm_root = NULL;
Simon Glass19c82052017-05-18 20:09:08 -0600159 ut_assertok(dm_init(false));
Joe Hershbergerb6227d32015-05-20 14:27:35 -0500160 dm_scan_platdata(false);
161 dm_scan_fdt(gd->fdt_blob, false);
162
Joe Hershberger7cccc662015-05-20 14:27:32 -0500163 return uts->fail_count ? CMD_RET_FAILURE : 0;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700164}
Joe Hershberger40441e02015-05-20 14:27:29 -0500165
166int do_ut_dm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
167{
168 const char *test_name = NULL;
169
170 if (argc > 1)
171 test_name = argv[1];
172
173 return dm_test_main(test_name);
174}