test/parser: Add pxe discovery tests

Add tests to cover pxlinux-style configuration autodiscovery.

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

diff --git a/test/parser/Makefile.am b/test/parser/Makefile.am
index 62770d2c950e05aa9ca915e21f2acefdc76eb3ab..b36889df12bdd4301bfb9014784b6984cc994cea 100644 (file)
--- 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-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
        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 (file)
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 (file)
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 (file)
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 (file)
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 (file)
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");
+}