test/parser: Add pxe discovery tests

Add tests to cover pxlinux-style configuration autodiscovery.

Signed-off-by: Jeremy Kerr <jk@ozlabs.org>

6 files changed, 204 insertions, 0 deletions

diff --git a/test/parser/Makefile.am b/test/parser/Makefile.am
index 62770d2..b36889d 100644
--- a/test/parser/Makefile.am
+++ b/test/parser/Makefile.am
@@ -69,6 +69,11 @@ TESTS = \
 	test-pxe-pathprefix-discover-mac \
 	test-pxe-path-resolve-relative \
 	test-pxe-path-resolve-absolute \
+	test-pxe-discover-bootfile-root \
+	test-pxe-discover-bootfile-subdir \
+	test-pxe-discover-bootfile-pathprefix \
+	test-pxe-discover-bootfile-relative-conffile \
+	test-pxe-discover-bootfile-absolute-conffile \
 	test-unresolved-remove
 
 $(TESTS): %: %.embedded-config.o
diff --git a/test/parser/test-pxe-discover-bootfile-absolute-conffile.c b/test/parser/test-pxe-discover-bootfile-absolute-conffile.c
new file mode 100644
index 0000000..8326690
--- /dev/null
+++ b/test/parser/test-pxe-discover-bootfile-absolute-conffile.c
@@ -0,0 +1,42 @@
+
+#include "parser-test.h"
+
+#if 0 /* PARSER_EMBEDDED_CONFIG */
+label linux
+kernel vmlinux
+initrd initrd
+#endif
+
+/**
+ * Test that a pxeconffile option (DHCP opt 209) takes precedence over
+ * configuration discovery, and is resolved as an absolute path, overriding
+ * any prefix in bootfile.
+ */
+
+void run_test(struct parser_test *test)
+{
+	struct discover_boot_option *opt;
+	struct discover_context *ctx;
+
+	/* fixme: we should have a normalised URL here */
+	test_read_conf_embedded_url(test, "tftp://host//dir1/conf");
+
+	test_set_event_source(test);
+	test_set_event_param(test->ctx->event, "bootfile", "dir2/binary");
+	test_set_event_param(test->ctx->event, "pxeconffile", "/dir1/conf");
+	test_set_event_param(test->ctx->event, "tftp", "host");
+
+	test_run_parser(test, "pxe");
+
+	ctx = test->ctx;
+
+	check_boot_option_count(ctx, 1);
+	opt = get_boot_option(ctx, 0);
+
+	check_name(opt, "linux");
+
+	check_resolved_url_resource(opt->boot_image,
+			"tftp://host/dir1/vmlinux");
+	check_resolved_url_resource(opt->initrd,
+			"tftp://host/dir1/initrd");
+}
diff --git a/test/parser/test-pxe-discover-bootfile-pathprefix.c b/test/parser/test-pxe-discover-bootfile-pathprefix.c
new file mode 100644
index 0000000..40a056a
--- /dev/null
+++ b/test/parser/test-pxe-discover-bootfile-pathprefix.c
@@ -0,0 +1,42 @@
+
+#include "parser-test.h"
+
+#if 0 /* PARSER_EMBEDDED_CONFIG */
+label linux
+kernel vmlinux
+initrd initrd
+#endif
+
+/**
+ * Test that a pxepathprefix option (DHCP opt 210) takes precedence over
+ * any prefix specified in the bootfile.
+ */
+
+void run_test(struct parser_test *test)
+{
+	struct discover_boot_option *opt;
+	struct discover_context *ctx;
+
+	test_read_conf_embedded_url(test,
+			"tftp://host/dir1/pxelinux.cfg/default");
+
+	test_set_event_source(test);
+	test_set_event_param(test->ctx->event, "bootfile", "dir2/binary");
+	test_set_event_param(test->ctx->event, "pxepathprefix", "dir1/");
+	test_set_event_param(test->ctx->event, "mac", "00:11:22:33:44:55");
+	test_set_event_param(test->ctx->event, "tftp", "host");
+
+	test_run_parser(test, "pxe");
+
+	ctx = test->ctx;
+
+	check_boot_option_count(ctx, 1);
+	opt = get_boot_option(ctx, 0);
+
+	check_name(opt, "linux");
+
+	check_resolved_url_resource(opt->boot_image,
+			"tftp://host/dir1/vmlinux");
+	check_resolved_url_resource(opt->initrd,
+			"tftp://host/dir1/initrd");
+}
diff --git a/test/parser/test-pxe-discover-bootfile-relative-conffile.c b/test/parser/test-pxe-discover-bootfile-relative-conffile.c
new file mode 100644
index 0000000..8e9ac82
--- /dev/null
+++ b/test/parser/test-pxe-discover-bootfile-relative-conffile.c
@@ -0,0 +1,38 @@
+
+#include "parser-test.h"
+
+#if 0 /* PARSER_EMBEDDED_CONFIG */
+label linux
+kernel vmlinux
+initrd initrd
+#endif
+
+/**
+ * Test that a pxeconffile option (DHCP opt 209) takes precedence over
+ * configuration discovery, and is resolved relative to bootfile
+ */
+
+void run_test(struct parser_test *test)
+{
+	struct discover_boot_option *opt;
+	struct discover_context *ctx;
+
+	test_read_conf_embedded_url(test, "tftp://host/dir/conf.text");
+
+	test_set_event_source(test);
+	test_set_event_param(test->ctx->event, "bootfile", "dir/binary");
+	test_set_event_param(test->ctx->event, "pxeconffile", "conf.text");
+	test_set_event_param(test->ctx->event, "tftp", "host");
+
+	test_run_parser(test, "pxe");
+
+	ctx = test->ctx;
+
+	check_boot_option_count(ctx, 1);
+	opt = get_boot_option(ctx, 0);
+
+	check_name(opt, "linux");
+
+	check_resolved_url_resource(opt->boot_image, "tftp://host/dir/vmlinux");
+	check_resolved_url_resource(opt->initrd, "tftp://host/dir/initrd");
+}
diff --git a/test/parser/test-pxe-discover-bootfile-root.c b/test/parser/test-pxe-discover-bootfile-root.c
new file mode 100644
index 0000000..e3c9338
--- /dev/null
+++ b/test/parser/test-pxe-discover-bootfile-root.c
@@ -0,0 +1,38 @@
+
+#include "parser-test.h"
+
+#if 0 /* PARSER_EMBEDDED_CONFIG */
+label linux
+kernel vmlinux
+initrd initrd
+#endif
+
+/**
+ * Test that we look for a configuration file under the same prefix as
+ * the DHCP bootfile parameter, which is in a top-level directory.
+ */
+
+void run_test(struct parser_test *test)
+{
+	struct discover_boot_option *opt;
+	struct discover_context *ctx;
+
+	test_read_conf_embedded_url(test, "tftp://host/pxelinux.cfg/default");
+
+	test_set_event_source(test);
+	test_set_event_param(test->ctx->event, "bootfile", "binary");
+	test_set_event_param(test->ctx->event, "mac", "00:11:22:33:44:55");
+	test_set_event_param(test->ctx->event, "tftp", "host");
+
+	test_run_parser(test, "pxe");
+
+	ctx = test->ctx;
+
+	check_boot_option_count(ctx, 1);
+	opt = get_boot_option(ctx, 0);
+
+	check_name(opt, "linux");
+
+	check_resolved_url_resource(opt->boot_image, "tftp://host/vmlinux");
+	check_resolved_url_resource(opt->initrd, "tftp://host/initrd");
+}
diff --git a/test/parser/test-pxe-discover-bootfile-subdir.c b/test/parser/test-pxe-discover-bootfile-subdir.c
new file mode 100644
index 0000000..ef1fc8d
--- /dev/null
+++ b/test/parser/test-pxe-discover-bootfile-subdir.c
@@ -0,0 +1,39 @@
+
+#include "parser-test.h"
+
+#if 0 /* PARSER_EMBEDDED_CONFIG */
+label linux
+kernel vmlinux
+initrd initrd
+#endif
+
+/**
+ * Test that we look for a configuration file under the same prefix as
+ * the DHCP bootfile parameter, which is in a subdirectory of the TFTP root.
+ */
+
+void run_test(struct parser_test *test)
+{
+	struct discover_boot_option *opt;
+	struct discover_context *ctx;
+
+	test_read_conf_embedded_url(test,
+			"tftp://host/dir/pxelinux.cfg/default");
+
+	test_set_event_source(test);
+	test_set_event_param(test->ctx->event, "bootfile", "dir/binary");
+	test_set_event_param(test->ctx->event, "mac", "00:11:22:33:44:55");
+	test_set_event_param(test->ctx->event, "tftp", "host");
+
+	test_run_parser(test, "pxe");
+
+	ctx = test->ctx;
+
+	check_boot_option_count(ctx, 1);
+	opt = get_boot_option(ctx, 0);
+
+	check_name(opt, "linux");
+
+	check_resolved_url_resource(opt->boot_image, "tftp://host/dir/vmlinux");
+	check_resolved_url_resource(opt->initrd, "tftp://host/dir/initrd");
+}