blob: 91bdda83ab36a2ee6600cf2132379d27f9f06c12 [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
33 ut_assertok(dm_init());
34 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
84 /*
85 * If we have no device tree, or it only has a root node, then these
86 * tests clearly aren't going to work...
87 */
88 if (!gd->fdt_blob || fdt_next_node(gd->fdt_blob, 0, NULL) < 0) {
89 puts("Please run with test device tree:\n"
Przemyslaw Marczakf64000c2015-05-13 13:38:34 +020090 " ./u-boot -d arch/sandbox/dts/test.dtb\n");
Simon Glass2e7d35d2014-02-26 15:59:21 -070091 ut_assert(gd->fdt_blob);
92 }
93
Simon Glass57f54d52015-03-25 12:23:04 -060094 if (!test_name)
95 printf("Running %d driver model tests\n", n_ents);
Simon Glass2e7d35d2014-02-26 15:59:21 -070096
Simon Glassc02790c2015-07-06 12:54:23 -060097 run_count = 0;
Simon Glass2e7d35d2014-02-26 15:59:21 -070098 for (test = tests; test < tests + n_ents; test++) {
Simon Glassc02790c2015-07-06 12:54:23 -060099 const char *name = test->name;
100
101 /* All tests have this prefix */
102 if (!strncmp(name, "dm_test_", 8))
103 name += 8;
104 if (test_name && strcmp(test_name, name))
Simon Glass57f54d52015-03-25 12:23:04 -0600105 continue;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700106 printf("Test: %s\n", test->name);
Simon Glassc02790c2015-07-06 12:54:23 -0600107 run_count++;
Joe Hershbergere721b882015-05-20 14:27:27 -0500108 ut_assertok(dm_test_init(uts));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700109
Joe Hershbergere721b882015-05-20 14:27:27 -0500110 uts->start = mallinfo();
Simon Glass2e7d35d2014-02-26 15:59:21 -0700111 if (test->flags & DM_TESTF_SCAN_PDATA)
Simon Glass00606d72014-07-23 06:55:03 -0600112 ut_assertok(dm_scan_platdata(false));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700113 if (test->flags & DM_TESTF_PROBE_TEST)
Joe Hershbergere721b882015-05-20 14:27:27 -0500114 ut_assertok(do_autoprobe(uts));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700115 if (test->flags & DM_TESTF_SCAN_FDT)
Simon Glass00606d72014-07-23 06:55:03 -0600116 ut_assertok(dm_scan_fdt(gd->fdt_blob, false));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700117
Simon Glass9ce8b402015-11-08 23:47:50 -0700118 /*
119 * Silence the console and rely on console reocrding to get
120 * our output.
121 */
122 console_record_reset();
123 if (!state->show_test_output)
124 gd->flags |= GD_FLG_SILENT;
Joe Hershberger7cccc662015-05-20 14:27:32 -0500125 test->func(uts);
Simon Glass9ce8b402015-11-08 23:47:50 -0700126 gd->flags &= ~GD_FLG_SILENT;
Simon Glass3884c982015-11-08 23:47:44 -0700127 state_set_skip_delays(false);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700128
Joe Hershbergere721b882015-05-20 14:27:27 -0500129 ut_assertok(dm_test_destroy(uts));
Simon Glass2e7d35d2014-02-26 15:59:21 -0700130 }
131
Simon Glassc02790c2015-07-06 12:54:23 -0600132 if (test_name && !run_count)
133 printf("Test '%s' not found\n", test_name);
134 else
135 printf("Failures: %d\n", uts->fail_count);
Simon Glass2e7d35d2014-02-26 15:59:21 -0700136
Joe Hershbergerb6227d32015-05-20 14:27:35 -0500137 gd->dm_root = NULL;
138 ut_assertok(dm_init());
139 dm_scan_platdata(false);
140 dm_scan_fdt(gd->fdt_blob, false);
141
Joe Hershberger7cccc662015-05-20 14:27:32 -0500142 return uts->fail_count ? CMD_RET_FAILURE : 0;
Simon Glass2e7d35d2014-02-26 15:59:21 -0700143}
Joe Hershberger40441e02015-05-20 14:27:29 -0500144
145int do_ut_dm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
146{
147 const char *test_name = NULL;
148
149 if (argc > 1)
150 test_name = argv[1];
151
152 return dm_test_main(test_name);
153}