xref: /openbmc/u-boot/test/dm/test-uclass.c (revision 83d290c56fab2d38cd1ab4c4cc7099559c1d5046)
1*83d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
22e7d35d2SSimon Glass /*
32e7d35d2SSimon Glass  * Copyright (c) 2013 Google, Inc
42e7d35d2SSimon Glass  *
52e7d35d2SSimon Glass  * (C) Copyright 2012
62e7d35d2SSimon Glass  * Pavel Herrmann <morpheus.ibis@gmail.com>
72e7d35d2SSimon Glass  */
82e7d35d2SSimon Glass 
92e7d35d2SSimon Glass #include <common.h>
102e7d35d2SSimon Glass #include <malloc.h>
112e7d35d2SSimon Glass #include <dm.h>
122e7d35d2SSimon Glass #include <errno.h>
132e7d35d2SSimon Glass #include <asm/io.h>
14e721b882SJoe Hershberger #include <dm/test.h>
152e7d35d2SSimon Glass #include <linux/list.h>
16e721b882SJoe Hershberger #include <test/ut.h>
172e7d35d2SSimon Glass 
18e721b882SJoe Hershberger static struct unit_test_state *uts = &global_dm_test_state;
192e7d35d2SSimon Glass 
test_ping(struct udevice * dev,int pingval,int * pingret)2054c5d08aSHeiko Schocher int test_ping(struct udevice *dev, int pingval, int *pingret)
212e7d35d2SSimon Glass {
222e7d35d2SSimon Glass 	const struct test_ops *ops = device_get_ops(dev);
232e7d35d2SSimon Glass 
242e7d35d2SSimon Glass 	if (!ops->ping)
252e7d35d2SSimon Glass 		return -ENOSYS;
262e7d35d2SSimon Glass 
272e7d35d2SSimon Glass 	return ops->ping(dev, pingval, pingret);
282e7d35d2SSimon Glass }
292e7d35d2SSimon Glass 
test_post_bind(struct udevice * dev)3054c5d08aSHeiko Schocher static int test_post_bind(struct udevice *dev)
312e7d35d2SSimon Glass {
32754e71e8SPrzemyslaw Marczak 	struct dm_test_perdev_uc_pdata *uc_pdata;
33754e71e8SPrzemyslaw Marczak 
342e7d35d2SSimon Glass 	dm_testdrv_op_count[DM_TEST_OP_POST_BIND]++;
35206d4d2bSSimon Glass 	ut_assert(!device_active(dev));
362e7d35d2SSimon Glass 
37754e71e8SPrzemyslaw Marczak 	uc_pdata = dev_get_uclass_platdata(dev);
38754e71e8SPrzemyslaw Marczak 	ut_assert(uc_pdata);
39754e71e8SPrzemyslaw Marczak 
40754e71e8SPrzemyslaw Marczak 	uc_pdata->intval1 = TEST_UC_PDATA_INTVAL1;
41754e71e8SPrzemyslaw Marczak 	uc_pdata->intval2 = TEST_UC_PDATA_INTVAL2;
42754e71e8SPrzemyslaw Marczak 	uc_pdata->intval3 = TEST_UC_PDATA_INTVAL3;
43754e71e8SPrzemyslaw Marczak 
442e7d35d2SSimon Glass 	return 0;
452e7d35d2SSimon Glass }
462e7d35d2SSimon Glass 
test_pre_unbind(struct udevice * dev)4754c5d08aSHeiko Schocher static int test_pre_unbind(struct udevice *dev)
482e7d35d2SSimon Glass {
492e7d35d2SSimon Glass 	dm_testdrv_op_count[DM_TEST_OP_PRE_UNBIND]++;
502e7d35d2SSimon Glass 
512e7d35d2SSimon Glass 	return 0;
522e7d35d2SSimon Glass }
532e7d35d2SSimon Glass 
test_pre_probe(struct udevice * dev)5402c07b37SSimon Glass static int test_pre_probe(struct udevice *dev)
5502c07b37SSimon Glass {
5602c07b37SSimon Glass 	struct dm_test_uclass_perdev_priv *priv = dev_get_uclass_priv(dev);
5702c07b37SSimon Glass 
5802c07b37SSimon Glass 	dm_testdrv_op_count[DM_TEST_OP_PRE_PROBE]++;
5902c07b37SSimon Glass 	ut_assert(priv);
60206d4d2bSSimon Glass 	ut_assert(device_active(dev));
6102c07b37SSimon Glass 
6202c07b37SSimon Glass 	return 0;
6302c07b37SSimon Glass }
6402c07b37SSimon Glass 
test_post_probe(struct udevice * dev)6554c5d08aSHeiko Schocher static int test_post_probe(struct udevice *dev)
662e7d35d2SSimon Glass {
6754c5d08aSHeiko Schocher 	struct udevice *prev = list_entry(dev->uclass_node.prev,
6854c5d08aSHeiko Schocher 					    struct udevice, uclass_node);
6954c5d08aSHeiko Schocher 
70e564f054SSimon Glass 	struct dm_test_uclass_perdev_priv *priv = dev_get_uclass_priv(dev);
712e7d35d2SSimon Glass 	struct uclass *uc = dev->uclass;
72e721b882SJoe Hershberger 	struct dm_test_state *dms = uts->priv;
732e7d35d2SSimon Glass 
742e7d35d2SSimon Glass 	dm_testdrv_op_count[DM_TEST_OP_POST_PROBE]++;
752e7d35d2SSimon Glass 	ut_assert(priv);
762e7d35d2SSimon Glass 	ut_assert(device_active(dev));
772e7d35d2SSimon Glass 	priv->base_add = 0;
782e7d35d2SSimon Glass 	if (dms->skip_post_probe)
792e7d35d2SSimon Glass 		return 0;
802e7d35d2SSimon Glass 	if (&prev->uclass_node != &uc->dev_head) {
812e7d35d2SSimon Glass 		struct dm_test_uclass_perdev_priv *prev_uc_priv
82e564f054SSimon Glass 				= dev_get_uclass_priv(prev);
832e7d35d2SSimon Glass 		struct dm_test_pdata *pdata = prev->platdata;
842e7d35d2SSimon Glass 
852e7d35d2SSimon Glass 		ut_assert(pdata);
862e7d35d2SSimon Glass 		ut_assert(prev_uc_priv);
872e7d35d2SSimon Glass 		priv->base_add = prev_uc_priv->base_add + pdata->ping_add;
882e7d35d2SSimon Glass 	}
892e7d35d2SSimon Glass 
902e7d35d2SSimon Glass 	return 0;
912e7d35d2SSimon Glass }
922e7d35d2SSimon Glass 
test_pre_remove(struct udevice * dev)9354c5d08aSHeiko Schocher static int test_pre_remove(struct udevice *dev)
942e7d35d2SSimon Glass {
952e7d35d2SSimon Glass 	dm_testdrv_op_count[DM_TEST_OP_PRE_REMOVE]++;
962e7d35d2SSimon Glass 
972e7d35d2SSimon Glass 	return 0;
982e7d35d2SSimon Glass }
992e7d35d2SSimon Glass 
test_init(struct uclass * uc)1002e7d35d2SSimon Glass static int test_init(struct uclass *uc)
1012e7d35d2SSimon Glass {
1022e7d35d2SSimon Glass 	dm_testdrv_op_count[DM_TEST_OP_INIT]++;
1032e7d35d2SSimon Glass 	ut_assert(uc->priv);
1042e7d35d2SSimon Glass 
1052e7d35d2SSimon Glass 	return 0;
1062e7d35d2SSimon Glass }
1072e7d35d2SSimon Glass 
test_destroy(struct uclass * uc)1082e7d35d2SSimon Glass static int test_destroy(struct uclass *uc)
1092e7d35d2SSimon Glass {
1102e7d35d2SSimon Glass 	dm_testdrv_op_count[DM_TEST_OP_DESTROY]++;
1112e7d35d2SSimon Glass 
1122e7d35d2SSimon Glass 	return 0;
1132e7d35d2SSimon Glass }
1142e7d35d2SSimon Glass 
1152e7d35d2SSimon Glass UCLASS_DRIVER(test) = {
1162e7d35d2SSimon Glass 	.name		= "test",
1172e7d35d2SSimon Glass 	.id		= UCLASS_TEST,
1182e7d35d2SSimon Glass 	.post_bind	= test_post_bind,
1192e7d35d2SSimon Glass 	.pre_unbind	= test_pre_unbind,
12002c07b37SSimon Glass 	.pre_probe	= test_pre_probe,
1212e7d35d2SSimon Glass 	.post_probe	= test_post_probe,
1222e7d35d2SSimon Glass 	.pre_remove	= test_pre_remove,
1232e7d35d2SSimon Glass 	.init		= test_init,
1242e7d35d2SSimon Glass 	.destroy	= test_destroy,
1252e7d35d2SSimon Glass 	.priv_auto_alloc_size	= sizeof(struct dm_test_uclass_priv),
1262e7d35d2SSimon Glass 	.per_device_auto_alloc_size = sizeof(struct dm_test_uclass_perdev_priv),
127754e71e8SPrzemyslaw Marczak 	.per_device_platdata_auto_alloc_size =
128754e71e8SPrzemyslaw Marczak 					sizeof(struct dm_test_perdev_uc_pdata),
1292e7d35d2SSimon Glass };
130