tal/talloc: new module for backending tal onto talloc.

[ccan] / ccan / tal / talloc / test / run-take.c

diff --git a/ccan/tal/talloc/test/run-take.c b/ccan/tal/talloc/test/run-take.c
new file mode 100644 (file)
index 0000000..7283c16
--- /dev/null
+++ b/ccan/tal/talloc/test/run-take.c
@@ -0,0 +1,56 @@
+#include <ccan/tal/talloc/talloc.h>
+#include <ccan/tal/talloc/talloc.c>
+#include <ccan/tap/tap.h>
+
+int main(void)
+{
+       char *parent, *c;
+
+       plan_tests(21);
+
+       /* We can take NULL. */
+       ok1(take(NULL) == NULL);
+       ok1(is_taken(NULL));
+       ok1(taken(NULL)); /* Undoes take() */
+       ok1(!is_taken(NULL));
+       ok1(!taken(NULL));
+
+       parent = tal(NULL, char);
+       ok1(parent);
+
+       ok1(take(parent) == parent);
+       ok1(is_taken(parent));
+       ok1(taken(parent)); /* Undoes take() */
+       ok1(!is_taken(parent));
+       ok1(!taken(parent));
+
+       c = tal(parent, char);
+       *c = 'h';
+       c = tal_dup(parent, char, take(c), 1, 0);
+       ok1(c[0] == 'h');
+       ok1(tal_parent(c) == parent);
+
+       c = tal_dup(parent, char, take(c), 1, 2);
+       ok1(c[0] == 'h');
+       strcpy(c, "hi");
+       ok1(tal_parent(c) == parent);
+
+       /* dup must reparent child. */
+       c = tal_dup(NULL, char, take(c), 1, 0);
+       ok1(c[0] == 'h');
+       ok1(tal_parent(c) == NULL);
+
+       /* No leftover allocations. */
+       tal_free(c);
+       ok1(talloc_total_blocks(parent) == 1);
+
+       tal_free(parent);
+       ok1(!taken_any());
+
+       /* NULL pass-through. */
+       c = NULL;
+       ok1(tal_dup(NULL, char, take(c), 5, 5) == NULL);
+       ok1(!taken_any());
+
+       return exit_status();
+}