1 //===- StringExtrasTest.cpp - Unit tests for String extras ----------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "llvm/ADT/StringExtras.h" 10 #include "llvm/Support/raw_ostream.h" 11 #include "gtest/gtest.h" 12 13 using namespace llvm; 14 15 TEST(StringExtrasTest, isPrint) { 16 EXPECT_FALSE(isPrint('\0')); 17 EXPECT_FALSE(isPrint('\t')); 18 EXPECT_TRUE(isPrint('0')); 19 EXPECT_TRUE(isPrint('a')); 20 EXPECT_TRUE(isPrint('A')); 21 EXPECT_TRUE(isPrint(' ')); 22 EXPECT_TRUE(isPrint('~')); 23 EXPECT_TRUE(isPrint('?')); 24 } 25 26 TEST(StringExtrasTest, isSpace) { 27 EXPECT_TRUE(isSpace(' ')); 28 EXPECT_TRUE(isSpace('\t')); 29 EXPECT_TRUE(isSpace('\n')); 30 EXPECT_TRUE(isSpace('\v')); 31 EXPECT_TRUE(isSpace('\f')); 32 EXPECT_TRUE(isSpace('\v')); 33 EXPECT_FALSE(isSpace('\0')); 34 EXPECT_FALSE(isSpace('_')); 35 } 36 37 TEST(StringExtrasTest, Join) { 38 std::vector<std::string> Items; 39 EXPECT_EQ("", join(Items.begin(), Items.end(), " <sep> ")); 40 41 Items = {"foo"}; 42 EXPECT_EQ("foo", join(Items.begin(), Items.end(), " <sep> ")); 43 44 Items = {"foo", "bar"}; 45 EXPECT_EQ("foo <sep> bar", join(Items.begin(), Items.end(), " <sep> ")); 46 47 Items = {"foo", "bar", "baz"}; 48 EXPECT_EQ("foo <sep> bar <sep> baz", 49 join(Items.begin(), Items.end(), " <sep> ")); 50 } 51 52 TEST(StringExtrasTest, JoinItems) { 53 const char *Foo = "foo"; 54 std::string Bar = "bar"; 55 llvm::StringRef Baz = "baz"; 56 char X = 'x'; 57 58 EXPECT_EQ("", join_items(" <sep> ")); 59 EXPECT_EQ("", join_items('/')); 60 61 EXPECT_EQ("foo", join_items(" <sep> ", Foo)); 62 EXPECT_EQ("foo", join_items('/', Foo)); 63 64 EXPECT_EQ("foo <sep> bar", join_items(" <sep> ", Foo, Bar)); 65 EXPECT_EQ("foo/bar", join_items('/', Foo, Bar)); 66 67 EXPECT_EQ("foo <sep> bar <sep> baz", join_items(" <sep> ", Foo, Bar, Baz)); 68 EXPECT_EQ("foo/bar/baz", join_items('/', Foo, Bar, Baz)); 69 70 EXPECT_EQ("foo <sep> bar <sep> baz <sep> x", 71 join_items(" <sep> ", Foo, Bar, Baz, X)); 72 73 EXPECT_EQ("foo/bar/baz/x", join_items('/', Foo, Bar, Baz, X)); 74 } 75 76 TEST(StringExtrasTest, ToAndFromHex) { 77 std::vector<uint8_t> OddBytes = {0x5, 0xBD, 0x0D, 0x3E, 0xCD}; 78 std::string OddStr = "05BD0D3ECD"; 79 StringRef OddData(reinterpret_cast<const char *>(OddBytes.data()), 80 OddBytes.size()); 81 EXPECT_EQ(OddStr, toHex(OddData)); 82 EXPECT_EQ(OddData, fromHex(StringRef(OddStr).drop_front())); 83 EXPECT_EQ(StringRef(OddStr).lower(), toHex(OddData, true)); 84 85 std::vector<uint8_t> EvenBytes = {0xA5, 0xBD, 0x0D, 0x3E, 0xCD}; 86 std::string EvenStr = "A5BD0D3ECD"; 87 StringRef EvenData(reinterpret_cast<const char *>(EvenBytes.data()), 88 EvenBytes.size()); 89 EXPECT_EQ(EvenStr, toHex(EvenData)); 90 EXPECT_EQ(EvenData, fromHex(EvenStr)); 91 EXPECT_EQ(StringRef(EvenStr).lower(), toHex(EvenData, true)); 92 } 93 94 TEST(StringExtrasTest, to_float) { 95 float F; 96 EXPECT_TRUE(to_float("4.7", F)); 97 EXPECT_FLOAT_EQ(4.7f, F); 98 99 double D; 100 EXPECT_TRUE(to_float("4.7", D)); 101 EXPECT_DOUBLE_EQ(4.7, D); 102 103 long double LD; 104 EXPECT_TRUE(to_float("4.7", LD)); 105 EXPECT_DOUBLE_EQ(4.7, LD); 106 107 EXPECT_FALSE(to_float("foo", F)); 108 EXPECT_FALSE(to_float("7.4 foo", F)); 109 EXPECT_FLOAT_EQ(4.7f, F); // F should be unchanged 110 } 111 112 TEST(StringExtrasTest, printLowerCase) { 113 std::string str; 114 raw_string_ostream OS(str); 115 printLowerCase("ABCdefg01234.,&!~`'}\"", OS); 116 EXPECT_EQ("abcdefg01234.,&!~`'}\"", OS.str()); 117 } 118 119 TEST(StringExtrasTest, printEscapedString) { 120 std::string str; 121 raw_string_ostream OS(str); 122 printEscapedString("ABCdef123&<>\\\"'\t", OS); 123 EXPECT_EQ("ABCdef123&<>\\\\\\22'\\09", OS.str()); 124 } 125 126 TEST(StringExtrasTest, printHTMLEscaped) { 127 std::string str; 128 raw_string_ostream OS(str); 129 printHTMLEscaped("ABCdef123&<>\"'", OS); 130 EXPECT_EQ("ABCdef123&<>"'", OS.str()); 131 } 132 133 TEST(StringExtras, ConvertToSnakeFromCamelCase) { 134 auto testConvertToSnakeCase = [](llvm::StringRef input, 135 llvm::StringRef expected) { 136 EXPECT_EQ(convertToSnakeFromCamelCase(input), expected.str()); 137 }; 138 139 testConvertToSnakeCase("OpName", "op_name"); 140 testConvertToSnakeCase("opName", "op_name"); 141 testConvertToSnakeCase("_OpName", "_op_name"); 142 testConvertToSnakeCase("Op_Name", "op_name"); 143 testConvertToSnakeCase("", ""); 144 testConvertToSnakeCase("A", "a"); 145 testConvertToSnakeCase("_", "_"); 146 testConvertToSnakeCase("a", "a"); 147 testConvertToSnakeCase("op_name", "op_name"); 148 testConvertToSnakeCase("_op_name", "_op_name"); 149 testConvertToSnakeCase("__op_name", "__op_name"); 150 testConvertToSnakeCase("op__name", "op__name"); 151 } 152 153 TEST(StringExtras, ConvertToCamelFromSnakeCase) { 154 auto testConvertToCamelCase = [](bool capitalizeFirst, llvm::StringRef input, 155 llvm::StringRef expected) { 156 EXPECT_EQ(convertToCamelFromSnakeCase(input, capitalizeFirst), 157 expected.str()); 158 }; 159 160 testConvertToCamelCase(false, "op_name", "opName"); 161 testConvertToCamelCase(false, "_op_name", "_opName"); 162 testConvertToCamelCase(false, "__op_name", "_OpName"); 163 testConvertToCamelCase(false, "op__name", "op_Name"); 164 testConvertToCamelCase(false, "", ""); 165 testConvertToCamelCase(false, "A", "A"); 166 testConvertToCamelCase(false, "_", "_"); 167 testConvertToCamelCase(false, "a", "a"); 168 testConvertToCamelCase(false, "OpName", "OpName"); 169 testConvertToCamelCase(false, "opName", "opName"); 170 testConvertToCamelCase(false, "_OpName", "_OpName"); 171 testConvertToCamelCase(false, "Op_Name", "Op_Name"); 172 testConvertToCamelCase(true, "op_name", "OpName"); 173 testConvertToCamelCase(true, "_op_name", "_opName"); 174 testConvertToCamelCase(true, "__op_name", "_OpName"); 175 testConvertToCamelCase(true, "op__name", "Op_Name"); 176 testConvertToCamelCase(true, "", ""); 177 testConvertToCamelCase(true, "A", "A"); 178 testConvertToCamelCase(true, "_", "_"); 179 testConvertToCamelCase(true, "a", "A"); 180 testConvertToCamelCase(true, "OpName", "OpName"); 181 testConvertToCamelCase(true, "_OpName", "_OpName"); 182 testConvertToCamelCase(true, "Op_Name", "Op_Name"); 183 testConvertToCamelCase(true, "opName", "OpName"); 184 } 185