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