summaryrefslogtreecommitdiffstats
path: root/test/dm/test-driver.c
diff options
context:
space:
mode:
Diffstat (limited to 'test/dm/test-driver.c')
-rw-r--r--test/dm/test-driver.c6
1 files changed, 4 insertions, 2 deletions
diff --git a/test/dm/test-driver.c b/test/dm/test-driver.c
index bc6a6e721d..d10af51147 100644
--- a/test/dm/test-driver.c
+++ b/test/dm/test-driver.c
@@ -12,11 +12,11 @@
#include <errno.h>
#include <malloc.h>
#include <dm/test.h>
-#include <dm/ut.h>
+#include <test/ut.h>
#include <asm/io.h>
int dm_testdrv_op_count[DM_TEST_OP_COUNT];
-static struct dm_test_state *dms = &global_test_state;
+static struct unit_test_state *uts = &global_dm_test_state;
static int testdrv_ping(struct udevice *dev, int pingval, int *pingret)
{
@@ -114,6 +114,8 @@ static int test_manual_bind(struct udevice *dev)
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));
OpenPOWER on IntegriCloud