tal/str: use tal/talloc backend #ifdef TAL_USE_TALLOC.
[ccan] / ccan / tal / str / test / run-take.c
1 #include <ccan/tal/str/str.h>
2 #include <ccan/tal/str/str.c>
3 #include <ccan/tap/tap.h>
4 #include "helper.h"
5
6 int main(void)
7 {
8         char *parent, *c;
9
10         plan_tests(14);
11
12         parent = tal(NULL, char);
13         ok1(parent);
14
15         c = tal_strdup(parent, "hello");
16
17         c = tal_strdup(parent, take(c));
18         ok1(strcmp(c, "hello") == 0);
19         ok1(tal_parent(c) == parent);
20
21         c = tal_strndup(parent, take(c), 5);
22         ok1(strcmp(c, "hello") == 0);
23         ok1(tal_parent(c) == parent);
24
25         c = tal_strndup(parent, take(c), 3);
26         ok1(strcmp(c, "hel") == 0);
27         ok1(tal_parent(c) == parent);
28         tal_free(c);
29
30         c = tal_strdup(parent, "hello %s");
31         c = tal_fmt(parent, take(c), "there");
32         ok1(strcmp(c, "hello there") == 0);
33         ok1(tal_parent(c) == parent);
34         /* No leftover allocations. */
35         tal_free(c);
36         ok1(no_children(parent));
37
38         tal_free(parent);
39         ok1(!taken_any());
40
41         /* NULL pass-through. */
42         c = NULL;
43         ok1(tal_strdup(NULL, take(c)) == NULL);
44         ok1(tal_strndup(NULL, take(c), 5) == NULL);
45         ok1(tal_fmt(NULL, take(c), 0) == NULL);
46
47         return exit_status();
48 }