Support numeric types as targets for folly::split
[folly.git] / folly / test / StringTest.cpp
1 /*
2  * Copyright 2014 Facebook, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *   http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include "folly/String.h"
18
19 #include <random>
20 #include <boost/algorithm/string.hpp>
21 #include <gtest/gtest.h>
22
23 #include "folly/Benchmark.h"
24
25 using namespace folly;
26 using namespace std;
27
28 TEST(StringPrintf, BasicTest) {
29   EXPECT_EQ("abc", stringPrintf("%s", "abc"));
30   EXPECT_EQ("abc", stringPrintf("%sbc", "a"));
31   EXPECT_EQ("abc", stringPrintf("a%sc", "b"));
32   EXPECT_EQ("abc", stringPrintf("ab%s", "c"));
33
34   EXPECT_EQ("abc", stringPrintf("abc"));
35 }
36
37 TEST(StringPrintf, NumericFormats) {
38   EXPECT_EQ("12", stringPrintf("%d", 12));
39   EXPECT_EQ("5000000000", stringPrintf("%ld", 5000000000UL));
40   EXPECT_EQ("5000000000", stringPrintf("%ld", 5000000000L));
41   EXPECT_EQ("-5000000000", stringPrintf("%ld", -5000000000L));
42   EXPECT_EQ("-1", stringPrintf("%d", 0xffffffff));
43   EXPECT_EQ("-1", stringPrintf("%ld", 0xffffffffffffffff));
44   EXPECT_EQ("-1", stringPrintf("%ld", 0xffffffffffffffffUL));
45
46   EXPECT_EQ("7.7", stringPrintf("%1.1f", 7.7));
47   EXPECT_EQ("7.7", stringPrintf("%1.1lf", 7.7));
48   EXPECT_EQ("7.70000000000000018",
49             stringPrintf("%.17f", 7.7));
50   EXPECT_EQ("7.70000000000000018",
51             stringPrintf("%.17lf", 7.7));
52 }
53
54 TEST(StringPrintf, Appending) {
55   string s;
56   stringAppendf(&s, "a%s", "b");
57   stringAppendf(&s, "%c", 'c');
58   EXPECT_EQ(s, "abc");
59   stringAppendf(&s, " %d", 123);
60   EXPECT_EQ(s, "abc 123");
61 }
62
63 TEST(StringPrintf, VariousSizes) {
64   // Test a wide variety of output sizes
65   for (int i = 0; i < 100; ++i) {
66     string expected(i + 1, 'a');
67     EXPECT_EQ("X" + expected + "X", stringPrintf("X%sX", expected.c_str()));
68   }
69
70   EXPECT_EQ("abc12345678910111213141516171819202122232425xyz",
71             stringPrintf("abc%d%d%d%d%d%d%d%d%d%d%d%d%d%d"
72                          "%d%d%d%d%d%d%d%d%d%d%dxyz",
73                          1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
74                          17, 18, 19, 20, 21, 22, 23, 24, 25));
75 }
76
77 TEST(StringPrintf, oldStringPrintfTests) {
78   EXPECT_EQ(string("a/b/c/d"),
79             stringPrintf("%s/%s/%s/%s", "a", "b", "c", "d"));
80
81   EXPECT_EQ(string("    5    10"),
82             stringPrintf("%5d %5d", 5, 10));
83
84   // check printing w/ a big buffer
85   for (int size = (1 << 8); size <= (1 << 15); size <<= 1) {
86     string a(size, 'z');
87     string b = stringPrintf("%s", a.c_str());
88     EXPECT_EQ(a.size(), b.size());
89   }
90 }
91
92 TEST(StringPrintf, oldStringAppendf) {
93   string s = "hello";
94   stringAppendf(&s, "%s/%s/%s/%s", "a", "b", "c", "d");
95   EXPECT_EQ(string("helloa/b/c/d"), s);
96 }
97
98 BENCHMARK(new_stringPrintfSmall, iters) {
99   for (int64_t i = 0; i < iters; ++i) {
100     int32_t x = int32_t(i);
101     int32_t y = int32_t(i + 1);
102     string s =
103       stringPrintf("msg msg msg msg msg msg msg msg:  %d, %d, %s",
104                    x, y, "hello");
105   }
106 }
107
108 TEST(Escape, cEscape) {
109   EXPECT_EQ("hello world", cEscape<std::string>("hello world"));
110   EXPECT_EQ("hello \\\\world\\\" goodbye",
111             cEscape<std::string>("hello \\world\" goodbye"));
112   EXPECT_EQ("hello\\nworld", cEscape<std::string>("hello\nworld"));
113   EXPECT_EQ("hello\\377\\376", cEscape<std::string>("hello\xff\xfe"));
114 }
115
116 TEST(Escape, cUnescape) {
117   EXPECT_EQ("hello world", cUnescape<std::string>("hello world"));
118   EXPECT_EQ("hello \\world\" goodbye",
119             cUnescape<std::string>("hello \\\\world\\\" goodbye"));
120   EXPECT_EQ("hello\nworld", cUnescape<std::string>("hello\\nworld"));
121   EXPECT_EQ("hello\nworld", cUnescape<std::string>("hello\\012world"));
122   EXPECT_EQ("hello\nworld", cUnescape<std::string>("hello\\x0aworld"));
123   EXPECT_EQ("hello\xff\xfe", cUnescape<std::string>("hello\\377\\376"));
124   EXPECT_EQ("hello\xff\xfe", cUnescape<std::string>("hello\\xff\\xfe"));
125
126   EXPECT_THROW({cUnescape<std::string>("hello\\");},
127                std::invalid_argument);
128   EXPECT_THROW({cUnescape<std::string>("hello\\x");},
129                std::invalid_argument);
130   EXPECT_THROW({cUnescape<std::string>("hello\\q");},
131                std::invalid_argument);
132 }
133
134 TEST(Escape, uriEscape) {
135   EXPECT_EQ("hello%2c%20%2fworld", uriEscape<std::string>("hello, /world"));
136   EXPECT_EQ("hello%2c%20/world", uriEscape<std::string>("hello, /world",
137                                                         UriEscapeMode::PATH));
138   EXPECT_EQ("hello%2c+%2fworld", uriEscape<std::string>("hello, /world",
139                                                         UriEscapeMode::QUERY));
140 }
141
142 TEST(Escape, uriUnescape) {
143   EXPECT_EQ("hello, /world", uriUnescape<std::string>("hello, /world"));
144   EXPECT_EQ("hello, /world", uriUnescape<std::string>("hello%2c%20%2fworld"));
145   EXPECT_EQ("hello,+/world", uriUnescape<std::string>("hello%2c+%2fworld"));
146   EXPECT_EQ("hello, /world", uriUnescape<std::string>("hello%2c+%2fworld",
147                                                       UriEscapeMode::QUERY));
148   EXPECT_EQ("hello/", uriUnescape<std::string>("hello%2f"));
149   EXPECT_EQ("hello/", uriUnescape<std::string>("hello%2F"));
150   EXPECT_THROW({uriUnescape<std::string>("hello%");},
151                std::invalid_argument);
152   EXPECT_THROW({uriUnescape<std::string>("hello%2");},
153                std::invalid_argument);
154   EXPECT_THROW({uriUnescape<std::string>("hello%2g");},
155                std::invalid_argument);
156 }
157
158 namespace {
159 void expectPrintable(StringPiece s) {
160   for (char c : s) {
161     EXPECT_LE(32, c);
162     EXPECT_GE(127, c);
163   }
164 }
165 }  // namespace
166
167 TEST(Escape, uriEscapeAllCombinations) {
168   char c[3];
169   c[2] = '\0';
170   StringPiece in(c, 2);
171   fbstring tmp;
172   fbstring out;
173   for (int i = 0; i < 256; ++i) {
174     c[0] = i;
175     for (int j = 0; j < 256; ++j) {
176       c[1] = j;
177       tmp.clear();
178       out.clear();
179       uriEscape(in, tmp);
180       expectPrintable(tmp);
181       uriUnescape(tmp, out);
182       EXPECT_EQ(in, out);
183     }
184   }
185 }
186
187 namespace {
188 bool isHex(int v) {
189   return ((v >= '0' && v <= '9') ||
190           (v >= 'A' && v <= 'F') ||
191           (v >= 'a' && v <= 'f'));
192 }
193 }  // namespace
194
195 TEST(Escape, uriUnescapePercentDecoding) {
196   char c[4] = {'%', '\0', '\0', '\0'};
197   StringPiece in(c, 3);
198   fbstring out;
199   unsigned int expected = 0;
200   for (int i = 0; i < 256; ++i) {
201     c[1] = i;
202     for (int j = 0; j < 256; ++j) {
203       c[2] = j;
204       if (isHex(i) && isHex(j)) {
205         out.clear();
206         uriUnescape(in, out);
207         EXPECT_EQ(1, out.size());
208         EXPECT_EQ(1, sscanf(c + 1, "%x", &expected));
209         unsigned char v = out[0];
210         EXPECT_EQ(expected, v);
211       } else {
212         EXPECT_THROW({uriUnescape(in, out);}, std::invalid_argument);
213       }
214     }
215   }
216 }
217
218 namespace {
219 fbstring cbmString;
220 fbstring cbmEscapedString;
221 fbstring cEscapedString;
222 fbstring cUnescapedString;
223 const size_t kCBmStringLength = 64 << 10;
224 const uint32_t kCPrintablePercentage = 90;
225
226 fbstring uribmString;
227 fbstring uribmEscapedString;
228 fbstring uriEscapedString;
229 fbstring uriUnescapedString;
230 const size_t kURIBmStringLength = 256;
231 const uint32_t kURIPassThroughPercentage = 50;
232
233 void initBenchmark() {
234   std::mt19937 rnd;
235
236   // C escape
237   std::uniform_int_distribution<uint32_t> printable(32, 126);
238   std::uniform_int_distribution<uint32_t> nonPrintable(0, 160);
239   std::uniform_int_distribution<uint32_t> percentage(0, 99);
240
241   cbmString.reserve(kCBmStringLength);
242   for (size_t i = 0; i < kCBmStringLength; ++i) {
243     unsigned char c;
244     if (percentage(rnd) < kCPrintablePercentage) {
245       c = printable(rnd);
246     } else {
247       c = nonPrintable(rnd);
248       // Generate characters in both non-printable ranges:
249       // 0..31 and 127..255
250       if (c >= 32) {
251         c += (126 - 32) + 1;
252       }
253     }
254     cbmString.push_back(c);
255   }
256
257   cbmEscapedString = cEscape<fbstring>(cbmString);
258
259   // URI escape
260   std::uniform_int_distribution<uint32_t> passthrough('a', 'z');
261   std::string encodeChars = " ?!\"',+[]";
262   std::uniform_int_distribution<uint32_t> encode(0, encodeChars.size() - 1);
263
264   uribmString.reserve(kURIBmStringLength);
265   for (size_t i = 0; i < kURIBmStringLength; ++i) {
266     unsigned char c;
267     if (percentage(rnd) < kURIPassThroughPercentage) {
268       c = passthrough(rnd);
269     } else {
270       c = encodeChars[encode(rnd)];
271     }
272     uribmString.push_back(c);
273   }
274
275   uribmEscapedString = uriEscape<fbstring>(uribmString);
276 }
277
278 BENCHMARK(BM_cEscape, iters) {
279   while (iters--) {
280     cEscapedString = cEscape<fbstring>(cbmString);
281     doNotOptimizeAway(cEscapedString.size());
282   }
283 }
284
285 BENCHMARK(BM_cUnescape, iters) {
286   while (iters--) {
287     cUnescapedString = cUnescape<fbstring>(cbmEscapedString);
288     doNotOptimizeAway(cUnescapedString.size());
289   }
290 }
291
292 BENCHMARK(BM_uriEscape, iters) {
293   while (iters--) {
294     uriEscapedString = uriEscape<fbstring>(uribmString);
295     doNotOptimizeAway(uriEscapedString.size());
296   }
297 }
298
299 BENCHMARK(BM_uriUnescape, iters) {
300   while (iters--) {
301     uriUnescapedString = uriUnescape<fbstring>(uribmEscapedString);
302     doNotOptimizeAway(uriUnescapedString.size());
303   }
304 }
305
306 }  // namespace
307
308 namespace {
309
310 double pow2(int exponent) {
311   return double(int64_t(1) << exponent);
312 }
313
314 }  // namespace
315
316 TEST(PrettyPrint, Basic) {
317   // check time printing
318   EXPECT_EQ(string("8.53e+07 s "), prettyPrint(85.3e6, PRETTY_TIME));
319   EXPECT_EQ(string("85.3 s "), prettyPrint(85.3, PRETTY_TIME));
320   EXPECT_EQ(string("85.3 ms"), prettyPrint(85.3e-3, PRETTY_TIME));
321   EXPECT_EQ(string("85.3 us"), prettyPrint(85.3e-6, PRETTY_TIME));
322   EXPECT_EQ(string("85.3 ns"), prettyPrint(85.3e-9, PRETTY_TIME));
323   EXPECT_EQ(string("85.3 ps"), prettyPrint(85.3e-12, PRETTY_TIME));
324   EXPECT_EQ(string("8.53e-14 s "), prettyPrint(85.3e-15, PRETTY_TIME));
325
326   EXPECT_EQ(string("0 s "), prettyPrint(0, PRETTY_TIME));
327   EXPECT_EQ(string("1 s "), prettyPrint(1.0, PRETTY_TIME));
328   EXPECT_EQ(string("1 ms"), prettyPrint(1.0e-3, PRETTY_TIME));
329   EXPECT_EQ(string("1 us"), prettyPrint(1.0e-6, PRETTY_TIME));
330   EXPECT_EQ(string("1 ns"), prettyPrint(1.0e-9, PRETTY_TIME));
331   EXPECT_EQ(string("1 ps"), prettyPrint(1.0e-12, PRETTY_TIME));
332
333   // check bytes printing
334   EXPECT_EQ(string("853 B "), prettyPrint(853., PRETTY_BYTES));
335   EXPECT_EQ(string("833 kB"), prettyPrint(853.e3, PRETTY_BYTES));
336   EXPECT_EQ(string("813.5 MB"), prettyPrint(853.e6, PRETTY_BYTES));
337   EXPECT_EQ(string("7.944 GB"), prettyPrint(8.53e9, PRETTY_BYTES));
338   EXPECT_EQ(string("794.4 GB"), prettyPrint(853.e9, PRETTY_BYTES));
339   EXPECT_EQ(string("775.8 TB"), prettyPrint(853.e12, PRETTY_BYTES));
340
341   EXPECT_EQ(string("0 B "), prettyPrint(0, PRETTY_BYTES));
342   EXPECT_EQ(string("1 B "), prettyPrint(pow2(0), PRETTY_BYTES));
343   EXPECT_EQ(string("1 kB"), prettyPrint(pow2(10), PRETTY_BYTES));
344   EXPECT_EQ(string("1 MB"), prettyPrint(pow2(20), PRETTY_BYTES));
345   EXPECT_EQ(string("1 GB"), prettyPrint(pow2(30), PRETTY_BYTES));
346   EXPECT_EQ(string("1 TB"), prettyPrint(pow2(40), PRETTY_BYTES));
347
348   EXPECT_EQ(string("853 B  "), prettyPrint(853., PRETTY_BYTES_IEC));
349   EXPECT_EQ(string("833 KiB"), prettyPrint(853.e3, PRETTY_BYTES_IEC));
350   EXPECT_EQ(string("813.5 MiB"), prettyPrint(853.e6, PRETTY_BYTES_IEC));
351   EXPECT_EQ(string("7.944 GiB"), prettyPrint(8.53e9, PRETTY_BYTES_IEC));
352   EXPECT_EQ(string("794.4 GiB"), prettyPrint(853.e9, PRETTY_BYTES_IEC));
353   EXPECT_EQ(string("775.8 TiB"), prettyPrint(853.e12, PRETTY_BYTES_IEC));
354
355   EXPECT_EQ(string("0 B  "), prettyPrint(0, PRETTY_BYTES_IEC));
356   EXPECT_EQ(string("1 B  "), prettyPrint(pow2(0), PRETTY_BYTES_IEC));
357   EXPECT_EQ(string("1 KiB"), prettyPrint(pow2(10), PRETTY_BYTES_IEC));
358   EXPECT_EQ(string("1 MiB"), prettyPrint(pow2(20), PRETTY_BYTES_IEC));
359   EXPECT_EQ(string("1 GiB"), prettyPrint(pow2(30), PRETTY_BYTES_IEC));
360   EXPECT_EQ(string("1 TiB"), prettyPrint(pow2(40), PRETTY_BYTES_IEC));
361
362   // check bytes metric printing
363   EXPECT_EQ(string("853 B "), prettyPrint(853., PRETTY_BYTES_METRIC));
364   EXPECT_EQ(string("853 kB"), prettyPrint(853.e3, PRETTY_BYTES_METRIC));
365   EXPECT_EQ(string("853 MB"), prettyPrint(853.e6, PRETTY_BYTES_METRIC));
366   EXPECT_EQ(string("8.53 GB"), prettyPrint(8.53e9, PRETTY_BYTES_METRIC));
367   EXPECT_EQ(string("853 GB"), prettyPrint(853.e9, PRETTY_BYTES_METRIC));
368   EXPECT_EQ(string("853 TB"), prettyPrint(853.e12, PRETTY_BYTES_METRIC));
369
370   EXPECT_EQ(string("0 B "), prettyPrint(0, PRETTY_BYTES_METRIC));
371   EXPECT_EQ(string("1 B "), prettyPrint(1.0, PRETTY_BYTES_METRIC));
372   EXPECT_EQ(string("1 kB"), prettyPrint(1.0e+3, PRETTY_BYTES_METRIC));
373   EXPECT_EQ(string("1 MB"), prettyPrint(1.0e+6, PRETTY_BYTES_METRIC));
374
375   EXPECT_EQ(string("1 GB"), prettyPrint(1.0e+9, PRETTY_BYTES_METRIC));
376   EXPECT_EQ(string("1 TB"), prettyPrint(1.0e+12, PRETTY_BYTES_METRIC));
377
378   // check metric-units (powers of 1000) printing
379   EXPECT_EQ(string("853  "), prettyPrint(853., PRETTY_UNITS_METRIC));
380   EXPECT_EQ(string("853 k"), prettyPrint(853.e3, PRETTY_UNITS_METRIC));
381   EXPECT_EQ(string("853 M"), prettyPrint(853.e6, PRETTY_UNITS_METRIC));
382   EXPECT_EQ(string("8.53 bil"), prettyPrint(8.53e9, PRETTY_UNITS_METRIC));
383   EXPECT_EQ(string("853 bil"), prettyPrint(853.e9, PRETTY_UNITS_METRIC));
384   EXPECT_EQ(string("853 tril"), prettyPrint(853.e12, PRETTY_UNITS_METRIC));
385
386   // check binary-units (powers of 1024) printing
387   EXPECT_EQ(string("0  "), prettyPrint(0, PRETTY_UNITS_BINARY));
388   EXPECT_EQ(string("1  "), prettyPrint(pow2(0), PRETTY_UNITS_BINARY));
389   EXPECT_EQ(string("1 k"), prettyPrint(pow2(10), PRETTY_UNITS_BINARY));
390   EXPECT_EQ(string("1 M"), prettyPrint(pow2(20), PRETTY_UNITS_BINARY));
391   EXPECT_EQ(string("1 G"), prettyPrint(pow2(30), PRETTY_UNITS_BINARY));
392   EXPECT_EQ(string("1 T"), prettyPrint(pow2(40), PRETTY_UNITS_BINARY));
393
394   EXPECT_EQ(string("1023  "),
395       prettyPrint(pow2(10) - 1, PRETTY_UNITS_BINARY));
396   EXPECT_EQ(string("1024 k"),
397       prettyPrint(pow2(20) - 1, PRETTY_UNITS_BINARY));
398   EXPECT_EQ(string("1024 M"),
399       prettyPrint(pow2(30) - 1, PRETTY_UNITS_BINARY));
400   EXPECT_EQ(string("1024 G"),
401       prettyPrint(pow2(40) - 1, PRETTY_UNITS_BINARY));
402
403   EXPECT_EQ(string("0   "), prettyPrint(0, PRETTY_UNITS_BINARY_IEC));
404   EXPECT_EQ(string("1   "), prettyPrint(pow2(0), PRETTY_UNITS_BINARY_IEC));
405   EXPECT_EQ(string("1 Ki"), prettyPrint(pow2(10), PRETTY_UNITS_BINARY_IEC));
406   EXPECT_EQ(string("1 Mi"), prettyPrint(pow2(20), PRETTY_UNITS_BINARY_IEC));
407   EXPECT_EQ(string("1 Gi"), prettyPrint(pow2(30), PRETTY_UNITS_BINARY_IEC));
408   EXPECT_EQ(string("1 Ti"), prettyPrint(pow2(40), PRETTY_UNITS_BINARY_IEC));
409
410   EXPECT_EQ(string("1023   "),
411       prettyPrint(pow2(10) - 1, PRETTY_UNITS_BINARY_IEC));
412   EXPECT_EQ(string("1024 Ki"),
413       prettyPrint(pow2(20) - 1, PRETTY_UNITS_BINARY_IEC));
414   EXPECT_EQ(string("1024 Mi"),
415       prettyPrint(pow2(30) - 1, PRETTY_UNITS_BINARY_IEC));
416   EXPECT_EQ(string("1024 Gi"),
417       prettyPrint(pow2(40) - 1, PRETTY_UNITS_BINARY_IEC));
418
419   // check that negative values work
420   EXPECT_EQ(string("-85.3 s "), prettyPrint(-85.3, PRETTY_TIME));
421   EXPECT_EQ(string("-85.3 ms"), prettyPrint(-85.3e-3, PRETTY_TIME));
422   EXPECT_EQ(string("-85.3 us"), prettyPrint(-85.3e-6, PRETTY_TIME));
423   EXPECT_EQ(string("-85.3 ns"), prettyPrint(-85.3e-9, PRETTY_TIME));
424 }
425
426 TEST(PrettyPrint, HexDump) {
427   std::string a("abc\x00\x02\xa0", 6);  // embedded NUL
428   EXPECT_EQ(
429     "00000000  61 62 63 00 02 a0                                 "
430     "|abc...          |\n",
431     hexDump(a.data(), a.size()));
432
433   a = "abcdefghijklmnopqrstuvwxyz";
434   EXPECT_EQ(
435     "00000000  61 62 63 64 65 66 67 68  69 6a 6b 6c 6d 6e 6f 70  "
436     "|abcdefghijklmnop|\n"
437     "00000010  71 72 73 74 75 76 77 78  79 7a                    "
438     "|qrstuvwxyz      |\n",
439     hexDump(a.data(), a.size()));
440 }
441
442 TEST(System, errnoStr) {
443   errno = EACCES;
444   EXPECT_EQ(EACCES, errno);
445   EXPECT_EQ(EACCES, errno);  // twice to make sure EXPECT_EQ doesn't change it
446
447   fbstring expected = strerror(ENOENT);
448
449   errno = EACCES;
450   EXPECT_EQ(expected, errnoStr(ENOENT));
451   // Ensure that errno isn't changed
452   EXPECT_EQ(EACCES, errno);
453
454   // Per POSIX, all errno values are positive, so -1 is invalid
455   errnoStr(-1);
456
457   // Ensure that errno isn't changed
458   EXPECT_EQ(EACCES, errno);
459 }
460
461 namespace folly_test {
462 struct ThisIsAVeryLongStructureName {
463 };
464 }  // namespace folly_test
465
466 #if FOLLY_HAVE_CPLUS_DEMANGLE_V3_CALLBACK
467 TEST(System, demangle) {
468   char expected[] = "folly_test::ThisIsAVeryLongStructureName";
469   EXPECT_STREQ(
470       expected,
471       demangle(typeid(folly_test::ThisIsAVeryLongStructureName)).c_str());
472
473   {
474     char buf[sizeof(expected)];
475     EXPECT_EQ(sizeof(expected) - 1,
476               demangle(typeid(folly_test::ThisIsAVeryLongStructureName),
477                        buf, sizeof(buf)));
478     EXPECT_STREQ(expected, buf);
479
480     EXPECT_EQ(sizeof(expected) - 1,
481               demangle(typeid(folly_test::ThisIsAVeryLongStructureName),
482                        buf, 11));
483     EXPECT_STREQ("folly_test", buf);
484   }
485 }
486 #endif
487
488 namespace {
489
490 template<template<class,class> class VectorType>
491 void splitTest() {
492   VectorType<string,std::allocator<string> > parts;
493
494   folly::split(',', "a,b,c", parts);
495   EXPECT_EQ(parts.size(), 3);
496   EXPECT_EQ(parts[0], "a");
497   EXPECT_EQ(parts[1], "b");
498   EXPECT_EQ(parts[2], "c");
499   parts.clear();
500
501   folly::split(',', StringPiece("a,b,c"), parts);
502   EXPECT_EQ(parts.size(), 3);
503   EXPECT_EQ(parts[0], "a");
504   EXPECT_EQ(parts[1], "b");
505   EXPECT_EQ(parts[2], "c");
506   parts.clear();
507
508   folly::split(',', string("a,b,c"), parts);
509   EXPECT_EQ(parts.size(), 3);
510   EXPECT_EQ(parts[0], "a");
511   EXPECT_EQ(parts[1], "b");
512   EXPECT_EQ(parts[2], "c");
513   parts.clear();
514
515   folly::split(',', "a,,c", parts);
516   EXPECT_EQ(parts.size(), 3);
517   EXPECT_EQ(parts[0], "a");
518   EXPECT_EQ(parts[1], "");
519   EXPECT_EQ(parts[2], "c");
520   parts.clear();
521
522   folly::split(',', string("a,,c"), parts);
523   EXPECT_EQ(parts.size(), 3);
524   EXPECT_EQ(parts[0], "a");
525   EXPECT_EQ(parts[1], "");
526   EXPECT_EQ(parts[2], "c");
527   parts.clear();
528
529   folly::split(',', "a,,c", parts, true);
530   EXPECT_EQ(parts.size(), 2);
531   EXPECT_EQ(parts[0], "a");
532   EXPECT_EQ(parts[1], "c");
533   parts.clear();
534
535   folly::split(',', string("a,,c"), parts, true);
536   EXPECT_EQ(parts.size(), 2);
537   EXPECT_EQ(parts[0], "a");
538   EXPECT_EQ(parts[1], "c");
539   parts.clear();
540
541   folly::split(',', string(",,a,,c,,,"), parts, true);
542   EXPECT_EQ(parts.size(), 2);
543   EXPECT_EQ(parts[0], "a");
544   EXPECT_EQ(parts[1], "c");
545   parts.clear();
546
547   // test multiple split w/o clear
548   folly::split(',', ",,a,,c,,,", parts, true);
549   EXPECT_EQ(parts.size(), 2);
550   EXPECT_EQ(parts[0], "a");
551   EXPECT_EQ(parts[1], "c");
552   folly::split(',', ",,a,,c,,,", parts, true);
553   EXPECT_EQ(parts.size(), 4);
554   EXPECT_EQ(parts[2], "a");
555   EXPECT_EQ(parts[3], "c");
556   parts.clear();
557
558   // test splits that with multi-line delimiter
559   folly::split("ab", "dabcabkdbkab", parts, true);
560   EXPECT_EQ(parts.size(), 3);
561   EXPECT_EQ(parts[0], "d");
562   EXPECT_EQ(parts[1], "c");
563   EXPECT_EQ(parts[2], "kdbk");
564   parts.clear();
565
566   string orig = "ab2342asdfv~~!";
567   folly::split("", orig, parts, true);
568   EXPECT_EQ(parts.size(), 1);
569   EXPECT_EQ(parts[0], orig);
570   parts.clear();
571
572   folly::split("452x;o38asfsajsdlfdf.j", "asfds", parts, true);
573   EXPECT_EQ(parts.size(), 1);
574   EXPECT_EQ(parts[0], "asfds");
575   parts.clear();
576
577   folly::split("a", "", parts, true);
578   EXPECT_EQ(parts.size(), 0);
579   parts.clear();
580
581   folly::split("a", "", parts);
582   EXPECT_EQ(parts.size(), 1);
583   EXPECT_EQ(parts[0], "");
584   parts.clear();
585
586   folly::split("a", StringPiece(), parts, true);
587   EXPECT_EQ(parts.size(), 0);
588   parts.clear();
589
590   folly::split("a", StringPiece(), parts);
591   EXPECT_EQ(parts.size(), 1);
592   EXPECT_EQ(parts[0], "");
593   parts.clear();
594
595   folly::split("a", "abcdefg", parts, true);
596   EXPECT_EQ(parts.size(), 1);
597   EXPECT_EQ(parts[0], "bcdefg");
598   parts.clear();
599
600   orig = "All, , your base, are , , belong to us";
601   folly::split(", ", orig, parts, true);
602   EXPECT_EQ(parts.size(), 4);
603   EXPECT_EQ(parts[0], "All");
604   EXPECT_EQ(parts[1], "your base");
605   EXPECT_EQ(parts[2], "are ");
606   EXPECT_EQ(parts[3], "belong to us");
607   parts.clear();
608   folly::split(", ", orig, parts);
609   EXPECT_EQ(parts.size(), 6);
610   EXPECT_EQ(parts[0], "All");
611   EXPECT_EQ(parts[1], "");
612   EXPECT_EQ(parts[2], "your base");
613   EXPECT_EQ(parts[3], "are ");
614   EXPECT_EQ(parts[4], "");
615   EXPECT_EQ(parts[5], "belong to us");
616   parts.clear();
617
618   orig = ", Facebook, rul,es!, ";
619   folly::split(", ", orig, parts, true);
620   EXPECT_EQ(parts.size(), 2);
621   EXPECT_EQ(parts[0], "Facebook");
622   EXPECT_EQ(parts[1], "rul,es!");
623   parts.clear();
624   folly::split(", ", orig, parts);
625   EXPECT_EQ(parts.size(), 4);
626   EXPECT_EQ(parts[0], "");
627   EXPECT_EQ(parts[1], "Facebook");
628   EXPECT_EQ(parts[2], "rul,es!");
629   EXPECT_EQ(parts[3], "");
630 }
631
632 template<template<class,class> class VectorType>
633 void piecesTest() {
634   VectorType<StringPiece,std::allocator<StringPiece> > pieces;
635   VectorType<StringPiece,std::allocator<StringPiece> > pieces2;
636
637   folly::split(',', "a,b,c", pieces);
638   EXPECT_EQ(pieces.size(), 3);
639   EXPECT_EQ(pieces[0], "a");
640   EXPECT_EQ(pieces[1], "b");
641   EXPECT_EQ(pieces[2], "c");
642
643   pieces.clear();
644
645   folly::split(',', "a,,c", pieces);
646   EXPECT_EQ(pieces.size(), 3);
647   EXPECT_EQ(pieces[0], "a");
648   EXPECT_EQ(pieces[1], "");
649   EXPECT_EQ(pieces[2], "c");
650   pieces.clear();
651
652   folly::split(',', "a,,c", pieces, true);
653   EXPECT_EQ(pieces.size(), 2);
654   EXPECT_EQ(pieces[0], "a");
655   EXPECT_EQ(pieces[1], "c");
656   pieces.clear();
657
658   folly::split(',', ",,a,,c,,,", pieces, true);
659   EXPECT_EQ(pieces.size(), 2);
660   EXPECT_EQ(pieces[0], "a");
661   EXPECT_EQ(pieces[1], "c");
662   pieces.clear();
663
664   // test multiple split w/o clear
665   folly::split(',', ",,a,,c,,,", pieces, true);
666   EXPECT_EQ(pieces.size(), 2);
667   EXPECT_EQ(pieces[0], "a");
668   EXPECT_EQ(pieces[1], "c");
669   folly::split(',', ",,a,,c,,,", pieces, true);
670   EXPECT_EQ(pieces.size(), 4);
671   EXPECT_EQ(pieces[2], "a");
672   EXPECT_EQ(pieces[3], "c");
673   pieces.clear();
674
675   // test multiple split rounds
676   folly::split(",", "a_b,c_d", pieces);
677   EXPECT_EQ(pieces.size(), 2);
678   EXPECT_EQ(pieces[0], "a_b");
679   EXPECT_EQ(pieces[1], "c_d");
680   folly::split("_", pieces[0], pieces2);
681   EXPECT_EQ(pieces2.size(), 2);
682   EXPECT_EQ(pieces2[0], "a");
683   EXPECT_EQ(pieces2[1], "b");
684   pieces2.clear();
685   folly::split("_", pieces[1], pieces2);
686   EXPECT_EQ(pieces2.size(), 2);
687   EXPECT_EQ(pieces2[0], "c");
688   EXPECT_EQ(pieces2[1], "d");
689   pieces.clear();
690   pieces2.clear();
691
692   // test splits that with multi-line delimiter
693   folly::split("ab", "dabcabkdbkab", pieces, true);
694   EXPECT_EQ(pieces.size(), 3);
695   EXPECT_EQ(pieces[0], "d");
696   EXPECT_EQ(pieces[1], "c");
697   EXPECT_EQ(pieces[2], "kdbk");
698   pieces.clear();
699
700   string orig = "ab2342asdfv~~!";
701   folly::split("", orig.c_str(), pieces, true);
702   EXPECT_EQ(pieces.size(), 1);
703   EXPECT_EQ(pieces[0], orig);
704   pieces.clear();
705
706   folly::split("452x;o38asfsajsdlfdf.j", "asfds", pieces, true);
707   EXPECT_EQ(pieces.size(), 1);
708   EXPECT_EQ(pieces[0], "asfds");
709   pieces.clear();
710
711   folly::split("a", "", pieces, true);
712   EXPECT_EQ(pieces.size(), 0);
713   pieces.clear();
714
715   folly::split("a", "", pieces);
716   EXPECT_EQ(pieces.size(), 1);
717   EXPECT_EQ(pieces[0], "");
718   pieces.clear();
719
720   folly::split("a", "abcdefg", pieces, true);
721   EXPECT_EQ(pieces.size(), 1);
722   EXPECT_EQ(pieces[0], "bcdefg");
723   pieces.clear();
724
725   orig = "All, , your base, are , , belong to us";
726   folly::split(", ", orig, pieces, true);
727   EXPECT_EQ(pieces.size(), 4);
728   EXPECT_EQ(pieces[0], "All");
729   EXPECT_EQ(pieces[1], "your base");
730   EXPECT_EQ(pieces[2], "are ");
731   EXPECT_EQ(pieces[3], "belong to us");
732   pieces.clear();
733   folly::split(", ", orig, pieces);
734   EXPECT_EQ(pieces.size(), 6);
735   EXPECT_EQ(pieces[0], "All");
736   EXPECT_EQ(pieces[1], "");
737   EXPECT_EQ(pieces[2], "your base");
738   EXPECT_EQ(pieces[3], "are ");
739   EXPECT_EQ(pieces[4], "");
740   EXPECT_EQ(pieces[5], "belong to us");
741   pieces.clear();
742
743   orig = ", Facebook, rul,es!, ";
744   folly::split(", ", orig, pieces, true);
745   EXPECT_EQ(pieces.size(), 2);
746   EXPECT_EQ(pieces[0], "Facebook");
747   EXPECT_EQ(pieces[1], "rul,es!");
748   pieces.clear();
749   folly::split(", ", orig, pieces);
750   EXPECT_EQ(pieces.size(), 4);
751   EXPECT_EQ(pieces[0], "");
752   EXPECT_EQ(pieces[1], "Facebook");
753   EXPECT_EQ(pieces[2], "rul,es!");
754   EXPECT_EQ(pieces[3], "");
755   pieces.clear();
756
757   const char* str = "a,b";
758   folly::split(',', StringPiece(str), pieces);
759   EXPECT_EQ(pieces.size(), 2);
760   EXPECT_EQ(pieces[0], "a");
761   EXPECT_EQ(pieces[1], "b");
762   EXPECT_EQ(pieces[0].start(), str);
763   EXPECT_EQ(pieces[1].start(), str + 2);
764
765   std::set<StringPiece> unique;
766   folly::splitTo<StringPiece>(":", "asd:bsd:asd:asd:bsd:csd::asd",
767     std::inserter(unique, unique.begin()), true);
768   EXPECT_EQ(unique.size(), 3);
769   if (unique.size() == 3) {
770     EXPECT_EQ(*unique.begin(), "asd");
771     EXPECT_EQ(*--unique.end(), "csd");
772   }
773
774   VectorType<fbstring,std::allocator<fbstring> > blah;
775   folly::split('-', "a-b-c-d-f-e", blah);
776   EXPECT_EQ(blah.size(), 6);
777 }
778
779 }
780
781 TEST(Split, split_vector) {
782   splitTest<std::vector>();
783 }
784 TEST(Split, split_fbvector) {
785   splitTest<folly::fbvector>();
786 }
787 TEST(Split, pieces_vector) {
788   piecesTest<std::vector>();
789 }
790 TEST(Split, pieces_fbvector) {
791   piecesTest<folly::fbvector>();
792 }
793
794 TEST(Split, fixed) {
795   StringPiece a, b, c, d;
796
797   EXPECT_TRUE(folly::split<false>('.', "a.b.c.d", a, b, c, d));
798   EXPECT_TRUE(folly::split<false>('.', "a.b.c", a, b, c));
799   EXPECT_TRUE(folly::split<false>('.', "a.b", a, b));
800   EXPECT_TRUE(folly::split<false>('.', "a", a));
801
802   EXPECT_TRUE(folly::split('.', "a.b.c.d", a, b, c, d));
803   EXPECT_TRUE(folly::split('.', "a.b.c", a, b, c));
804   EXPECT_TRUE(folly::split('.', "a.b", a, b));
805   EXPECT_TRUE(folly::split('.', "a", a));
806
807   EXPECT_TRUE(folly::split<false>('.', "a.b.c", a, b, c));
808   EXPECT_EQ("a", a);
809   EXPECT_EQ("b", b);
810   EXPECT_EQ("c", c);
811   EXPECT_FALSE(folly::split<false>('.', "a.b", a, b, c));
812   EXPECT_TRUE(folly::split<false>('.', "a.b.c", a, b));
813   EXPECT_EQ("a", a);
814   EXPECT_EQ("b.c", b);
815
816   EXPECT_TRUE(folly::split('.', "a.b.c", a, b, c));
817   EXPECT_EQ("a", a);
818   EXPECT_EQ("b", b);
819   EXPECT_EQ("c", c);
820   EXPECT_FALSE(folly::split('.', "a.b.c", a, b));
821   EXPECT_FALSE(folly::split('.', "a.b", a, b, c));
822
823   EXPECT_TRUE(folly::split<false>('.', "a.b", a, b));
824   EXPECT_EQ("a", a);
825   EXPECT_EQ("b", b);
826   EXPECT_FALSE(folly::split<false>('.', "a", a, b));
827   EXPECT_TRUE(folly::split<false>('.', "a.b", a));
828   EXPECT_EQ("a.b", a);
829
830   EXPECT_TRUE(folly::split('.', "a.b", a, b));
831   EXPECT_EQ("a", a);
832   EXPECT_EQ("b", b);
833   EXPECT_FALSE(folly::split('.', "a", a, b));
834   EXPECT_FALSE(folly::split('.', "a.b", a));
835 }
836
837 TEST(Split, fixed_convert) {
838   StringPiece a, d;
839   int b;
840   double c;
841
842   EXPECT_TRUE(folly::split(':', "a:13:14.7:b", a, b, c, d));
843   EXPECT_EQ("a", a);
844   EXPECT_EQ(13, b);
845   EXPECT_NEAR(14.7, c, 1e-10);
846   EXPECT_EQ("b", d);
847
848   EXPECT_TRUE(folly::split<false>(':', "b:14:15.3:c", a, b, c, d));
849   EXPECT_EQ("b", a);
850   EXPECT_EQ(14, b);
851   EXPECT_NEAR(15.3, c, 1e-10);
852   EXPECT_EQ("c", d);
853
854   EXPECT_FALSE(folly::split(':', "a:13:14.7:b", a, b, d));
855
856   EXPECT_TRUE(folly::split<false>(':', "a:13:14.7:b", a, b, d));
857   EXPECT_EQ("a", a);
858   EXPECT_EQ(13, b);
859   EXPECT_EQ("14.7:b", d);
860
861   EXPECT_THROW(folly::split<false>(':', "a:13:14.7:b", a, b, c),
862                std::range_error);
863 }
864
865 TEST(String, join) {
866   string output;
867
868   std::vector<int> empty = { };
869   join(":", empty, output);
870   EXPECT_TRUE(output.empty());
871
872   std::vector<std::string> input1 = { "1", "23", "456", "" };
873   join(':', input1, output);
874   EXPECT_EQ(output, "1:23:456:");
875   output = join(':', input1);
876   EXPECT_EQ(output, "1:23:456:");
877
878   auto input2 = { 1, 23, 456 };
879   join("-*-", input2, output);
880   EXPECT_EQ(output, "1-*-23-*-456");
881   output = join("-*-", input2);
882   EXPECT_EQ(output, "1-*-23-*-456");
883
884   auto input3 = { 'f', 'a', 'c', 'e', 'b', 'o', 'o', 'k' };
885   join("", input3, output);
886   EXPECT_EQ(output, "facebook");
887
888   join("_", { "", "f", "a", "c", "e", "b", "o", "o", "k", "" }, output);
889   EXPECT_EQ(output, "_f_a_c_e_b_o_o_k_");
890 }
891
892 TEST(String, hexlify) {
893   string input1 = "0123";
894   string output1;
895   EXPECT_TRUE(hexlify(input1, output1));
896   EXPECT_EQ(output1, "30313233");
897
898   fbstring input2 = "abcdefg";
899   input2[1] = 0;
900   input2[3] = 0xff;
901   input2[5] = 0xb6;
902   fbstring output2;
903   EXPECT_TRUE(hexlify(input2, output2));
904   EXPECT_EQ(output2, "610063ff65b667");
905 }
906
907 TEST(String, unhexlify) {
908   string input1 = "30313233";
909   string output1;
910   EXPECT_TRUE(unhexlify(input1, output1));
911   EXPECT_EQ(output1, "0123");
912
913   fbstring input2 = "610063ff65b667";
914   fbstring output2;
915   EXPECT_TRUE(unhexlify(input2, output2));
916   EXPECT_EQ(output2.size(), 7);
917   EXPECT_EQ(output2[0], 'a');
918   EXPECT_EQ(output2[1], 0);
919   EXPECT_EQ(output2[2], 'c');
920   EXPECT_EQ(output2[3] & 0xff, 0xff);
921   EXPECT_EQ(output2[4], 'e');
922   EXPECT_EQ(output2[5] & 0xff, 0xb6);
923   EXPECT_EQ(output2[6], 'g');
924
925   string input3 = "x";
926   string output3;
927   EXPECT_FALSE(unhexlify(input3, output3));
928
929   string input4 = "xy";
930   string output4;
931   EXPECT_FALSE(unhexlify(input4, output4));
932 }
933
934 TEST(String, backslashify) {
935   EXPECT_EQ("abc", string("abc"));
936   EXPECT_EQ("abc", backslashify(string("abc")));
937   EXPECT_EQ("abc\\r", backslashify(string("abc\r")));
938   EXPECT_EQ("abc\\x0d", backslashify(string("abc\r"), true));
939   EXPECT_EQ("\\0\\0", backslashify(string(2, '\0')));
940 }
941
942 TEST(String, humanify) {
943   // Simple cases; output is obvious.
944   EXPECT_EQ("abc", humanify(string("abc")));
945   EXPECT_EQ("abc\\\\r", humanify(string("abc\\r")));
946   EXPECT_EQ("0xff", humanify(string("\xff")));
947   EXPECT_EQ("abc\\xff", humanify(string("abc\xff")));
948   EXPECT_EQ("abc\\b", humanify(string("abc\b")));
949   EXPECT_EQ("0x00", humanify(string(1, '\0')));
950   EXPECT_EQ("0x0000", humanify(string(2, '\0')));
951
952
953   // Mostly printable, so backslash!  80, 60, and 40% printable, respectively
954   EXPECT_EQ("aaaa\\xff", humanify(string("aaaa\xff")));
955   EXPECT_EQ("aaa\\xff\\xff", humanify(string("aaa\xff\xff")));
956   EXPECT_EQ("aa\\xff\\xff\\xff", humanify(string("aa\xff\xff\xff")));
957
958   // 20% printable, and the printable portion isn't the prefix; hexify!
959   EXPECT_EQ("0xff61ffffff", humanify(string("\xff" "a\xff\xff\xff")));
960
961   // Same as previous, except swap first two chars; prefix is
962   // printable and within the threshold, so backslashify.
963   EXPECT_EQ("a\\xff\\xff\\xff\\xff", humanify(string("a\xff\xff\xff\xff")));
964
965   // Just too much unprintable; hex, despite prefix.
966   EXPECT_EQ("0x61ffffffffff", humanify(string("a\xff\xff\xff\xff\xff")));
967 }
968
969 //////////////////////////////////////////////////////////////////////
970
971 BENCHMARK(splitOnSingleChar, iters) {
972   static const std::string line = "one:two:three:four";
973   for (int i = 0; i < iters << 4; ++i) {
974     std::vector<StringPiece> pieces;
975     folly::split(':', line, pieces);
976   }
977 }
978
979 BENCHMARK(splitOnSingleCharFixed, iters) {
980   static const std::string line = "one:two:three:four";
981   for (int i = 0; i < iters << 4; ++i) {
982     StringPiece a, b, c, d;
983     folly::split(':', line, a, b, c, d);
984   }
985 }
986
987 BENCHMARK(splitOnSingleCharFixedAllowExtra, iters) {
988   static const std::string line = "one:two:three:four";
989   for (int i = 0; i < iters << 4; ++i) {
990     StringPiece a, b, c, d;
991     folly::split<false>(':', line, a, b, c, d);
992   }
993 }
994
995 BENCHMARK(splitStr, iters) {
996   static const std::string line = "one-*-two-*-three-*-four";
997   for (int i = 0; i < iters << 4; ++i) {
998     std::vector<StringPiece> pieces;
999     folly::split("-*-", line, pieces);
1000   }
1001 }
1002
1003 BENCHMARK(splitStrFixed, iters) {
1004   static const std::string line = "one-*-two-*-three-*-four";
1005   for (int i = 0; i < iters << 4; ++i) {
1006     StringPiece a, b, c, d;
1007     folly::split("-*-", line, a, b, c, d);
1008   }
1009 }
1010
1011 BENCHMARK(boost_splitOnSingleChar, iters) {
1012   static const std::string line = "one:two:three:four";
1013   bool(*pred)(char) = [] (char c) -> bool { return c == ':'; };
1014   for (int i = 0; i < iters << 4; ++i) {
1015     std::vector<boost::iterator_range<std::string::const_iterator> > pieces;
1016     boost::split(pieces, line, pred);
1017   }
1018 }
1019
1020 BENCHMARK(joinCharStr, iters) {
1021   static const std::vector<std::string> input = {
1022     "one", "two", "three", "four", "five", "six", "seven" };
1023   for (int i = 0; i < iters << 4; ++i) {
1024     std::string output;
1025     folly::join(':', input, output);
1026   }
1027 }
1028
1029 BENCHMARK(joinStrStr, iters) {
1030   static const std::vector<std::string> input = {
1031     "one", "two", "three", "four", "five", "six", "seven" };
1032   for (int i = 0; i < iters << 4; ++i) {
1033     std::string output;
1034     folly::join(":", input, output);
1035   }
1036 }
1037
1038 BENCHMARK(joinInt, iters) {
1039   static const auto input = {
1040     123, 456, 78910, 1112, 1314, 151, 61718 };
1041   for (int i = 0; i < iters << 4; ++i) {
1042     std::string output;
1043     folly::join(":", input, output);
1044   }
1045 }
1046
1047 int main(int argc, char *argv[]) {
1048   testing::InitGoogleTest(&argc, argv);
1049   google::ParseCommandLineFlags(&argc, &argv, true);
1050   auto ret = RUN_ALL_TESTS();
1051   if (!ret) {
1052     initBenchmark();
1053     if (FLAGS_benchmark) {
1054       folly::runBenchmarks();
1055     }
1056   }
1057   return ret;
1058 }