[u-boot.git] / test / dm / test-fdt.c

/*
 * Copyright (c) 2013 Google, Inc
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <dm.h>
#include <errno.h>
#include <fdtdec.h>
#include <malloc.h>
#include <asm/io.h>
#include <dm/test.h>
#include <dm/root.h>
#include <dm/device-internal.h>
#include <dm/uclass-internal.h>
#include <dm/util.h>
#include <test/ut.h>

DECLARE_GLOBAL_DATA_PTR;

static int testfdt_drv_ping(struct udevice *dev, int pingval, int *pingret)
{
	const struct dm_test_pdata *pdata = dev->platdata;
	struct dm_test_priv *priv = dev_get_priv(dev);

	*pingret = pingval + pdata->ping_add;
	priv->ping_total += *pingret;

	return 0;
}

static const struct test_ops test_ops = {
	.ping = testfdt_drv_ping,
};

static int testfdt_ofdata_to_platdata(struct udevice *dev)
{
	struct dm_test_pdata *pdata = dev_get_platdata(dev);

	pdata->ping_add = fdtdec_get_int(gd->fdt_blob, dev_of_offset(dev),
					"ping-add", -1);
	pdata->base = fdtdec_get_addr(gd->fdt_blob, dev_of_offset(dev),
				      "ping-expect");

	return 0;
}

static int testfdt_drv_probe(struct udevice *dev)
{
	struct dm_test_priv *priv = dev_get_priv(dev);

	priv->ping_total += DM_TEST_START_TOTAL;

	/*
	 * If this device is on a bus, the uclass_flag will be set before
	 * calling this function. This is used by
	 * dm_test_bus_child_pre_probe_uclass().
	 */
	priv->uclass_total += priv->uclass_flag;

	return 0;
}

static const struct udevice_id testfdt_ids[] = {
	{
		.compatible = "denx,u-boot-fdt-test",
		.data = DM_TEST_TYPE_FIRST },
	{
		.compatible = "google,another-fdt-test",
		.data = DM_TEST_TYPE_SECOND },
	{ }
};

U_BOOT_DRIVER(testfdt_drv) = {
	.name	= "testfdt_drv",
	.of_match	= testfdt_ids,
	.id	= UCLASS_TEST_FDT,
	.ofdata_to_platdata = testfdt_ofdata_to_platdata,
	.probe	= testfdt_drv_probe,
	.ops	= &test_ops,
	.priv_auto_alloc_size = sizeof(struct dm_test_priv),
	.platdata_auto_alloc_size = sizeof(struct dm_test_pdata),
};

/* From here is the testfdt uclass code */
int testfdt_ping(struct udevice *dev, int pingval, int *pingret)
{
	const struct test_ops *ops = device_get_ops(dev);

	if (!ops->ping)
		return -ENOSYS;

	return ops->ping(dev, pingval, pingret);
}

UCLASS_DRIVER(testfdt) = {
	.name		= "testfdt",
	.id		= UCLASS_TEST_FDT,
	.flags		= DM_UC_FLAG_SEQ_ALIAS,
};

struct dm_testprobe_pdata {
	int probe_err;
};

static int testprobe_drv_probe(struct udevice *dev)
{
	struct dm_testprobe_pdata *pdata = dev_get_platdata(dev);

	return pdata->probe_err;
}

static const struct udevice_id testprobe_ids[] = {
	{ .compatible = "denx,u-boot-probe-test" },
	{ }
};

U_BOOT_DRIVER(testprobe_drv) = {
	.name	= "testprobe_drv",
	.of_match	= testprobe_ids,
	.id	= UCLASS_TEST_PROBE,
	.probe	= testprobe_drv_probe,
	.platdata_auto_alloc_size	= sizeof(struct dm_testprobe_pdata),
};

UCLASS_DRIVER(testprobe) = {
	.name		= "testprobe",
	.id		= UCLASS_TEST_PROBE,
	.flags		= DM_UC_FLAG_SEQ_ALIAS,
};

