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