2 * Copyright (c) 2014 Google, Inc
4 * SPDX-License-Identifier: GPL-2.0+
14 DECLARE_GLOBAL_DATA_PTR;
16 static int testbus_drv_probe(struct udevice *dev)
18 return dm_scan_fdt_node(dev, gd->fdt_blob, dev->of_offset, false);
21 static const struct udevice_id testbus_ids[] = {
23 .compatible = "denx,u-boot-test-bus",
24 .data = DM_TEST_TYPE_FIRST },
28 U_BOOT_DRIVER(testbus_drv) = {
29 .name = "testbus_drv",
30 .of_match = testbus_ids,
31 .id = UCLASS_TEST_BUS,
32 .probe = testbus_drv_probe,
33 .priv_auto_alloc_size = sizeof(struct dm_test_priv),
34 .platdata_auto_alloc_size = sizeof(struct dm_test_pdata),
37 UCLASS_DRIVER(testbus) = {
39 .id = UCLASS_TEST_BUS,
42 /* Test that we can probe for children */
43 static int dm_test_bus_children(struct dm_test_state *dms)
49 ut_assertok(uclass_get(UCLASS_TEST_FDT, &uc));
50 ut_asserteq(num_devices, list_count_items(&uc->dev_head));
52 /* Probe the bus, which should yield 3 more devices */
53 ut_assertok(uclass_get_device(UCLASS_TEST_BUS, 0, &bus));
56 ut_assertok(uclass_get(UCLASS_TEST_FDT, &uc));
57 ut_asserteq(num_devices, list_count_items(&uc->dev_head));
59 ut_assert(!dm_check_devices(dms, num_devices));
63 DM_TEST(dm_test_bus_children, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
65 /* Test our functions for accessing children */
66 static int dm_test_bus_children_funcs(struct dm_test_state *dms)
68 const void *blob = gd->fdt_blob;
69 struct udevice *bus, *dev;
72 ut_assertok(uclass_get_device(UCLASS_TEST_BUS, 0, &bus));
74 /* device_get_child() */
75 ut_assertok(device_get_child(bus, 0, &dev));
76 ut_asserteq(-ENODEV, device_get_child(bus, 4, &dev));
77 ut_assertok(device_get_child_by_seq(bus, 5, &dev));
78 ut_assert(dev->flags & DM_FLAG_ACTIVATED);
79 ut_asserteq_str("c-test@5", dev->name);
81 /* Device with sequence number 0 should be accessible */
82 ut_asserteq(-ENODEV, device_find_child_by_seq(bus, -1, true, &dev));
83 ut_assertok(device_find_child_by_seq(bus, 0, true, &dev));
84 ut_assert(!(dev->flags & DM_FLAG_ACTIVATED));
85 ut_asserteq(-ENODEV, device_find_child_by_seq(bus, 0, false, &dev));
86 ut_assertok(device_get_child_by_seq(bus, 0, &dev));
87 ut_assert(dev->flags & DM_FLAG_ACTIVATED);
89 /* There is no device with sequence number 2 */
90 ut_asserteq(-ENODEV, device_find_child_by_seq(bus, 2, false, &dev));
91 ut_asserteq(-ENODEV, device_find_child_by_seq(bus, 2, true, &dev));
92 ut_asserteq(-ENODEV, device_get_child_by_seq(bus, 2, &dev));
94 /* Looking for something that is not a child */
95 node = fdt_path_offset(blob, "/junk");
96 ut_asserteq(-ENODEV, device_find_child_by_of_offset(bus, node, &dev));
97 node = fdt_path_offset(blob, "/d-test");
98 ut_asserteq(-ENODEV, device_find_child_by_of_offset(bus, node, &dev));
100 /* Find a valid child */
101 node = fdt_path_offset(blob, "/some-bus/c-test@1");
102 ut_assertok(device_find_child_by_of_offset(bus, node, &dev));
103 ut_assert(!(dev->flags & DM_FLAG_ACTIVATED));
104 ut_assertok(device_get_child_by_of_offset(bus, node, &dev));
105 ut_assert(dev->flags & DM_FLAG_ACTIVATED);
109 DM_TEST(dm_test_bus_children_funcs, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);