mirror of
https://github.com/u-boot/u-boot.git
synced 2024-11-28 15:13:31 +08:00
e721b882e9
Separate the ability to define tests and assert status of test functions from the dm tests so they can be used more consistently throughout all tests. Signed-off-by: Joe Hershberger <joe.hershberger@ni.com> Reviewed-by: Simon Glass <sjg@chromium.org>
160 lines
3.2 KiB
C
160 lines
3.2 KiB
C
/*
|
|
* Copyright (c) 2013 Google, Inc
|
|
*
|
|
* (C) Copyright 2012
|
|
* Pavel Herrmann <morpheus.ibis@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0+
|
|
*/
|
|
|
|
#include <common.h>
|
|
#include <dm.h>
|
|
#include <errno.h>
|
|
#include <malloc.h>
|
|
#include <dm/test.h>
|
|
#include <test/ut.h>
|
|
#include <asm/io.h>
|
|
|
|
int dm_testdrv_op_count[DM_TEST_OP_COUNT];
|
|
static struct unit_test_state *uts = &global_dm_test_state;
|
|
|
|
static int testdrv_ping(struct udevice *dev, int pingval, int *pingret)
|
|
{
|
|
const struct dm_test_pdata *pdata = dev_get_platdata(dev);
|
|
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 = testdrv_ping,
|
|
};
|
|
|
|
static int test_bind(struct udevice *dev)
|
|
{
|
|
/* Private data should not be allocated */
|
|
ut_assert(!dev_get_priv(dev));
|
|
|
|
dm_testdrv_op_count[DM_TEST_OP_BIND]++;
|
|
return 0;
|
|
}
|
|
|
|
static int test_probe(struct udevice *dev)
|
|
{
|
|
struct dm_test_priv *priv = dev_get_priv(dev);
|
|
|
|
/* Private data should be allocated */
|
|
ut_assert(priv);
|
|
|
|
dm_testdrv_op_count[DM_TEST_OP_PROBE]++;
|
|
priv->ping_total += DM_TEST_START_TOTAL;
|
|
return 0;
|
|
}
|
|
|
|
static int test_remove(struct udevice *dev)
|
|
{
|
|
/* Private data should still be allocated */
|
|
ut_assert(dev_get_priv(dev));
|
|
|
|
dm_testdrv_op_count[DM_TEST_OP_REMOVE]++;
|
|
return 0;
|
|
}
|
|
|
|
static int test_unbind(struct udevice *dev)
|
|
{
|
|
/* Private data should not be allocated */
|
|
ut_assert(!dev->priv);
|
|
|
|
dm_testdrv_op_count[DM_TEST_OP_UNBIND]++;
|
|
return 0;
|
|
}
|
|
|
|
U_BOOT_DRIVER(test_drv) = {
|
|
.name = "test_drv",
|
|
.id = UCLASS_TEST,
|
|
.ops = &test_ops,
|
|
.bind = test_bind,
|
|
.probe = test_probe,
|
|
.remove = test_remove,
|
|
.unbind = test_unbind,
|
|
.priv_auto_alloc_size = sizeof(struct dm_test_priv),
|
|
};
|
|
|
|
U_BOOT_DRIVER(test2_drv) = {
|
|
.name = "test2_drv",
|
|
.id = UCLASS_TEST,
|
|
.ops = &test_ops,
|
|
.bind = test_bind,
|
|
.probe = test_probe,
|
|
.remove = test_remove,
|
|
.unbind = test_unbind,
|
|
.priv_auto_alloc_size = sizeof(struct dm_test_priv),
|
|
};
|
|
|
|
static int test_manual_drv_ping(struct udevice *dev, int pingval, int *pingret)
|
|
{
|
|
*pingret = pingval + 2;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static const struct test_ops test_manual_ops = {
|
|
.ping = test_manual_drv_ping,
|
|
};
|
|
|
|
static int test_manual_bind(struct udevice *dev)
|
|
{
|
|
dm_testdrv_op_count[DM_TEST_OP_BIND]++;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int test_manual_probe(struct udevice *dev)
|
|
{
|
|
struct dm_test_state *dms = uts->priv;
|
|
|
|
dm_testdrv_op_count[DM_TEST_OP_PROBE]++;
|
|
if (!dms->force_fail_alloc)
|
|
dev->priv = calloc(1, sizeof(struct dm_test_priv));
|
|
if (!dev->priv)
|
|
return -ENOMEM;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int test_manual_remove(struct udevice *dev)
|
|
{
|
|
dm_testdrv_op_count[DM_TEST_OP_REMOVE]++;
|
|
return 0;
|
|
}
|
|
|
|
static int test_manual_unbind(struct udevice *dev)
|
|
{
|
|
dm_testdrv_op_count[DM_TEST_OP_UNBIND]++;
|
|
return 0;
|
|
}
|
|
|
|
U_BOOT_DRIVER(test_manual_drv) = {
|
|
.name = "test_manual_drv",
|
|
.id = UCLASS_TEST,
|
|
.ops = &test_manual_ops,
|
|
.bind = test_manual_bind,
|
|
.probe = test_manual_probe,
|
|
.remove = test_manual_remove,
|
|
.unbind = test_manual_unbind,
|
|
};
|
|
|
|
U_BOOT_DRIVER(test_pre_reloc_drv) = {
|
|
.name = "test_pre_reloc_drv",
|
|
.id = UCLASS_TEST,
|
|
.ops = &test_manual_ops,
|
|
.bind = test_manual_bind,
|
|
.probe = test_manual_probe,
|
|
.remove = test_manual_remove,
|
|
.unbind = test_manual_unbind,
|
|
.flags = DM_FLAG_PRE_RELOC,
|
|
};
|