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