discover/grub2: Implement save_env command

[petitboot] / test / parser / test-grub2-save-env.c

diff --git a/test/parser/test-grub2-save-env.c b/test/parser/test-grub2-save-env.c
new file mode 100644 (file)
index 0000000..7a2938f
--- /dev/null
+++ b/test/parser/test-grub2-save-env.c
@@ -0,0 +1,109 @@
+
+#include <string.h>
+
+#include <array-size/array-size.h>
+#include <talloc/talloc.h>
+
+#include "parser-test.h"
+
+static const char *envsig = "# GRUB Environment Block\n";
+
+struct env_test {
+       const char *name;
+       const char *env_before;
+       const char *script;
+       const char *env_after;
+} tests[] = {
+       {
+               "init",
+               "######",
+               "a=xxx\nsave_env a\n",
+               "a=xxx\n"
+       },
+       {
+               "append",
+               "q=q\nr=r\n######",
+               "a=xxx\nsave_env a\n",
+               "q=q\nr=r\na=xxx\n"
+       },
+       {
+               "expand",
+               "q=q\na=x\nr=r\n##",
+               "a=xxx\nsave_env a\n",
+               "q=q\na=xxx\nr=r\n",
+       },
+       {
+               "reduce",
+               "q=q\na=xxx\nr=r\n",
+               "a=x\nsave_env a\n",
+               "q=q\na=x\nr=r\n##",
+       },
+       {
+               "invalid-insert",
+               "q=q\n---\nr=r\n",
+               "a=x\nsave_env a\n",
+               "q=q\na=x\nr=r\n",
+       },
+       {
+               "invalid-shift",
+               "q=q\n--\nr=r\n#",
+               "a=x\nsave_env a\n",
+               "q=q\na=x\nr=r\n",
+       },
+       {
+               "invalid-reduce",
+               "q=q\n----\nr=r\n",
+               "a=x\nsave_env a\n",
+               "q=q\na=x\nr=r\n#",
+       },
+       {
+               "dup-replace-first",
+               "q=q\na=y\nr=r\na=z",
+               "a=x\nsave_env a\n",
+               "q=q\na=x\nr=r\na=z",
+       },
+       {
+               "nospace-add",
+               "q=q\nr=r\n###",
+               "a=x\nsave_env a\n",
+               "q=q\nr=r\n###",
+       },
+       {
+               "nospace-replace",
+               "q=q\na=x\nr=r\n#",
+               "a=xxx\nsave_env a\n",
+               "q=q\na=x\nr=r\n#",
+       },
+};
+
+static void run_env_test(struct parser_test *test, struct env_test *envtest)
+{
+       const char *env_before, *env_after;
+
+       env_before = talloc_asprintf(test, "%s%s", envsig, envtest->env_before);
+       env_after  = talloc_asprintf(test, "%s%s", envsig, envtest->env_after);
+
+       test_add_file_data(test, test->ctx->device, "/boot/grub/grubenv",
+                       env_before, strlen(env_before));
+
+       __test_read_conf_data(test, envtest->script, strlen(envtest->script));
+       test_run_parser(test, "grub2");
+
+       check_file_contents(test, test->ctx->device, "/boot/grub/grubenv",
+                       env_after, strlen(env_after));
+}
+
+void run_test(struct parser_test *test)
+{
+       struct env_test *env_test;
+       unsigned int i;
+
+       for (i = 0; i < ARRAY_SIZE(tests); i++) {
+               env_test = &tests[i];
+               printf("test %s: ", env_test->name);
+               fflush(stdout);
+               run_env_test(test, env_test);
+               printf("OK\n");
+               fflush(stdout);
+       }
+}