1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Copyright (C) 2020 Sean Anderson <seanga2@gmail.com>
4 */
5
6 #include <common.h>
7 #include <dm.h>
8 #include <dm/test.h>
9 #include <dm/device-internal.h>
10 #include <test/ut.h>
11 #include <asm/clk.h>
12 #include <asm/power-domain.h>
13
14 /* These must match the ids in the device tree */
15 #define TEST_CLOCK_ID 4
16 #define TEST_POWER_ID 1
17
dm_test_simple_pm_bus(struct unit_test_state * uts)18 static int dm_test_simple_pm_bus(struct unit_test_state *uts)
19 {
20 struct udevice *power;
21 struct udevice *clock;
22 struct udevice *bus;
23
24 ut_assertok(uclass_get_device_by_name(UCLASS_POWER_DOMAIN,
25 "power-domain", &power));
26 ut_assertok(uclass_get_device_by_name(UCLASS_CLK, "clk-sbox",
27 &clock));
28 ut_asserteq(0, sandbox_power_domain_query(power, TEST_POWER_ID));
29 ut_asserteq(0, sandbox_clk_query_enable(clock, TEST_CLOCK_ID));
30
31 ut_assertok(uclass_get_device_by_name(UCLASS_SIMPLE_BUS, "pm-bus-test",
32 &bus));
33 ut_asserteq(1, sandbox_power_domain_query(power, TEST_POWER_ID));
34 ut_asserteq(1, sandbox_clk_query_enable(clock, TEST_CLOCK_ID));
35
36 ut_assertok(device_remove(bus, DM_REMOVE_NORMAL));
37 /* must re-probe since device_remove also removes the power domain */
38 ut_assertok(uclass_get_device_by_name(UCLASS_POWER_DOMAIN,
39 "power-domain", &power));
40 ut_asserteq(0, sandbox_power_domain_query(power, TEST_POWER_ID));
41 ut_asserteq(0, sandbox_clk_query_enable(clock, TEST_CLOCK_ID));
42
43 return 0;
44 }
45 DM_TEST(dm_test_simple_pm_bus, UT_TESTF_SCAN_FDT);
46