test: Mark all driver model tests with a flag

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

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

diff --git a/include/dm/test.h b/include/dm/test.h
index 6ac6672..dfbc82c 100644 (file)
--- 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 3fdaa2b..2758550 100644 (file)
--- 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),
 };
 
 /**