2 * Copyright 2016 Facebook, Inc.
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 #include <folly/String.h>
19 #include <boost/regex.hpp>
20 #include <gtest/gtest.h>
22 using namespace folly;
25 TEST(StringPrintf, BasicTest) {
26 EXPECT_EQ("abc", stringPrintf("%s", "abc"));
27 EXPECT_EQ("abc", stringPrintf("%sbc", "a"));
28 EXPECT_EQ("abc", stringPrintf("a%sc", "b"));
29 EXPECT_EQ("abc", stringPrintf("ab%s", "c"));
31 EXPECT_EQ("abc", stringPrintf("abc"));
34 TEST(StringPrintf, NumericFormats) {
35 EXPECT_EQ("12", stringPrintf("%d", 12));
36 EXPECT_EQ("2000000000", stringPrintf("%ld", 2000000000ULL));
37 EXPECT_EQ("2000000000", stringPrintf("%ld", 2000000000L));
38 EXPECT_EQ("-2000000000", stringPrintf("%ld", -2000000000L));
39 EXPECT_EQ("5000000000", stringPrintf("%lld", 5000000000ULL));
40 EXPECT_EQ("5000000000", stringPrintf("%lld", 5000000000LL));
41 EXPECT_EQ("-5000000000", stringPrintf("%lld", -5000000000LL));
42 EXPECT_EQ("-1", stringPrintf("%d", 0xffffffff));
43 EXPECT_EQ("-1", stringPrintf("%ld", 0xffffffffffffffff));
44 EXPECT_EQ("-1", stringPrintf("%ld", 0xffffffffffffffffUL));
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));
54 TEST(StringPrintf, Appending) {
56 stringAppendf(&s, "a%s", "b");
57 stringAppendf(&s, "%c", 'c');
59 stringAppendf(&s, " %d", 123);
60 EXPECT_EQ(s, "abc 123");
63 void vprintfCheck(const char* expected, const char* fmt, ...) {
65 va_start(apOrig, fmt);
75 // Check both APIs for calling stringVPrintf()
76 EXPECT_EQ(expected, stringVPrintf(fmt, ap));
81 stringVPrintf(&out, fmt, ap);
84 EXPECT_EQ(expected, out);
86 // Check stringVAppendf() as well
87 std::string prefix = "foobar";
89 EXPECT_EQ(prefix + expected, stringVAppendf(&out, fmt, ap));
94 void vprintfError(const char* fmt, ...) {
101 // OSX's sprintf family does not return a negative number on a bad format
102 // string, but Linux does. It's unclear to me which behavior is more
104 #ifdef HAVE_VSNPRINTF_ERRORS
105 EXPECT_THROW({stringVPrintf(fmt, ap);},
110 TEST(StringPrintf, VPrintf) {
111 vprintfCheck("foo", "%s", "foo");
112 vprintfCheck("long string requiring reallocation 1 2 3 0x12345678",
113 "%s %s %d %d %d %#x",
114 "long string", "requiring reallocation", 1, 2, 3, 0x12345678);
115 vprintfError("bogus%", "foo");
118 TEST(StringPrintf, VariousSizes) {
119 // Test a wide variety of output sizes, making sure to cross the
120 // vsnprintf buffer boundary implementation detail.
121 for (int i = 0; i < 4096; ++i) {
122 string expected(i + 1, 'a');
123 expected = "X" + expected + "X";
124 string result = stringPrintf("%s", expected.c_str());
125 EXPECT_EQ(expected.size(), result.size());
126 EXPECT_EQ(expected, result);
129 EXPECT_EQ("abc12345678910111213141516171819202122232425xyz",
130 stringPrintf("abc%d%d%d%d%d%d%d%d%d%d%d%d%d%d"
131 "%d%d%d%d%d%d%d%d%d%d%dxyz",
132 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
133 17, 18, 19, 20, 21, 22, 23, 24, 25));
136 TEST(StringPrintf, oldStringPrintfTests) {
137 EXPECT_EQ(string("a/b/c/d"),
138 stringPrintf("%s/%s/%s/%s", "a", "b", "c", "d"));
140 EXPECT_EQ(string(" 5 10"),
141 stringPrintf("%5d %5d", 5, 10));
143 // check printing w/ a big buffer
144 for (int size = (1 << 8); size <= (1 << 15); size <<= 1) {
146 string b = stringPrintf("%s", a.c_str());
147 EXPECT_EQ(a.size(), b.size());
151 TEST(StringPrintf, oldStringAppendf) {
153 stringAppendf(&s, "%s/%s/%s/%s", "a", "b", "c", "d");
154 EXPECT_EQ(string("helloa/b/c/d"), s);
157 TEST(Escape, cEscape) {
158 EXPECT_EQ("hello world", cEscape<std::string>("hello world"));
159 EXPECT_EQ("hello \\\\world\\\" goodbye",
160 cEscape<std::string>("hello \\world\" goodbye"));
161 EXPECT_EQ("hello\\nworld", cEscape<std::string>("hello\nworld"));
162 EXPECT_EQ("hello\\377\\376", cEscape<std::string>("hello\xff\xfe"));
165 TEST(Escape, cUnescape) {
166 EXPECT_EQ("hello world", cUnescape<std::string>("hello world"));
167 EXPECT_EQ("hello \\world\" goodbye",
168 cUnescape<std::string>("hello \\\\world\\\" goodbye"));
169 EXPECT_EQ("hello\nworld", cUnescape<std::string>("hello\\nworld"));
170 EXPECT_EQ("hello\nworld", cUnescape<std::string>("hello\\012world"));
171 EXPECT_EQ("hello\nworld", cUnescape<std::string>("hello\\x0aworld"));
172 EXPECT_EQ("hello\xff\xfe", cUnescape<std::string>("hello\\377\\376"));
173 EXPECT_EQ("hello\xff\xfe", cUnescape<std::string>("hello\\xff\\xfe"));
175 EXPECT_THROW({cUnescape<std::string>("hello\\");},
176 std::invalid_argument);
177 EXPECT_THROW({cUnescape<std::string>("hello\\x");},
178 std::invalid_argument);
179 EXPECT_THROW({cUnescape<std::string>("hello\\q");},
180 std::invalid_argument);
183 TEST(Escape, uriEscape) {
184 EXPECT_EQ("hello%2c%20%2fworld", uriEscape<std::string>("hello, /world"));
185 EXPECT_EQ("hello%2c%20/world", uriEscape<std::string>("hello, /world",
186 UriEscapeMode::PATH));
187 EXPECT_EQ("hello%2c+%2fworld", uriEscape<std::string>("hello, /world",
188 UriEscapeMode::QUERY));
190 "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ-_.~",
191 uriEscape<std::string>(
192 "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ-_.~")
196 TEST(Escape, uriUnescape) {
197 EXPECT_EQ("hello, /world", uriUnescape<std::string>("hello, /world"));
198 EXPECT_EQ("hello, /world", uriUnescape<std::string>("hello%2c%20%2fworld"));
199 EXPECT_EQ("hello,+/world", uriUnescape<std::string>("hello%2c+%2fworld"));
200 EXPECT_EQ("hello, /world", uriUnescape<std::string>("hello%2c+%2fworld",
201 UriEscapeMode::QUERY));
202 EXPECT_EQ("hello/", uriUnescape<std::string>("hello%2f"));
203 EXPECT_EQ("hello/", uriUnescape<std::string>("hello%2F"));
204 EXPECT_THROW({uriUnescape<std::string>("hello%");},
205 std::invalid_argument);
206 EXPECT_THROW({uriUnescape<std::string>("hello%2");},
207 std::invalid_argument);
208 EXPECT_THROW({uriUnescape<std::string>("hello%2g");},
209 std::invalid_argument);
213 void expectPrintable(StringPiece s) {
221 TEST(Escape, uriEscapeAllCombinations) {
224 StringPiece in(c, 2);
227 for (int i = 0; i < 256; ++i) {
229 for (int j = 0; j < 256; ++j) {
234 expectPrintable(tmp);
235 uriUnescape(tmp, out);
243 return ((v >= '0' && v <= '9') ||
244 (v >= 'A' && v <= 'F') ||
245 (v >= 'a' && v <= 'f'));
249 TEST(Escape, uriUnescapePercentDecoding) {
250 char c[4] = {'%', '\0', '\0', '\0'};
251 StringPiece in(c, 3);
253 unsigned int expected = 0;
254 for (int i = 0; i < 256; ++i) {
256 for (int j = 0; j < 256; ++j) {
258 if (isHex(i) && isHex(j)) {
260 uriUnescape(in, out);
261 EXPECT_EQ(1, out.size());
262 EXPECT_EQ(1, sscanf(c + 1, "%x", &expected));
263 unsigned char v = out[0];
264 EXPECT_EQ(expected, v);
266 EXPECT_THROW({uriUnescape(in, out);}, std::invalid_argument);
274 double pow2(int exponent) {
275 return double(int64_t(1) << exponent);
279 struct PrettyTestCase{
280 std::string prettyString;
282 PrettyType prettyType;
285 PrettyTestCase prettyTestCases[] =
287 {string("8.53e+07 s "), 85.3e6, PRETTY_TIME},
288 {string("8.53e+07 s "), 85.3e6, PRETTY_TIME},
289 {string("85.3 ms"), 85.3e-3, PRETTY_TIME},
290 {string("85.3 us"), 85.3e-6, PRETTY_TIME},
291 {string("85.3 ns"), 85.3e-9, PRETTY_TIME},
292 {string("85.3 ps"), 85.3e-12, PRETTY_TIME},
293 {string("8.53e-14 s "), 85.3e-15, PRETTY_TIME},
295 {string("0 s "), 0, PRETTY_TIME},
296 {string("1 s "), 1.0, PRETTY_TIME},
297 {string("1 ms"), 1.0e-3, PRETTY_TIME},
298 {string("1 us"), 1.0e-6, PRETTY_TIME},
299 {string("1 ns"), 1.0e-9, PRETTY_TIME},
300 {string("1 ps"), 1.0e-12, PRETTY_TIME},
302 // check bytes printing
303 {string("853 B "), 853., PRETTY_BYTES},
304 {string("833 kB"), 853.e3, PRETTY_BYTES},
305 {string("813.5 MB"), 853.e6, PRETTY_BYTES},
306 {string("7.944 GB"), 8.53e9, PRETTY_BYTES},
307 {string("794.4 GB"), 853.e9, PRETTY_BYTES},
308 {string("775.8 TB"), 853.e12, PRETTY_BYTES},
310 {string("0 B "), 0, PRETTY_BYTES},
311 {string("1 B "), pow2(0), PRETTY_BYTES},
312 {string("1 kB"), pow2(10), PRETTY_BYTES},
313 {string("1 MB"), pow2(20), PRETTY_BYTES},
314 {string("1 GB"), pow2(30), PRETTY_BYTES},
315 {string("1 TB"), pow2(40), PRETTY_BYTES},
317 {string("853 B "), 853., PRETTY_BYTES_IEC},
318 {string("833 KiB"), 853.e3, PRETTY_BYTES_IEC},
319 {string("813.5 MiB"), 853.e6, PRETTY_BYTES_IEC},
320 {string("7.944 GiB"), 8.53e9, PRETTY_BYTES_IEC},
321 {string("794.4 GiB"), 853.e9, PRETTY_BYTES_IEC},
322 {string("775.8 TiB"), 853.e12, PRETTY_BYTES_IEC},
324 {string("0 B "), 0, PRETTY_BYTES_IEC},
325 {string("1 B "), pow2(0), PRETTY_BYTES_IEC},
326 {string("1 KiB"), pow2(10), PRETTY_BYTES_IEC},
327 {string("1 MiB"), pow2(20), PRETTY_BYTES_IEC},
328 {string("1 GiB"), pow2(30), PRETTY_BYTES_IEC},
329 {string("1 TiB"), pow2(40), PRETTY_BYTES_IEC},
331 // check bytes metric printing
332 {string("853 B "), 853., PRETTY_BYTES_METRIC},
333 {string("853 kB"), 853.e3, PRETTY_BYTES_METRIC},
334 {string("853 MB"), 853.e6, PRETTY_BYTES_METRIC},
335 {string("8.53 GB"), 8.53e9, PRETTY_BYTES_METRIC},
336 {string("853 GB"), 853.e9, PRETTY_BYTES_METRIC},
337 {string("853 TB"), 853.e12, PRETTY_BYTES_METRIC},
339 {string("0 B "), 0, PRETTY_BYTES_METRIC},
340 {string("1 B "), 1.0, PRETTY_BYTES_METRIC},
341 {string("1 kB"), 1.0e+3, PRETTY_BYTES_METRIC},
342 {string("1 MB"), 1.0e+6, PRETTY_BYTES_METRIC},
344 {string("1 GB"), 1.0e+9, PRETTY_BYTES_METRIC},
345 {string("1 TB"), 1.0e+12, PRETTY_BYTES_METRIC},
347 // check metric-units (powers of 1000) printing
348 {string("853 "), 853., PRETTY_UNITS_METRIC},
349 {string("853 k"), 853.e3, PRETTY_UNITS_METRIC},
350 {string("853 M"), 853.e6, PRETTY_UNITS_METRIC},
351 {string("8.53 bil"), 8.53e9, PRETTY_UNITS_METRIC},
352 {string("853 bil"), 853.e9, PRETTY_UNITS_METRIC},
353 {string("853 tril"), 853.e12, PRETTY_UNITS_METRIC},
355 // check binary-units (powers of 1024) printing
356 {string("0 "), 0, PRETTY_UNITS_BINARY},
357 {string("1 "), pow2(0), PRETTY_UNITS_BINARY},
358 {string("1 k"), pow2(10), PRETTY_UNITS_BINARY},
359 {string("1 M"), pow2(20), PRETTY_UNITS_BINARY},
360 {string("1 G"), pow2(30), PRETTY_UNITS_BINARY},
361 {string("1 T"), pow2(40), PRETTY_UNITS_BINARY},
363 {string("1023 "), pow2(10) - 1, PRETTY_UNITS_BINARY},
364 {string("1024 k"), pow2(20) - 1, PRETTY_UNITS_BINARY},
365 {string("1024 M"), pow2(30) - 1, PRETTY_UNITS_BINARY},
366 {string("1024 G"), pow2(40) - 1, PRETTY_UNITS_BINARY},
368 {string("0 "), 0, PRETTY_UNITS_BINARY_IEC},
369 {string("1 "), pow2(0), PRETTY_UNITS_BINARY_IEC},
370 {string("1 Ki"), pow2(10), PRETTY_UNITS_BINARY_IEC},
371 {string("1 Mi"), pow2(20), PRETTY_UNITS_BINARY_IEC},
372 {string("1 Gi"), pow2(30), PRETTY_UNITS_BINARY_IEC},
373 {string("1 Ti"), pow2(40), PRETTY_UNITS_BINARY_IEC},
375 {string("1023 "), pow2(10) - 1, PRETTY_UNITS_BINARY_IEC},
376 {string("1024 Ki"), pow2(20) - 1, PRETTY_UNITS_BINARY_IEC},
377 {string("1024 Mi"), pow2(30) - 1, PRETTY_UNITS_BINARY_IEC},
378 {string("1024 Gi"), pow2(40) - 1, PRETTY_UNITS_BINARY_IEC},
380 //check border SI cases
382 {string("1 Y"), 1e24, PRETTY_SI},
383 {string("10 Y"), 1e25, PRETTY_SI},
384 {string("1 y"), 1e-24, PRETTY_SI},
385 {string("10 y"), 1e-23, PRETTY_SI},
387 // check that negative values work
388 {string("-85.3 s "), -85.3, PRETTY_TIME},
389 {string("-85.3 ms"), -85.3e-3, PRETTY_TIME},
390 {string("-85.3 us"), -85.3e-6, PRETTY_TIME},
391 {string("-85.3 ns"), -85.3e-9, PRETTY_TIME},
393 {string("endoftest"), 0, PRETTY_NUM_TYPES}
396 TEST(PrettyPrint, Basic) {
397 for (int i = 0; prettyTestCases[i].prettyType != PRETTY_NUM_TYPES; ++i){
398 const PrettyTestCase& prettyTest = prettyTestCases[i];
399 EXPECT_EQ(prettyTest.prettyString,
400 prettyPrint(prettyTest.realValue, prettyTest.prettyType));
404 TEST(PrettyToDouble, Basic) {
405 // check manually created tests
406 for (int i = 0; prettyTestCases[i].prettyType != PRETTY_NUM_TYPES; ++i){
407 PrettyTestCase testCase = prettyTestCases[i];
408 PrettyType formatType = testCase.prettyType;
409 double x = testCase.realValue;
410 std::string testString = testCase.prettyString;
411 double recoveredX = 0;
413 recoveredX = prettyToDouble(testString, formatType);
414 } catch (std::range_error &ex){
417 double relativeError = fabs(x) < 1e-5 ? (x-recoveredX) :
418 (x - recoveredX) / x;
419 EXPECT_NEAR(0, relativeError, 1e-3);
422 // checks for compatibility with prettyPrint over the whole parameter space
423 for (int i = 0 ; i < PRETTY_NUM_TYPES; ++i){
424 PrettyType formatType = static_cast<PrettyType>(i);
425 for (double x = 1e-18; x < 1e40; x *= 1.9){
426 bool addSpace = static_cast<PrettyType> (i) == PRETTY_SI;
427 for (int it = 0; it < 2; ++it, addSpace = true){
428 double recoveredX = 0;
430 recoveredX = prettyToDouble(prettyPrint(x, formatType, addSpace),
432 } catch (std::range_error &ex){
435 double relativeError = (x - recoveredX) / x;
436 EXPECT_NEAR(0, relativeError, 1e-3);
441 // check for incorrect values
442 EXPECT_THROW(prettyToDouble("10Mx", PRETTY_SI), std::range_error);
443 EXPECT_THROW(prettyToDouble("10 Mx", PRETTY_SI), std::range_error);
444 EXPECT_THROW(prettyToDouble("10 M x", PRETTY_SI), std::range_error);
446 StringPiece testString = "10Mx";
447 EXPECT_DOUBLE_EQ(prettyToDouble(&testString, PRETTY_UNITS_METRIC), 10e6);
448 EXPECT_EQ(testString, "x");
451 TEST(PrettyPrint, HexDump) {
452 std::string a("abc\x00\x02\xa0", 6); // embedded NUL
454 "00000000 61 62 63 00 02 a0 "
456 hexDump(a.data(), a.size()));
458 a = "abcdefghijklmnopqrstuvwxyz";
460 "00000000 61 62 63 64 65 66 67 68 69 6a 6b 6c 6d 6e 6f 70 "
461 "|abcdefghijklmnop|\n"
462 "00000010 71 72 73 74 75 76 77 78 79 7a "
464 hexDump(a.data(), a.size()));
467 TEST(System, errnoStr) {
469 EXPECT_EQ(EACCES, errno);
470 EXPECT_EQ(EACCES, errno); // twice to make sure EXPECT_EQ doesn't change it
472 fbstring expected = strerror(ENOENT);
475 EXPECT_EQ(expected, errnoStr(ENOENT));
476 // Ensure that errno isn't changed
477 EXPECT_EQ(EACCES, errno);
479 // Per POSIX, all errno values are positive, so -1 is invalid
482 // Ensure that errno isn't changed
483 EXPECT_EQ(EACCES, errno);
488 template<template<class,class> class VectorType>
490 VectorType<string,std::allocator<string> > parts;
492 folly::split(',', "a,b,c", parts);
493 EXPECT_EQ(parts.size(), 3);
494 EXPECT_EQ(parts[0], "a");
495 EXPECT_EQ(parts[1], "b");
496 EXPECT_EQ(parts[2], "c");
499 folly::split(',', StringPiece("a,b,c"), parts);
500 EXPECT_EQ(parts.size(), 3);
501 EXPECT_EQ(parts[0], "a");
502 EXPECT_EQ(parts[1], "b");
503 EXPECT_EQ(parts[2], "c");
506 folly::split(',', string("a,b,c"), parts);
507 EXPECT_EQ(parts.size(), 3);
508 EXPECT_EQ(parts[0], "a");
509 EXPECT_EQ(parts[1], "b");
510 EXPECT_EQ(parts[2], "c");
513 folly::split(',', "a,,c", parts);
514 EXPECT_EQ(parts.size(), 3);
515 EXPECT_EQ(parts[0], "a");
516 EXPECT_EQ(parts[1], "");
517 EXPECT_EQ(parts[2], "c");
520 folly::split(',', string("a,,c"), parts);
521 EXPECT_EQ(parts.size(), 3);
522 EXPECT_EQ(parts[0], "a");
523 EXPECT_EQ(parts[1], "");
524 EXPECT_EQ(parts[2], "c");
527 folly::split(',', "a,,c", parts, true);
528 EXPECT_EQ(parts.size(), 2);
529 EXPECT_EQ(parts[0], "a");
530 EXPECT_EQ(parts[1], "c");
533 folly::split(',', string("a,,c"), parts, true);
534 EXPECT_EQ(parts.size(), 2);
535 EXPECT_EQ(parts[0], "a");
536 EXPECT_EQ(parts[1], "c");
539 folly::split(',', string(",,a,,c,,,"), parts, true);
540 EXPECT_EQ(parts.size(), 2);
541 EXPECT_EQ(parts[0], "a");
542 EXPECT_EQ(parts[1], "c");
545 // test multiple split w/o clear
546 folly::split(',', ",,a,,c,,,", parts, true);
547 EXPECT_EQ(parts.size(), 2);
548 EXPECT_EQ(parts[0], "a");
549 EXPECT_EQ(parts[1], "c");
550 folly::split(',', ",,a,,c,,,", parts, true);
551 EXPECT_EQ(parts.size(), 4);
552 EXPECT_EQ(parts[2], "a");
553 EXPECT_EQ(parts[3], "c");
556 // test splits that with multi-line delimiter
557 folly::split("ab", "dabcabkdbkab", parts, true);
558 EXPECT_EQ(parts.size(), 3);
559 EXPECT_EQ(parts[0], "d");
560 EXPECT_EQ(parts[1], "c");
561 EXPECT_EQ(parts[2], "kdbk");
564 // test last part is shorter than the delimiter
565 folly::split("bc", "abcd", parts, true);
566 EXPECT_EQ(parts.size(), 2);
567 EXPECT_EQ(parts[0], "a");
568 EXPECT_EQ(parts[1], "d");
571 string orig = "ab2342asdfv~~!";
572 folly::split("", orig, parts, true);
573 EXPECT_EQ(parts.size(), 1);
574 EXPECT_EQ(parts[0], orig);
577 folly::split("452x;o38asfsajsdlfdf.j", "asfds", parts, true);
578 EXPECT_EQ(parts.size(), 1);
579 EXPECT_EQ(parts[0], "asfds");
582 folly::split("a", "", parts, true);
583 EXPECT_EQ(parts.size(), 0);
586 folly::split("a", "", parts);
587 EXPECT_EQ(parts.size(), 1);
588 EXPECT_EQ(parts[0], "");
591 folly::split("a", StringPiece(), parts, true);
592 EXPECT_EQ(parts.size(), 0);
595 folly::split("a", StringPiece(), parts);
596 EXPECT_EQ(parts.size(), 1);
597 EXPECT_EQ(parts[0], "");
600 folly::split("a", "abcdefg", parts, true);
601 EXPECT_EQ(parts.size(), 1);
602 EXPECT_EQ(parts[0], "bcdefg");
605 orig = "All, , your base, are , , belong to us";
606 folly::split(", ", orig, parts, true);
607 EXPECT_EQ(parts.size(), 4);
608 EXPECT_EQ(parts[0], "All");
609 EXPECT_EQ(parts[1], "your base");
610 EXPECT_EQ(parts[2], "are ");
611 EXPECT_EQ(parts[3], "belong to us");
613 folly::split(", ", orig, parts);
614 EXPECT_EQ(parts.size(), 6);
615 EXPECT_EQ(parts[0], "All");
616 EXPECT_EQ(parts[1], "");
617 EXPECT_EQ(parts[2], "your base");
618 EXPECT_EQ(parts[3], "are ");
619 EXPECT_EQ(parts[4], "");
620 EXPECT_EQ(parts[5], "belong to us");
623 orig = ", Facebook, rul,es!, ";
624 folly::split(", ", orig, parts, true);
625 EXPECT_EQ(parts.size(), 2);
626 EXPECT_EQ(parts[0], "Facebook");
627 EXPECT_EQ(parts[1], "rul,es!");
629 folly::split(", ", orig, parts);
630 EXPECT_EQ(parts.size(), 4);
631 EXPECT_EQ(parts[0], "");
632 EXPECT_EQ(parts[1], "Facebook");
633 EXPECT_EQ(parts[2], "rul,es!");
634 EXPECT_EQ(parts[3], "");
637 template<template<class,class> class VectorType>
639 VectorType<StringPiece,std::allocator<StringPiece> > pieces;
640 VectorType<StringPiece,std::allocator<StringPiece> > pieces2;
642 folly::split(',', "a,b,c", pieces);
643 EXPECT_EQ(pieces.size(), 3);
644 EXPECT_EQ(pieces[0], "a");
645 EXPECT_EQ(pieces[1], "b");
646 EXPECT_EQ(pieces[2], "c");
650 folly::split(',', "a,,c", pieces);
651 EXPECT_EQ(pieces.size(), 3);
652 EXPECT_EQ(pieces[0], "a");
653 EXPECT_EQ(pieces[1], "");
654 EXPECT_EQ(pieces[2], "c");
657 folly::split(',', "a,,c", pieces, true);
658 EXPECT_EQ(pieces.size(), 2);
659 EXPECT_EQ(pieces[0], "a");
660 EXPECT_EQ(pieces[1], "c");
663 folly::split(',', ",,a,,c,,,", pieces, true);
664 EXPECT_EQ(pieces.size(), 2);
665 EXPECT_EQ(pieces[0], "a");
666 EXPECT_EQ(pieces[1], "c");
669 // test multiple split w/o clear
670 folly::split(',', ",,a,,c,,,", pieces, true);
671 EXPECT_EQ(pieces.size(), 2);
672 EXPECT_EQ(pieces[0], "a");
673 EXPECT_EQ(pieces[1], "c");
674 folly::split(',', ",,a,,c,,,", pieces, true);
675 EXPECT_EQ(pieces.size(), 4);
676 EXPECT_EQ(pieces[2], "a");
677 EXPECT_EQ(pieces[3], "c");
680 // test multiple split rounds
681 folly::split(",", "a_b,c_d", pieces);
682 EXPECT_EQ(pieces.size(), 2);
683 EXPECT_EQ(pieces[0], "a_b");
684 EXPECT_EQ(pieces[1], "c_d");
685 folly::split("_", pieces[0], pieces2);
686 EXPECT_EQ(pieces2.size(), 2);
687 EXPECT_EQ(pieces2[0], "a");
688 EXPECT_EQ(pieces2[1], "b");
690 folly::split("_", pieces[1], pieces2);
691 EXPECT_EQ(pieces2.size(), 2);
692 EXPECT_EQ(pieces2[0], "c");
693 EXPECT_EQ(pieces2[1], "d");
697 // test splits that with multi-line delimiter
698 folly::split("ab", "dabcabkdbkab", pieces, true);
699 EXPECT_EQ(pieces.size(), 3);
700 EXPECT_EQ(pieces[0], "d");
701 EXPECT_EQ(pieces[1], "c");
702 EXPECT_EQ(pieces[2], "kdbk");
705 string orig = "ab2342asdfv~~!";
706 folly::split("", orig.c_str(), pieces, true);
707 EXPECT_EQ(pieces.size(), 1);
708 EXPECT_EQ(pieces[0], orig);
711 folly::split("452x;o38asfsajsdlfdf.j", "asfds", pieces, true);
712 EXPECT_EQ(pieces.size(), 1);
713 EXPECT_EQ(pieces[0], "asfds");
716 folly::split("a", "", pieces, true);
717 EXPECT_EQ(pieces.size(), 0);
720 folly::split("a", "", pieces);
721 EXPECT_EQ(pieces.size(), 1);
722 EXPECT_EQ(pieces[0], "");
725 folly::split("a", "abcdefg", pieces, true);
726 EXPECT_EQ(pieces.size(), 1);
727 EXPECT_EQ(pieces[0], "bcdefg");
730 orig = "All, , your base, are , , belong to us";
731 folly::split(", ", orig, pieces, true);
732 EXPECT_EQ(pieces.size(), 4);
733 EXPECT_EQ(pieces[0], "All");
734 EXPECT_EQ(pieces[1], "your base");
735 EXPECT_EQ(pieces[2], "are ");
736 EXPECT_EQ(pieces[3], "belong to us");
738 folly::split(", ", orig, pieces);
739 EXPECT_EQ(pieces.size(), 6);
740 EXPECT_EQ(pieces[0], "All");
741 EXPECT_EQ(pieces[1], "");
742 EXPECT_EQ(pieces[2], "your base");
743 EXPECT_EQ(pieces[3], "are ");
744 EXPECT_EQ(pieces[4], "");
745 EXPECT_EQ(pieces[5], "belong to us");
748 orig = ", Facebook, rul,es!, ";
749 folly::split(", ", orig, pieces, true);
750 EXPECT_EQ(pieces.size(), 2);
751 EXPECT_EQ(pieces[0], "Facebook");
752 EXPECT_EQ(pieces[1], "rul,es!");
754 folly::split(", ", orig, pieces);
755 EXPECT_EQ(pieces.size(), 4);
756 EXPECT_EQ(pieces[0], "");
757 EXPECT_EQ(pieces[1], "Facebook");
758 EXPECT_EQ(pieces[2], "rul,es!");
759 EXPECT_EQ(pieces[3], "");
762 const char* str = "a,b";
763 folly::split(',', StringPiece(str), pieces);
764 EXPECT_EQ(pieces.size(), 2);
765 EXPECT_EQ(pieces[0], "a");
766 EXPECT_EQ(pieces[1], "b");
767 EXPECT_EQ(pieces[0].start(), str);
768 EXPECT_EQ(pieces[1].start(), str + 2);
770 std::set<StringPiece> unique;
771 folly::splitTo<StringPiece>(":", "asd:bsd:asd:asd:bsd:csd::asd",
772 std::inserter(unique, unique.begin()), true);
773 EXPECT_EQ(unique.size(), 3);
774 if (unique.size() == 3) {
775 EXPECT_EQ(*unique.begin(), "asd");
776 EXPECT_EQ(*--unique.end(), "csd");
779 VectorType<fbstring,std::allocator<fbstring> > blah;
780 folly::split('-', "a-b-c-d-f-e", blah);
781 EXPECT_EQ(blah.size(), 6);
786 TEST(Split, split_vector) {
787 splitTest<std::vector>();
789 TEST(Split, split_fbvector) {
790 splitTest<folly::fbvector>();
792 TEST(Split, pieces_vector) {
793 piecesTest<std::vector>();
795 TEST(Split, pieces_fbvector) {
796 piecesTest<folly::fbvector>();
800 StringPiece a, b, c, d;
802 EXPECT_TRUE(folly::split<false>('.', "a.b.c.d", a, b, c, d));
803 EXPECT_TRUE(folly::split<false>('.', "a.b.c", a, b, c));
804 EXPECT_TRUE(folly::split<false>('.', "a.b", a, b));
805 EXPECT_TRUE(folly::split<false>('.', "a", a));
807 EXPECT_TRUE(folly::split('.', "a.b.c.d", a, b, c, d));
808 EXPECT_TRUE(folly::split('.', "a.b.c", a, b, c));
809 EXPECT_TRUE(folly::split('.', "a.b", a, b));
810 EXPECT_TRUE(folly::split('.', "a", a));
812 EXPECT_TRUE(folly::split<false>('.', "a.b.c", a, b, c));
816 EXPECT_FALSE(folly::split<false>('.', "a.b", a, b, c));
817 EXPECT_TRUE(folly::split<false>('.', "a.b.c", a, b));
821 EXPECT_TRUE(folly::split('.', "a.b.c", a, b, c));
825 EXPECT_FALSE(folly::split('.', "a.b.c", a, b));
826 EXPECT_FALSE(folly::split('.', "a.b", a, b, c));
828 EXPECT_TRUE(folly::split<false>('.', "a.b", a, b));
831 EXPECT_FALSE(folly::split<false>('.', "a", a, b));
832 EXPECT_TRUE(folly::split<false>('.', "a.b", a));
835 EXPECT_TRUE(folly::split('.', "a.b", a, b));
838 EXPECT_FALSE(folly::split('.', "a", a, b));
839 EXPECT_FALSE(folly::split('.', "a.b", a));
842 TEST(Split, std_string_fixed) {
843 std::string a, b, c, d;
845 EXPECT_TRUE(folly::split<false>('.', "a.b.c.d", a, b, c, d));
846 EXPECT_TRUE(folly::split<false>('.', "a.b.c", a, b, c));
847 EXPECT_TRUE(folly::split<false>('.', "a.b", a, b));
848 EXPECT_TRUE(folly::split<false>('.', "a", a));
850 EXPECT_TRUE(folly::split('.', "a.b.c.d", a, b, c, d));
851 EXPECT_TRUE(folly::split('.', "a.b.c", a, b, c));
852 EXPECT_TRUE(folly::split('.', "a.b", a, b));
853 EXPECT_TRUE(folly::split('.', "a", a));
855 EXPECT_TRUE(folly::split<false>('.', "a.b.c", a, b, c));
859 EXPECT_FALSE(folly::split<false>('.', "a.b", a, b, c));
860 EXPECT_TRUE(folly::split<false>('.', "a.b.c", a, b));
864 EXPECT_TRUE(folly::split('.', "a.b.c", a, b, c));
868 EXPECT_FALSE(folly::split('.', "a.b.c", a, b));
869 EXPECT_FALSE(folly::split('.', "a.b", a, b, c));
871 EXPECT_TRUE(folly::split<false>('.', "a.b", a, b));
874 EXPECT_FALSE(folly::split<false>('.', "a", a, b));
875 EXPECT_TRUE(folly::split<false>('.', "a.b", a));
878 EXPECT_TRUE(folly::split('.', "a.b", a, b));
881 EXPECT_FALSE(folly::split('.', "a", a, b));
882 EXPECT_FALSE(folly::split('.', "a.b", a));
885 TEST(Split, fixed_convert) {
890 EXPECT_TRUE(folly::split(':', "a:13:14.7:b", a, b, c, d));
893 EXPECT_NEAR(14.7, c, 1e-10);
896 EXPECT_TRUE(folly::split<false>(':', "b:14:15.3:c", a, b, c, d));
899 EXPECT_NEAR(15.3, c, 1e-10);
902 EXPECT_FALSE(folly::split(':', "a:13:14.7:b", a, b, d));
904 EXPECT_TRUE(folly::split<false>(':', "a:13:14.7:b", a, b, d));
907 EXPECT_EQ("14.7:b", d);
909 EXPECT_THROW(folly::split<false>(':', "a:13:14.7:b", a, b, c),
916 std::vector<int> empty = { };
917 join(":", empty, output);
918 EXPECT_TRUE(output.empty());
920 std::vector<std::string> input1 = { "1", "23", "456", "" };
921 join(':', input1, output);
922 EXPECT_EQ(output, "1:23:456:");
923 output = join(':', input1);
924 EXPECT_EQ(output, "1:23:456:");
926 auto input2 = { 1, 23, 456 };
927 join("-*-", input2, output);
928 EXPECT_EQ(output, "1-*-23-*-456");
929 output = join("-*-", input2);
930 EXPECT_EQ(output, "1-*-23-*-456");
932 auto input3 = { 'f', 'a', 'c', 'e', 'b', 'o', 'o', 'k' };
933 join("", input3, output);
934 EXPECT_EQ(output, "facebook");
936 join("_", { "", "f", "a", "c", "e", "b", "o", "o", "k", "" }, output);
937 EXPECT_EQ(output, "_f_a_c_e_b_o_o_k_");
939 output = join("", input3.begin(), input3.end());
940 EXPECT_EQ(output, "facebook");
943 TEST(String, hexlify) {
944 string input1 = "0123";
946 EXPECT_TRUE(hexlify(input1, output1));
947 EXPECT_EQ(output1, "30313233");
949 fbstring input2 = "abcdefg";
954 EXPECT_TRUE(hexlify(input2, output2));
955 EXPECT_EQ(output2, "610063ff65b667");
958 TEST(String, unhexlify) {
959 string input1 = "30313233";
961 EXPECT_TRUE(unhexlify(input1, output1));
962 EXPECT_EQ(output1, "0123");
964 fbstring input2 = "610063ff65b667";
966 EXPECT_TRUE(unhexlify(input2, output2));
967 EXPECT_EQ(output2.size(), 7);
968 EXPECT_EQ(output2[0], 'a');
969 EXPECT_EQ(output2[1], 0);
970 EXPECT_EQ(output2[2], 'c');
971 EXPECT_EQ(output2[3] & 0xff, 0xff);
972 EXPECT_EQ(output2[4], 'e');
973 EXPECT_EQ(output2[5] & 0xff, 0xb6);
974 EXPECT_EQ(output2[6], 'g');
978 EXPECT_FALSE(unhexlify(input3, output3));
980 string input4 = "xy";
982 EXPECT_FALSE(unhexlify(input4, output4));
985 TEST(String, backslashify) {
986 EXPECT_EQ("abc", string("abc"));
987 EXPECT_EQ("abc", backslashify(string("abc")));
988 EXPECT_EQ("abc\\r", backslashify(string("abc\r")));
989 EXPECT_EQ("abc\\x0d", backslashify(string("abc\r"), true));
990 EXPECT_EQ("\\0\\0", backslashify(string(2, '\0')));
993 TEST(String, humanify) {
994 // Simple cases; output is obvious.
995 EXPECT_EQ("abc", humanify(string("abc")));
996 EXPECT_EQ("abc\\\\r", humanify(string("abc\\r")));
997 EXPECT_EQ("0xff", humanify(string("\xff")));
998 EXPECT_EQ("abc\\xff", humanify(string("abc\xff")));
999 EXPECT_EQ("abc\\b", humanify(string("abc\b")));
1000 EXPECT_EQ("0x00", humanify(string(1, '\0')));
1001 EXPECT_EQ("0x0000", humanify(string(2, '\0')));
1004 // Mostly printable, so backslash! 80, 60, and 40% printable, respectively
1005 EXPECT_EQ("aaaa\\xff", humanify(string("aaaa\xff")));
1006 EXPECT_EQ("aaa\\xff\\xff", humanify(string("aaa\xff\xff")));
1007 EXPECT_EQ("aa\\xff\\xff\\xff", humanify(string("aa\xff\xff\xff")));
1009 // 20% printable, and the printable portion isn't the prefix; hexify!
1010 EXPECT_EQ("0xff61ffffff", humanify(string("\xff" "a\xff\xff\xff")));
1012 // Same as previous, except swap first two chars; prefix is
1013 // printable and within the threshold, so backslashify.
1014 EXPECT_EQ("a\\xff\\xff\\xff\\xff", humanify(string("a\xff\xff\xff\xff")));
1016 // Just too much unprintable; hex, despite prefix.
1017 EXPECT_EQ("0x61ffffffffff", humanify(string("a\xff\xff\xff\xff\xff")));
1023 * Copy bytes from src to somewhere in the buffer referenced by dst. The
1024 * actual starting position of the copy will be the first address in the
1025 * destination buffer whose address mod 8 is equal to the src address mod 8.
1026 * The caller is responsible for ensuring that the destination buffer has
1027 * enough extra space to accommodate the shifted copy.
1029 char* copyWithSameAlignment(char* dst, const char* src, size_t length) {
1030 const char* originalDst = dst;
1031 size_t dstOffset = size_t(dst) & 0x7;
1032 size_t srcOffset = size_t(src) & 0x7;
1033 while (dstOffset != srcOffset) {
1038 CHECK(dst <= originalDst + 7);
1039 CHECK((size_t(dst) & 0x7) == (size_t(src) & 0x7));
1040 memcpy(dst, src, length);
1044 void testToLowerAscii(Range<const char*> src) {
1045 // Allocate extra space so we can make copies that start at the
1046 // same alignment (byte, word, quadword, etc) as the source buffer.
1047 char controlBuf[src.size() + 7];
1048 char* control = copyWithSameAlignment(controlBuf, src.begin(), src.size());
1050 char testBuf[src.size() + 7];
1051 char* test = copyWithSameAlignment(testBuf, src.begin(), src.size());
1053 for (size_t i = 0; i < src.size(); i++) {
1054 control[i] = tolower(control[i]);
1056 toLowerAscii(test, src.size());
1057 for (size_t i = 0; i < src.size(); i++) {
1058 EXPECT_EQ(control[i], test[i]);
1064 TEST(String, toLowerAsciiAligned) {
1065 static const size_t kSize = 256;
1067 for (size_t i = 0; i < kSize; i++) {
1068 input[i] = (char)(i & 0xff);
1070 testToLowerAscii(Range<const char*>(input, kSize));
1073 TEST(String, toLowerAsciiUnaligned) {
1074 static const size_t kSize = 256;
1076 for (size_t i = 0; i < kSize; i++) {
1077 input[i] = (char)(i & 0xff);
1079 // Test input buffers of several lengths to exercise all the
1080 // cases: buffer at the start/middle/end of an aligned block, plus
1081 // buffers that span multiple aligned blocks. The longest test input
1082 // is 3 unaligned bytes + 4 32-bit aligned bytes + 8 64-bit aligned
1083 // + 4 32-bit aligned + 3 unaligned = 22 bytes.
1084 for (size_t length = 1; length < 23; length++) {
1085 for (size_t offset = 0; offset + length <= kSize; offset++) {
1086 testToLowerAscii(Range<const char*>(input + offset, length));
1091 TEST(String, whitespace) {
1093 EXPECT_EQ("kavabanga",
1094 trimWhitespace("kavabanga"));
1095 EXPECT_EQ("kavabanga",
1096 trimWhitespace("kavabanga \t \n "));
1097 EXPECT_EQ("kavabanga",
1098 trimWhitespace(" \t \r \n \n kavabanga"));
1099 EXPECT_EQ("kavabanga",
1100 trimWhitespace("\t \r \n kavabanga \t \n "));
1101 EXPECT_EQ("kavabanga",
1102 trimWhitespace(" \t \r \n \n kavabanga"));
1103 EXPECT_EQ("kavabanga",
1104 trimWhitespace("\t \r \n kavabanga \t \n "));
1106 ltrimWhitespace(rtrimWhitespace("kavabanga")),
1107 rtrimWhitespace(ltrimWhitespace("kavabanga")));
1109 ltrimWhitespace(rtrimWhitespace("kavabanga \r\t\n")),
1110 rtrimWhitespace(ltrimWhitespace("kavabanga \r\t\n")));
1111 EXPECT_EQ("", trimWhitespace("\t \r \n \t \n "));
1112 EXPECT_EQ("", trimWhitespace(""));
1113 EXPECT_EQ("", trimWhitespace("\t"));
1114 EXPECT_EQ("", trimWhitespace("\r"));
1115 EXPECT_EQ("", trimWhitespace("\n"));
1116 EXPECT_EQ("", trimWhitespace("\t "));
1117 EXPECT_EQ("", trimWhitespace("\r "));
1118 EXPECT_EQ("", trimWhitespace("\n "));
1119 EXPECT_EQ("", trimWhitespace(" \t"));
1120 EXPECT_EQ("", trimWhitespace(" \r"));
1121 EXPECT_EQ("", trimWhitespace(" \n"));
1124 EXPECT_EQ("kavabanga", ltrimWhitespace("\t kavabanga"));
1125 EXPECT_EQ("kavabanga \r\n", ltrimWhitespace("\t kavabanga \r\n"));
1126 EXPECT_EQ("", ltrimWhitespace("\r "));
1127 EXPECT_EQ("", ltrimWhitespace("\n "));
1128 EXPECT_EQ("", ltrimWhitespace("\r "));
1131 EXPECT_EQ("\t kavabanga", rtrimWhitespace("\t kavabanga"));
1132 EXPECT_EQ("\t kavabanga", rtrimWhitespace("\t kavabanga \r\n"));
1133 EXPECT_EQ("", rtrimWhitespace("\r "));
1134 EXPECT_EQ("", rtrimWhitespace("\n "));
1135 EXPECT_EQ("", rtrimWhitespace("\r "));
1138 TEST(String, stripLeftMargin_empty) {
1139 auto input = R"TEXT(
1142 EXPECT_EQ(expected, stripLeftMargin(input));
1145 TEST(String, stripLeftMargin_one_line) {
1146 auto input = R"TEXT(
1149 auto expected = "hi there bob!\n";
1150 EXPECT_EQ(expected, stripLeftMargin(input));
1153 TEST(String, stripLeftMargin_two_lines) {
1154 auto input = R"TEXT(
1158 auto expected = "hi there bob!\nnice weather today!\n";
1159 EXPECT_EQ(expected, stripLeftMargin(input));
1162 TEST(String, stripLeftMargin_three_lines_uneven) {
1163 auto input = R"TEXT(
1168 auto expected = " hi there bob!\nnice weather today!\n so long!\n";
1169 EXPECT_EQ(expected, stripLeftMargin(input));
1172 TEST(String, stripLeftMargin_preceding_blank_lines) {
1173 auto input = R"TEXT(
1178 auto expected = "\n\nhi there bob!\n";
1179 EXPECT_EQ(expected, stripLeftMargin(input));
1182 TEST(String, stripLeftMargin_succeeding_blank_lines) {
1183 auto input = R"TEXT(
1188 auto expected = "hi there bob!\n\n\n";
1189 EXPECT_EQ(expected, stripLeftMargin(input));
1192 TEST(String, stripLeftMargin_interstitial_undented_whiteline) {
1193 // using ~ as a marker
1194 string input = R"TEXT(
1199 input = boost::regex_replace(input, boost::regex(" +~"), "");
1200 EXPECT_EQ("\n hi there bob!\n\n so long!\n ", input);
1201 auto expected = "hi there bob!\n\nso long!\n";
1202 EXPECT_EQ(expected, stripLeftMargin(input));
1205 TEST(String, stripLeftMargin_interstitial_dedented_whiteline) {
1206 // using ~ as a marker
1207 string input = R"TEXT(
1212 input = boost::regex_replace(input, boost::regex("~"), "");
1213 EXPECT_EQ("\n hi there bob!\n \n so long!\n ", input);
1214 auto expected = "hi there bob!\n\nso long!\n";
1215 EXPECT_EQ(expected, stripLeftMargin(input));
1218 TEST(String, stripLeftMargin_interstitial_equidented_whiteline) {
1219 // using ~ as a marker
1220 string input = R"TEXT(
1225 input = boost::regex_replace(input, boost::regex("~"), "");
1226 EXPECT_EQ("\n hi there bob!\n \n so long!\n ", input);
1227 auto expected = "hi there bob!\n\nso long!\n";
1228 EXPECT_EQ(expected, stripLeftMargin(input));
1231 TEST(String, stripLeftMargin_interstitial_indented_whiteline) {
1232 // using ~ as a marker
1233 string input = R"TEXT(
1238 input = boost::regex_replace(input, boost::regex("~"), "");
1239 EXPECT_EQ("\n hi there bob!\n \n so long!\n ", input);
1240 auto expected = "hi there bob!\n \nso long!\n";
1241 EXPECT_EQ(expected, stripLeftMargin(input));
1244 const folly::StringPiece kTestUTF8 = "This is \U0001F602 stuff!";
1246 TEST(UTF8StringPiece, valid_utf8) {
1247 folly::StringPiece sp = kTestUTF8;
1248 UTF8StringPiece utf8 = sp;
1249 // utf8.size() not available since it's not a random-access range
1250 EXPECT_EQ(16, utf8.walk_size());
1253 TEST(UTF8StringPiece, valid_suffix) {
1254 UTF8StringPiece utf8 = kTestUTF8.subpiece(8);
1255 EXPECT_EQ(8, utf8.walk_size());
1258 TEST(UTF8StringPiece, empty_mid_codepoint) {
1259 UTF8StringPiece utf8 = kTestUTF8.subpiece(9, 0); // okay since it's empty
1260 EXPECT_EQ(0, utf8.walk_size());
1263 TEST(UTF8StringPiece, invalid_mid_codepoint) {
1264 EXPECT_THROW(UTF8StringPiece(kTestUTF8.subpiece(9, 1)), std::out_of_range);