TestString.cpp 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886
  1. /*
  2. * Copyright (c) 2022, Andreas Kling <kling@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. // This is included first on purpose. We specifically do not want LibTest to override VERIFY here so
  7. // that we can actually test that some String factory methods cause a crash with invalid input.
  8. #include <AK/String.h>
  9. #include <LibTest/TestCase.h>
  10. #include <AK/MemoryStream.h>
  11. #include <AK/StringBuilder.h>
  12. #include <AK/Try.h>
  13. #include <AK/Utf8View.h>
  14. #include <AK/Vector.h>
  15. TEST_CASE(construct_empty)
  16. {
  17. String empty;
  18. EXPECT(empty.is_empty());
  19. EXPECT_EQ(empty.bytes().size(), 0u);
  20. EXPECT_EQ(empty, ""sv);
  21. auto empty2 = MUST(""_string);
  22. EXPECT(empty2.is_empty());
  23. EXPECT_EQ(empty, empty2);
  24. auto empty3 = MUST(String::from_utf8(""sv));
  25. EXPECT(empty3.is_empty());
  26. EXPECT_EQ(empty, empty3);
  27. }
  28. TEST_CASE(move_assignment)
  29. {
  30. String string1 = MUST("hello"_string);
  31. string1 = MUST("friends!"_string);
  32. EXPECT_EQ(string1, "friends!"sv);
  33. }
  34. TEST_CASE(short_strings)
  35. {
  36. #ifdef AK_ARCH_64_BIT
  37. auto string1 = MUST(String::from_utf8("abcdefg"sv));
  38. EXPECT_EQ(string1.is_short_string(), true);
  39. EXPECT_EQ(string1.bytes().size(), 7u);
  40. EXPECT_EQ(string1.bytes_as_string_view(), "abcdefg"sv);
  41. constexpr auto string2 = String::from_utf8_short_string("abcdefg"sv);
  42. EXPECT_EQ(string2.is_short_string(), true);
  43. EXPECT_EQ(string2.bytes().size(), 7u);
  44. EXPECT_EQ(string2, string1);
  45. auto string3 = MUST("abcdefg"_string);
  46. EXPECT_EQ(string3.is_short_string(), true);
  47. EXPECT_EQ(string3.bytes().size(), 7u);
  48. EXPECT_EQ(string3, string1);
  49. constexpr auto string4 = "abcdefg"_short_string;
  50. EXPECT_EQ(string4.is_short_string(), true);
  51. EXPECT_EQ(string4.bytes().size(), 7u);
  52. EXPECT_EQ(string4, string1);
  53. #else
  54. auto string1 = MUST(String::from_utf8("abc"sv));
  55. EXPECT_EQ(string1.is_short_string(), true);
  56. EXPECT_EQ(string1.bytes().size(), 3u);
  57. EXPECT_EQ(string1.bytes_as_string_view(), "abc"sv);
  58. constexpr auto string2 = String::from_utf8_short_string("abc"sv);
  59. EXPECT_EQ(string2.is_short_string(), true);
  60. EXPECT_EQ(string2.bytes().size(), 3u);
  61. EXPECT_EQ(string2, string1);
  62. auto string3 = MUST("abc"_string);
  63. EXPECT_EQ(string3.is_short_string(), true);
  64. EXPECT_EQ(string3.bytes().size(), 3u);
  65. EXPECT_EQ(string3, string1);
  66. constexpr auto string4 = "abc"_short_string;
  67. EXPECT_EQ(string4.is_short_string(), true);
  68. EXPECT_EQ(string4.bytes().size(), 3u);
  69. EXPECT_EQ(string4, string1);
  70. #endif
  71. }
  72. TEST_CASE(long_strings)
  73. {
  74. auto string = MUST(String::from_utf8("abcdefgh"sv));
  75. EXPECT_EQ(string.is_short_string(), false);
  76. EXPECT_EQ(string.bytes().size(), 8u);
  77. EXPECT_EQ(string.bytes_as_string_view(), "abcdefgh"sv);
  78. }
  79. TEST_CASE(long_streams)
  80. {
  81. {
  82. u8 bytes[64] = {};
  83. constexpr auto test_view = "Well, hello friends"sv;
  84. FixedMemoryStream stream(Bytes { bytes, sizeof(bytes) });
  85. MUST(stream.write_until_depleted(test_view.bytes()));
  86. MUST(stream.seek(0));
  87. auto string = MUST(String::from_stream(stream, test_view.length()));
  88. EXPECT_EQ(string.is_short_string(), false);
  89. EXPECT_EQ(string.bytes().size(), 19u);
  90. EXPECT_EQ(string.bytes_as_string_view(), test_view);
  91. }
  92. {
  93. AllocatingMemoryStream stream;
  94. MUST(stream.write_until_depleted(("abc"sv).bytes()));
  95. auto string = MUST(String::from_stream(stream, 3u));
  96. EXPECT_EQ(string.is_short_string(), true);
  97. EXPECT_EQ(string.bytes().size(), 3u);
  98. EXPECT_EQ(string.bytes_as_string_view(), "abc"sv);
  99. }
  100. {
  101. AllocatingMemoryStream stream;
  102. MUST(stream.write_until_depleted(("0123456789"sv).bytes()));
  103. auto string = MUST(String::from_stream(stream, 9u));
  104. EXPECT_EQ(string.is_short_string(), false);
  105. EXPECT_EQ(string.bytes().size(), 9u);
  106. EXPECT_EQ(string.bytes_as_string_view(), "012345678"sv);
  107. }
  108. {
  109. AllocatingMemoryStream stream;
  110. MUST(stream.write_value(0xffffffff));
  111. MUST(stream.write_value(0xffffffff));
  112. MUST(stream.write_value(0xffffffff));
  113. auto error_or_string = String::from_stream(stream, stream.used_buffer_size());
  114. EXPECT_EQ(error_or_string.is_error(), true);
  115. }
  116. }
  117. TEST_CASE(invalid_utf8)
  118. {
  119. auto string1 = String::from_utf8("long string \xf4\x8f\xbf\xc0"sv); // U+110000
  120. EXPECT(string1.is_error());
  121. EXPECT(string1.error().string_literal().contains("Input was not valid UTF-8"sv));
  122. auto string2 = String::from_utf8("\xf4\xa1\xb0\xbd"sv); // U+121C3D
  123. EXPECT(string2.is_error());
  124. EXPECT(string2.error().string_literal().contains("Input was not valid UTF-8"sv));
  125. AllocatingMemoryStream stream;
  126. MUST(stream.write_value<u8>(0xf4));
  127. MUST(stream.write_value<u8>(0xa1));
  128. MUST(stream.write_value<u8>(0xb0));
  129. MUST(stream.write_value<u8>(0xbd));
  130. auto string3 = String::from_stream(stream, stream.used_buffer_size());
  131. EXPECT_EQ(string3.is_error(), true);
  132. EXPECT(string3.error().string_literal().contains("Input was not valid UTF-8"sv));
  133. }
  134. TEST_CASE(from_code_points)
  135. {
  136. for (u32 code_point = 0; code_point < 0x80; ++code_point) {
  137. auto string = String::from_code_point(code_point);
  138. auto ch = static_cast<char>(code_point);
  139. StringView view { &ch, 1 };
  140. EXPECT_EQ(string, view);
  141. }
  142. auto string = String::from_code_point(0x10ffff);
  143. EXPECT_EQ(string, "\xF4\x8F\xBF\xBF"sv);
  144. EXPECT_CRASH("Creating a string from an invalid code point", [] {
  145. String::from_code_point(0xffffffff);
  146. return Test::Crash::Failure::DidNotCrash;
  147. });
  148. }
  149. TEST_CASE(substring)
  150. {
  151. auto superstring = MUST("Hello I am a long string"_string);
  152. auto short_substring = MUST(superstring.substring_from_byte_offset(0, 5));
  153. EXPECT_EQ(short_substring, "Hello"sv);
  154. auto long_substring = MUST(superstring.substring_from_byte_offset(0, 10));
  155. EXPECT_EQ(long_substring, "Hello I am"sv);
  156. }
  157. TEST_CASE(substring_with_shared_superstring)
  158. {
  159. auto superstring = MUST("Hello I am a long string"_string);
  160. auto substring1 = MUST(superstring.substring_from_byte_offset_with_shared_superstring(0, 5));
  161. EXPECT_EQ(substring1, "Hello"sv);
  162. auto substring2 = MUST(superstring.substring_from_byte_offset_with_shared_superstring(0, 10));
  163. EXPECT_EQ(substring2, "Hello I am"sv);
  164. }
  165. TEST_CASE(code_points)
  166. {
  167. auto string = MUST("🦬🪒"_string);
  168. Vector<u32> code_points;
  169. for (auto code_point : string.code_points())
  170. code_points.append(code_point);
  171. EXPECT_EQ(code_points[0], 0x1f9acu);
  172. EXPECT_EQ(code_points[1], 0x1fa92u);
  173. }
  174. TEST_CASE(string_builder)
  175. {
  176. StringBuilder builder;
  177. builder.append_code_point(0x1f9acu);
  178. builder.append_code_point(0x1fa92u);
  179. auto string = MUST(builder.to_string());
  180. EXPECT_EQ(string, "🦬🪒"sv);
  181. EXPECT_EQ(string.bytes().size(), 8u);
  182. }
  183. TEST_CASE(ak_format)
  184. {
  185. auto foo = MUST(String::formatted("Hello {}", MUST("friends"_string)));
  186. EXPECT_EQ(foo, "Hello friends"sv);
  187. }
  188. TEST_CASE(replace)
  189. {
  190. {
  191. auto haystack = MUST("Hello enemies"_string);
  192. auto result = MUST(haystack.replace("enemies"sv, "friends"sv, ReplaceMode::All));
  193. EXPECT_EQ(result, "Hello friends"sv);
  194. }
  195. {
  196. auto base_title = MUST("anon@courage:~"_string);
  197. auto result = MUST(base_title.replace("[*]"sv, "(*)"sv, ReplaceMode::FirstOnly));
  198. EXPECT_EQ(result, "anon@courage:~"sv);
  199. }
  200. }
  201. TEST_CASE(reverse)
  202. {
  203. auto test_reverse = [](auto test, auto expected) {
  204. auto string = MUST(String::from_utf8(test));
  205. auto result = MUST(string.reverse());
  206. EXPECT_EQ(result, expected);
  207. };
  208. test_reverse(""sv, ""sv);
  209. test_reverse("a"sv, "a"sv);
  210. test_reverse("ab"sv, "ba"sv);
  211. test_reverse("ab cd ef"sv, "fe dc ba"sv);
  212. test_reverse("😀"sv, "😀"sv);
  213. test_reverse("ab😀cd"sv, "dc😀ba"sv);
  214. }
  215. TEST_CASE(to_lowercase)
  216. {
  217. {
  218. auto string = MUST("Aa"_string);
  219. auto result = MUST(string.to_lowercase());
  220. EXPECT_EQ(result, "aa"sv);
  221. }
  222. {
  223. auto string = MUST("Ωω"_string);
  224. auto result = MUST(string.to_lowercase());
  225. EXPECT_EQ(result, "ωω"sv);
  226. }
  227. {
  228. auto string = MUST("İi̇"_string);
  229. auto result = MUST(string.to_lowercase());
  230. EXPECT_EQ(result, "i̇i̇"sv);
  231. }
  232. }
  233. TEST_CASE(to_uppercase)
  234. {
  235. {
  236. auto string = MUST("Aa"_string);
  237. auto result = MUST(string.to_uppercase());
  238. EXPECT_EQ(result, "AA"sv);
  239. }
  240. {
  241. auto string = MUST("Ωω"_string);
  242. auto result = MUST(string.to_uppercase());
  243. EXPECT_EQ(result, "ΩΩ"sv);
  244. }
  245. {
  246. auto string = MUST("ʼn"_string);
  247. auto result = MUST(string.to_uppercase());
  248. EXPECT_EQ(result, "ʼN"sv);
  249. }
  250. }
  251. TEST_CASE(to_titlecase)
  252. {
  253. {
  254. auto string = MUST("foo bar baz"_string);
  255. auto result = MUST(string.to_titlecase());
  256. EXPECT_EQ(result, "Foo Bar Baz"sv);
  257. }
  258. {
  259. auto string = MUST("foo \n \r bar \t baz"_string);
  260. auto result = MUST(string.to_titlecase());
  261. EXPECT_EQ(result, "Foo \n \r Bar \t Baz"sv);
  262. }
  263. {
  264. auto string = MUST("f\"oo\" b'ar'"_string);
  265. auto result = MUST(string.to_titlecase());
  266. EXPECT_EQ(result, "F\"Oo\" B'ar'"sv);
  267. }
  268. {
  269. auto string = MUST("123dollars"_string);
  270. auto result = MUST(string.to_titlecase());
  271. EXPECT_EQ(result, "123Dollars"sv);
  272. }
  273. }
  274. TEST_CASE(equals_ignoring_case)
  275. {
  276. {
  277. String string1 {};
  278. String string2 {};
  279. EXPECT(string1.equals_ignoring_case(string2));
  280. }
  281. {
  282. auto string1 = MUST("abcd"_string);
  283. auto string2 = MUST("ABCD"_string);
  284. auto string3 = MUST("AbCd"_string);
  285. auto string4 = MUST("dcba"_string);
  286. auto string5 = MUST("abce"_string);
  287. auto string6 = MUST("abc"_string);
  288. EXPECT(string1.equals_ignoring_case(string2));
  289. EXPECT(string1.equals_ignoring_case(string3));
  290. EXPECT(!string1.equals_ignoring_case(string4));
  291. EXPECT(!string1.equals_ignoring_case(string5));
  292. EXPECT(!string1.equals_ignoring_case(string6));
  293. EXPECT(string2.equals_ignoring_case(string1));
  294. EXPECT(string2.equals_ignoring_case(string3));
  295. EXPECT(!string2.equals_ignoring_case(string4));
  296. EXPECT(!string2.equals_ignoring_case(string5));
  297. EXPECT(!string2.equals_ignoring_case(string6));
  298. EXPECT(string3.equals_ignoring_case(string1));
  299. EXPECT(string3.equals_ignoring_case(string2));
  300. EXPECT(!string3.equals_ignoring_case(string4));
  301. EXPECT(!string3.equals_ignoring_case(string5));
  302. EXPECT(!string3.equals_ignoring_case(string6));
  303. }
  304. {
  305. auto string1 = MUST("\u00DF"_string); // LATIN SMALL LETTER SHARP S
  306. auto string2 = MUST("SS"_string);
  307. auto string3 = MUST("Ss"_string);
  308. auto string4 = MUST("ss"_string);
  309. auto string5 = MUST("S"_string);
  310. auto string6 = MUST("s"_string);
  311. EXPECT(string1.equals_ignoring_case(string2));
  312. EXPECT(string1.equals_ignoring_case(string3));
  313. EXPECT(string1.equals_ignoring_case(string4));
  314. EXPECT(!string1.equals_ignoring_case(string5));
  315. EXPECT(!string1.equals_ignoring_case(string6));
  316. EXPECT(string2.equals_ignoring_case(string1));
  317. EXPECT(string2.equals_ignoring_case(string3));
  318. EXPECT(string2.equals_ignoring_case(string4));
  319. EXPECT(!string2.equals_ignoring_case(string5));
  320. EXPECT(!string2.equals_ignoring_case(string6));
  321. EXPECT(string3.equals_ignoring_case(string1));
  322. EXPECT(string3.equals_ignoring_case(string2));
  323. EXPECT(string3.equals_ignoring_case(string4));
  324. EXPECT(!string3.equals_ignoring_case(string5));
  325. EXPECT(!string3.equals_ignoring_case(string6));
  326. EXPECT(string4.equals_ignoring_case(string1));
  327. EXPECT(string4.equals_ignoring_case(string2));
  328. EXPECT(string4.equals_ignoring_case(string3));
  329. EXPECT(!string4.equals_ignoring_case(string5));
  330. EXPECT(!string4.equals_ignoring_case(string6));
  331. }
  332. {
  333. auto string1 = MUST("Ab\u00DFCd\u00DFeF"_string);
  334. auto string2 = MUST("ABSSCDSSEF"_string);
  335. auto string3 = MUST("absscdssef"_string);
  336. auto string4 = MUST("aBSscDsSEf"_string);
  337. auto string5 = MUST("Ab\u00DFCd\u00DFeg"_string);
  338. auto string6 = MUST("Ab\u00DFCd\u00DFe"_string);
  339. EXPECT(string1.equals_ignoring_case(string1));
  340. EXPECT(string1.equals_ignoring_case(string2));
  341. EXPECT(string1.equals_ignoring_case(string3));
  342. EXPECT(string1.equals_ignoring_case(string4));
  343. EXPECT(!string1.equals_ignoring_case(string5));
  344. EXPECT(!string1.equals_ignoring_case(string6));
  345. EXPECT(string2.equals_ignoring_case(string1));
  346. EXPECT(string2.equals_ignoring_case(string2));
  347. EXPECT(string2.equals_ignoring_case(string3));
  348. EXPECT(string2.equals_ignoring_case(string4));
  349. EXPECT(!string2.equals_ignoring_case(string5));
  350. EXPECT(!string2.equals_ignoring_case(string6));
  351. EXPECT(string3.equals_ignoring_case(string1));
  352. EXPECT(string3.equals_ignoring_case(string2));
  353. EXPECT(string3.equals_ignoring_case(string3));
  354. EXPECT(string3.equals_ignoring_case(string4));
  355. EXPECT(!string3.equals_ignoring_case(string5));
  356. EXPECT(!string3.equals_ignoring_case(string6));
  357. EXPECT(string4.equals_ignoring_case(string1));
  358. EXPECT(string4.equals_ignoring_case(string2));
  359. EXPECT(string4.equals_ignoring_case(string3));
  360. EXPECT(string4.equals_ignoring_case(string4));
  361. EXPECT(!string4.equals_ignoring_case(string5));
  362. EXPECT(!string4.equals_ignoring_case(string6));
  363. }
  364. }
  365. TEST_CASE(is_one_of)
  366. {
  367. auto foo = MUST("foo"_string);
  368. auto bar = MUST("bar"_string);
  369. EXPECT(foo.is_one_of(foo));
  370. EXPECT(foo.is_one_of(foo, bar));
  371. EXPECT(foo.is_one_of(bar, foo));
  372. EXPECT(!foo.is_one_of(bar));
  373. EXPECT(!bar.is_one_of("foo"sv));
  374. EXPECT(bar.is_one_of("foo"sv, "bar"sv));
  375. EXPECT(bar.is_one_of("bar"sv, "foo"sv));
  376. EXPECT(bar.is_one_of("bar"sv));
  377. }
  378. TEST_CASE(split)
  379. {
  380. {
  381. auto test = MUST("foo bar baz"_string);
  382. auto parts = MUST(test.split(' '));
  383. EXPECT_EQ(parts.size(), 3u);
  384. EXPECT_EQ(parts[0], "foo");
  385. EXPECT_EQ(parts[1], "bar");
  386. EXPECT_EQ(parts[2], "baz");
  387. }
  388. {
  389. auto test = MUST("ωΣ2ωΣω"_string);
  390. auto parts = MUST(test.split(0x03A3u));
  391. EXPECT_EQ(parts.size(), 3u);
  392. EXPECT_EQ(parts[0], "ω"sv);
  393. EXPECT_EQ(parts[1], "2ω"sv);
  394. EXPECT_EQ(parts[2], "ω"sv);
  395. }
  396. }
  397. TEST_CASE(find_byte_offset)
  398. {
  399. {
  400. String string {};
  401. auto index1 = string.find_byte_offset(0);
  402. EXPECT(!index1.has_value());
  403. auto index2 = string.find_byte_offset(""sv);
  404. EXPECT(!index2.has_value());
  405. }
  406. {
  407. auto string = MUST("foo"_string);
  408. auto index1 = string.find_byte_offset('f');
  409. EXPECT_EQ(index1, 0u);
  410. auto index2 = string.find_byte_offset('o');
  411. EXPECT_EQ(index2, 1u);
  412. auto index3 = string.find_byte_offset('o', *index2 + 1);
  413. EXPECT_EQ(index3, 2u);
  414. auto index4 = string.find_byte_offset('b');
  415. EXPECT(!index4.has_value());
  416. }
  417. {
  418. auto string = MUST("foo"_string);
  419. auto index1 = string.find_byte_offset("fo"sv);
  420. EXPECT_EQ(index1, 0u);
  421. auto index2 = string.find_byte_offset("oo"sv);
  422. EXPECT_EQ(index2, 1u);
  423. auto index3 = string.find_byte_offset("o"sv, *index2 + 1);
  424. EXPECT_EQ(index3, 2u);
  425. auto index4 = string.find_byte_offset("fooo"sv);
  426. EXPECT(!index4.has_value());
  427. }
  428. {
  429. auto string = MUST("ωΣωΣω"_string);
  430. auto index1 = string.find_byte_offset(0x03C9U);
  431. EXPECT_EQ(index1, 0u);
  432. auto index2 = string.find_byte_offset(0x03A3u);
  433. EXPECT_EQ(index2, 2u);
  434. auto index3 = string.find_byte_offset(0x03C9U, 2);
  435. EXPECT_EQ(index3, 4u);
  436. auto index4 = string.find_byte_offset(0x03A3u, 4);
  437. EXPECT_EQ(index4, 6u);
  438. auto index5 = string.find_byte_offset(0x03C9U, 6);
  439. EXPECT_EQ(index5, 8u);
  440. }
  441. {
  442. auto string = MUST("ωΣωΣω"_string);
  443. auto index1 = string.find_byte_offset("ω"sv);
  444. EXPECT_EQ(index1, 0u);
  445. auto index2 = string.find_byte_offset("Σ"sv);
  446. EXPECT_EQ(index2, 2u);
  447. auto index3 = string.find_byte_offset("ω"sv, 2);
  448. EXPECT_EQ(index3, 4u);
  449. auto index4 = string.find_byte_offset("Σ"sv, 4);
  450. EXPECT_EQ(index4, 6u);
  451. auto index5 = string.find_byte_offset("ω"sv, 6);
  452. EXPECT_EQ(index5, 8u);
  453. }
  454. }
  455. TEST_CASE(repeated)
  456. {
  457. {
  458. auto string1 = MUST(String::repeated('a', 0));
  459. EXPECT(string1.is_short_string());
  460. EXPECT(string1.is_empty());
  461. auto string2 = MUST(String::repeated(0x03C9U, 0));
  462. EXPECT(string2.is_short_string());
  463. EXPECT(string2.is_empty());
  464. auto string3 = MUST(String::repeated(0x10300, 0));
  465. EXPECT(string3.is_short_string());
  466. EXPECT(string3.is_empty());
  467. }
  468. {
  469. auto string1 = MUST(String::repeated('a', 1));
  470. EXPECT(string1.is_short_string());
  471. EXPECT_EQ(string1.bytes_as_string_view().length(), 1u);
  472. EXPECT_EQ(string1, "a"sv);
  473. auto string2 = MUST(String::repeated(0x03C9U, 1));
  474. EXPECT(string2.is_short_string());
  475. EXPECT_EQ(string2.bytes_as_string_view().length(), 2u);
  476. EXPECT_EQ(string2, "ω"sv);
  477. auto string3 = MUST(String::repeated(0x10300, 1));
  478. #ifdef AK_ARCH_64_BIT
  479. EXPECT(string3.is_short_string());
  480. #else
  481. EXPECT(!string3.is_short_string());
  482. #endif
  483. EXPECT_EQ(string3.bytes_as_string_view().length(), 4u);
  484. EXPECT_EQ(string3, "𐌀"sv);
  485. }
  486. {
  487. auto string1 = MUST(String::repeated('a', 3));
  488. EXPECT(string1.is_short_string());
  489. EXPECT_EQ(string1.bytes_as_string_view().length(), 3u);
  490. EXPECT_EQ(string1, "aaa"sv);
  491. auto string2 = MUST(String::repeated(0x03C9U, 3));
  492. #ifdef AK_ARCH_64_BIT
  493. EXPECT(string2.is_short_string());
  494. #else
  495. EXPECT(!string2.is_short_string());
  496. #endif
  497. EXPECT_EQ(string2.bytes_as_string_view().length(), 6u);
  498. EXPECT_EQ(string2, "ωωω"sv);
  499. auto string3 = MUST(String::repeated(0x10300, 3));
  500. EXPECT(!string3.is_short_string());
  501. EXPECT_EQ(string3.bytes_as_string_view().length(), 12u);
  502. EXPECT_EQ(string3, "𐌀𐌀𐌀"sv);
  503. }
  504. {
  505. auto string1 = MUST(String::repeated('a', 10));
  506. EXPECT(!string1.is_short_string());
  507. EXPECT_EQ(string1.bytes_as_string_view().length(), 10u);
  508. EXPECT_EQ(string1, "aaaaaaaaaa"sv);
  509. auto string2 = MUST(String::repeated(0x03C9U, 10));
  510. EXPECT(!string2.is_short_string());
  511. EXPECT_EQ(string2.bytes_as_string_view().length(), 20u);
  512. EXPECT_EQ(string2, "ωωωωωωωωωω"sv);
  513. auto string3 = MUST(String::repeated(0x10300, 10));
  514. EXPECT(!string3.is_short_string());
  515. EXPECT_EQ(string3.bytes_as_string_view().length(), 40u);
  516. EXPECT_EQ(string3, "𐌀𐌀𐌀𐌀𐌀𐌀𐌀𐌀𐌀𐌀"sv);
  517. }
  518. EXPECT_CRASH("Creating a string from an invalid code point", [] {
  519. (void)String::repeated(0xffffffff, 1);
  520. return Test::Crash::Failure::DidNotCrash;
  521. });
  522. }
  523. TEST_CASE(join)
  524. {
  525. auto string1 = MUST(String::join(',', Vector<i32> {}));
  526. EXPECT(string1.is_empty());
  527. auto string2 = MUST(String::join(',', Array { 1 }));
  528. EXPECT_EQ(string2, "1"sv);
  529. auto string3 = MUST(String::join(':', Array { 1 }, "[{}]"sv));
  530. EXPECT_EQ(string3, "[1]"sv);
  531. auto string4 = MUST(String::join(',', Array { 1, 2, 3 }));
  532. EXPECT_EQ(string4, "1,2,3"sv);
  533. auto string5 = MUST(String::join(',', Array { 1, 2, 3 }, "[{}]"sv));
  534. EXPECT_EQ(string5, "[1],[2],[3]"sv);
  535. auto string6 = MUST(String::join("!!!"_short_string, Array { "foo"sv, "bar"sv, "baz"sv }));
  536. EXPECT_EQ(string6, "foo!!!bar!!!baz"sv);
  537. auto string7 = MUST(String::join(" - "sv, Array { 1, 16, 256, 4096 }, "[{:#04x}]"sv));
  538. EXPECT_EQ(string7, "[0x0001] - [0x0010] - [0x0100] - [0x1000]"sv);
  539. }
  540. TEST_CASE(trim)
  541. {
  542. {
  543. String string {};
  544. auto result = MUST(string.trim(" "sv, TrimMode::Both));
  545. EXPECT(result.is_empty());
  546. result = MUST(string.trim(" "sv, TrimMode::Left));
  547. EXPECT(result.is_empty());
  548. result = MUST(string.trim(" "sv, TrimMode::Right));
  549. EXPECT(result.is_empty());
  550. }
  551. {
  552. auto string = MUST("word"_string);
  553. auto result = MUST(string.trim(" "sv, TrimMode::Both));
  554. EXPECT_EQ(result, "word"sv);
  555. result = MUST(string.trim(" "sv, TrimMode::Left));
  556. EXPECT_EQ(result, "word"sv);
  557. result = MUST(string.trim(" "sv, TrimMode::Right));
  558. EXPECT_EQ(result, "word"sv);
  559. }
  560. {
  561. auto string = MUST(" word"_string);
  562. auto result = MUST(string.trim(" "sv, TrimMode::Both));
  563. EXPECT_EQ(result, "word"sv);
  564. result = MUST(string.trim(" "sv, TrimMode::Left));
  565. EXPECT_EQ(result, "word"sv);
  566. result = MUST(string.trim(" "sv, TrimMode::Right));
  567. EXPECT_EQ(result, " word"sv);
  568. }
  569. {
  570. auto string = MUST("word "_string);
  571. auto result = MUST(string.trim(" "sv, TrimMode::Both));
  572. EXPECT_EQ(result, "word"sv);
  573. result = MUST(string.trim(" "sv, TrimMode::Left));
  574. EXPECT_EQ(result, "word "sv);
  575. result = MUST(string.trim(" "sv, TrimMode::Right));
  576. EXPECT_EQ(result, "word"sv);
  577. }
  578. {
  579. auto string = MUST(" word "_string);
  580. auto result = MUST(string.trim(" "sv, TrimMode::Both));
  581. EXPECT_EQ(result, "word"sv);
  582. result = MUST(string.trim(" "sv, TrimMode::Left));
  583. EXPECT_EQ(result, "word "sv);
  584. result = MUST(string.trim(" "sv, TrimMode::Right));
  585. EXPECT_EQ(result, " word"sv);
  586. }
  587. {
  588. auto string = MUST(" word "_string);
  589. auto result = MUST(string.trim("\t"sv, TrimMode::Both));
  590. EXPECT_EQ(result, " word "sv);
  591. result = MUST(string.trim("\t"sv, TrimMode::Left));
  592. EXPECT_EQ(result, " word "sv);
  593. result = MUST(string.trim("\t"sv, TrimMode::Right));
  594. EXPECT_EQ(result, " word "sv);
  595. }
  596. {
  597. auto string = MUST("ωΣωΣω"_string);
  598. auto result = MUST(string.trim("ω"sv, TrimMode::Both));
  599. EXPECT_EQ(result, "ΣωΣ"sv);
  600. result = MUST(string.trim("ω"sv, TrimMode::Left));
  601. EXPECT_EQ(result, "ΣωΣω"sv);
  602. result = MUST(string.trim("ω"sv, TrimMode::Right));
  603. EXPECT_EQ(result, "ωΣωΣ"sv);
  604. }
  605. {
  606. auto string = MUST("ωΣωΣω"_string);
  607. auto result = MUST(string.trim("ωΣ"sv, TrimMode::Both));
  608. EXPECT(result.is_empty());
  609. result = MUST(string.trim("ωΣ"sv, TrimMode::Left));
  610. EXPECT(result.is_empty());
  611. result = MUST(string.trim("ωΣ"sv, TrimMode::Right));
  612. EXPECT(result.is_empty());
  613. }
  614. {
  615. auto string = MUST("ωΣωΣω"_string);
  616. auto result = MUST(string.trim("Σω"sv, TrimMode::Both));
  617. EXPECT(result.is_empty());
  618. result = MUST(string.trim("Σω"sv, TrimMode::Left));
  619. EXPECT(result.is_empty());
  620. result = MUST(string.trim("Σω"sv, TrimMode::Right));
  621. EXPECT(result.is_empty());
  622. }
  623. }
  624. TEST_CASE(contains)
  625. {
  626. EXPECT(!String {}.contains({}));
  627. EXPECT(!String {}.contains(" "sv));
  628. EXPECT(!String {}.contains(0));
  629. EXPECT("a"_short_string.contains("a"sv));
  630. EXPECT(!"a"_short_string.contains({}));
  631. EXPECT(!"a"_short_string.contains("b"sv));
  632. EXPECT(!"a"_short_string.contains("ab"sv));
  633. EXPECT("a"_short_string.contains(0x0061));
  634. EXPECT(!"a"_short_string.contains(0x0062));
  635. EXPECT("abc"_short_string.contains("a"sv));
  636. EXPECT("abc"_short_string.contains("b"sv));
  637. EXPECT("abc"_short_string.contains("c"sv));
  638. EXPECT("abc"_short_string.contains("ab"sv));
  639. EXPECT("abc"_short_string.contains("bc"sv));
  640. EXPECT("abc"_short_string.contains("abc"sv));
  641. EXPECT(!"abc"_short_string.contains({}));
  642. EXPECT(!"abc"_short_string.contains("ac"sv));
  643. EXPECT(!"abc"_short_string.contains("abcd"sv));
  644. EXPECT("abc"_short_string.contains(0x0061));
  645. EXPECT("abc"_short_string.contains(0x0062));
  646. EXPECT("abc"_short_string.contains(0x0063));
  647. EXPECT(!"abc"_short_string.contains(0x0064));
  648. auto emoji = MUST("😀"_string);
  649. EXPECT(emoji.contains("\xF0"sv));
  650. EXPECT(emoji.contains("\x9F"sv));
  651. EXPECT(emoji.contains("\x98"sv));
  652. EXPECT(emoji.contains("\x80"sv));
  653. EXPECT(emoji.contains("\xF0\x9F"sv));
  654. EXPECT(emoji.contains("\xF0\x9F\x98"sv));
  655. EXPECT(emoji.contains("\xF0\x9F\x98\x80"sv));
  656. EXPECT(emoji.contains("\x9F\x98\x80"sv));
  657. EXPECT(emoji.contains("\x98\x80"sv));
  658. EXPECT(!emoji.contains("a"sv));
  659. EXPECT(!emoji.contains("🙃"sv));
  660. EXPECT(emoji.contains(0x1F600));
  661. EXPECT(!emoji.contains(0x1F643));
  662. }
  663. TEST_CASE(starts_with)
  664. {
  665. EXPECT(String {}.starts_with_bytes({}));
  666. EXPECT(!String {}.starts_with_bytes(" "sv));
  667. EXPECT(!String {}.starts_with(0));
  668. EXPECT("a"_short_string.starts_with_bytes({}));
  669. EXPECT("a"_short_string.starts_with_bytes("a"sv));
  670. EXPECT(!"a"_short_string.starts_with_bytes("b"sv));
  671. EXPECT(!"a"_short_string.starts_with_bytes("ab"sv));
  672. EXPECT("a"_short_string.starts_with(0x0061));
  673. EXPECT(!"a"_short_string.starts_with(0x0062));
  674. EXPECT("abc"_short_string.starts_with_bytes({}));
  675. EXPECT("abc"_short_string.starts_with_bytes("a"sv));
  676. EXPECT("abc"_short_string.starts_with_bytes("ab"sv));
  677. EXPECT("abc"_short_string.starts_with_bytes("abc"sv));
  678. EXPECT(!"abc"_short_string.starts_with_bytes("b"sv));
  679. EXPECT(!"abc"_short_string.starts_with_bytes("bc"sv));
  680. EXPECT("abc"_short_string.starts_with(0x0061));
  681. EXPECT(!"abc"_short_string.starts_with(0x0062));
  682. EXPECT(!"abc"_short_string.starts_with(0x0063));
  683. auto emoji = MUST("😀🙃"_string);
  684. EXPECT(emoji.starts_with_bytes("\xF0"sv));
  685. EXPECT(emoji.starts_with_bytes("\xF0\x9F"sv));
  686. EXPECT(emoji.starts_with_bytes("\xF0\x9F\x98"sv));
  687. EXPECT(emoji.starts_with_bytes("\xF0\x9F\x98\x80"sv));
  688. EXPECT(emoji.starts_with_bytes("\xF0\x9F\x98\x80\xF0"sv));
  689. EXPECT(emoji.starts_with_bytes("\xF0\x9F\x98\x80\xF0\x9F"sv));
  690. EXPECT(emoji.starts_with_bytes("\xF0\x9F\x98\x80\xF0\x9F\x99"sv));
  691. EXPECT(emoji.starts_with_bytes("\xF0\x9F\x98\x80\xF0\x9F\x99\x83"sv));
  692. EXPECT(!emoji.starts_with_bytes("a"sv));
  693. EXPECT(!emoji.starts_with_bytes("🙃"sv));
  694. EXPECT(emoji.starts_with(0x1F600));
  695. EXPECT(!emoji.starts_with(0x1F643));
  696. }
  697. TEST_CASE(ends_with)
  698. {
  699. EXPECT(String {}.ends_with_bytes({}));
  700. EXPECT(!String {}.ends_with_bytes(" "sv));
  701. EXPECT(!String {}.ends_with(0));
  702. EXPECT("a"_short_string.ends_with_bytes({}));
  703. EXPECT("a"_short_string.ends_with_bytes("a"sv));
  704. EXPECT(!"a"_short_string.ends_with_bytes("b"sv));
  705. EXPECT(!"a"_short_string.ends_with_bytes("ba"sv));
  706. EXPECT("a"_short_string.ends_with(0x0061));
  707. EXPECT(!"a"_short_string.ends_with(0x0062));
  708. EXPECT("abc"_short_string.ends_with_bytes({}));
  709. EXPECT("abc"_short_string.ends_with_bytes("c"sv));
  710. EXPECT("abc"_short_string.ends_with_bytes("bc"sv));
  711. EXPECT("abc"_short_string.ends_with_bytes("abc"sv));
  712. EXPECT(!"abc"_short_string.ends_with_bytes("b"sv));
  713. EXPECT(!"abc"_short_string.ends_with_bytes("ab"sv));
  714. EXPECT("abc"_short_string.ends_with(0x0063));
  715. EXPECT(!"abc"_short_string.ends_with(0x0062));
  716. EXPECT(!"abc"_short_string.ends_with(0x0061));
  717. auto emoji = MUST("😀🙃"_string);
  718. EXPECT(emoji.ends_with_bytes("\x83"sv));
  719. EXPECT(emoji.ends_with_bytes("\x99\x83"sv));
  720. EXPECT(emoji.ends_with_bytes("\x9F\x99\x83"sv));
  721. EXPECT(emoji.ends_with_bytes("\xF0\x9F\x99\x83"sv));
  722. EXPECT(emoji.ends_with_bytes("\x80\xF0\x9F\x99\x83"sv));
  723. EXPECT(emoji.ends_with_bytes("\x98\x80\xF0\x9F\x99\x83"sv));
  724. EXPECT(emoji.ends_with_bytes("\x9F\x98\x80\xF0\x9F\x99\x83"sv));
  725. EXPECT(emoji.ends_with_bytes("\xF0\x9F\x98\x80\xF0\x9F\x99\x83"sv));
  726. EXPECT(!emoji.ends_with_bytes("a"sv));
  727. EXPECT(!emoji.ends_with_bytes("😀"sv));
  728. EXPECT(emoji.ends_with(0x1F643));
  729. EXPECT(!emoji.ends_with(0x1F600));
  730. }