From b6227d3993db8f1763a8bd377b189e7b77cfe885 Mon Sep 17 00:00:00 2001
From: Joe Hershberger <joe.hershberger@ni.com>
Date: Wed, 20 May 2015 14:27:35 -0500
Subject: [PATCH] test: dm: Recover the driver model tree after tests

Put the driver model for the system back into a good state after
completing the DM testing.

Signed-off-by: Joe Hershberger <joe.hershberger@ni.com>
Acked-by: Simon Glass <sjg@chromium.org>
---
 test/dm/test-main.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/test/dm/test-main.c b/test/dm/test-main.c
index 53c3a6ebbd..0477d2fa73 100644
--- a/test/dm/test-main.c
+++ b/test/dm/test-main.c
@@ -111,6 +111,11 @@ static int dm_test_main(const char *test_name)
 
 	printf("Failures: %d\n", uts->fail_count);
 
+	gd->dm_root = NULL;
+	ut_assertok(dm_init());
+	dm_scan_platdata(false);
+	dm_scan_fdt(gd->fdt_blob, false);
+
 	return uts->fail_count ? CMD_RET_FAILURE : 0;
 }
 
-- 
2.39.5