Add a flag for driver model tests, so we can do special processing for
them.

Signed-off-by: Simon Glass <s...@chromium.org>
---

(no changes since v1)

 include/dm/test.h   | 3 ++-
 include/test/test.h | 2 ++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/include/dm/test.h b/include/dm/test.h
index 6ac6672cd6f..dfbc82c756d 100644
--- a/include/dm/test.h
+++ b/include/dm/test.h
@@ -143,7 +143,8 @@ struct dm_test_state {
 };
 
 /* Declare a new driver model test */
-#define DM_TEST(_name, _flags) UNIT_TEST(_name, _flags, dm_test)
+#define DM_TEST(_name, _flags) \
+       UNIT_TEST(_name, UT_TESTF_DM | (_flags), dm_test)
 
 /*
  * struct sandbox_sdl_plat - Platform data for the SDL video driver
diff --git a/include/test/test.h b/include/test/test.h
index 3fdaa2b5e51..27585507d8c 100644
--- a/include/test/test.h
+++ b/include/test/test.h
@@ -36,6 +36,8 @@ enum {
        UT_TESTF_FLAT_TREE      = BIT(3),       /* test needs flat DT */
        UT_TESTF_LIVE_TREE      = BIT(4),       /* needs live device tree */
        UT_TESTF_CONSOLE_REC    = BIT(5),       /* needs console recording */
+       /* do extra driver model init and uninit */
+       UT_TESTF_DM             = BIT(6),
 };
 
 /**
-- 
2.30.1.766.gb4fecdf3b7-goog

Reply via email to