test/parser: Add check_boot_option_count helper & get_boot_option

[petitboot] / test / parser / utils.c

diff --git a/test/parser/utils.c b/test/parser/utils.c
index 16c415a1ebd7477f27baa2fcfbd96cf5362fab0a..019ba633e23b8de944dca06d906b8e3d4423aa31 100644 (file)
--- a/test/parser/utils.c
+++ b/test/parser/utils.c
@@ -85,6 +85,13 @@ void test_fini(struct parser_test *test)
        talloc_free(test);
 }
 
+void __test_read_conf_data(struct parser_test *test,
+               const char *buf, size_t len)
+{
+       test->conf.size = len;
+       test->conf.buf = talloc_memdup(test, buf, len);
+}
+
 void test_read_conf_file(struct parser_test *test, const char *filename)
 {
        struct stat stat;
@@ -124,7 +131,51 @@ int test_run_parser(struct parser_test *test, const char *parser_name)
                        continue;
                test->ctx->parser = parser;
                rc = parser->parse(test->ctx, test->conf.buf, test->conf.size);
+               break;
        }
 
+       if (i == n_parsers)
+               errx(EXIT_FAILURE, "%s: parser '%s' not found",
+                               __func__, 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);
+}