1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (c) 2016, NVIDIA CORPORATION.
4  */
5 
6 #include <dm.h>
7 #include <malloc.h>
8 #include <dm/test.h>
9 #include <asm/power-domain.h>
10 #include <test/test.h>
11 #include <test/ut.h>
12 
13 /* This must match the specifier for power-domains in the DT node */
14 #define TEST_POWER_DOMAIN 2
15 
dm_test_power_domain(struct unit_test_state * uts)16 static int dm_test_power_domain(struct unit_test_state *uts)
17 {
18 	struct udevice *dev_power_domain;
19 	struct udevice *dev_test;
20 
21 	ut_assertok(uclass_get_device_by_name(UCLASS_POWER_DOMAIN,
22 					      "power-domain",
23 					      &dev_power_domain));
24 	ut_asserteq(0, sandbox_power_domain_query(dev_power_domain, 0));
25 	ut_asserteq(0, sandbox_power_domain_query(dev_power_domain,
26 						  TEST_POWER_DOMAIN));
27 
28 	ut_assertok(uclass_get_device_by_name(UCLASS_MISC, "power-domain-test",
29 					      &dev_test));
30 	ut_asserteq(0, sandbox_power_domain_query(dev_power_domain,
31 						  TEST_POWER_DOMAIN));
32 	ut_assertok(sandbox_power_domain_test_get(dev_test));
33 
34 	ut_assertok(sandbox_power_domain_test_on(dev_test));
35 	ut_asserteq(0, sandbox_power_domain_query(dev_power_domain, 0));
36 	ut_asserteq(1, sandbox_power_domain_query(dev_power_domain,
37 						  TEST_POWER_DOMAIN));
38 	ut_asserteq(-EALREADY, sandbox_power_domain_test_on_ll(dev_test));
39 	ut_asserteq(1, sandbox_power_domain_query(dev_power_domain,
40 						  TEST_POWER_DOMAIN));
41 	ut_asserteq(-EBUSY, sandbox_power_domain_test_off_ll(dev_test));
42 	ut_asserteq(1, sandbox_power_domain_query(dev_power_domain,
43 						  TEST_POWER_DOMAIN));
44 
45 	ut_assertok(sandbox_power_domain_test_off(dev_test));
46 	ut_asserteq(0, sandbox_power_domain_query(dev_power_domain, 0));
47 	ut_asserteq(0, sandbox_power_domain_query(dev_power_domain,
48 						  TEST_POWER_DOMAIN));
49 	ut_asserteq(-EALREADY, sandbox_power_domain_test_off_ll(dev_test));
50 	ut_asserteq(0, sandbox_power_domain_query(dev_power_domain,
51 						  TEST_POWER_DOMAIN));
52 
53 	ut_assertok(sandbox_power_domain_test_free(dev_test));
54 
55 	return 0;
56 }
57 DM_TEST(dm_test_power_domain, UTF_SCAN_FDT);
58