blob: 62a0c692c897c7c36fc81153044a3e2d92cc4a02 [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Simon Glass2e7d35d2014-02-26 15:59:21 -07002/*
3 * Copyright (c) 2013 Google, Inc
Simon Glass2e7d35d2014-02-26 15:59:21 -07004 */
5
6#include <common.h>
Joe Hershberger40441e02015-05-20 14:27:29 -05007#include <command.h>
Simon Glass24b852a2015-11-08 23:47:45 -07008#include <console.h>
Simon Glass2e7d35d2014-02-26 15:59:21 -07009#include <dm.h>
10#include <errno.h>
Simon Glass756ac0b2014-10-04 11:29:50 -060011#include <malloc.h>
Simon Glass3884c982015-11-08 23:47:44 -070012#include <asm/state.h>
Simon Glass2e7d35d2014-02-26 15:59:21 -070013#include <dm/test.h>
14#include <dm/root.h>
15#include <dm/uclass-internal.h>
Joe Hershbergere721b882015-05-20 14:27:27 -050016#include <test/ut.h>
Simon Glass2e7d35d2014-02-26 15:59:21 -070017
18DECLARE_GLOBAL_DATA_PTR;
19
Joe Hershbergere721b882015-05-20 14:27:27 -050020struct unit_test_state global_dm_test_state;
21static struct dm_test_state _global_priv_dm_test_state;
Simon Glass2e7d35d2014-02-26 15:59:21 -070022
23/* Get ready for testing */
Simon Glassc166c472017-05-18 20:09:16 -060024static int dm_test_init(struct unit_test_state *uts, bool of_live)
Simon Glass2e7d35d2014-02-26 15:59:21 -070025{
Joe Hershbergere721b882015-05-20 14:27:27 -050026 struct dm_test_state *dms = uts->priv;
27
Simon Glass2e7d35d2014-02-26 15:59:21 -070028 memset(dms, '\0', sizeof(*dms));
29 gd->dm_root = NULL;
30 memset(dm_testdrv_op_count, '\0', sizeof(dm_testdrv_op_count));
Simon Glass34b744b2017-05-18 20:09:13 -060031 state_reset_for_test(state_get_current());
Simon Glass2e7d35d2014-02-26 15:59:21 -070032
Simon Glassc166c472017-05-18 20:09:16 -060033#ifdef CONFIG_OF_LIVE
34 /* Determine whether to make the live tree available */
35 gd->of_root = of_live ? uts->of_root : NULL;
36#endif
37 ut_assertok(dm_init(of_live));
Simon Glass2e7d35d2014-02-26 15:59:21 -070038 dms->root = dm_root();
39
40 return 0;
41}
42
43/* Ensure all the test devices are probed */
Joe Hershbergere721b882015-05-20 14:27:27 -050044static int do_autoprobe(struct unit_test_state *uts)
Simon Glass2e7d35d2014-02-26 15:59:21 -070045{
Heiko Schocher54c5d082014-05-22 12:43:05 +020046 struct udevice *dev;
Simon Glass2e7d35d2014-02-26 15:59:21 -070047 int ret;
48
49 /* Scanning the uclass is enough to probe all the devices */
50 for (ret = uclass_first_device(UCLASS_TEST, &dev);
51 dev;
52 ret = uclass_next_device(&dev))
53 ;
54
55 return ret;
56}
57
Joe Hershbergere721b882015-05-20 14:27:27 -050058static int dm_test_destroy(struct unit_test_state *uts)
Simon Glass2e7d35d2014-02-26 15:59:21 -070059{
60 int id;
61
62 for (id = 0; id < UCLASS_COUNT; id++) {
63 struct uclass *uc;
64
65 /*
66 * If the uclass doesn't exist we don't want to create it. So
Simon Glass22327002019-09-25 08:55:57 -060067 * check that here before we call uclass_find_device().
Simon Glass2e7d35d2014-02-26 15:59:21 -070068 */
69 uc = uclass_find(id);
70 if (!uc)
71 continue;
72 ut_assertok(uclass_destroy(uc));
73 }
74
75 return 0;
76}
77
Simon Glassc166c472017-05-18 20:09:16 -060078static int dm_do_test(struct unit_test_state *uts, struct unit_test *test,
79 bool of_live)
Simon Glassf86db102017-05-18 20:09:14 -060080{
81 struct sandbox_state *state = state_get_current();
Simon Glass801587b2017-05-18 20:09:15 -060082 const char *fname = strrchr(test->file, '/') + 1;
Simon Glassf86db102017-05-18 20:09:14 -060083
Simon Glassc166c472017-05-18 20:09:16 -060084 printf("Test: %s: %s%s\n", test->name, fname,
85 !of_live ? " (flat tree)" : "");
86 ut_assertok(dm_test_init(uts, of_live));
Simon Glassf86db102017-05-18 20:09:14 -060087
88 uts->start = mallinfo();
89 if (test->flags & DM_TESTF_SCAN_PDATA)
90 ut_assertok(dm_scan_platdata(false));
91 if (test->flags & DM_TESTF_PROBE_TEST)
92 ut_assertok(do_autoprobe(uts));
93 if (test->flags & DM_TESTF_SCAN_FDT)
Patrice Chotardee87a092017-09-04 14:55:57 +020094 ut_assertok(dm_extended_scan_fdt(gd->fdt_blob, false));
Simon Glassf86db102017-05-18 20:09:14 -060095
96 /*
Michal Simek96ffe872018-08-20 08:03:22 +020097 * Silence the console and rely on console recording to get
Simon Glassf86db102017-05-18 20:09:14 -060098 * our output.
99 */
Simon Glasscfccff82020-01-27 08:49:55 -0700100 console_record_reset_enable();
Simon Glassf86db102017-05-18 20:09:14 -0600101 if (!state->show_test_output)
102 gd->flags |= GD_FLG_SILENT;
103 test->func(uts);
Simon Glasscfccff82020-01-27 08:49:55 -0700104 gd->flags &= ~(GD_FLG_SILENT | GD_FLG_RECORD);
Simon Glassf86db102017-05-18 20:09:14 -0600105 state_set_skip_delays(false);
106
107 ut_assertok(dm_test_destroy(uts));
108
109 return 0;
110}
111
Simon Glass6fb2f572017-05-18 20:09:17 -0600112/**
113 * dm_test_run_on_flattree() - Check if we should run a test with flat DT
114 *
115 * This skips long/slow tests where there is not much value in running a flat
116 * DT test in addition to a live DT test.
117 *
118 * @return true to run the given test on the flat device tree
119 */
120static bool dm_test_run_on_flattree(struct unit_test *test)
121{
122 const char *fname = strrchr(test->file, '/') + 1;
123
124 return !strstr(fname, "video") || strstr(test->name, "video_base");
125}
126
Joe Hershberger40441e02015-05-20 14:27:29 -0500127static int dm_test_main(const char *test_name)
Simon Glass2e7d35d2014-02-26 15:59:21 -0700128{
Joe Hershbergere721b882015-05-20 14:27:27 -0500129 struct unit_test *tests = ll_entry_start(struct unit_test, dm_test);
130 const int n_ents = ll_entry_count(struct unit_test, dm_test);
131 struct unit_test_state *uts = &global_dm_test_state;
Joe Hershbergere721b882015-05-20 14:27:27 -0500132 struct unit_test *test;
Simon Glassad950472019-09-25 08:55:52 -0600133 int found;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700134
Simon Glassc166c472017-05-18 20:09:16 -0600135 uts->priv = &_global_priv_dm_test_state;
Stephen Warren26e1bec2016-01-27 23:57:46 -0700136 uts->fail_count = 0;
137
Simon Glass2e7d35d2014-02-26 15:59:21 -0700138 /*
139 * If we have no device tree, or it only has a root node, then these
140 * tests clearly aren't going to work...
141 */
142 if (!gd->fdt_blob || fdt_next_node(gd->fdt_blob, 0, NULL) < 0) {
143 puts("Please run with test device tree:\n"
Przemyslaw Marczakf64000c2015-05-13 13:38:34 +0200144 " ./u-boot -d arch/sandbox/dts/test.dtb\n");
Simon Glass2e7d35d2014-02-26 15:59:21 -0700145 ut_assert(gd->fdt_blob);
146 }
147
Simon Glass57f54d52015-03-25 12:23:04 -0600148 if (!test_name)
149 printf("Running %d driver model tests\n", n_ents);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700150
Simon Glassad950472019-09-25 08:55:52 -0600151 found = 0;
Simon Glassc166c472017-05-18 20:09:16 -0600152#ifdef CONFIG_OF_LIVE
153 uts->of_root = gd->of_root;
154#endif
Simon Glass2e7d35d2014-02-26 15:59:21 -0700155 for (test = tests; test < tests + n_ents; test++) {
Simon Glassc02790c2015-07-06 12:54:23 -0600156 const char *name = test->name;
Simon Glass6fb2f572017-05-18 20:09:17 -0600157 int runs;
Simon Glassc02790c2015-07-06 12:54:23 -0600158
159 /* All tests have this prefix */
160 if (!strncmp(name, "dm_test_", 8))
161 name += 8;
162 if (test_name && strcmp(test_name, name))
Simon Glass57f54d52015-03-25 12:23:04 -0600163 continue;
Simon Glass6fb2f572017-05-18 20:09:17 -0600164
165 /* Run with the live tree if possible */
166 runs = 0;
167 if (IS_ENABLED(CONFIG_OF_LIVE)) {
168 if (!(test->flags & DM_TESTF_FLAT_TREE)) {
169 ut_assertok(dm_do_test(uts, test, true));
170 runs++;
171 }
172 }
173
174 /*
175 * Run with the flat tree if we couldn't run it with live tree,
176 * or it is a core test.
177 */
178 if (!(test->flags & DM_TESTF_LIVE_TREE) &&
179 (!runs || dm_test_run_on_flattree(test))) {
180 ut_assertok(dm_do_test(uts, test, false));
181 runs++;
182 }
Simon Glassad950472019-09-25 08:55:52 -0600183 found++;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700184 }
185
Simon Glassad950472019-09-25 08:55:52 -0600186 if (test_name && !found)
Simon Glassc02790c2015-07-06 12:54:23 -0600187 printf("Test '%s' not found\n", test_name);
188 else
189 printf("Failures: %d\n", uts->fail_count);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700190
Simon Glassfe9a9672019-09-25 08:55:51 -0600191 /* Put everything back to normal so that sandbox works as expected */
192#ifdef CONFIG_OF_LIVE
193 gd->of_root = uts->of_root;
194#endif
Joe Hershbergerb6227d32015-05-20 14:27:35 -0500195 gd->dm_root = NULL;
Simon Glassfe9a9672019-09-25 08:55:51 -0600196 ut_assertok(dm_init(IS_ENABLED(CONFIG_OF_LIVE)));
Joe Hershbergerb6227d32015-05-20 14:27:35 -0500197 dm_scan_platdata(false);
198 dm_scan_fdt(gd->fdt_blob, false);
199
Joe Hershberger7cccc662015-05-20 14:27:32 -0500200 return uts->fail_count ? CMD_RET_FAILURE : 0;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700201}
Joe Hershberger40441e02015-05-20 14:27:29 -0500202
Simon Glass09140112020-05-10 11:40:03 -0600203int do_ut_dm(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
Joe Hershberger40441e02015-05-20 14:27:29 -0500204{
205 const char *test_name = NULL;
206
207 if (argc > 1)
208 test_name = argv[1];
209
210 return dm_test_main(test_name);
211}