rfc822: Retrieve header fields by name

[ccan] / ccan / rfc822 / test / run-hdr-of-name.c

diff --git a/ccan/rfc822/test/run-hdr-of-name.c b/ccan/rfc822/test/run-hdr-of-name.c
new file mode 100644 (file)
index 0000000..aa98ac5
--- /dev/null
+++ b/ccan/rfc822/test/run-hdr-of-name.c
@@ -0,0 +1,84 @@
+#include <ccan/foreach/foreach.h>
+#include <ccan/failtest/failtest_override.h>
+#include <ccan/failtest/failtest.h>
+#include <stdlib.h>
+#include <string.h>
+
+#define CCAN_RFC822_DEBUG
+
+#include <ccan/rfc822/rfc822.h>
+
+#include <ccan/rfc822/rfc822.c>
+
+#include "testdata.h"
+#include "helper.h"
+
+static void test_hdrbyname(const struct aexample *e, const char *buf, size_t len,
+                          const char *exname, int crlf)
+{
+       struct rfc822_msg *msg;
+       struct rfc822_header *h, *hx;
+       int i, j;
+
+       msg = rfc822_start(NULL, buf, len);
+       allocation_failure_check();
+       ok(msg, "opened %s", exname);
+
+       for (i = 0; i < e->nhdrs; i++) {
+               struct testhdr *eh = &e->hdrs[i];
+
+               h = rfc822_first_header_of_name(msg, eh->name);
+               hx = rfc822_next_header_of_name(msg, NULL, eh->name);
+               ok1(h == hx);
+
+               for (j = 0; h && j < eh->index; j++)
+                       h = rfc822_next_header_of_name(msg, h, eh->name);
+               ok(h, "header \"%s\" (#%d) exists", eh->name, eh->index);
+               if (!h)
+                       break;
+               check_header(msg, h, eh->name, eh->val, eh->errors, crlf);
+
+               h = rfc822_next_header_of_name(msg, h, eh->name);
+               ok1((eh->index != eh->last) ^ !h);
+       }
+
+       h = rfc822_first_header_of_name(msg, NULL);
+       ok(!h, "Never match NULL name");
+
+       rfc822_free(msg);
+       allocation_failure_check();
+}
+
+int main(int argc, char *argv[])
+{
+       struct aexample *e;
+
+       /* This is how many tests you plan to run */
+       plan_tests(6*num_aexamples() + 14*num_aexample_hdrs());
+
+       failtest_setup(argc, argv);
+
+       for_each_aexample(e) {
+               int crlf;
+
+               foreach_int(crlf, 0, 1) {
+                       const char *buf;
+                       size_t len;
+                       char exname[256];
+
+                       sprintf(exname, "%s[%s]", e->name, NLT(crlf));
+
+                       buf = assemble_msg(e, &len, crlf);
+                       ok((buf), "assembled %s", exname);
+                       if (!buf)
+                               continue;
+
+                       test_hdrbyname(e, buf, len, exname, crlf);
+
+                       talloc_free(buf);
+               }
+       }
+
+       /* This exits depending on whether all tests passed */
+       failtest_exit(exit_status());
+}