StringRefTest.cpp revision 437ef0cfb74102208d6e96ff7e7f1204835e66f3
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" 14using namespace llvm; 15 16namespace llvm { 17 18std::ostream &operator<<(std::ostream &OS, const StringRef &S) { 19 OS << S.str(); 20 return OS; 21} 22 23std::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 31namespace { 32TEST(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 39TEST(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 46TEST(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("aad", 3)); 59 EXPECT_EQ( 0, StringRef("aab").compare("aab", 3)); 60 EXPECT_EQ( 1, StringRef("aab").compare("aaa", 3)); 61 EXPECT_EQ(-1, StringRef("aab").compare("aabb", 4)); 62 EXPECT_EQ( 1, StringRef("aab").compare("aa", 3)); 63 EXPECT_EQ( 1, StringRef("\xFF").compare("\1", 3)); 64 EXPECT_EQ( 0, StringRef("aab").compare("aad", 2)); 65 EXPECT_EQ( 0, StringRef("aab").compare("aab", 2)); 66 EXPECT_EQ( 0, StringRef("aab").compare("aab", 4)); 67 EXPECT_EQ( 0, StringRef("aab").compare("aaa", 2)); 68 EXPECT_EQ( 0, StringRef("aab").compare("aabb", 3)); 69 EXPECT_EQ( 0, StringRef("aab").compare("aa", 2)); 70 71 EXPECT_EQ(-1, StringRef("AaB").compare_lower("aAd")); 72 EXPECT_EQ( 0, StringRef("AaB").compare_lower("aab")); 73 EXPECT_EQ( 1, StringRef("AaB").compare_lower("AAA")); 74 EXPECT_EQ(-1, StringRef("AaB").compare_lower("aaBb")); 75 EXPECT_EQ( 1, StringRef("AaB").compare_lower("aA")); 76 EXPECT_EQ( 1, StringRef("\xFF").compare_lower("\1")); 77 78 EXPECT_EQ(-1, StringRef("aab").compare_numeric("aad")); 79 EXPECT_EQ( 0, StringRef("aab").compare_numeric("aab")); 80 EXPECT_EQ( 1, StringRef("aab").compare_numeric("aaa")); 81 EXPECT_EQ(-1, StringRef("aab").compare_numeric("aabb")); 82 EXPECT_EQ( 1, StringRef("aab").compare_numeric("aa")); 83 EXPECT_EQ(-1, StringRef("1").compare_numeric("10")); 84 EXPECT_EQ( 0, StringRef("10").compare_numeric("10")); 85 EXPECT_EQ( 0, StringRef("10a").compare_numeric("10a")); 86 EXPECT_EQ( 1, StringRef("2").compare_numeric("1")); 87 EXPECT_EQ( 0, StringRef("llvm_v1i64_ty").compare_numeric("llvm_v1i64_ty")); 88 EXPECT_EQ( 1, StringRef("\xFF").compare_numeric("\1")); 89} 90 91TEST(StringRefTest, Operators) { 92 EXPECT_EQ("", StringRef()); 93 EXPECT_TRUE(StringRef("aab") < StringRef("aad")); 94 EXPECT_FALSE(StringRef("aab") < StringRef("aab")); 95 EXPECT_TRUE(StringRef("aab") <= StringRef("aab")); 96 EXPECT_FALSE(StringRef("aab") <= StringRef("aaa")); 97 EXPECT_TRUE(StringRef("aad") > StringRef("aab")); 98 EXPECT_FALSE(StringRef("aab") > StringRef("aab")); 99 EXPECT_TRUE(StringRef("aab") >= StringRef("aab")); 100 EXPECT_FALSE(StringRef("aaa") >= StringRef("aab")); 101 EXPECT_EQ(StringRef("aab"), StringRef("aab")); 102 EXPECT_FALSE(StringRef("aab") == StringRef("aac")); 103 EXPECT_FALSE(StringRef("aab") != StringRef("aab")); 104 EXPECT_TRUE(StringRef("aab") != StringRef("aac")); 105 EXPECT_EQ('a', StringRef("aab")[1]); 106} 107 108TEST(StringRefTest, Substr) { 109 StringRef Str("hello"); 110 EXPECT_EQ("lo", Str.substr(3)); 111 EXPECT_EQ("", Str.substr(100)); 112 EXPECT_EQ("hello", Str.substr(0, 100)); 113 EXPECT_EQ("o", Str.substr(4, 10)); 114} 115 116TEST(StringRefTest, Slice) { 117 StringRef Str("hello"); 118 EXPECT_EQ("l", Str.slice(2, 3)); 119 EXPECT_EQ("ell", Str.slice(1, 4)); 120 EXPECT_EQ("llo", Str.slice(2, 100)); 121 EXPECT_EQ("", Str.slice(2, 1)); 122 EXPECT_EQ("", Str.slice(10, 20)); 123} 124 125TEST(StringRefTest, Split) { 126 StringRef Str("hello"); 127 EXPECT_EQ(std::make_pair(StringRef("hello"), StringRef("")), 128 Str.split('X')); 129 EXPECT_EQ(std::make_pair(StringRef("h"), StringRef("llo")), 130 Str.split('e')); 131 EXPECT_EQ(std::make_pair(StringRef(""), StringRef("ello")), 132 Str.split('h')); 133 EXPECT_EQ(std::make_pair(StringRef("he"), StringRef("lo")), 134 Str.split('l')); 135 EXPECT_EQ(std::make_pair(StringRef("hell"), StringRef("")), 136 Str.split('o')); 137 138 EXPECT_EQ(std::make_pair(StringRef("hello"), StringRef("")), 139 Str.rsplit('X')); 140 EXPECT_EQ(std::make_pair(StringRef("h"), StringRef("llo")), 141 Str.rsplit('e')); 142 EXPECT_EQ(std::make_pair(StringRef(""), StringRef("ello")), 143 Str.rsplit('h')); 144 EXPECT_EQ(std::make_pair(StringRef("hel"), StringRef("o")), 145 Str.rsplit('l')); 146 EXPECT_EQ(std::make_pair(StringRef("hell"), StringRef("")), 147 Str.rsplit('o')); 148} 149 150TEST(StringRefTest, Split2) { 151 SmallVector<StringRef, 5> parts; 152 SmallVector<StringRef, 5> expected; 153 154 expected.push_back("ab"); expected.push_back("c"); 155 StringRef(",ab,,c,").split(parts, ",", -1, false); 156 EXPECT_TRUE(parts == expected); 157 158 expected.clear(); parts.clear(); 159 expected.push_back(""); expected.push_back("ab"); expected.push_back(""); 160 expected.push_back("c"); expected.push_back(""); 161 StringRef(",ab,,c,").split(parts, ",", -1, true); 162 EXPECT_TRUE(parts == expected); 163 164 expected.clear(); parts.clear(); 165 expected.push_back(""); 166 StringRef("").split(parts, ",", -1, true); 167 EXPECT_TRUE(parts == expected); 168 169 expected.clear(); parts.clear(); 170 StringRef("").split(parts, ",", -1, false); 171 EXPECT_TRUE(parts == expected); 172 173 expected.clear(); parts.clear(); 174 StringRef(",").split(parts, ",", -1, false); 175 EXPECT_TRUE(parts == expected); 176 177 expected.clear(); parts.clear(); 178 expected.push_back(""); expected.push_back(""); 179 StringRef(",").split(parts, ",", -1, true); 180 EXPECT_TRUE(parts == expected); 181 182 expected.clear(); parts.clear(); 183 expected.push_back("a"); expected.push_back("b"); 184 StringRef("a,b").split(parts, ",", -1, true); 185 EXPECT_TRUE(parts == expected); 186 187 // Test MaxSplit 188 expected.clear(); parts.clear(); 189 expected.push_back("a,,b,c"); 190 StringRef("a,,b,c").split(parts, ",", 0, true); 191 EXPECT_TRUE(parts == expected); 192 193 expected.clear(); parts.clear(); 194 expected.push_back("a,,b,c"); 195 StringRef("a,,b,c").split(parts, ",", 0, false); 196 EXPECT_TRUE(parts == expected); 197 198 expected.clear(); parts.clear(); 199 expected.push_back("a"); expected.push_back(",b,c"); 200 StringRef("a,,b,c").split(parts, ",", 1, true); 201 EXPECT_TRUE(parts == expected); 202 203 expected.clear(); parts.clear(); 204 expected.push_back("a"); expected.push_back(",b,c"); 205 StringRef("a,,b,c").split(parts, ",", 1, false); 206 EXPECT_TRUE(parts == expected); 207 208 expected.clear(); parts.clear(); 209 expected.push_back("a"); expected.push_back(""); expected.push_back("b,c"); 210 StringRef("a,,b,c").split(parts, ",", 2, true); 211 EXPECT_TRUE(parts == expected); 212 213 expected.clear(); parts.clear(); 214 expected.push_back("a"); expected.push_back("b,c"); 215 StringRef("a,,b,c").split(parts, ",", 2, false); 216 EXPECT_TRUE(parts == expected); 217 218 expected.clear(); parts.clear(); 219 expected.push_back("a"); expected.push_back(""); expected.push_back("b"); 220 expected.push_back("c"); 221 StringRef("a,,b,c").split(parts, ",", 3, true); 222 EXPECT_TRUE(parts == expected); 223 224 expected.clear(); parts.clear(); 225 expected.push_back("a"); expected.push_back("b"); expected.push_back("c"); 226 StringRef("a,,b,c").split(parts, ",", 3, false); 227 EXPECT_TRUE(parts == expected); 228} 229 230TEST(StringRefTest, StartsWith) { 231 StringRef Str("hello"); 232 EXPECT_TRUE(Str.startswith("he")); 233 EXPECT_FALSE(Str.startswith("helloworld")); 234 EXPECT_FALSE(Str.startswith("hi")); 235} 236 237TEST(StringRefTest, EndsWith) { 238 StringRef Str("hello"); 239 EXPECT_TRUE(Str.endswith("lo")); 240 EXPECT_FALSE(Str.endswith("helloworld")); 241 EXPECT_FALSE(Str.endswith("worldhello")); 242 EXPECT_FALSE(Str.endswith("so")); 243} 244 245TEST(StringRefTest, Find) { 246 StringRef Str("hello"); 247 EXPECT_EQ(2U, Str.find('l')); 248 EXPECT_EQ(StringRef::npos, Str.find('z')); 249 EXPECT_EQ(StringRef::npos, Str.find("helloworld")); 250 EXPECT_EQ(0U, Str.find("hello")); 251 EXPECT_EQ(1U, Str.find("ello")); 252 EXPECT_EQ(StringRef::npos, Str.find("zz")); 253 EXPECT_EQ(2U, Str.find("ll", 2)); 254 EXPECT_EQ(StringRef::npos, Str.find("ll", 3)); 255 256 EXPECT_EQ(3U, Str.rfind('l')); 257 EXPECT_EQ(StringRef::npos, Str.rfind('z')); 258 EXPECT_EQ(StringRef::npos, Str.rfind("helloworld")); 259 EXPECT_EQ(0U, Str.rfind("hello")); 260 EXPECT_EQ(1U, Str.rfind("ello")); 261 EXPECT_EQ(StringRef::npos, Str.rfind("zz")); 262 263 EXPECT_EQ(2U, Str.find_first_of('l')); 264 EXPECT_EQ(1U, Str.find_first_of("el")); 265 EXPECT_EQ(StringRef::npos, Str.find_first_of("xyz")); 266 267 EXPECT_EQ(1U, Str.find_first_not_of('h')); 268 EXPECT_EQ(4U, Str.find_first_not_of("hel")); 269 EXPECT_EQ(StringRef::npos, Str.find_first_not_of("hello")); 270} 271 272TEST(StringRefTest, Count) { 273 StringRef Str("hello"); 274 EXPECT_EQ(2U, Str.count('l')); 275 EXPECT_EQ(1U, Str.count('o')); 276 EXPECT_EQ(0U, Str.count('z')); 277 EXPECT_EQ(0U, Str.count("helloworld")); 278 EXPECT_EQ(1U, Str.count("hello")); 279 EXPECT_EQ(1U, Str.count("ello")); 280 EXPECT_EQ(0U, Str.count("zz")); 281} 282 283TEST(StringRefTest, EditDistance) { 284 StringRef Str("hello"); 285 EXPECT_EQ(2U, Str.edit_distance("hill")); 286} 287 288TEST(StringRefTest, Misc) { 289 std::string Storage; 290 raw_string_ostream OS(Storage); 291 OS << StringRef("hello"); 292 EXPECT_EQ("hello", OS.str()); 293} 294 295} // end anonymous namespace 296