int dm_check_devices(struct unit_test_state *uts, int num_devices)
{
	struct udevice *dev;
	int ret;
	int i;

	/*
	 * Now check that the ping adds are what we expect. This is using the
	 * ping-add property in each node.
	 */
	for (i = 0; i < num_devices; i++) {
		uint32_t base;

		ret = uclass_get_device(UCLASS_TEST_FDT, i, &dev);
		ut_assert(!ret);

		/*
		 * Get the 'ping-expect' property, which tells us what the
		 * ping add should be. We don't use the platdata because we
		 * want to test the code that sets that up
		 * (testfdt_drv_probe()).
		 */
		base = fdtdec_get_addr(gd->fdt_blob, dev_of_offset(dev),
				       "ping-expect");
		debug("dev=%d, base=%d: %s\n", i, base,
		      fdt_get_name(gd->fdt_blob, dev_of_offset(dev), NULL));

		ut_assert(!dm_check_operations(uts, dev, base,
					       dev_get_priv(dev)));
	}

	return 0;
}

/* Test that FDT-based binding works correctly */
static int dm_test_fdt(struct unit_test_state *uts)
{
	const int num_devices = 6;
	struct udevice *dev;
	struct uclass *uc;
	int ret;
	int i;

	ret = dm_scan_fdt(gd->fdt_blob, false);
	ut_assert(!ret);

	ret = uclass_get(UCLASS_TEST_FDT, &uc);
	ut_assert(!ret);

	/* These are num_devices compatible root-level device tree nodes */
	ut_asserteq(num_devices, list_count_items(&uc->dev_head));

	/* Each should have platform data but no private data */
	for (i = 0; i < num_devices; i++) {
		ret = uclass_find_device(UCLASS_TEST_FDT, i, &dev);
		ut_assert(!ret);
		ut_assert(!dev_get_priv(dev));
		ut_assert(dev->platdata);
	}

	ut_assertok(dm_check_devices(uts, num_devices));

	return 0;
}
DM_TEST(dm_test_fdt, 0);

static int dm_test_fdt_pre_reloc(struct unit_test_state *uts)
{
	struct uclass *uc;
	int ret;

	ret = dm_scan_fdt(gd->fdt_blob, true);
	ut_assert(!ret);

	ret = uclass_get(UCLASS_TEST_FDT, &uc);
	ut_assert(!ret);

	/* These is only one pre-reloc device */
	ut_asserteq(1, list_count_items(&uc->dev_head));

	return 0;
}
DM_TEST(dm_test_fdt_pre_reloc, 0);

/* Test that sequence numbers are allocated properly */
static int dm_test_fdt_uclass_seq(struct unit_test_state *uts)
{
	struct udevice *dev;

	/* A few basic santiy tests */
	ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_FDT, 3, true, &dev));
	ut_asserteq_str("b-test", dev->name);

	ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_FDT, 8, true, &dev));
	ut_asserteq_str("a-test", dev->name);

	ut_asserteq(-ENODEV, uclass_find_device_by_seq(UCLASS_TEST_FDT, 5,
						       true, &dev));
	ut_asserteq_ptr(NULL, dev);

	/* Test aliases */
	ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT, 6, &dev));
	ut_asserteq_str("e-test", dev->name);

	ut_asserteq(-ENODEV, uclass_find_device_by_seq(UCLASS_TEST_FDT, 7,
						       true, &dev));

	/*
	 * Note that c-test nodes are not probed since it is not a top-level
	 * node
	 */
	ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT, 3, &dev));
	ut_asserteq_str("b-test", dev->name);

	/*
	 * d-test wants sequence number 3 also, but it can't have it because
	 * b-test gets it first.
	 */
	ut_assertok(uclass_get_device(UCLASS_TEST_FDT, 2, &dev));
	ut_asserteq_str("d-test", dev->name);

	/* d-test actually gets 0 */
	ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT, 0, &dev));
	ut_asserteq_str("d-test", dev->name);

	/* initially no one wants seq 1 */
	ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_TEST_FDT, 1,
						      &dev));
	ut_assertok(uclass_get_device(UCLASS_TEST_FDT, 0, &dev));
	ut_assertok(uclass_get_device(UCLASS_TEST_FDT, 4, &dev));

	/* But now that it is probed, we can find it */
	ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT, 1, &dev));
	ut_asserteq_str("f-test", dev->name);

	return 0;
}
DM_TEST(dm_test_fdt_uclass_seq, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);

