test/parser: Add resource check helpers

[petitboot] / test / parser / utils.c

diff --git a/test/parser/utils.c b/test/parser/utils.c
index 69b0006d1f25c6263b8f58f76e02ad6cd466ab36..fb558cf2c3ec457be3d0ef63a8f399bb06b3f343 100644 (file)
--- a/test/parser/utils.c
+++ b/test/parser/utils.c
@@ -140,3 +140,111 @@ 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);
+}
+
+void __check_args(struct discover_boot_option *opt, const char *args,
+               const char *file, int line)
+{
+       int rc;
+
+       if (!opt->option->boot_args) {
+               fprintf(stderr, "%s%d: arg check failed\n", file, line);
+               fprintf(stderr, "  no arguments parsed\n");
+               fprintf(stderr, "  expected '%s'\n", args);
+               exit(EXIT_FAILURE);
+       }
+
+       rc = strcmp(opt->option->boot_args, args);
+       if (rc) {
+               fprintf(stderr, "%s%d: arg check failed\n", file, line);
+               fprintf(stderr, "  got      '%s'\n", opt->option->boot_args);
+               fprintf(stderr, "  expected '%s'\n", args);
+               exit(EXIT_FAILURE);
+       }
+}
+
+void __check_name(struct discover_boot_option *opt, const char *name,
+               const char *file, int line)
+{
+       int rc;
+
+       rc = strcmp(opt->option->name, name);
+       if (rc) {
+               fprintf(stderr, "%s%d: name check failed\n", file, line);
+               fprintf(stderr, "  got      '%s'\n", opt->option->name);
+               fprintf(stderr, "  expected '%s'\n", name);
+               exit(EXIT_FAILURE);
+       }
+}
+
+void __check_resolved_local_resource(struct resource *res,
+               struct discover_device *dev, const char *local_path,
+               const char *file, int line)
+{
+       const char *exp_url, *got_url;
+
+       if (!res)
+               errx(EXIT_FAILURE, "%s:%d: No resource", file, line);
+
+       if (!res->resolved)
+               errx(EXIT_FAILURE, "%s:%d: Resource is not resolved",
+                               file, line);
+
+       exp_url = talloc_asprintf(res, "file://%s%s",
+                       dev->mount_path, local_path);
+       got_url = pb_url_to_string(res->url);
+
+       if (strcmp(got_url, exp_url)) {
+               errx(EXIT_FAILURE,
+                               "%s:%d Resource mismatch: got %s, expected %s",
+                               file, line, got_url, exp_url);
+       }
+}
+
+void __check_unresolved_resource(struct resource *res,
+               const char *file, int line)
+{
+       if (!res)
+               errx(EXIT_FAILURE, "%s:%d: No resource", file, line);
+
+       if (res->resolved)
+               errx(EXIT_FAILURE, "%s:%d: Resource is resolved", file, line);
+}