blob: 012bf4cab56ef43fbfa605c7bac9a32eebd4baa1 [file] [log] [blame]
Simon Glasse4fb8632016-03-13 08:22:36 -06001/*
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 <usb.h>
10#include <asm/state.h>
11#include <dm/test.h>
12#include <test/ut.h>
13
14DECLARE_GLOBAL_DATA_PTR;
15
16/* Test that block devices can be created */
17static int dm_test_blk_base(struct unit_test_state *uts)
18{
19 struct udevice *blk, *usb_blk, *dev;
20
21 /* Make sure there are no block devices */
22 ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_BLK, 0, &blk));
23
24 /* Create two, one the parent of the other */
25 ut_assertok(blk_create_device(gd->dm_root, "sandbox_host_blk", "test",
26 IF_TYPE_HOST, 1, 512, 1024, &blk));
27 ut_assertok(blk_create_device(blk, "usb_storage_blk", "test",
28 IF_TYPE_USB, 3, 512, 1024, &usb_blk));
29
30 /* Check we can find them */
31 ut_asserteq(-ENODEV, blk_get_device(IF_TYPE_HOST, 0, &dev));
32 ut_assertok(blk_get_device(IF_TYPE_HOST, 1, &dev));
33 ut_asserteq_ptr(blk, dev);
34
35 ut_asserteq(-ENODEV, blk_get_device(IF_TYPE_USB, 0, &dev));
36 ut_assertok(blk_get_device(IF_TYPE_USB, 3, &dev));
37 ut_asserteq_ptr(usb_blk, dev);
38
39 /* Check we can iterate */
40 ut_assertok(blk_first_device(IF_TYPE_HOST, &dev));
41 ut_asserteq_ptr(blk, dev);
42 ut_asserteq(-ENODEV, blk_next_device(&dev));
43
44 ut_assertok(blk_first_device(IF_TYPE_USB, &dev));
45 ut_asserteq_ptr(usb_blk, dev);
46 ut_asserteq(-ENODEV, blk_next_device(&dev));
47
48 return 0;
49}
50DM_TEST(dm_test_blk_base, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
51
52static int count_blk_devices(void)
53{
54 struct udevice *blk;
55 struct uclass *uc;
56 int count = 0;
57 int ret;
58
59 ret = uclass_get(UCLASS_BLK, &uc);
60 if (ret)
61 return ret;
62
63 uclass_foreach_dev(blk, uc)
64 count++;
65
66 return count;
67}
68
69/* Test that block devices work correctly with USB */
70static int dm_test_blk_usb(struct unit_test_state *uts)
71{
72 struct udevice *usb_dev, *dev;
73 struct blk_desc *dev_desc;
74
75 /* Get a flash device */
76 state_set_skip_delays(true);
77 ut_assertok(usb_init());
78 ut_assertok(uclass_get_device(UCLASS_MASS_STORAGE, 0, &usb_dev));
79 ut_assertok(blk_get_device_by_str("usb", "0", &dev_desc));
80
81 /* The parent should be a block device */
82 ut_assertok(blk_get_device(IF_TYPE_USB, 0, &dev));
83 ut_asserteq_ptr(usb_dev, dev_get_parent(dev));
84
85 /* Check we have one block device for each mass storage device */
Simon Glassafa2c312016-05-01 13:52:43 -060086 ut_asserteq(4, count_blk_devices());
Simon Glasse4fb8632016-03-13 08:22:36 -060087
88 /* Now go around again, making sure the old devices were unbound */
89 ut_assertok(usb_stop());
90 ut_assertok(usb_init());
Simon Glassafa2c312016-05-01 13:52:43 -060091 ut_asserteq(4, count_blk_devices());
Simon Glasse4fb8632016-03-13 08:22:36 -060092 ut_assertok(usb_stop());
93
94 return 0;
95}
96DM_TEST(dm_test_blk_usb, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);