/* Test that we can find a device by device tree offset */
static int dm_test_fdt_offset(struct unit_test_state *uts)
{
	const void *blob = gd->fdt_blob;
	struct udevice *dev;
	int node;

	node = fdt_path_offset(blob, "/e-test");
	ut_assert(node > 0);
	ut_assertok(uclass_get_device_by_of_offset(UCLASS_TEST_FDT, node,
						   &dev));
	ut_asserteq_str("e-test", dev->name);

	/* This node should not be bound */
	node = fdt_path_offset(blob, "/junk");
	ut_assert(node > 0);
	ut_asserteq(-ENODEV, uclass_get_device_by_of_offset(UCLASS_TEST_FDT,
							    node, &dev));

	/* This is not a top level node so should not be probed */
	node = fdt_path_offset(blob, "/some-bus/c-test@5");
	ut_assert(node > 0);
	ut_asserteq(-ENODEV, uclass_get_device_by_of_offset(UCLASS_TEST_FDT,
							    node, &dev));

	return 0;
}
DM_TEST(dm_test_fdt_offset,
	DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT | DM_TESTF_FLAT_TREE);

/**
 * Test various error conditions with uclass_first_device() and
 * uclass_next_device()
 */
static int dm_test_first_next_device(struct unit_test_state *uts)
{
	struct dm_testprobe_pdata *pdata;
	struct udevice *dev, *parent = NULL;
	int count;
	int ret;

	/* There should be 4 devices */
	for (ret = uclass_first_device(UCLASS_TEST_PROBE, &dev), count = 0;
	     dev;
	     ret = uclass_next_device(&dev)) {
		count++;
		parent = dev_get_parent(dev);
		}
	ut_assertok(ret);
	ut_asserteq(4, count);

	/* Remove them and try again, with an error on the second one */
	ut_assertok(uclass_get_device(UCLASS_TEST_PROBE, 1, &dev));
	pdata = dev_get_platdata(dev);
	pdata->probe_err = -ENOMEM;
	device_remove(parent, DM_REMOVE_NORMAL);
	ut_assertok(uclass_first_device(UCLASS_TEST_PROBE, &dev));
	ut_asserteq(-ENOMEM, uclass_next_device(&dev));
	ut_asserteq_ptr(dev, NULL);

	/* Now an error on the first one */
	ut_assertok(uclass_get_device(UCLASS_TEST_PROBE, 0, &dev));
	pdata = dev_get_platdata(dev);
	pdata->probe_err = -ENOENT;
	device_remove(parent, DM_REMOVE_NORMAL);
	ut_asserteq(-ENOENT, uclass_first_device(UCLASS_TEST_PROBE, &dev));

	return 0;
}
DM_TEST(dm_test_first_next_device, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
Commit	Line	Data
2e7d35d2 SG	1	/*
	2	* Copyright (c) 2013 Google, Inc
	3	*
	4	* SPDX-License-Identifier: GPL-2.0+
	5	*/
	6
	7	#include <common.h>
	8	#include <dm.h>
	9	#include <errno.h>
	10	#include <fdtdec.h>
	11	#include <malloc.h>
	12	#include <asm/io.h>
	13	#include <dm/test.h>
	14	#include <dm/root.h>
98561572	15	#include <dm/device-internal.h>
2e7d35d2 SG	16	#include <dm/uclass-internal.h>
2e7d35d2 SG	17	#include <dm/util.h>
e721b882	18	#include <test/ut.h>
2e7d35d2 SG	19
	20	DECLARE_GLOBAL_DATA_PTR;
	21
54c5d08a	22	static int testfdt_drv_ping(struct udevice dev, int pingval, int pingret)
2e7d35d2 SG	23	{
	24	const struct dm_test_pdata *pdata = dev->platdata;
	25	struct dm_test_priv *priv = dev_get_priv(dev);
	26
	27	*pingret = pingval + pdata->ping_add;
	28	priv->ping_total += *pingret;
	29
	30	return 0;
	31	}
	32
	33	static const struct test_ops test_ops = {
	34	.ping = testfdt_drv_ping,
	35	};
	36
54c5d08a	37	static int testfdt_ofdata_to_platdata(struct udevice *dev)
2e7d35d2 SG	38	{
	39	struct dm_test_pdata *pdata = dev_get_platdata(dev);
	40
e160f7d4	41	pdata->ping_add = fdtdec_get_int(gd->fdt_blob, dev_of_offset(dev),
2e7d35d2	42	"ping-add", -1);
e160f7d4	43	pdata->base = fdtdec_get_addr(gd->fdt_blob, dev_of_offset(dev),
eb9ef5fe	44	"ping-expect");
2e7d35d2 SG	45
	46	return 0;
	47	}
	48
54c5d08a	49	static int testfdt_drv_probe(struct udevice *dev)
2e7d35d2 SG	50	{
	51	struct dm_test_priv *priv = dev_get_priv(dev);
	52
	53	priv->ping_total += DM_TEST_START_TOTAL;
	54
83c7e434 SG	55	/*
	56	* If this device is on a bus, the uclass_flag will be set before
	57	* calling this function. This is used by
	58	* dm_test_bus_child_pre_probe_uclass().
	59	*/
	60	priv->uclass_total += priv->uclass_flag;
	61
2e7d35d2 SG	62	return 0;
	63	}
	64
ae7f4513	65	static const struct udevice_id testfdt_ids[] = {
2e7d35d2 SG	66	{
	67	.compatible = "denx,u-boot-fdt-test",
	68	.data = DM_TEST_TYPE_FIRST },
	69	{
	70	.compatible = "google,another-fdt-test",
	71	.data = DM_TEST_TYPE_SECOND },
	72	{ }
	73	};
	74
	75	U_BOOT_DRIVER(testfdt_drv) = {
	76	.name = "testfdt_drv",
	77	.of_match = testfdt_ids,
	78	.id = UCLASS_TEST_FDT,
	79	.ofdata_to_platdata = testfdt_ofdata_to_platdata,
	80	.probe = testfdt_drv_probe,
	81	.ops = &test_ops,
	82	.priv_auto_alloc_size = sizeof(struct dm_test_priv),
	83	.platdata_auto_alloc_size = sizeof(struct dm_test_pdata),
	84	};
	85
	86	/* From here is the testfdt uclass code */
54c5d08a	87	int testfdt_ping(struct udevice dev, int pingval, int pingret)
2e7d35d2 SG	88	{
	89	const struct test_ops *ops = device_get_ops(dev);
	90
	91	if (!ops->ping)
	92	return -ENOSYS;
	93
	94	return ops->ping(dev, pingval, pingret);
	95	}
	96
	97	UCLASS_DRIVER(testfdt) = {
	98	.name = "testfdt",
	99	.id = UCLASS_TEST_FDT,
9cc36a2b	100	.flags = DM_UC_FLAG_SEQ_ALIAS,
2e7d35d2 SG	101	};
2e7d35d2 SG	102
98561572 SG	103	struct dm_testprobe_pdata {
	104	int probe_err;
	105	};
	106
	107	static int testprobe_drv_probe(struct udevice *dev)
	108	{
	109	struct dm_testprobe_pdata *pdata = dev_get_platdata(dev);
	110
	111	return pdata->probe_err;
	112	}
	113
	114	static const struct udevice_id testprobe_ids[] = {
	115	{ .compatible = "denx,u-boot-probe-test" },
	116	{ }
	117	};
	118
	119	U_BOOT_DRIVER(testprobe_drv) = {
	120	.name = "testprobe_drv",
	121	.of_match = testprobe_ids,
	122	.id = UCLASS_TEST_PROBE,
	123	.probe = testprobe_drv_probe,
	124	.platdata_auto_alloc_size = sizeof(struct dm_testprobe_pdata),
	125	};
	126
	127	UCLASS_DRIVER(testprobe) = {
	128	.name = "testprobe",
	129	.id = UCLASS_TEST_PROBE,
	130	.flags = DM_UC_FLAG_SEQ_ALIAS,
	131	};
	132
e721b882	133	int dm_check_devices(struct unit_test_state *uts, int num_devices)
2e7d35d2	134	{
54c5d08a	135	struct udevice *dev;
2e7d35d2 SG	136	int ret;
	137	int i;
	138
2e7d35d2 SG	139	/*
	140	* Now check that the ping adds are what we expect. This is using the
	141	* ping-add property in each node.
	142	*/
1ca7e206	143	for (i = 0; i < num_devices; i++) {
2e7d35d2 SG	144	uint32_t base;
	145
	146	ret = uclass_get_device(UCLASS_TEST_FDT, i, &dev);
	147	ut_assert(!ret);
	148
	149	/*
eb9ef5fe SG	150	* Get the 'ping-expect' property, which tells us what the
	151	* ping add should be. We don't use the platdata because we
	152	* want to test the code that sets that up
	153	* (testfdt_drv_probe()).
2e7d35d2	154	*/
e160f7d4	155	base = fdtdec_get_addr(gd->fdt_blob, dev_of_offset(dev),
eb9ef5fe	156	"ping-expect");
2e7d35d2	157	debug("dev=%d, base=%d: %s\n", i, base,
e160f7d4	158	fdt_get_name(gd->fdt_blob, dev_of_offset(dev), NULL));
2e7d35d2	159
e721b882	160	ut_assert(!dm_check_operations(uts, dev, base,
2e7d35d2 SG	161	dev_get_priv(dev)));
	162	}
	163
	164	return 0;
	165	}
1ca7e206 SG	166
1ca7e206 SG	167	/* Test that FDT-based binding works correctly */
e721b882	168	static int dm_test_fdt(struct unit_test_state *uts)
1ca7e206	169	{
9cc36a2b	170	const int num_devices = 6;
1ca7e206 SG	171	struct udevice *dev;
	172	struct uclass *uc;
	173	int ret;
	174	int i;
	175
	176	ret = dm_scan_fdt(gd->fdt_blob, false);
	177	ut_assert(!ret);
	178
	179	ret = uclass_get(UCLASS_TEST_FDT, &uc);
	180	ut_assert(!ret);
	181
	182	/* These are num_devices compatible root-level device tree nodes */
	183	ut_asserteq(num_devices, list_count_items(&uc->dev_head));
	184
f8a85449	185	/* Each should have platform data but no private data */
1ca7e206 SG	186	for (i = 0; i < num_devices; i++) {
	187	ret = uclass_find_device(UCLASS_TEST_FDT, i, &dev);
	188	ut_assert(!ret);
	189	ut_assert(!dev_get_priv(dev));
f8a85449	190	ut_assert(dev->platdata);
1ca7e206 SG	191	}
1ca7e206 SG	192
e721b882	193	ut_assertok(dm_check_devices(uts, num_devices));
1ca7e206 SG	194
	195	return 0;
	196	}
2e7d35d2	197	DM_TEST(dm_test_fdt, 0);
00606d7e	198
e721b882	199	static int dm_test_fdt_pre_reloc(struct unit_test_state *uts)
00606d7e SG	200	{
	201	struct uclass *uc;
	202	int ret;
	203
	204	ret = dm_scan_fdt(gd->fdt_blob, true);
	205	ut_assert(!ret);
	206
	207	ret = uclass_get(UCLASS_TEST_FDT, &uc);
	208	ut_assert(!ret);
	209
	210	/* These is only one pre-reloc device */
	211	ut_asserteq(1, list_count_items(&uc->dev_head));
	212
	213	return 0;
	214	}
	215	DM_TEST(dm_test_fdt_pre_reloc, 0);
5a66a8ff SG	216
5a66a8ff SG	217	/* Test that sequence numbers are allocated properly */
e721b882	218	static int dm_test_fdt_uclass_seq(struct unit_test_state *uts)
5a66a8ff SG	219	{
	220	struct udevice *dev;
	221
	222	/* A few basic santiy tests */
	223	ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_FDT, 3, true, &dev));
	224	ut_asserteq_str("b-test", dev->name);
	225
