generator: Allow generators to take arguments

[ccan] / ccan / generator / test / example-gens.c

diff --git a/ccan/generator/test/example-gens.c b/ccan/generator/test/example-gens.c
index c55401166409ebd857a92d9ade170abdd399778f..ec3157d77591de399d05cd043ab658ab2f9bd303 100644 (file)
--- a/ccan/generator/test/example-gens.c
+++ b/ccan/generator/test/example-gens.c
@@ -1,3 +1,5 @@
+#include <stdio.h>
+
 #include <ccan/generator/generator.h>
 
 #include "example-gens.h"
@@ -8,3 +10,19 @@ generator_def(gen1, int)
        generator_yield(3);
        generator_yield(17);
 }
+
+generator_def(gen2, int, int, base)
+{
+       generator_yield(base + 1);
+       generator_yield(base + 3);
+       generator_yield(base + 17);
+}
+
+generator_def(gen3, const char *, const char *, str, int, count)
+{
+       int i;
+
+       for (i = 0; i < count; i++)
+               generator_yield(str);
+}
+