[PATCH 4/4] test: Add tests for the bootmeth set command

[Martyn Welch]

We have added a "set" sub command to bootmeth, add some tests to check
it's operation.

Signed-off-by: Martyn Welch <martyn.welch at collabora.com>
---
 test/boot/bootmeth.c | 47 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/test/boot/bootmeth.c b/test/boot/bootmeth.c
index 113b789ea7..659f386228 100644
--- a/test/boot/bootmeth.c
+++ b/test/boot/bootmeth.c
@@ -130,6 +130,53 @@ static int bootmeth_cmd_order_glob(struct unit_test_state *uts)
 }
 BOOTSTD_TEST(bootmeth_cmd_order_glob, UT_TESTF_DM | UT_TESTF_SCAN_FDT);
 
+/* Check 'bootmeth set' command */
+static int bootmeth_cmd_set(struct unit_test_state *uts)
+{
+	/* Check we can enable extlinux fallback */
+	console_record_reset_enable();
+	ut_assertok(run_command("bootmeth set extlinux fallback 1", 0));
+	ut_assert_console_end();
+
+	/* Check we can disable extlinux fallback */
+	console_record_reset_enable();
+	ut_assertok(run_command("bootmeth set extlinux fallback 0", 0));
+	ut_assert_console_end();
+
+	/* Check extlinux fallback unexpected value */
+	console_record_reset_enable();
+	ut_asserteq(1, run_command("bootmeth set extlinux fallback fred", 0));
+	ut_assert_nextline("Unexpected value 'fred'");
+	ut_assert_nextline("Failed (err=-22)");
+	ut_assert_console_end();
+
+	/* Check that we need to provide right number of parameters */
+	ut_asserteq(1, run_command("bootmeth set extlinux fallback", 0));
+	ut_assert_nextline("Required parameters not provided");
+	ut_assert_console_end();
+
+	/* Check that we need to provide a valid bootmethod */
+	ut_asserteq(1, run_command("bootmeth set fred fallback 0", 0));
+	ut_assert_nextline("Unknown bootmeth 'fred'");
+	ut_assert_nextline("Failed (err=-19)");
+	ut_assert_console_end();
+
+	/* Check that we need to provide a valid property */
+	ut_asserteq(1, run_command("bootmeth set extlinux fred 0", 0));
+	ut_assert_nextline("Invalid option");
+	ut_assert_nextline("Failed (err=-22)");
+	ut_assert_console_end();
+
+	/* Check that we need to provide a bootmeth that supports properties */
+	ut_asserteq(1, run_command("bootmeth set efi fallback 0", 0));
+	ut_assert_nextline("set_property not found");
+	ut_assert_nextline("Failed (err=-19)");
+	ut_assert_console_end();
+
+	return 0;
+}
+BOOTSTD_TEST(bootmeth_cmd_set, UT_TESTF_DM | UT_TESTF_SCAN_FDT);
+
 /* Check 'bootmeths' env var */
 static int bootmeth_env(struct unit_test_state *uts)
 {
-- 
2.45.2