X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=test%2Fparser%2Futils.c;h=43479b1a88aea8cbd3c25a91e578d47ca18eb524;hp=a1d0ad24b590d9938dcdd12ed00b3548137d1e97;hb=22b9ca8f4b9d28b4405576b39a87d3d0c945dd07;hpb=a5a787c5d170708be5d15ee9418520917653a104

diff --git a/test/parser/utils.c b/test/parser/utils.c
index a1d0ad2..43479b1 100644
--- a/test/parser/utils.c
+++ b/test/parser/utils.c
@@ -131,7 +131,155 @@ 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;
 }
+
+bool resource_resolve(struct device_handler *handler, struct parser *parser,
+		struct resource *resource)
+{
+	if (!resource)
+		return true;
+	if (resource->resolved)
+		return true;
+
+	assert(parser);
+	assert(parser->resolve_resource);
+
+	return parser->resolve_resource(handler, resource);
+}
+
+void boot_option_resolve(struct device_handler *handler,
+		struct discover_boot_option *opt)
+{
+	resource_resolve(handler, opt->source, opt->boot_image);
+	resource_resolve(handler, opt->source, opt->initrd);
+	resource_resolve(handler, opt->source, opt->icon);
+}
+
+extern void device_handler_add_device(struct device_handler *handler,
+		struct discover_device *dev);
+
+void test_hotplug_device(struct parser_test *test, struct discover_device *dev)
+{
+	struct discover_boot_option *opt;
+
+	device_handler_add_device(test->handler, dev);
+
+	list_for_each_entry(&test->ctx->boot_options, opt, list)
+		boot_option_resolve(test->handler, opt);
+}
+
+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);
+}