1 //===- llvm/unittest/ADT/StringRefTest.cpp - StringRef unit tests ---------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "gtest/gtest.h" 11 #include "llvm/ADT/StringRef.h" 12 #include "llvm/ADT/SmallVector.h" 13 #include "llvm/Support/raw_ostream.h" 14 using namespace llvm; 15 16 namespace llvm { 17 18 std::ostream &operator<<(std::ostream &OS, const StringRef &S) { 19 OS << S.str(); 20 return OS; 21 } 22 23 std::ostream &operator<<(std::ostream &OS, 24 const std::pair<StringRef, StringRef> &P) { 25 OS << "(" << P.first << ", " << P.second << ")"; 26 return OS; 27 } 28 29 } 30 31 namespace { 32 TEST(StringRefTest, Construction) { 33 EXPECT_EQ("", StringRef()); 34 EXPECT_EQ("hello", StringRef("hello")); 35 EXPECT_EQ("hello", StringRef("hello world", 5)); 36 EXPECT_EQ("hello", StringRef(std::string("hello"))); 37 } 38 39 TEST(StringRefTest, Iteration) { 40 StringRef S("hello"); 41 const char *p = "hello"; 42 for (const char *it = S.begin(), *ie = S.end(); it != ie; ++it, ++p) 43 EXPECT_EQ(*it, *p); 44 } 45 46 TEST(StringRefTest, StringOps) { 47 const char *p = "hello"; 48 EXPECT_EQ(p, StringRef(p, 0).data()); 49 EXPECT_TRUE(StringRef().empty()); 50 EXPECT_EQ((size_t) 5, StringRef("hello").size()); 51 EXPECT_EQ(-1, StringRef("aab").compare("aad")); 52 EXPECT_EQ( 0, StringRef("aab").compare("aab")); 53 EXPECT_EQ( 1, StringRef("aab").compare("aaa")); 54 EXPECT_EQ(-1, StringRef("aab").compare("aabb")); 55 EXPECT_EQ( 1, StringRef("aab").compare("aa")); 56 EXPECT_EQ( 1, StringRef("\xFF").compare("\1")); 57 58 EXPECT_EQ(-1, StringRef("AaB").compare_lower("aAd")); 59 EXPECT_EQ( 0, StringRef("AaB").compare_lower("aab")); 60 EXPECT_EQ( 1, StringRef("AaB").compare_lower("AAA")); 61 EXPECT_EQ(-1, StringRef("AaB").compare_lower("aaBb")); 62 EXPECT_EQ( 1, StringRef("AaB").compare_lower("aA")); 63 EXPECT_EQ( 1, StringRef("\xFF").compare_lower("\1")); 64 65 EXPECT_EQ(-1, StringRef("aab").compare_numeric("aad")); 66 EXPECT_EQ( 0, StringRef("aab").compare_numeric("aab")); 67 EXPECT_EQ( 1, StringRef("aab").compare_numeric("aaa")); 68 EXPECT_EQ(-1, StringRef("aab").compare_numeric("aabb")); 69 EXPECT_EQ( 1, StringRef("aab").compare_numeric("aa")); 70 EXPECT_EQ(-1, StringRef("1").compare_numeric("10")); 71 EXPECT_EQ( 0, StringRef("10").compare_numeric("10")); 72 EXPECT_EQ( 0, StringRef("10a").compare_numeric("10a")); 73 EXPECT_EQ( 1, StringRef("2").compare_numeric("1")); 74 EXPECT_EQ( 0, StringRef("llvm_v1i64_ty").compare_numeric("llvm_v1i64_ty")); 75 EXPECT_EQ( 1, StringRef("\xFF").compare_numeric("\1")); 76 EXPECT_EQ( 1, StringRef("V16").compare_numeric("V1_q0")); 77 EXPECT_EQ(-1, StringRef("V1_q0").compare_numeric("V16")); 78 EXPECT_EQ(-1, StringRef("V8_q0").compare_numeric("V16")); 79 EXPECT_EQ( 1, StringRef("V16").compare_numeric("V8_q0")); 80 EXPECT_EQ(-1, StringRef("V1_q0").compare_numeric("V8_q0")); 81 EXPECT_EQ( 1, StringRef("V8_q0").compare_numeric("V1_q0")); 82 } 83 84 TEST(StringRefTest, Operators) { 85 EXPECT_EQ("", StringRef()); 86 EXPECT_TRUE(StringRef("aab") < StringRef("aad")); 87 EXPECT_FALSE(StringRef("aab") < StringRef("aab")); 88 EXPECT_TRUE(StringRef("aab") <= StringRef("aab")); 89 EXPECT_FALSE(StringRef("aab") <= StringRef("aaa")); 90 EXPECT_TRUE(StringRef("aad") > StringRef("aab")); 91 EXPECT_FALSE(StringRef("aab") > StringRef("aab")); 92 EXPECT_TRUE(StringRef("aab") >= StringRef("aab")); 93 EXPECT_FALSE(StringRef("aaa") >= StringRef("aab")); 94 EXPECT_EQ(StringRef("aab"), StringRef("aab")); 95 EXPECT_FALSE(StringRef("aab") == StringRef("aac")); 96 EXPECT_FALSE(StringRef("aab") != StringRef("aab")); 97 EXPECT_TRUE(StringRef("aab") != StringRef("aac")); 98 EXPECT_EQ('a', StringRef("aab")[1]); 99 } 100 101 TEST(StringRefTest, Substr) { 102 StringRef Str("hello"); 103 EXPECT_EQ("lo", Str.substr(3)); 104 EXPECT_EQ("", Str.substr(100)); 105 EXPECT_EQ("hello", Str.substr(0, 100)); 106 EXPECT_EQ("o", Str.substr(4, 10)); 107 } 108 109 TEST(StringRefTest, Slice) { 110 StringRef Str("hello"); 111 EXPECT_EQ("l", Str.slice(2, 3)); 112 EXPECT_EQ("ell", Str.slice(1, 4)); 113 EXPECT_EQ("llo", Str.slice(2, 100)); 114 EXPECT_EQ("", Str.slice(2, 1)); 115 EXPECT_EQ("", Str.slice(10, 20)); 116 } 117 118 TEST(StringRefTest, Split) { 119 StringRef Str("hello"); 120 EXPECT_EQ(std::make_pair(StringRef("hello"), StringRef("")), 121 Str.split('X')); 122 EXPECT_EQ(std::make_pair(StringRef("h"), StringRef("llo")), 123 Str.split('e')); 124 EXPECT_EQ(std::make_pair(StringRef(""), StringRef("ello")), 125 Str.split('h')); 126 EXPECT_EQ(std::make_pair(StringRef("he"), StringRef("lo")), 127 Str.split('l')); 128 EXPECT_EQ(std::make_pair(StringRef("hell"), StringRef("")), 129 Str.split('o')); 130 131 EXPECT_EQ(std::make_pair(StringRef("hello"), StringRef("")), 132 Str.rsplit('X')); 133 EXPECT_EQ(std::make_pair(StringRef("h"), StringRef("llo")), 134 Str.rsplit('e')); 135 EXPECT_EQ(std::make_pair(StringRef(""), StringRef("ello")), 136 Str.rsplit('h')); 137 EXPECT_EQ(std::make_pair(StringRef("hel"), StringRef("o")), 138 Str.rsplit('l')); 139 EXPECT_EQ(std::make_pair(StringRef("hell"), StringRef("")), 140 Str.rsplit('o')); 141 } 142 143 TEST(StringRefTest, Split2) { 144 SmallVector<StringRef, 5> parts; 145 SmallVector<StringRef, 5> expected; 146 147 expected.push_back("ab"); expected.push_back("c"); 148 StringRef(",ab,,c,").split(parts, ",", -1, false); 149 EXPECT_TRUE(parts == expected); 150 151 expected.clear(); parts.clear(); 152 expected.push_back(""); expected.push_back("ab"); expected.push_back(""); 153 expected.push_back("c"); expected.push_back(""); 154 StringRef(",ab,,c,").split(parts, ",", -1, true); 155 EXPECT_TRUE(parts == expected); 156 157 expected.clear(); parts.clear(); 158 expected.push_back(""); 159 StringRef("").split(parts, ",", -1, true); 160 EXPECT_TRUE(parts == expected); 161 162 expected.clear(); parts.clear(); 163 StringRef("").split(parts, ",", -1, false); 164 EXPECT_TRUE(parts == expected); 165 166 expected.clear(); parts.clear(); 167 StringRef(",").split(parts, ",", -1, false); 168 EXPECT_TRUE(parts == expected); 169 170 expected.clear(); parts.clear(); 171 expected.push_back(""); expected.push_back(""); 172 StringRef(",").split(parts, ",", -1, true); 173 EXPECT_TRUE(parts == expected); 174 175 expected.clear(); parts.clear(); 176 expected.push_back("a"); expected.push_back("b"); 177 StringRef("a,b").split(parts, ",", -1, true); 178 EXPECT_TRUE(parts == expected); 179 180 // Test MaxSplit 181 expected.clear(); parts.clear(); 182 expected.push_back("a,,b,c"); 183 StringRef("a,,b,c").split(parts, ",", 0, true); 184 EXPECT_TRUE(parts == expected); 185 186 expected.clear(); parts.clear(); 187 expected.push_back("a,,b,c"); 188 StringRef("a,,b,c").split(parts, ",", 0, false); 189 EXPECT_TRUE(parts == expected); 190 191 expected.clear(); parts.clear(); 192 expected.push_back("a"); expected.push_back(",b,c"); 193 StringRef("a,,b,c").split(parts, ",", 1, true); 194 EXPECT_TRUE(parts == expected); 195 196 expected.clear(); parts.clear(); 197 expected.push_back("a"); expected.push_back(",b,c"); 198 StringRef("a,,b,c").split(parts, ",", 1, false); 199 EXPECT_TRUE(parts == expected); 200 201 expected.clear(); parts.clear(); 202 expected.push_back("a"); expected.push_back(""); expected.push_back("b,c"); 203 StringRef("a,,b,c").split(parts, ",", 2, true); 204 EXPECT_TRUE(parts == expected); 205 206 expected.clear(); parts.clear(); 207 expected.push_back("a"); expected.push_back("b,c"); 208 StringRef("a,,b,c").split(parts, ",", 2, false); 209 EXPECT_TRUE(parts == expected); 210 211 expected.clear(); parts.clear(); 212 expected.push_back("a"); expected.push_back(""); expected.push_back("b"); 213 expected.push_back("c"); 214 StringRef("a,,b,c").split(parts, ",", 3, true); 215 EXPECT_TRUE(parts == expected); 216 217 expected.clear(); parts.clear(); 218 expected.push_back("a"); expected.push_back("b"); expected.push_back("c"); 219 StringRef("a,,b,c").split(parts, ",", 3, false); 220 EXPECT_TRUE(parts == expected); 221 } 222 223 TEST(StringRefTest, StartsWith) { 224 StringRef Str("hello"); 225 EXPECT_TRUE(Str.startswith("he")); 226 EXPECT_FALSE(Str.startswith("helloworld")); 227 EXPECT_FALSE(Str.startswith("hi")); 228 } 229 230 TEST(StringRefTest, EndsWith) { 231 StringRef Str("hello"); 232 EXPECT_TRUE(Str.endswith("lo")); 233 EXPECT_FALSE(Str.endswith("helloworld")); 234 EXPECT_FALSE(Str.endswith("worldhello")); 235 EXPECT_FALSE(Str.endswith("so")); 236 } 237 238 TEST(StringRefTest, Find) { 239 StringRef Str("hello"); 240 EXPECT_EQ(2U, Str.find('l')); 241 EXPECT_EQ(StringRef::npos, Str.find('z')); 242 EXPECT_EQ(StringRef::npos, Str.find("helloworld")); 243 EXPECT_EQ(0U, Str.find("hello")); 244 EXPECT_EQ(1U, Str.find("ello")); 245 EXPECT_EQ(StringRef::npos, Str.find("zz")); 246 EXPECT_EQ(2U, Str.find("ll", 2)); 247 EXPECT_EQ(StringRef::npos, Str.find("ll", 3)); 248 249 EXPECT_EQ(3U, Str.rfind('l')); 250 EXPECT_EQ(StringRef::npos, Str.rfind('z')); 251 EXPECT_EQ(StringRef::npos, Str.rfind("helloworld")); 252 EXPECT_EQ(0U, Str.rfind("hello")); 253 EXPECT_EQ(1U, Str.rfind("ello")); 254 EXPECT_EQ(StringRef::npos, Str.rfind("zz")); 255 256 EXPECT_EQ(2U, Str.find_first_of('l')); 257 EXPECT_EQ(1U, Str.find_first_of("el")); 258 EXPECT_EQ(StringRef::npos, Str.find_first_of("xyz")); 259 260 EXPECT_EQ(1U, Str.find_first_not_of('h')); 261 EXPECT_EQ(4U, Str.find_first_not_of("hel")); 262 EXPECT_EQ(StringRef::npos, Str.find_first_not_of("hello")); 263 } 264 265 TEST(StringRefTest, Count) { 266 StringRef Str("hello"); 267 EXPECT_EQ(2U, Str.count('l')); 268 EXPECT_EQ(1U, Str.count('o')); 269 EXPECT_EQ(0U, Str.count('z')); 270 EXPECT_EQ(0U, Str.count("helloworld")); 271 EXPECT_EQ(1U, Str.count("hello")); 272 EXPECT_EQ(1U, Str.count("ello")); 273 EXPECT_EQ(0U, Str.count("zz")); 274 } 275 276 TEST(StringRefTest, EditDistance) { 277 StringRef Str("hello"); 278 EXPECT_EQ(2U, Str.edit_distance("hill")); 279 } 280 281 TEST(StringRefTest, Misc) { 282 std::string Storage; 283 raw_string_ostream OS(Storage); 284 OS << StringRef("hello"); 285 EXPECT_EQ("hello", OS.str()); 286 } 287 288 } // end anonymous namespace 289
