2 * Copyright (C) 2015 Google, Inc
4 * SPDX-License-Identifier: GPL-2.0+
12 #include <asm/state.h>
14 #include <dm/device-internal.h>
16 #include <dm/uclass-internal.h>
19 /* Test that sandbox USB works correctly */
20 static int dm_test_usb_base(struct unit_test_state *uts)
24 ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_USB, 0, &bus));
25 ut_assertok(uclass_get_device(UCLASS_USB, 0, &bus));
26 ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_USB, 2, &bus));
30 DM_TEST(dm_test_usb_base, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
33 * Test that we can use the flash stick. This is more of a functional test. It
34 * covers scanning the bug, setting up a hub and a flash stick and reading
35 * data from the flash stick.
37 static int dm_test_usb_flash(struct unit_test_state *uts)
40 struct blk_desc *dev_desc;
43 state_set_skip_delays(true);
44 ut_assertok(usb_init());
45 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 0, &dev));
46 ut_assertok(blk_get_device_by_str("usb", "0", &dev_desc));
48 /* Read a few blocks and look for the string we expect */
49 ut_asserteq(512, dev_desc->blksz);
50 memset(cmp, '\0', sizeof(cmp));
51 ut_asserteq(2, blk_dread(dev_desc, 0, 2, cmp));
52 ut_assertok(strcmp(cmp, "this is a test"));
53 ut_assertok(usb_stop());
57 DM_TEST(dm_test_usb_flash, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
59 /* test that we can handle multiple storage devices */
60 static int dm_test_usb_multi(struct unit_test_state *uts)
64 state_set_skip_delays(true);
65 ut_assertok(usb_init());
66 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 0, &dev));
67 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 1, &dev));
68 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 2, &dev));
69 ut_assertok(usb_stop());
73 DM_TEST(dm_test_usb_multi, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
75 static int count_usb_devices(void)
82 ret = uclass_get(UCLASS_USB_HUB, &uc);
86 uclass_foreach_dev(hub, uc) {
90 for (device_find_first_child(hub, &dev);
92 device_find_next_child(&dev)) {
100 /* test that no USB devices are found after we stop the stack */
101 static int dm_test_usb_stop(struct unit_test_state *uts)
105 /* Scan and check that all devices are present */
106 state_set_skip_delays(true);
107 ut_assertok(usb_init());
108 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 0, &dev));
109 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 1, &dev));
110 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 2, &dev));
111 ut_asserteq(6, count_usb_devices());
112 ut_assertok(usb_stop());
113 ut_asserteq(0, count_usb_devices());
117 DM_TEST(dm_test_usb_stop, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
119 static int dm_test_usb_keyb(struct unit_test_state *uts)
123 state_set_skip_delays(true);
124 ut_assertok(usb_init());
126 /* Initially there should be no characters */
127 ut_asserteq(0, tstc());
129 ut_assertok(uclass_get_device_by_name(UCLASS_USB_EMUL, "keyb",
133 * Add a string to the USB keyboard buffer - it should appear in
136 ut_assertok(sandbox_usb_keyb_add_string(dev, "ab"));
137 ut_asserteq(1, tstc());
138 ut_asserteq('a', getc());
139 ut_asserteq(1, tstc());
140 ut_asserteq('b', getc());
141 ut_asserteq(0, tstc());
143 ut_assertok(usb_stop());
147 DM_TEST(dm_test_usb_keyb, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);