1
2
3
4
5
6
7
8#include <common.h>
9#include <dm.h>
10#include <fdtdec.h>
11#include <i2c.h>
12#include <asm/state.h>
13#include <asm/test.h>
14#include <dm/device-internal.h>
15#include <dm/test.h>
16#include <dm/uclass-internal.h>
17#include <dm/util.h>
18#include <test/ut.h>
19
20static const int busnum;
21static const int chip = 0x2c;
22
23
24static int dm_test_i2c_find(struct unit_test_state *uts)
25{
26 struct udevice *bus, *dev;
27 const int no_chip = 0x10;
28
29 ut_asserteq(-ENODEV, uclass_find_device_by_seq(UCLASS_I2C, busnum,
30 false, &bus));
31
32
33
34
35
36 ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
37 ut_assertok(dm_i2c_probe(bus, chip, 0, &dev));
38 ut_asserteq(-ENOENT, dm_i2c_probe(bus, no_chip, 0, &dev));
39 ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_I2C, 1, &bus));
40
41 return 0;
42}
43DM_TEST(dm_test_i2c_find, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
44
45static int dm_test_i2c_read_write(struct unit_test_state *uts)
46{
47 struct udevice *bus, *dev;
48 uint8_t buf[5];
49
50 ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
51 ut_assertok(i2c_get_chip(bus, chip, 1, &dev));
52 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
53 ut_assertok(memcmp(buf, "\0\0\0\0\0", sizeof(buf)));
54 ut_assertok(dm_i2c_write(dev, 2, (uint8_t *)"AB", 2));
55 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
56 ut_assertok(memcmp(buf, "\0\0AB\0", sizeof(buf)));
57
58 return 0;
59}
60DM_TEST(dm_test_i2c_read_write, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
61
62static int dm_test_i2c_speed(struct unit_test_state *uts)
63{
64 struct udevice *bus, *dev;
65 uint8_t buf[5];
66
67 ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
68
69
70 sandbox_i2c_set_test_mode(bus, true);
71 ut_assertok(i2c_get_chip(bus, chip, 1, &dev));
72 ut_assertok(dm_i2c_set_bus_speed(bus, 100000));
73 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
74 ut_assertok(dm_i2c_set_bus_speed(bus, 400000));
75 ut_asserteq(400000, dm_i2c_get_bus_speed(bus));
76 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
77 ut_asserteq(-EINVAL, dm_i2c_write(dev, 0, buf, 5));
78 sandbox_i2c_set_test_mode(bus, false);
79
80 return 0;
81}
82DM_TEST(dm_test_i2c_speed, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
83
84static int dm_test_i2c_offset_len(struct unit_test_state *uts)
85{
86 struct udevice *bus, *dev;
87 uint8_t buf[5];
88
89 ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
90 ut_assertok(i2c_get_chip(bus, chip, 1, &dev));
91 ut_assertok(i2c_set_chip_offset_len(dev, 1));
92 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
93
94
95 ut_asserteq(-EINVAL, i2c_set_chip_offset_len(dev, 5));
96
97 return 0;
98}
99DM_TEST(dm_test_i2c_offset_len, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
100
101static int dm_test_i2c_probe_empty(struct unit_test_state *uts)
102{
103 struct udevice *bus, *dev;
104
105 ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
106
107
108 sandbox_i2c_set_test_mode(bus, true);
109 ut_assertok(dm_i2c_probe(bus, SANDBOX_I2C_TEST_ADDR, 0, &dev));
110 sandbox_i2c_set_test_mode(bus, false);
111
112 return 0;
113}
114DM_TEST(dm_test_i2c_probe_empty, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
115
116static int dm_test_i2c_bytewise(struct unit_test_state *uts)
117{
118 struct udevice *bus, *dev;
119 struct udevice *eeprom;
120 uint8_t buf[5];
121
122 ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
123 ut_assertok(i2c_get_chip(bus, chip, 1, &dev));
124 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
125 ut_assertok(memcmp(buf, "\0\0\0\0\0", sizeof(buf)));
126
127
128 ut_assertok(uclass_first_device(UCLASS_I2C_EMUL, &eeprom));
129 ut_assertnonnull(eeprom);
130 sandbox_i2c_eeprom_set_test_mode(eeprom, SIE_TEST_MODE_SINGLE_BYTE);
131
132
133 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
134 ut_assertok(memcmp(buf, "\0\xff\xff\xff\xff", sizeof(buf)));
135
136
137 ut_assertok(i2c_set_chip_flags(dev, DM_I2C_CHIP_RD_ADDRESS));
138 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
139 ut_assertok(memcmp(buf, "\0\0\0\0\0", sizeof(buf)));
140
141
142 ut_assertok(i2c_set_chip_flags(dev, 0));
143 ut_assertok(dm_i2c_write(dev, 2, (uint8_t *)"AB", 2));
144 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
145 ut_assertok(memcmp(buf, "\0\xff\xff\xff\xff", sizeof(buf)));
146
147
148 ut_assertok(i2c_set_chip_flags(dev, DM_I2C_CHIP_RD_ADDRESS));
149 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
150 ut_assertok(memcmp(buf, "\0\0A\0\0\0", sizeof(buf)));
151
152
153 ut_assertok(i2c_set_chip_flags(dev, DM_I2C_CHIP_WR_ADDRESS));
154 ut_assertok(dm_i2c_write(dev, 2, (uint8_t *)"AB", 2));
155 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
156 ut_assertok(memcmp(buf, "\0\xff\xff\xff\xff", sizeof(buf)));
157
158 ut_assertok(i2c_set_chip_flags(dev, DM_I2C_CHIP_WR_ADDRESS |
159 DM_I2C_CHIP_RD_ADDRESS));
160 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
161 ut_assertok(memcmp(buf, "\0\0AB\0\0", sizeof(buf)));
162
163
164 sandbox_i2c_eeprom_set_test_mode(eeprom, SIE_TEST_MODE_NONE);
165 ut_assertok(i2c_set_chip_flags(dev, 0));
166
167 return 0;
168}
169DM_TEST(dm_test_i2c_bytewise, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
170
171static int dm_test_i2c_offset(struct unit_test_state *uts)
172{
173 struct udevice *eeprom;
174 struct udevice *dev;
175 uint8_t buf[5];
176
177 ut_assertok(i2c_get_chip_for_busnum(busnum, chip, 1, &dev));
178
179
180 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
181 ut_assertok(uclass_first_device(UCLASS_I2C_EMUL, &eeprom));
182
183
184 sandbox_i2c_eeprom_set_offset_len(eeprom, 0);
185 ut_assertok(i2c_set_chip_offset_len(dev, 0));
186 ut_assertok(dm_i2c_write(dev, 10 , (uint8_t *)"AB", 2));
187 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
188 ut_assertok(memcmp(buf, "AB\0\0\0\0", sizeof(buf)));
189
190
191 sandbox_i2c_eeprom_set_offset_len(eeprom, 1);
192 ut_assertok(i2c_set_chip_offset_len(dev, 1));
193 ut_assertok(dm_i2c_write(dev, 2, (uint8_t *)"AB", 2));
194 ut_assertok(dm_i2c_read(dev, 0, buf, 5));
195 ut_assertok(memcmp(buf, "ABAB\0", sizeof(buf)));
196
197
198 sandbox_i2c_eeprom_set_offset_len(eeprom, 2);
199 ut_assertok(i2c_set_chip_offset_len(dev, 2));
200 ut_assertok(dm_i2c_write(dev, 0x210, (uint8_t *)"AB", 2));
201 ut_assertok(dm_i2c_read(dev, 0x210, buf, 5));
202 ut_assertok(memcmp(buf, "AB\0\0\0", sizeof(buf)));
203
204
205 sandbox_i2c_eeprom_set_offset_len(eeprom, 2);
206 ut_assertok(i2c_set_chip_offset_len(dev, 2));
207 ut_assertok(dm_i2c_write(dev, 0x410, (uint8_t *)"AB", 2));
208 ut_assertok(dm_i2c_read(dev, 0x410, buf, 5));
209 ut_assertok(memcmp(buf, "AB\0\0\0", sizeof(buf)));
210
211
212 sandbox_i2c_eeprom_set_offset_len(eeprom, 2);
213 ut_assertok(i2c_set_chip_offset_len(dev, 2));
214 ut_assertok(dm_i2c_write(dev, 0x420, (uint8_t *)"AB", 2));
215 ut_assertok(dm_i2c_read(dev, 0x420, buf, 5));
216 ut_assertok(memcmp(buf, "AB\0\0\0", sizeof(buf)));
217
218
219 sandbox_i2c_eeprom_set_offset_len(eeprom, 1);
220
221 return 0;
222}
223DM_TEST(dm_test_i2c_offset, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
224