X-Git-Url: http://git.ozlabs.org/?a=blobdiff_plain;f=ccan%2Ftal%2Ftalloc%2Ftest%2Frun-expand.c;fp=ccan%2Ftal%2Ftalloc%2Ftest%2Frun-expand.c;h=4708768c8bc9b374ebcc10c832c5767bf5a6a4e2;hb=3c164cd21a13bd3904117f6a33e06945f0f1850c;hp=0000000000000000000000000000000000000000;hpb=d3cbb4cbbc38c96c3a20a7e4e602be7327f25202;p=ccan

diff --git a/ccan/tal/talloc/test/run-expand.c b/ccan/tal/talloc/test/run-expand.c
new file mode 100644
index 00000000..4708768c
--- /dev/null
+++ b/ccan/tal/talloc/test/run-expand.c
@@ -0,0 +1,35 @@
+#include <ccan/tal/talloc/talloc.h>
+#include <ccan/tal/talloc/talloc.c>
+#include <ccan/tap/tap.h>
+
+int main(void)
+{
+	int *a;
+	const int arr[] = { 1, 2 };
+
+	plan_tests(14);
+	talloc_enable_null_tracking_no_autofree();
+
+	a = tal_arrz(NULL, int, 1);
+	ok1(a);
+
+	ok1(tal_expand(&a, arr, 2));
+	ok1(tal_count(a) == 3);
+	ok1(a[0] == 0);
+	ok1(a[1] == 1);
+	ok1(a[2] == 2);
+
+	ok1(tal_expand(&a, take(tal_arrz(NULL, int, 1)), 1));
+	ok1(tal_count(a) == 4);
+	ok1(a[0] == 0);
+	ok1(a[1] == 1);
+	ok1(a[2] == 2);
+	ok1(a[3] == 0);
+	ok1(talloc_total_blocks(NULL) == 2);
+	ok1(talloc_total_blocks(a) == 1);
+
+	tal_free(a);
+
+	talloc_disable_null_tracking();
+	return exit_status();
+}