3 #include <ccan/bytestring/bytestring.h>
4 #include <ccan/tap/tap.h>
6 #include <ccan/bytestring/bytestring.c>
8 #define TEST_STRING "test string"
9 #define TEST_STRING_2 "abc\0def"
11 const char str1[] = TEST_STRING;
12 const char *str2 = TEST_STRING;
16 struct bytestring bs, bs1, bs2, bs3, bs4, bs5, bs6, bs7;
19 /* This is how many tests you plan to run */
22 bs = bytestring(str1, sizeof(str1) - 1);
24 ok1(bs.len == (sizeof(str1) - 1));
26 bs1 = BYTESTRING(TEST_STRING);
27 ok1(bytestring_eq(bs, bs1));
29 bs2 = BYTESTRING(TEST_STRING_2);
32 bs3 = bytestring_from_string(str2);
33 ok1(bytestring_eq(bs3, bs));
35 bs4 = bytestring_from_string(TEST_STRING_2);
38 bs5 = bytestring_from_string(NULL);
41 ok1(bytestring_eq(bs5, bytestring_NULL));
43 ok1(bytestring_byte(bs2, 0) == 'a');
44 ok1(bytestring_byte(bs2, 1) == 'b');
45 ok1(bytestring_byte(bs2, 2) == 'c');
46 ok1(bytestring_byte(bs2, 3) == '\0');
47 ok1(bytestring_byte(bs2, 4) == 'd');
48 ok1(bytestring_byte(bs2, 5) == 'e');
49 ok1(bytestring_byte(bs2, 6) == 'f');
51 ok1(bytestring_eq(bytestring_slice(bs, 0, 4), BYTESTRING("test")));
52 ok1(bytestring_eq(bytestring_slice(bs, 5, 8), BYTESTRING("str")));
53 ok1(bytestring_eq(bytestring_slice(bs2, 2, 5), BYTESTRING("c\0d")));
54 ok1(bytestring_eq(bytestring_slice(bs2, 0, -1U), bs2));
55 ok1(bytestring_eq(bytestring_slice(bs2, 10, 20), bytestring_NULL));
56 ok1(bytestring_eq(bytestring_slice(bs2, 2, 1), bytestring_NULL));
58 ok1(bytestring_starts(bs, BYTESTRING("test")));
59 ok1(bytestring_ends(bs, BYTESTRING("string")));
60 ok1(bytestring_starts(bs2, BYTESTRING("abc")));
61 ok1(bytestring_starts(bs2, BYTESTRING("abc\0")));
62 ok1(bytestring_ends(bs2, BYTESTRING("def")));
63 ok1(bytestring_ends(bs2, BYTESTRING("\0def")));
64 ok1(!bytestring_starts(bs2, BYTESTRING("def")));
65 ok1(!bytestring_ends(bs2, BYTESTRING("abc")));
67 ok1(bytestring_index(bs1, ' ') == (bs1.ptr + 4));
68 ok1(bytestring_index(bs1, 't') == (bs1.ptr + 0));
69 ok1(bytestring_index(bs1, 0) == NULL);
70 ok1(bytestring_index(bs2, 0) == (bs2.ptr + 3));
71 ok1(bytestring_index(bs2, 'f') == (bs2.ptr + 6));
72 ok1(bytestring_index(bs2, 'q') == NULL);
74 ok1(bytestring_rindex(bs1, ' ') == (bs1.ptr + 4));
75 ok1(bytestring_rindex(bs1, 't') == (bs1.ptr + 6));
76 ok1(bytestring_rindex(bs1, 0) == NULL);
77 ok1(bytestring_rindex(bs2, 0) == (bs2.ptr + 3));
78 ok1(bytestring_rindex(bs2, 'f') == (bs2.ptr + 6));
79 ok1(bytestring_rindex(bs2, 'q') == NULL);
81 bs6 = BYTESTRING("string");
82 ok1(bytestring_eq(bytestring_bytestring(bs1, bs6),
83 bytestring(bs1.ptr + 5, 6)));
84 bs6 = BYTESTRING("c\0d");
85 ok1(bytestring_eq(bytestring_bytestring(bs2, bs6),
86 bytestring(bs2.ptr + 2, 3)));
87 bs6 = BYTESTRING("c\0e");
88 ok1(bytestring_bytestring(bs2, bs6).ptr == NULL);
89 ok1(bytestring_eq(bytestring_bytestring(bs1, bytestring_NULL),
90 bytestring(bs1.ptr, 0)));
91 ok1(bytestring_eq(bytestring_bytestring(bs2, bytestring_NULL),
92 bytestring(bs2.ptr, 0)));
95 ok1(bytestring_spn(bs1, BYTESTRING("est")) == 4);
96 ok1(bytestring_cspn(bs1, BYTESTRING(" ")) == 4);
98 ok1(bytestring_spn(bs2, BYTESTRING("z")) == 0);
99 ok1(bytestring_cspn(bs2, BYTESTRING("\0")) == 3);
101 ok1(bytestring_spn(bs1, BYTESTRING("eginrst ")) == bs1.len);
102 ok1(bytestring_cspn(bs2, BYTESTRING("z")) == bs2.len);
104 bs7 = bytestring_splitchr_first(bs, ' ');
105 ok1(bs7.ptr == bs.ptr);
106 ok1(bytestring_eq(bs7, BYTESTRING("test")));
107 bs7 = bytestring_splitchr_next(bs, ' ', bs7);
108 ok1(bs7.ptr == bs.ptr + 5);
109 ok1(bytestring_eq(bs7, BYTESTRING("string")));
110 bs7 = bytestring_splitchr_next(bs, ' ', bs7);
112 bs7 = bytestring_splitchr_next(bs, ' ', bs7);
115 bs7 = bytestring_splitchr_first(bs2, '\0');
116 ok1(bs7.ptr = bs2.ptr);
117 ok1(bytestring_eq(bs7, BYTESTRING("abc")));
118 bs7 = bytestring_splitchr_next(bs2, '\0', bs7);
119 ok1(bs7.ptr == bs2.ptr + 4);
120 ok1(bytestring_eq(bs7, BYTESTRING("def")));
121 bs7 = bytestring_splitchr_next(bs2, '\0', bs7);
123 bs7 = bytestring_splitchr_next(bs2, ' ', bs7);
126 bs7 = bytestring_splitchr_first(bs, 's');
127 ok1(bs7.ptr == bs.ptr);
128 ok1(bytestring_eq(bs7, BYTESTRING("te")));
129 bs7 = bytestring_splitchr_next(bs, 's', bs7);
130 ok1(bs7.ptr == bs.ptr + 3);
131 ok1(bytestring_eq(bs7, BYTESTRING("t ")));
132 bs7 = bytestring_splitchr_next(bs, 's', bs7);
133 ok1(bs7.ptr == bs.ptr + 6);
134 ok1(bytestring_eq(bs7, BYTESTRING("tring")));
135 bs7 = bytestring_splitchr_next(bs, 's', bs7);
137 bs7 = bytestring_splitchr_next(bs, ' ', bs7);
140 bs7 = bytestring_splitchr_first(bs2, 'f');
141 ok1(bs7.ptr = bs2.ptr);
142 ok1(bytestring_eq(bs7, BYTESTRING("abc\0de")));
143 bs7 = bytestring_splitchr_next(bs2, 'f', bs7);
144 ok1(bs7.ptr == bs2.ptr + 7);
146 bs7 = bytestring_splitchr_next(bs2, 'f', bs7);
148 bs7 = bytestring_splitchr_next(bs2, 'f', bs7);
151 bs7 = bytestring_splitchr_first(BYTESTRING(""), 'q');
155 bytestring_foreach_splitchr(bs7, BYTESTRING(" "), ' ') {
157 ok1(bs7.ptr && !bs7.len);
162 bytestring_foreach_splitchr(bs7, BYTESTRING("\0\0\0"), '\0') {
164 ok1(bs7.ptr && !bs7.len);
168 bs7 = bytestring_splitchrs_first(bs, BYTESTRING(" \0"));
169 ok1(bs7.ptr == bs.ptr);
170 ok1(bytestring_eq(bs7, BYTESTRING("test")));
171 bs7 = bytestring_splitchrs_next(bs, BYTESTRING(" \0"), bs7);
172 ok1(bs7.ptr == bs.ptr + 5);
173 ok1(bytestring_eq(bs7, BYTESTRING("string")));
174 bs7 = bytestring_splitchrs_next(bs, BYTESTRING(" \0"), bs7);
176 bs7 = bytestring_splitchrs_next(bs, BYTESTRING(" \0"), bs7);
179 bs7 = bytestring_splitchrs_first(bs2, BYTESTRING(" \0"));
180 ok1(bs7.ptr == bs2.ptr);
181 ok1(bytestring_eq(bs7, BYTESTRING("abc")));
182 bs7 = bytestring_splitchrs_next(bs2, BYTESTRING(" \0"), bs7);
183 ok1(bs7.ptr == bs2.ptr + 4);
184 ok1(bytestring_eq(bs7, BYTESTRING("def")));
185 bs7 = bytestring_splitchrs_next(bs2, BYTESTRING(" \0"), bs7);
187 bs7 = bytestring_splitchrs_next(bs2, BYTESTRING(" \0"), bs7);
190 bs7 = bytestring_splitchrs_first(BYTESTRING(""), BYTESTRING(" \0"));
194 bytestring_foreach_splitchrs(bs7, BYTESTRING(" \0 \0 "),
197 ok1(bs7.ptr && !bs7.len);
201 bs7 = bytestring_splitstr_first(bs, BYTESTRING("t "));
202 ok1(bs7.ptr == bs.ptr);
203 ok1(bytestring_eq(bs7, BYTESTRING("tes")));
204 bs7 = bytestring_splitstr_next(bs, BYTESTRING("t "), bs7);
205 ok1(bs7.ptr == bs.ptr + 5);
206 ok1(bytestring_eq(bs7, BYTESTRING("string")));
207 bs7 = bytestring_splitstr_next(bs, BYTESTRING("t "), bs7);
209 bs7 = bytestring_splitstr_next(bs, BYTESTRING("t "), bs7);
212 bs7 = bytestring_splitstr_first(bs2, BYTESTRING("abc"));
213 ok1(bs7.ptr == bs2.ptr);
215 bs7 = bytestring_splitstr_next(bs2, BYTESTRING("abc"), bs7);
216 ok1(bs7.ptr == bs2.ptr + 3);
217 ok1(bytestring_eq(bs7, BYTESTRING("\0def")));
218 bs7 = bytestring_splitstr_next(bs2, BYTESTRING("t "), bs7);
220 bs7 = bytestring_splitstr_next(bs2, BYTESTRING("t "), bs7);
223 bs7 = bytestring_splitstr_first(BYTESTRING(""), BYTESTRING(""));
227 bytestring_foreach_splitstr(bs7, BYTESTRING("foofoo"),
230 ok1(bs7.ptr && !bs7.len);
234 /* This exits depending on whether all tests passed */
235 return exit_status();