9cc36a2b	226	ut_assertok(uclass_find_device_by_seq(UCLASS_TEST_FDT, 8, true, &dev));
5a66a8ff SG	227	ut_asserteq_str("a-test", dev->name);
	228
	229	ut_asserteq(-ENODEV, uclass_find_device_by_seq(UCLASS_TEST_FDT, 5,
	230	true, &dev));
	231	ut_asserteq_ptr(NULL, dev);
	232
	233	/* Test aliases */
	234	ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT, 6, &dev));
	235	ut_asserteq_str("e-test", dev->name);
	236
	237	ut_asserteq(-ENODEV, uclass_find_device_by_seq(UCLASS_TEST_FDT, 7,
	238	true, &dev));
	239
1ca7e206 SG	240	/*
	241	* Note that c-test nodes are not probed since it is not a top-level
	242	* node
	243	*/
5a66a8ff SG	244	ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT, 3, &dev));
	245	ut_asserteq_str("b-test", dev->name);
	246
	247	/*
	248	* d-test wants sequence number 3 also, but it can't have it because
	249	* b-test gets it first.
	250	*/
	251	ut_assertok(uclass_get_device(UCLASS_TEST_FDT, 2, &dev));
	252	ut_asserteq_str("d-test", dev->name);
	253
	254	/* d-test actually gets 0 */
	255	ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT, 0, &dev));
	256	ut_asserteq_str("d-test", dev->name);
	257
	258	/* initially no one wants seq 1 */
	259	ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_TEST_FDT, 1,
	260	&dev));
	261	ut_assertok(uclass_get_device(UCLASS_TEST_FDT, 0, &dev));
