From: Simon Glass <sjg@chromium.org>
Date: Wed, 29 Jul 2020 01:41:13 +0000 (-0600)
Subject: test: Add a flag for tests that need console recording
X-Git-Url: http://git.dujemihanovic.xyz/?a=commitdiff_plain;h=132644f56ebb7399fb139d6b9c9e54ef0c218ea9;p=u-boot.git

test: Add a flag for tests that need console recording

Allow tests that need console recording to be marked, so they can be
skipped if it is not available.

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

diff --git a/include/test/test.h b/include/test/test.h
index ff92c39006..67c7d69d48 100644
--- a/include/test/test.h
+++ b/include/test/test.h
@@ -35,6 +35,7 @@ enum {
 	UT_TESTF_SCAN_FDT	= BIT(2),	/* scan device tree */
 	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 */
 };
 
 /**
diff --git a/test/cmd_ut.c b/test/cmd_ut.c
index cc9543c315..1963f3792c 100644
--- a/test/cmd_ut.c
+++ b/test/cmd_ut.c
@@ -6,6 +6,7 @@
 
 #include <common.h>
 #include <command.h>
+#include <console.h>
 #include <test/suites.h>
 #include <test/test.h>
 
@@ -34,6 +35,15 @@ int cmd_ut_category(const char *name, const char *prefix,
 			continue;
 		printf("Test: %s\n", test->name);
 
+		if (test->flags & UT_TESTF_CONSOLE_REC) {
+			int ret = console_record_reset_enable();
+
+			if (ret) {
+				printf("Skipping: Console recording disabled\n");
+				continue;
+			}
+		}
+
 		uts.start = mallinfo();
 
 		test->func(&uts);