1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2015 Mateusz Kulikowski <mateusz.kulikowski@gmail.com>
4  */
5 
6 #include <fdtdec.h>
7 #include <dm.h>
8 #include <malloc.h>
9 #include <dm/device.h>
10 #include <dm/root.h>
11 #include <dm/test.h>
12 #include <dm/util.h>
13 #include <power/pmic.h>
14 #include <spmi/spmi.h>
15 #include <asm/gpio.h>
16 #include <test/test.h>
17 #include <test/ut.h>
18 
19 /* Test if bus children got probed properly*/
dm_test_spmi_probe(struct unit_test_state * uts)20 static int dm_test_spmi_probe(struct unit_test_state *uts)
21 {
22 	const char *name = "spmi@0";
23 	struct udevice *bus, *dev;
24 
25 	ut_assertok(uclass_get_device(UCLASS_SPMI, 0, &bus));
26 
27 	/* Check bus name */
28 	ut_asserteq_str(name, bus->name);
29 
30 	/* Check that it has some devices */
31 	ut_asserteq(device_has_children(bus), true);
32 
33 	ut_assertok(device_find_first_child(bus, &dev));
34 
35 	/* There should be at least one child */
36 	ut_assertnonnull(dev);
37 
38 	/* Check that only PMICs are connected to the bus */
39 	while (dev) {
40 		ut_asserteq(device_get_uclass_id(dev), UCLASS_PMIC);
41 		device_find_next_child(&dev);
42 	}
43 
44 	return 0;
45 }
46 DM_TEST(dm_test_spmi_probe, UTF_SCAN_FDT);
47 
48 /* Test if it's possible to read bus directly and indirectly */
dm_test_spmi_access(struct unit_test_state * uts)49 static int dm_test_spmi_access(struct unit_test_state *uts)
50 {
51 	const char *pmic_name = "pm8916@0";
52 	struct udevice *bus, *pmic;
53 
54 	ut_assertok(uclass_get_device(UCLASS_SPMI, 0, &bus));
55 
56 	ut_assertok(device_get_child(bus, 0, &pmic));
57 
58 	/* Sanity check if it's proper PMIC */
59 	ut_asserteq_str(pmic_name, pmic->name);
60 
61 	/* Read PMIC ID reg using SPMI bus - it assumes it has slaveID == 0*/
62 	ut_asserteq(spmi_reg_read(bus, 0, 0xC0, 0x4), 0x10);
63 	ut_asserteq(spmi_reg_read(bus, 0, 0xC0, 0x5), 0x5);
64 
65 	/* Read ID reg via pmic interface */
66 	ut_asserteq(pmic_reg_read(pmic, 0xC004), 0x10);
67 	ut_asserteq(pmic_reg_read(pmic, 0xC005), 0x5);
68 
69 	return 0;
70 }
71 DM_TEST(dm_test_spmi_access, UTF_SCAN_FDT);
72 
73 /* Test if it's possible to access GPIO that should be in pmic */
dm_test_spmi_access_peripheral(struct unit_test_state * uts)74 static int dm_test_spmi_access_peripheral(struct unit_test_state *uts)
75 {
76 	struct udevice *dev;
77 	unsigned int offset, gpio;
78 	const char *name;
79 	int offset_count;
80 
81 	/* Get second pin of PMIC GPIO */
82 	ut_assertok(gpio_lookup_name("pmic1", &dev, &offset, &gpio));
83 
84 	/* Check if PMIC is parent */
85 	ut_asserteq(device_get_uclass_id(dev->parent), UCLASS_PMIC);
86 
87 	/* This should be second gpio */
88 	ut_asserteq(1, offset);
89 
90 	name = gpio_get_bank_info(dev, &offset_count);
91 
92 	/* Check bank name */
93 	ut_asserteq_str("pmic", name);
94 	/* Check pin count */
95 	ut_asserteq(4, offset_count);
96 
97 	ut_assertok(gpio_request(gpio, "testing"));
98 
99 	/* Try to set/clear gpio */
100 	ut_assertok(gpio_direction_output(gpio, 0));
101 	ut_asserteq(gpio_get_value(gpio), 0);
102 	ut_assertok(gpio_direction_output(gpio, 1));
103 	ut_asserteq(gpio_get_value(gpio), 1);
104 	ut_assertok(gpio_direction_input(gpio));
105 	ut_asserteq(gpio_get_value(gpio), 1);
106 
107 	ut_assertok(gpio_free(gpio));
108 
109 	return 0;
110 }
111 DM_TEST(dm_test_spmi_access_peripheral, UTF_SCAN_FDT);
112