9cc36a2b	262	ut_assertok(uclass_get_device(UCLASS_TEST_FDT, 4, &dev));
5a66a8ff SG	263
	264	/* But now that it is probed, we can find it */
	265	ut_assertok(uclass_get_device_by_seq(UCLASS_TEST_FDT, 1, &dev));
9cc36a2b	266	ut_asserteq_str("f-test", dev->name);
5a66a8ff SG	267
	268	return 0;
	269	}
	270	DM_TEST(dm_test_fdt_uclass_seq, DM_TESTF_SCAN_PDATA \| DM_TESTF_SCAN_FDT);
f4cdead2 SG	271
f4cdead2 SG	272	/* Test that we can find a device by device tree offset */
e721b882	273	static int dm_test_fdt_offset(struct unit_test_state *uts)
f4cdead2 SG	274	{
	275	const void *blob = gd->fdt_blob;
	276	struct udevice *dev;
	277	int node;
	278
	279	node = fdt_path_offset(blob, "/e-test");
	280	ut_assert(node > 0);
	281	ut_assertok(uclass_get_device_by_of_offset(UCLASS_TEST_FDT, node,
	282	&dev));
	283	ut_asserteq_str("e-test", dev->name);
	284
	285	/* This node should not be bound */
	286	node = fdt_path_offset(blob, "/junk");
	287	ut_assert(node > 0);
	288	ut_asserteq(-ENODEV, uclass_get_device_by_of_offset(UCLASS_TEST_FDT,
	289	node, &dev));
	290
	291	/* This is not a top level node so should not be probed */
1ca7e206	292	node = fdt_path_offset(blob, "/some-bus/c-test@5");
f4cdead2 SG	293	ut_assert(node > 0);
	294	ut_asserteq(-ENODEV, uclass_get_device_by_of_offset(UCLASS_TEST_FDT,
	295	node, &dev));
	296
	297	return 0;
	298	}
