xref: /openbmc/u-boot/test/dm/test-uclass.c (revision c54473cb25968be53e6b47527d5970b03a3cf41d)
1 /*
2  * Copyright (c) 2013 Google, Inc
3  *
4  * (C) Copyright 2012
5  * Pavel Herrmann <morpheus.ibis@gmail.com>
6  *
7  * SPDX-License-Identifier:	GPL-2.0+
8  */
9 
10 #include <common.h>
11 #include <malloc.h>
12 #include <dm.h>
13 #include <errno.h>
14 #include <dm/test.h>
15 #include <dm/ut.h>
16 #include <asm/io.h>
17 #include <linux/list.h>
18 
19 static struct dm_test_state *dms = &global_test_state;
20 
21 int test_ping(struct udevice *dev, int pingval, int *pingret)
22 {
23 	const struct test_ops *ops = device_get_ops(dev);
24 
25 	if (!ops->ping)
26 		return -ENOSYS;
27 
28 	return ops->ping(dev, pingval, pingret);
29 }
30 
31 static int test_post_bind(struct udevice *dev)
32 {
33 	dm_testdrv_op_count[DM_TEST_OP_POST_BIND]++;
34 
35 	return 0;
36 }
37 
38 static int test_pre_unbind(struct udevice *dev)
39 {
40 	dm_testdrv_op_count[DM_TEST_OP_PRE_UNBIND]++;
41 
42 	return 0;
43 }
44 
45 static int test_pre_probe(struct udevice *dev)
46 {
47 	struct dm_test_uclass_perdev_priv *priv = dev_get_uclass_priv(dev);
48 
49 	dm_testdrv_op_count[DM_TEST_OP_PRE_PROBE]++;
50 	ut_assert(priv);
51 	ut_assert(!device_active(dev));
52 
53 	return 0;
54 }
55 
56 static int test_post_probe(struct udevice *dev)
57 {
58 	struct udevice *prev = list_entry(dev->uclass_node.prev,
59 					    struct udevice, uclass_node);
60 
61 	struct dm_test_uclass_perdev_priv *priv = dev_get_uclass_priv(dev);
62 	struct uclass *uc = dev->uclass;
63 
64 	dm_testdrv_op_count[DM_TEST_OP_POST_PROBE]++;
65 	ut_assert(priv);
66 	ut_assert(device_active(dev));
67 	priv->base_add = 0;
68 	if (dms->skip_post_probe)
69 		return 0;
70 	if (&prev->uclass_node != &uc->dev_head) {
71 		struct dm_test_uclass_perdev_priv *prev_uc_priv
72 				= dev_get_uclass_priv(prev);
73 		struct dm_test_pdata *pdata = prev->platdata;
74 
75 		ut_assert(pdata);
76 		ut_assert(prev_uc_priv);
77 		priv->base_add = prev_uc_priv->base_add + pdata->ping_add;
78 	}
79 
80 	return 0;
81 }
82 
83 static int test_pre_remove(struct udevice *dev)
84 {
85 	dm_testdrv_op_count[DM_TEST_OP_PRE_REMOVE]++;
86 
87 	return 0;
88 }
89 
90 static int test_init(struct uclass *uc)
91 {
92 	dm_testdrv_op_count[DM_TEST_OP_INIT]++;
93 	ut_assert(uc->priv);
94 
95 	return 0;
96 }
97 
98 static int test_destroy(struct uclass *uc)
99 {
100 	dm_testdrv_op_count[DM_TEST_OP_DESTROY]++;
101 
102 	return 0;
103 }
104 
105 UCLASS_DRIVER(test) = {
106 	.name		= "test",
107 	.id		= UCLASS_TEST,
108 	.post_bind	= test_post_bind,
109 	.pre_unbind	= test_pre_unbind,
110 	.pre_probe	= test_pre_probe,
111 	.post_probe	= test_post_probe,
112 	.pre_remove	= test_pre_remove,
113 	.init		= test_init,
114 	.destroy	= test_destroy,
115 	.priv_auto_alloc_size	= sizeof(struct dm_test_uclass_priv),
116 	.per_device_auto_alloc_size = sizeof(struct dm_test_uclass_perdev_priv),
117 };
118