blob: 10d2706377fafc70419eaaff39133b26191897aa [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 */
Joe Hershbergere721b882015-05-20 14:27:27 -050025static int dm_test_init(struct unit_test_state *uts)
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 Glass19c82052017-05-18 20:09:08 -060034 ut_assertok(dm_init(false));
Simon Glass2e7d35d2014-02-26 15:59:21 -070035 dms->root = dm_root();
36
37 return 0;
38}
39
40/* Ensure all the test devices are probed */
Joe Hershbergere721b882015-05-20 14:27:27 -050041static int do_autoprobe(struct unit_test_state *uts)
Simon Glass2e7d35d2014-02-26 15:59:21 -070042{
Heiko Schocher54c5d082014-05-22 12:43:05 +020043 struct udevice *dev;
Simon Glass2e7d35d2014-02-26 15:59:21 -070044 int ret;
45
46 /* Scanning the uclass is enough to probe all the devices */
47 for (ret = uclass_first_device(UCLASS_TEST, &dev);
48 dev;
49 ret = uclass_next_device(&dev))
50 ;
51
52 return ret;
53}
54
Joe Hershbergere721b882015-05-20 14:27:27 -050055static int dm_test_destroy(struct unit_test_state *uts)
Simon Glass2e7d35d2014-02-26 15:59:21 -070056{
57 int id;
58
59 for (id = 0; id < UCLASS_COUNT; id++) {
60 struct uclass *uc;
61
62 /*
63 * If the uclass doesn't exist we don't want to create it. So
64 * check that here before we call uclass_find_device()/
65 */
66 uc = uclass_find(id);
67 if (!uc)
68 continue;
69 ut_assertok(uclass_destroy(uc));
70 }
71
72 return 0;
73}
74
Simon Glassf86db102017-05-18 20:09:14 -060075static int dm_do_test(struct unit_test_state *uts, struct unit_test *test)
76{
77 struct sandbox_state *state = state_get_current();
Simon Glass801587b2017-05-18 20:09:15 -060078 const char *fname = strrchr(test->file, '/') + 1;
Simon Glassf86db102017-05-18 20:09:14 -060079
Simon Glass801587b2017-05-18 20:09:15 -060080 printf("Test: %s: %s\n", test->name, fname);
Simon Glassf86db102017-05-18 20:09:14 -060081 ut_assertok(dm_test_init(uts));
82
83 uts->start = mallinfo();
84 if (test->flags & DM_TESTF_SCAN_PDATA)
85 ut_assertok(dm_scan_platdata(false));
86 if (test->flags & DM_TESTF_PROBE_TEST)
87 ut_assertok(do_autoprobe(uts));
88 if (test->flags & DM_TESTF_SCAN_FDT)
89 ut_assertok(dm_scan_fdt(gd->fdt_blob, false));
90
91 /*
92 * Silence the console and rely on console reocrding to get
93 * our output.
94 */
95 console_record_reset();
96 if (!state->show_test_output)
97 gd->flags |= GD_FLG_SILENT;
98 test->func(uts);
99 gd->flags &= ~GD_FLG_SILENT;
100 state_set_skip_delays(false);
101
102 ut_assertok(dm_test_destroy(uts));
103
104 return 0;
105}
106
Joe Hershberger40441e02015-05-20 14:27:29 -0500107static int dm_test_main(const char *test_name)
Simon Glass2e7d35d2014-02-26 15:59:21 -0700108{
Joe Hershbergere721b882015-05-20 14:27:27 -0500109 struct unit_test *tests = ll_entry_start(struct unit_test, dm_test);
110 const int n_ents = ll_entry_count(struct unit_test, dm_test);
111 struct unit_test_state *uts = &global_dm_test_state;
112 uts->priv = &_global_priv_dm_test_state;
113 struct unit_test *test;
Simon Glassc02790c2015-07-06 12:54:23 -0600114 int run_count;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700115
Stephen Warren26e1bec2016-01-27 23:57:46 -0700116 uts->fail_count = 0;
117
Simon Glass2e7d35d2014-02-26 15:59:21 -0700118 /*
119 * If we have no device tree, or it only has a root node, then these
120 * tests clearly aren't going to work...
121 */
122 if (!gd->fdt_blob || fdt_next_node(gd->fdt_blob, 0, NULL) < 0) {
123 puts("Please run with test device tree:\n"
Przemyslaw Marczakf64000c2015-05-13 13:38:34 +0200124 " ./u-boot -d arch/sandbox/dts/test.dtb\n");
Simon Glass2e7d35d2014-02-26 15:59:21 -0700125 ut_assert(gd->fdt_blob);
126 }
127
Simon Glass57f54d52015-03-25 12:23:04 -0600128 if (!test_name)
129 printf("Running %d driver model tests\n", n_ents);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700130
Simon Glassc02790c2015-07-06 12:54:23 -0600131 run_count = 0;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700132 for (test = tests; test < tests + n_ents; test++) {
Simon Glassc02790c2015-07-06 12:54:23 -0600133 const char *name = test->name;
134
135 /* All tests have this prefix */
136 if (!strncmp(name, "dm_test_", 8))
137 name += 8;
138 if (test_name && strcmp(test_name, name))
Simon Glass57f54d52015-03-25 12:23:04 -0600139 continue;
Simon Glassf86db102017-05-18 20:09:14 -0600140 ut_assertok(dm_do_test(uts, test));
Simon Glassc02790c2015-07-06 12:54:23 -0600141 run_count++;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700142 }
143
Simon Glassc02790c2015-07-06 12:54:23 -0600144 if (test_name && !run_count)
145 printf("Test '%s' not found\n", test_name);
146 else
147 printf("Failures: %d\n", uts->fail_count);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700148
Joe Hershbergerb6227d32015-05-20 14:27:35 -0500149 gd->dm_root = NULL;
Simon Glass19c82052017-05-18 20:09:08 -0600150 ut_assertok(dm_init(false));
Joe Hershbergerb6227d32015-05-20 14:27:35 -0500151 dm_scan_platdata(false);
152 dm_scan_fdt(gd->fdt_blob, false);
153
Joe Hershberger7cccc662015-05-20 14:27:32 -0500154 return uts->fail_count ? CMD_RET_FAILURE : 0;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700155}
Joe Hershberger40441e02015-05-20 14:27:29 -0500156
157int do_ut_dm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
158{
159 const char *test_name = NULL;
160
161 if (argc > 1)
162 test_name = argv[1];
163
164 return dm_test_main(test_name);
165}