generator: Generators for C

[ccan] / ccan / generator / test / api.c

diff --git a/ccan/generator/test/api.c b/ccan/generator/test/api.c
new file mode 100644 (file)
index 0000000..a5770fd
--- /dev/null
+++ b/ccan/generator/test/api.c
@@ -0,0 +1,62 @@
+#include <ccan/generator/generator.h>
+#include <ccan/tap/tap.h>
+#include <ccan/str/str.h>
+
+#include "example-gens.h"
+
+generator_def_static(genx, const char *)
+{
+       generator_yield("one");
+       generator_yield("two");
+       generator_yield("three");
+       generator_yield("four");
+}
+
+static void test1(void)
+{
+       generator_t(int) state1 = gen1();
+       int *ret;
+
+       ok1((ret = generator_next(state1)) != NULL);
+       ok1(*ret == 1);
+       ok1((ret = generator_next(state1)) != NULL);
+       ok1(*ret == 3);
+       ok1((ret = generator_next(state1)) != NULL);
+       ok1(*ret == 17);
+       ok1((ret = generator_next(state1)) == NULL);
+
+       /* Test that things don't go bad if we try to invoke an
+        * already completed generator */
+       ok1((ret = generator_next(state1)) == NULL);
+
+       generator_free(state1);
+}
+
+static void testx(void)
+{
+       generator_t(const char *) statex = genx();
+       const char *val;
+
+       ok1(generator_next_val(val, statex));
+       ok1(streq(val, "one"));
+       ok1(generator_next_val(val, statex));
+       ok1(streq(val, "two"));
+       ok1(generator_next_val(val, statex));
+       ok1(streq(val, "three"));
+       ok1(generator_next_val(val, statex));
+       ok1(streq(val, "four"));
+       ok1(!generator_next_val(val, statex));
+       generator_free(statex);
+}
+
+int main(void)
+{
+       /* This is how many tests you plan to run */
+       plan_tests(8 + 9);
+
+       test1();
+       testx();
+
+       /* This exits depending on whether all tests passed */
+       return exit_status();
+}