git.ozlabs.org Git - ccan/blob - ccan/bytestring/test/run.c

   1 #include "config.h"
   2
   3 #include <ccan/bytestring/bytestring.h>
   4 #include <ccan/tap/tap.h>
   5
   6 #include <ccan/bytestring/bytestring.c>
   7
   8 #define TEST_STRING     "test string"
   9 #define TEST_STRING_2   "abc\0def"
  10
  11 const char str1[] = TEST_STRING;
  12 const char *str2 = TEST_STRING;
  13
  14 int main(void)
  15 {
  16         struct bytestring bs, bs1, bs2, bs3, bs4, bs5, bs6, bs7;
  17         int n;
  18
  19         /* This is how many tests you plan to run */
  20         plan_tests(126);
  21
  22         bs = bytestring(str1, sizeof(str1) - 1);
  23         ok1(bs.ptr == str1);
  24         ok1(bs.len == (sizeof(str1) - 1));
  25
  26         bs1 = BYTESTRING(TEST_STRING);
  27         ok1(bytestring_eq(bs, bs1));
  28
  29         bs2 = BYTESTRING(TEST_STRING_2);
  30         ok1(bs2.len == 7);
  31
  32         bs3 = bytestring_from_string(str2);
  33         ok1(bytestring_eq(bs3, bs));
  34
  35         bs4 = bytestring_from_string(TEST_STRING_2);
  36         ok1(bs4.len == 3);
  37
  38         bs5 = bytestring_from_string(NULL);
  39         ok1(bs5.len == 0);
  40         ok1(bs5.ptr == NULL);
  41         ok1(bytestring_eq(bs5, bytestring_NULL));
  42
  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');
  50
  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));
  57
  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")));
  66
  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);
  73
  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);
  80
  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)));
  93
  94
  95         ok1(bytestring_spn(bs1, BYTESTRING("est")) == 4);
  96         ok1(bytestring_cspn(bs1, BYTESTRING(" ")) == 4);
  97
  98         ok1(bytestring_spn(bs2, BYTESTRING("z")) == 0);
  99         ok1(bytestring_cspn(bs2, BYTESTRING("\0")) == 3);
 100
 101         ok1(bytestring_spn(bs1, BYTESTRING("eginrst ")) == bs1.len);
 102         ok1(bytestring_cspn(bs2, BYTESTRING("z")) == bs2.len);
 103
 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);
 111         ok1(!bs7.ptr);
 112         bs7 = bytestring_splitchr_next(bs, ' ', bs7);
 113         ok1(!bs7.ptr);
 114
 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);
 122         ok1(!bs7.ptr);
 123         bs7 = bytestring_splitchr_next(bs2, ' ', bs7);
 124         ok1(!bs7.ptr);
 125
 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);
 136         ok1(!bs7.ptr);
 137         bs7 = bytestring_splitchr_next(bs, ' ', bs7);
 138         ok1(!bs7.ptr);
 139
 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);
 145         ok1(bs7.len == 0);
 146         bs7 = bytestring_splitchr_next(bs2, 'f', bs7);
 147         ok1(!bs7.ptr);
 148         bs7 = bytestring_splitchr_next(bs2, 'f', bs7);
 149         ok1(!bs7.ptr);
 150
 151         bs7 = bytestring_splitchr_first(BYTESTRING(""), 'q');
 152         ok1(!bs7.ptr);
 153
 154         n = 0;
 155         bytestring_foreach_splitchr(bs7, BYTESTRING("  "), ' ') {
 156                 n++;
 157                 ok1(bs7.ptr && !bs7.len);
 158         }
 159         ok1(n == 3);
 160
 161         n = 0;
 162         bytestring_foreach_splitchr(bs7, BYTESTRING("\0\0\0"), '\0') {
 163                 n++;
 164                 ok1(bs7.ptr && !bs7.len);
 165         }
 166         ok1(n == 4);
 167
 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);
 175         ok1(!bs7.ptr);
 176         bs7 = bytestring_splitchrs_next(bs, BYTESTRING(" \0"), bs7);
 177         ok1(!bs7.ptr);
 178
 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);
 186         ok1(!bs7.ptr);
 187         bs7 = bytestring_splitchrs_next(bs2, BYTESTRING(" \0"), bs7);
 188         ok1(!bs7.ptr);
 189
 190         bs7 = bytestring_splitchrs_first(BYTESTRING(""), BYTESTRING(" \0"));
 191         ok1(!bs7.ptr);
 192
 193         n = 0;
 194         bytestring_foreach_splitchrs(bs7, BYTESTRING(" \0 \0 "),
 195                                      BYTESTRING("\0 ")) {
 196                 n++;
 197                 ok1(bs7.ptr && !bs7.len);
 198         }
 199         ok1(n == 6);
 200
 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);
 208         ok1(!bs7.ptr);
 209         bs7 = bytestring_splitstr_next(bs, BYTESTRING("t "), bs7);
 210         ok1(!bs7.ptr);
 211
 212         bs7 = bytestring_splitstr_first(bs2, BYTESTRING("abc"));
 213         ok1(bs7.ptr == bs2.ptr);
 214         ok1(bs7.len == 0);
 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);
 219         ok1(!bs7.ptr);
 220         bs7 = bytestring_splitstr_next(bs2, BYTESTRING("t "), bs7);
 221         ok1(!bs7.ptr);
 222
 223         bs7 = bytestring_splitstr_first(BYTESTRING(""), BYTESTRING(""));
 224         ok1(!bs7.ptr);
 225
 226         n = 0;
 227         bytestring_foreach_splitstr(bs7, BYTESTRING("foofoo"),
 228                                      BYTESTRING("foo")) {
 229                 n++;
 230                 ok1(bs7.ptr && !bs7.len);
 231         }
 232         ok1(n == 3);
 233
 234         /* This exits depending on whether all tests passed */
 235         return exit_status();
 236 }