X-Git-Url: http://git.ozlabs.org/?a=blobdiff_plain;f=ccan%2Ftal%2Fstr%2Ftest%2Frun-string.c;h=5246a8aa0513e0f44c0d1f864be9d7fdf087444d;hb=a5ebf3146a57b7db05f6b6e2de0f2689a1b06751;hp=4f8899de2acabe843ffe321fd06f4a34024d1858;hpb=680aa496521811e7c60afb940bcae5024f4c6adb;p=ccan

diff --git a/ccan/tal/str/test/run-string.c b/ccan/tal/str/test/run-string.c
index 4f8899de..5246a8aa 100644
--- a/ccan/tal/str/test/run-string.c
+++ b/ccan/tal/str/test/run-string.c
@@ -6,7 +6,7 @@ int main(void)
 {
 	char *parent, *c;
 
-	plan_tests(13);
+	plan_tests(27);
 
 	parent = tal(NULL, char);
 	ok1(parent);
@@ -14,16 +14,19 @@ int main(void)
 	c = tal_strdup(parent, "hello");
 	ok1(strcmp(c, "hello") == 0);
 	ok1(tal_parent(c) == parent);
+	tal_free(c);
 
 	c = tal_strndup(parent, "hello", 3);
 	ok1(strcmp(c, "hel") == 0);
 	ok1(tal_parent(c) == parent);
+	tal_free(c);
 
 	c = tal_typechk_(parent, char *);
 	c = tal_dup(parent, char, "hello", 6, 0);
 	ok1(strcmp(c, "hello") == 0);
 	ok1(strcmp(tal_name(c), "char[]") == 0);
 	ok1(tal_parent(c) == parent);
+	tal_free(c);
 
 	/* Now with an extra byte. */
 	c = tal_dup(parent, char, "hello", 6, 1);
@@ -31,10 +34,42 @@ int main(void)
 	ok1(strcmp(tal_name(c), "char[]") == 0);
 	ok1(tal_parent(c) == parent);
 	strcat(c, "x");
+	tal_free(c);
 
 	c = tal_asprintf(parent, "hello %s", "there");
 	ok1(strcmp(c, "hello there") == 0);
 	ok1(tal_parent(c) == parent);
+	tal_free(c);
+
+	c = tal_strcat(parent, "hello ", "there");
+	ok1(strcmp(c, "hello there") == 0);
+	ok1(tal_parent(c) == parent);
+
+	/* Make sure take works correctly. */
+	c = tal_strcat(parent, take(c), " again");
+	ok1(strcmp(c, "hello there again") == 0);
+	ok1(tal_parent(c) == parent);
+	ok1(tal_first(parent) == c && !tal_next(parent, c));
+
+	c = tal_strcat(parent, "And ", take(c));
+	ok1(strcmp(c, "And hello there again") == 0);
+	ok1(tal_parent(c) == parent);
+	ok1(tal_first(parent) == c && !tal_next(parent, c));
+
+	/* NULL pass through works... */
+	c = tal_strcat(parent, take(NULL), take(c));
+	ok1(!c);
+	ok1(!tal_first(parent));
+
+	c = tal_strcat(parent, take(tal_strdup(parent, "hi")),
+		       take(NULL));
+	ok1(!c);
+	ok1(!tal_first(parent));
+
+	c = tal_strcat(parent, take(NULL), take(NULL));
+	ok1(!c);
+	ok1(!tal_first(parent));
+
 	tal_free(parent);
 
 	return exit_status();