X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=test%2Fparser%2Futils.c;h=f6939828337cfde8aa3ce31b8f1687cea2542e34;hp=9a6b2e145061969eabaf62a9d96ebc72823825df;hb=98a3faddc9e01e2bfc9c267f32fce0173678a40a;hpb=7d20772cc7b4d258ee878b4cb30b313f14502dc7

diff --git a/test/parser/utils.c b/test/parser/utils.c
index 9a6b2e1..f693982 100644
--- a/test/parser/utils.c
+++ b/test/parser/utils.c
@@ -32,26 +32,31 @@ void __register_parser(struct parser *parser)
 	list_add(&parsers, &i->list);
 }
 
+static void __attribute__((destructor)) __cleanup_parsers(void)
+{
+	struct p_item *item, *tmp;
+
+	list_for_each_entry_safe(&parsers, item, tmp, list)
+		talloc_free(item);
+}
+
 static struct discover_device *test_create_device_simple(
-		struct discover_context *ctx)
+		struct parser_test *test)
 {
 	static int dev_idx;
 	char name[10];
 
 	sprintf(name, "__test%d", dev_idx++);
 
-	return test_create_device(ctx, name);
+	return test_create_device(test, name);
 }
 
-struct discover_device *test_create_device(struct discover_context *ctx,
+struct discover_device *test_create_device(struct parser_test *test,
 		const char *name)
 {
 	struct discover_device *dev;
 
-	dev = talloc_zero(ctx, struct discover_device);
-	dev->device = talloc_zero(dev, struct device);
-
-	list_init(&dev->boot_options);
+	dev = discover_device_create(test->handler, name);
 
 	dev->device->id = talloc_strdup(dev, name);
 	dev->device_path = talloc_asprintf(dev, "/dev/%s", name);
@@ -68,17 +73,21 @@ static struct discover_context *test_create_context(struct parser_test *test)
 	assert(ctx);
 
 	list_init(&ctx->boot_options);
-	ctx->device = test_create_device_simple(ctx);
+	ctx->device = test_create_device_simple(test);
+	device_handler_add_device(test->handler, ctx->device);
 
 	return ctx;
 }
 
+extern struct config *test_config_init(struct parser_test *test);
+
 struct parser_test *test_init(void)
 {
 	struct parser_test *test;
 
 	test = talloc_zero(NULL, struct parser_test);
-	test->handler = device_handler_init(NULL, 0);
+	test->config = test_config_init(test);
+	test->handler = device_handler_init(NULL, NULL, 0);
 	test->ctx = test_create_context(test);
 
 	return test;
@@ -125,6 +134,12 @@ void test_read_conf_file(struct parser_test *test, const char *filename)
 	talloc_free(path);
 }
 
+void test_set_conf_source(struct parser_test *test, const char *url)
+{
+	test->ctx->conf_url = pb_url_parse(test, url);
+	assert(test->ctx->conf_url);
+}
+
 int test_run_parser(struct parser_test *test, const char *parser_name)
 {
 	struct p_item* i;
@@ -161,9 +176,6 @@ void boot_option_resolve(struct device_handler *handler,
 	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;
@@ -194,10 +206,19 @@ void __check_boot_option_count(struct discover_context *ctx, int count,
 		const char *file, int line)
 {
 	struct discover_boot_option *opt;
-	int i = 0;
+	int defaults = 0, i = 0;
 
-	list_for_each_entry(&ctx->boot_options, opt, list)
+	list_for_each_entry(&ctx->boot_options, opt, list) {
 		i++;
+		if (opt->option->is_default)
+			defaults++;
+	}
+
+	if (defaults > 1) {
+		fprintf(stderr, "%s:%d: parser returned multiple default "
+				"options\n", file, line);
+		exit(EXIT_FAILURE);
+	}
 
 	if (i == count)
 		return;
@@ -218,6 +239,9 @@ void __check_args(struct discover_boot_option *opt, const char *args,
 {
 	int rc;
 
+	if (!opt->option->boot_args && !args)
+		return;
+
 	if (!opt->option->boot_args) {
 		fprintf(stderr, "%s:%d: arg check failed\n", file, line);
 		fprintf(stderr, "  no arguments parsed\n");
@@ -248,6 +272,16 @@ void __check_name(struct discover_boot_option *opt, const char *name,
 	}
 }
 
+void __check_is_default(struct discover_boot_option *opt,
+		const char *file, int line)
+{
+	if (opt->option->is_default)
+		return;
+
+	fprintf(stderr, "%s:%d: default check failed\n", file, line);
+	exit(EXIT_FAILURE);
+}
+
 void __check_resolved_local_resource(struct resource *res,
 		struct discover_device *dev, const char *local_path,
 		const char *file, int line)
@@ -273,6 +307,26 @@ void __check_resolved_local_resource(struct resource *res,
 	}
 }
 
+void __check_resolved_url_resource(struct resource *res,
+		const char *url, const char *file, int line)
+{
+	char *res_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);
+
+	res_url = pb_url_to_string(res->url);
+	if (strcmp(url, res_url)) {
+		fprintf(stderr, "%s:%d: Resource mismatch\n", file, line);
+		fprintf(stderr, "  got      '%s'\n", res_url);
+		fprintf(stderr, "  expected '%s'\n", url);
+		exit(EXIT_FAILURE);
+	}
+}
 void __check_unresolved_resource(struct resource *res,
 		const char *file, int line)
 {