dm: power-domain: query power domain after get device

This is to test power_domain_on in device_probe.
If the device has a power-domain property, enable it
when probe the device. So add the test to check
whether it is powered on or not.

Signed-off-by: Peng Fan <peng.fan@nxp.com>
Reviewed-by: Simon Glass <sjg@chromium.org>

diff --git a/test/dm/power-domain.c b/test/dm/power-domain.c
index a1e1df2..4831821 100644 (file)
--- a/test/dm/power-domain.c
+++ b/test/dm/power-domain.c
@@ -26,6 +26,8 @@ static int dm_test_power_domain(struct unit_test_state *uts)
 
        ut_assertok(uclass_get_device_by_name(UCLASS_MISC, "power-domain-test",
                                              &dev_test));
+       ut_asserteq(1, sandbox_power_domain_query(dev_power_domain,
+                                                 TEST_POWER_DOMAIN));
        ut_assertok(sandbox_power_domain_test_get(dev_test));
 
        ut_assertok(sandbox_power_domain_test_on(dev_test));