dm: led: Add support for getting the state of an LED

[people/ms/u-boot.git] / test / dm / led.c

diff --git a/test/dm/led.c b/test/dm/led.c
index 8ee075cf1ca75e61818668dbc51696da56431178..68aa39bd4d0809ae08761f2dd4a0b80d4306545c 100644 (file)
--- a/test/dm/led.c
+++ b/test/dm/led.c
@@ -41,10 +41,13 @@ static int dm_test_led_gpio(struct unit_test_state *uts)
        ut_assertok(uclass_get_device(UCLASS_LED, 1, &dev));
        ut_assertok(uclass_get_device(UCLASS_GPIO, 1, &gpio));
        ut_asserteq(0, sandbox_gpio_get_value(gpio, offset));
-       led_set_on(dev, 1);
+       ut_assertok(led_set_state(dev, LEDST_ON));
        ut_asserteq(1, sandbox_gpio_get_value(gpio, offset));
-       led_set_on(dev, 0);
+       ut_asserteq(LEDST_ON, led_get_state(dev));
+
+       ut_assertok(led_set_state(dev, LEDST_OFF));
        ut_asserteq(0, sandbox_gpio_get_value(gpio, offset));
+       ut_asserteq(LEDST_OFF, led_get_state(dev));
 
        return 0;
 }