New talloc_set for auto-cleanup.

[ccan] / ccan / talloc / test / run-talloc_set.c

diff --git a/ccan/talloc/test/run-talloc_set.c b/ccan/talloc/test/run-talloc_set.c
new file mode 100644 (file)
index 0000000..d7b4961
--- /dev/null
+++ b/ccan/talloc/test/run-talloc_set.c
@@ -0,0 +1,36 @@
+#include "talloc/talloc.c"
+#include "tap/tap.h"
+#include <assert.h>
+
+int main(void)
+{
+       char *c;
+       int *i;
+
+       plan_tests(12);
+
+       /* Set C to a valid pointer, with correct parent. */
+       talloc_set(&c, NULL);
+       ok1(c >= (char *)(intptr_t)getpagesize());
+       ok1(talloc_parent(c) == NULL);
+
+       /* Free it, should blatt c. */
+       talloc_free(c);
+       ok1(c);
+       ok1(c < (char *)(intptr_t)getpagesize());
+
+       /* Same test, indirect. */
+       talloc_set(&i, NULL);
+       talloc_set(&c, i);
+       ok1(c >= (char *)(intptr_t)getpagesize());
+       ok1(i >= (int *)(intptr_t)getpagesize());
+       ok1(talloc_parent(i) == NULL);
+       ok1(talloc_parent(c) == i);
+       talloc_free(i);
+       ok1(c);
+       ok1(c < (char *)(intptr_t)getpagesize());
+       ok1(i);
+       ok1(i < (int *)(intptr_t)getpagesize());
+
+       return exit_status();
+}