dm: led: Add support for getting the state of an LED
[oweals/u-boot.git] / test / dm / led.c
1 /*
2  * Copyright (C) 2015 Google, Inc
3  *
4  * SPDX-License-Identifier:     GPL-2.0+
5  */
6
7 #include <common.h>
8 #include <dm.h>
9 #include <led.h>
10 #include <asm/gpio.h>
11 #include <dm/test.h>
12 #include <test/ut.h>
13
14 DECLARE_GLOBAL_DATA_PTR;
15
16 /* Base test of the led uclass */
17 static int dm_test_led_base(struct unit_test_state *uts)
18 {
19         struct udevice *dev;
20
21         /* Get the top-level device */
22         ut_assertok(uclass_get_device(UCLASS_LED, 0, &dev));
23         ut_assertok(uclass_get_device(UCLASS_LED, 1, &dev));
24         ut_assertok(uclass_get_device(UCLASS_LED, 2, &dev));
25         ut_asserteq(-ENODEV, uclass_get_device(UCLASS_LED, 3, &dev));
26
27         return 0;
28 }
29 DM_TEST(dm_test_led_base, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
30
31 /* Test of the led uclass using the led_gpio driver */
32 static int dm_test_led_gpio(struct unit_test_state *uts)
33 {
34         const int offset = 1;
35         struct udevice *dev, *gpio;
36
37         /*
38          * Check that we can manipulate an LED. LED 1 is connected to GPIO
39          * bank gpio_a, offset 1.
40          */
41         ut_assertok(uclass_get_device(UCLASS_LED, 1, &dev));
42         ut_assertok(uclass_get_device(UCLASS_GPIO, 1, &gpio));
43         ut_asserteq(0, sandbox_gpio_get_value(gpio, offset));
44         ut_assertok(led_set_state(dev, LEDST_ON));
45         ut_asserteq(1, sandbox_gpio_get_value(gpio, offset));
46         ut_asserteq(LEDST_ON, led_get_state(dev));
47
48         ut_assertok(led_set_state(dev, LEDST_OFF));
49         ut_asserteq(0, sandbox_gpio_get_value(gpio, offset));
50         ut_asserteq(LEDST_OFF, led_get_state(dev));
51
52         return 0;
53 }
54 DM_TEST(dm_test_led_gpio, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
55
56 /* Test obtaining an LED by label */
57 static int dm_test_led_label(struct unit_test_state *uts)
58 {
59         struct udevice *dev, *cmp;
60
61         ut_assertok(led_get_by_label("sandbox:red", &dev));
62         ut_asserteq(1, device_active(dev));
63         ut_assertok(uclass_get_device(UCLASS_LED, 1, &cmp));
64         ut_asserteq_ptr(dev, cmp);
65
66         ut_assertok(led_get_by_label("sandbox:green", &dev));
67         ut_asserteq(1, device_active(dev));
68         ut_assertok(uclass_get_device(UCLASS_LED, 2, &cmp));
69         ut_asserteq_ptr(dev, cmp);
70
71         ut_asserteq(-ENODEV, led_get_by_label("sandbox:blue", &dev));
72
73         return 0;
74 }
75 DM_TEST(dm_test_led_label, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);