test/parser: Add check_boot_option_count helper & get_boot_option

Add a helper function to check the expected boot option counts, and
print the boot option details if the check fails.

Signed-off-by: Jeremy Kerr <jk@ozlabs.org>

1 files changed, 39 insertions, 0 deletions

diff --git a/test/parser/utils.c b/test/parser/utils.c
index 69b0006..019ba63 100644
--- a/test/parser/utils.c
+++ b/test/parser/utils.c
@@ -140,3 +140,42 @@ int test_run_parser(struct parser_test *test, const char *parser_name)
 
 	return rc;
 }
+
+struct discover_boot_option *get_boot_option(struct discover_context *ctx,
+		int idx)
+{
+	struct discover_boot_option *opt;
+	int i = 0;
+
+	list_for_each_entry(&ctx->boot_options, opt, list) {
+		if (i++ == idx)
+			return opt;
+	}
+
+	assert(0);
+
+	return NULL;
+}
+
+void __check_boot_option_count(struct discover_context *ctx, int count,
+		const char *file, int line)
+{
+	struct discover_boot_option *opt;
+	int i = 0;
+
+	list_for_each_entry(&ctx->boot_options, opt, list)
+		i++;
+
+	if (i == count)
+		return;
+
+	fprintf(stderr, "%s:%d: boot option count check failed\n", file, line);
+	fprintf(stderr, "expected %d options, got %d:\n", count, i);
+
+	i = 1;
+	list_for_each_entry(&ctx->boot_options, opt, list)
+		fprintf(stderr, "  %2d: %s [%s]\n", i++, opt->option->name,
+				opt->option->id);
+
+	exit(EXIT_FAILURE);
+}