blob: 67c0082fb841fff3a7f20c43df38fd1c6bd15958 [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));
32
Simon Glass19c82052017-05-18 20:09:08 -060033 ut_assertok(dm_init(false));
Simon Glass2e7d35d2014-02-26 15:59:21 -070034 dms->root = dm_root();
35
36 return 0;
37}
38
39/* Ensure all the test devices are probed */
Joe Hershbergere721b882015-05-20 14:27:27 -050040static int do_autoprobe(struct unit_test_state *uts)
Simon Glass2e7d35d2014-02-26 15:59:21 -070041{
Heiko Schocher54c5d082014-05-22 12:43:05 +020042 struct udevice *dev;
Simon Glass2e7d35d2014-02-26 15:59:21 -070043 int ret;
44
45 /* Scanning the uclass is enough to probe all the devices */
46 for (ret = uclass_first_device(UCLASS_TEST, &dev);
47 dev;
48 ret = uclass_next_device(&dev))
49 ;
50
51 return ret;
52}
53
Joe Hershbergere721b882015-05-20 14:27:27 -050054static int dm_test_destroy(struct unit_test_state *uts)
Simon Glass2e7d35d2014-02-26 15:59:21 -070055{
56 int id;
57
58 for (id = 0; id < UCLASS_COUNT; id++) {
59 struct uclass *uc;
60
61 /*
62 * If the uclass doesn't exist we don't want to create it. So
63 * check that here before we call uclass_find_device()/
64 */
65 uc = uclass_find(id);
66 if (!uc)
67 continue;
68 ut_assertok(uclass_destroy(uc));
69 }
70
71 return 0;
72}
73
Joe Hershberger40441e02015-05-20 14:27:29 -050074static int dm_test_main(const char *test_name)
Simon Glass2e7d35d2014-02-26 15:59:21 -070075{
Joe Hershbergere721b882015-05-20 14:27:27 -050076 struct unit_test *tests = ll_entry_start(struct unit_test, dm_test);
77 const int n_ents = ll_entry_count(struct unit_test, dm_test);
78 struct unit_test_state *uts = &global_dm_test_state;
Simon Glass9ce8b402015-11-08 23:47:50 -070079 struct sandbox_state *state = state_get_current();
Joe Hershbergere721b882015-05-20 14:27:27 -050080 uts->priv = &_global_priv_dm_test_state;
81 struct unit_test *test;
Simon Glassc02790c2015-07-06 12:54:23 -060082 int run_count;
Simon Glass2e7d35d2014-02-26 15:59:21 -070083
Stephen Warren26e1bec2016-01-27 23:57:46 -070084 uts->fail_count = 0;
85
Simon Glass2e7d35d2014-02-26 15:59:21 -070086 /*
87 * If we have no device tree, or it only has a root node, then these
88 * tests clearly aren't going to work...
89 */
90 if (!gd->fdt_blob || fdt_next_node(gd->fdt_blob, 0, NULL) < 0) {
91 puts("Please run with test device tree:\n"
Przemyslaw Marczakf64000c2015-05-13 13:38:34 +020092 " ./u-boot -d arch/sandbox/dts/test.dtb\n");
Simon Glass2e7d35d2014-02-26 15:59:21 -070093 ut_assert(gd->fdt_blob);
94 }
95
Simon Glass57f54d52015-03-25 12:23:04 -060096 if (!test_name)
97 printf("Running %d driver model tests\n", n_ents);
Simon Glass2e7d35d2014-02-26 15:59:21 -070098
Simon Glassc02790c2015-07-06 12:54:23 -060099 run_count = 0;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700100 for (test = tests; test < tests + n_ents; test++) {
Simon Glassc02790c2015-07-06 12:54:23 -0600101 const char *name = test->name;
102
103 /* All tests have this prefix */
104 if (!strncmp(name, "dm_test_", 8))
105 name += 8;
106 if (test_name && strcmp(test_name, name))
Simon Glass57f54d52015-03-25 12:23:04 -0600107 continue;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700108 printf("Test: %s\n", test->name);
Simon Glassc02790c2015-07-06 12:54:23 -0600109 run_count++;
Joe Hershbergere721b882015-05-20 14:27:27 -0500110 ut_assertok(dm_test_init(uts));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700111
Joe Hershbergere721b882015-05-20 14:27:27 -0500112 uts->start = mallinfo();
Simon Glass2e7d35d2014-02-26 15:59:21 -0700113 if (test->flags & DM_TESTF_SCAN_PDATA)
Simon Glass00606d72014-07-23 06:55:03 -0600114 ut_assertok(dm_scan_platdata(false));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700115 if (test->flags & DM_TESTF_PROBE_TEST)
Joe Hershbergere721b882015-05-20 14:27:27 -0500116 ut_assertok(do_autoprobe(uts));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700117 if (test->flags & DM_TESTF_SCAN_FDT)
Simon Glass00606d72014-07-23 06:55:03 -0600118 ut_assertok(dm_scan_fdt(gd->fdt_blob, false));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700119
Simon Glass9ce8b402015-11-08 23:47:50 -0700120 /*
121 * Silence the console and rely on console reocrding to get
122 * our output.
123 */
124 console_record_reset();
125 if (!state->show_test_output)
126 gd->flags |= GD_FLG_SILENT;
Joe Hershberger7cccc662015-05-20 14:27:32 -0500127 test->func(uts);
Simon Glass9ce8b402015-11-08 23:47:50 -0700128 gd->flags &= ~GD_FLG_SILENT;
Simon Glass3884c982015-11-08 23:47:44 -0700129 state_set_skip_delays(false);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700130
Joe Hershbergere721b882015-05-20 14:27:27 -0500131 ut_assertok(dm_test_destroy(uts));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700132 }
133
Simon Glassc02790c2015-07-06 12:54:23 -0600134 if (test_name && !run_count)
135 printf("Test '%s' not found\n", test_name);
136 else
137 printf("Failures: %d\n", uts->fail_count);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700138
Joe Hershbergerb6227d32015-05-20 14:27:35 -0500139 gd->dm_root = NULL;
Simon Glass19c82052017-05-18 20:09:08 -0600140 ut_assertok(dm_init(false));
Joe Hershbergerb6227d32015-05-20 14:27:35 -0500141 dm_scan_platdata(false);
142 dm_scan_fdt(gd->fdt_blob, false);
143
Joe Hershberger7cccc662015-05-20 14:27:32 -0500144 return uts->fail_count ? CMD_RET_FAILURE : 0;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700145}
Joe Hershberger40441e02015-05-20 14:27:29 -0500146
147int do_ut_dm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
148{
149 const char *test_name = NULL;
150
151 if (argc > 1)
152 test_name = argv[1];
153
154 return dm_test_main(test_name);
155}