dm: blk: Allow finding block devices without probing

[oweals/u-boot.git] / test / dm / blk.c

diff --git a/test/dm/blk.c b/test/dm/blk.c
index 012bf4cab56ef43fbfa605c7bac9a32eebd4baa1..3e34336fae1d2db895fda0ccd4e5b96204b55c44 100644 (file)
--- a/test/dm/blk.c
+++ b/test/dm/blk.c
@@ -94,3 +94,24 @@ static int dm_test_blk_usb(struct unit_test_state *uts)
        return 0;
 }
 DM_TEST(dm_test_blk_usb, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
+
+/* Test that we can find block devices without probing them */
+static int dm_test_blk_find(struct unit_test_state *uts)
+{
+       struct udevice *blk, *dev;
+
+       ut_assertok(blk_create_device(gd->dm_root, "sandbox_host_blk", "test",
+                                     IF_TYPE_HOST, 1, 512, 1024, &blk));
+       ut_asserteq(-ENODEV, blk_find_device(IF_TYPE_HOST, 0, &dev));
+       ut_assertok(blk_find_device(IF_TYPE_HOST, 1, &dev));
+       ut_asserteq_ptr(blk, dev);
+       ut_asserteq(false, device_active(dev));
+
+       /* Now activate it */
+       ut_assertok(blk_get_device(IF_TYPE_HOST, 1, &dev));
+       ut_asserteq_ptr(blk, dev);
+       ut_asserteq(true, device_active(dev));
+
+       return 0;
+}
+DM_TEST(dm_test_blk_find, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);