86b54ece SG	299	DM_TEST(dm_test_fdt_offset,
86b54ece SG	300	DM_TESTF_SCAN_PDATA \| DM_TESTF_SCAN_FDT \| DM_TESTF_FLAT_TREE);
98561572 SG	301
	302	/**
	303	* Test various error conditions with uclass_first_device() and
	304	* uclass_next_device()
	305	*/
	306	static int dm_test_first_next_device(struct unit_test_state *uts)
	307	{
	308	struct dm_testprobe_pdata *pdata;
	309	struct udevice dev, parent = NULL;
	310	int count;
	311	int ret;
	312
	313	/* There should be 4 devices */
	314	for (ret = uclass_first_device(UCLASS_TEST_PROBE, &dev), count = 0;
	315	dev;
	316	ret = uclass_next_device(&dev)) {
	317	count++;
	318	parent = dev_get_parent(dev);
	319	}
	320	ut_assertok(ret);
	321	ut_asserteq(4, count);
	322
	323	/* Remove them and try again, with an error on the second one */
	324	ut_assertok(uclass_get_device(UCLASS_TEST_PROBE, 1, &dev));
	325	pdata = dev_get_platdata(dev);
	326	pdata->probe_err = -ENOMEM;
	327	device_remove(parent, DM_REMOVE_NORMAL);
	328	ut_assertok(uclass_first_device(UCLASS_TEST_PROBE, &dev));
	329	ut_asserteq(-ENOMEM, uclass_next_device(&dev));
	330	ut_asserteq_ptr(dev, NULL);
	331
	332	/* Now an error on the first one */
	333	ut_assertok(uclass_get_device(UCLASS_TEST_PROBE, 0, &dev));
	334	pdata = dev_get_platdata(dev);
	335	pdata->probe_err = -ENOENT;
	336	device_remove(parent, DM_REMOVE_NORMAL);
	337	ut_asserteq(-ENOENT, uclass_first_device(UCLASS_TEST_PROBE, &dev));
	338
	339	return 0;
	340	}
	341	DM_TEST(dm_test_first_next_device, DM_TESTF_SCAN_PDATA \| DM_TESTF_SCAN_FDT);