From 31998343c5eb5533ef85b97f5bb2ffe200eb2ad0 Mon Sep 17 00:00:00 2001
From: Jeremy Kerr <jk@ozlabs.org>
Date: Wed, 15 May 2013 16:36:34 +0800
Subject: test/parser: Add resource check helpers

Add check_resolved_local_resource and check_unresolved_resource to
check the resources returned from parsers.

Signed-off-by: Jeremy Kerr <jk@ozlabs.org>
---
 test/parser/parser-test.h | 19 +++++++++++++++++++
 test/parser/utils.c       | 34 ++++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+)

diff --git a/test/parser/parser-test.h b/test/parser/parser-test.h
index 0baad94..3a4382a 100644
--- a/test/parser/parser-test.h
+++ b/test/parser/parser-test.h
@@ -71,4 +71,23 @@ void __check_name(struct discover_boot_option *opt, const char *name,
 #define check_name(opt, name) \
 	__check_name(opt, name, __FILE__, __LINE__)
 
+/**
+ * Check that a resource (@res) is present, resolved, and has a local path
+ * (within @dev's mount point) of @path.
+ */
+#define check_resolved_local_resource(res, dev, path) \
+	__check_resolved_local_resource(res, dev, path, __FILE__, __LINE__)
+
+void __check_resolved_local_resource(struct resource *res,
+		struct discover_device *dev, const char *local_path,
+		const char *file, int line);
+
+/**
+ * Check that a resource (@res) is present but not resolved
+ */
+void __check_unresolved_resource(struct resource *res,
+		const char *file, int line);
+#define check_unresolved_resource(res) \
+	__check_unresolved_resource(res, __FILE__, __LINE__)
+
 #endif /* PARSER_TEST_H */
diff --git a/test/parser/utils.c b/test/parser/utils.c
index 8ac2060..fb558cf 100644
--- a/test/parser/utils.c
+++ b/test/parser/utils.c
@@ -214,3 +214,37 @@ void __check_name(struct discover_boot_option *opt, const char *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);
+}
-- 
cgit v1.2.1