1 //===- llvm/unittest/ADT/SmallStringTest.cpp ------------------------------===//
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 // SmallString unit tests.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "llvm/ADT/SmallString.h"
15 #include "gtest/gtest.h"
16 #include <climits>
17 #include <cstring>
18 #include <stdarg.h>
19
20 using namespace llvm;
21
22 namespace {
23
24 // Test fixture class
25 class SmallStringTest : public testing::Test {
26 protected:
27 typedef SmallString<40> StringType;
28
29 StringType theString;
30
assertEmpty(StringType & v)31 void assertEmpty(StringType & v) {
32 // Size tests
33 EXPECT_EQ(0u, v.size());
34 EXPECT_TRUE(v.empty());
35 // Iterator tests
36 EXPECT_TRUE(v.begin() == v.end());
37 }
38 };
39
40 // New string test.
TEST_F(SmallStringTest,EmptyStringTest)41 TEST_F(SmallStringTest, EmptyStringTest) {
42 SCOPED_TRACE("EmptyStringTest");
43 assertEmpty(theString);
44 EXPECT_TRUE(theString.rbegin() == theString.rend());
45 }
46
TEST_F(SmallStringTest,AssignRepeated)47 TEST_F(SmallStringTest, AssignRepeated) {
48 theString.assign(3, 'a');
49 EXPECT_EQ(3u, theString.size());
50 EXPECT_STREQ("aaa", theString.c_str());
51 }
52
TEST_F(SmallStringTest,AssignIterPair)53 TEST_F(SmallStringTest, AssignIterPair) {
54 StringRef abc = "abc";
55 theString.assign(abc.begin(), abc.end());
56 EXPECT_EQ(3u, theString.size());
57 EXPECT_STREQ("abc", theString.c_str());
58 }
59
TEST_F(SmallStringTest,AssignStringRef)60 TEST_F(SmallStringTest, AssignStringRef) {
61 StringRef abc = "abc";
62 theString.assign(abc);
63 EXPECT_EQ(3u, theString.size());
64 EXPECT_STREQ("abc", theString.c_str());
65 }
66
TEST_F(SmallStringTest,AssignSmallVector)67 TEST_F(SmallStringTest, AssignSmallVector) {
68 StringRef abc = "abc";
69 SmallVector<char, 10> abcVec(abc.begin(), abc.end());
70 theString.assign(abcVec);
71 EXPECT_EQ(3u, theString.size());
72 EXPECT_STREQ("abc", theString.c_str());
73 }
74
TEST_F(SmallStringTest,AppendIterPair)75 TEST_F(SmallStringTest, AppendIterPair) {
76 StringRef abc = "abc";
77 theString.append(abc.begin(), abc.end());
78 theString.append(abc.begin(), abc.end());
79 EXPECT_EQ(6u, theString.size());
80 EXPECT_STREQ("abcabc", theString.c_str());
81 }
82
TEST_F(SmallStringTest,AppendStringRef)83 TEST_F(SmallStringTest, AppendStringRef) {
84 StringRef abc = "abc";
85 theString.append(abc);
86 theString.append(abc);
87 EXPECT_EQ(6u, theString.size());
88 EXPECT_STREQ("abcabc", theString.c_str());
89 }
90
TEST_F(SmallStringTest,AppendSmallVector)91 TEST_F(SmallStringTest, AppendSmallVector) {
92 StringRef abc = "abc";
93 SmallVector<char, 10> abcVec(abc.begin(), abc.end());
94 theString.append(abcVec);
95 theString.append(abcVec);
96 EXPECT_EQ(6u, theString.size());
97 EXPECT_STREQ("abcabc", theString.c_str());
98 }
99
TEST_F(SmallStringTest,Substr)100 TEST_F(SmallStringTest, Substr) {
101 theString = "hello";
102 EXPECT_EQ("lo", theString.substr(3));
103 EXPECT_EQ("", theString.substr(100));
104 EXPECT_EQ("hello", theString.substr(0, 100));
105 EXPECT_EQ("o", theString.substr(4, 10));
106 }
107
TEST_F(SmallStringTest,Slice)108 TEST_F(SmallStringTest, Slice) {
109 theString = "hello";
110 EXPECT_EQ("l", theString.slice(2, 3));
111 EXPECT_EQ("ell", theString.slice(1, 4));
112 EXPECT_EQ("llo", theString.slice(2, 100));
113 EXPECT_EQ("", theString.slice(2, 1));
114 EXPECT_EQ("", theString.slice(10, 20));
115 }
116
TEST_F(SmallStringTest,Find)117 TEST_F(SmallStringTest, Find) {
118 theString = "hello";
119 EXPECT_EQ(2U, theString.find('l'));
120 EXPECT_EQ(StringRef::npos, theString.find('z'));
121 EXPECT_EQ(StringRef::npos, theString.find("helloworld"));
122 EXPECT_EQ(0U, theString.find("hello"));
123 EXPECT_EQ(1U, theString.find("ello"));
124 EXPECT_EQ(StringRef::npos, theString.find("zz"));
125 EXPECT_EQ(2U, theString.find("ll", 2));
126 EXPECT_EQ(StringRef::npos, theString.find("ll", 3));
127 EXPECT_EQ(0U, theString.find(""));
128
129 EXPECT_EQ(3U, theString.rfind('l'));
130 EXPECT_EQ(StringRef::npos, theString.rfind('z'));
131 EXPECT_EQ(StringRef::npos, theString.rfind("helloworld"));
132 EXPECT_EQ(0U, theString.rfind("hello"));
133 EXPECT_EQ(1U, theString.rfind("ello"));
134 EXPECT_EQ(StringRef::npos, theString.rfind("zz"));
135
136 EXPECT_EQ(2U, theString.find_first_of('l'));
137 EXPECT_EQ(1U, theString.find_first_of("el"));
138 EXPECT_EQ(StringRef::npos, theString.find_first_of("xyz"));
139
140 EXPECT_EQ(1U, theString.find_first_not_of('h'));
141 EXPECT_EQ(4U, theString.find_first_not_of("hel"));
142 EXPECT_EQ(StringRef::npos, theString.find_first_not_of("hello"));
143
144 theString = "hellx xello hell ello world foo bar hello";
145 EXPECT_EQ(36U, theString.find("hello"));
146 EXPECT_EQ(28U, theString.find("foo"));
147 EXPECT_EQ(12U, theString.find("hell", 2));
148 EXPECT_EQ(0U, theString.find(""));
149 }
150
TEST_F(SmallStringTest,Count)151 TEST_F(SmallStringTest, Count) {
152 theString = "hello";
153 EXPECT_EQ(2U, theString.count('l'));
154 EXPECT_EQ(1U, theString.count('o'));
155 EXPECT_EQ(0U, theString.count('z'));
156 EXPECT_EQ(0U, theString.count("helloworld"));
157 EXPECT_EQ(1U, theString.count("hello"));
158 EXPECT_EQ(1U, theString.count("ello"));
159 EXPECT_EQ(0U, theString.count("zz"));
160 }
161
TEST_F(SmallStringTest,Realloc)162 TEST_F(SmallStringTest, Realloc) {
163 theString = "abcd";
164 theString.reserve(100);
165 EXPECT_EQ("abcd", theString);
166 unsigned const N = 100000;
167 theString.reserve(N);
168 for (unsigned i = 0; i < N - 4; ++i)
169 theString.push_back('y');
170 EXPECT_EQ("abcdyyy", theString.slice(0, 7));
171 }
172
TEST(StringRefTest,Comparisons)173 TEST(StringRefTest, Comparisons) {
174 EXPECT_EQ(-1, SmallString<10>("aab").compare("aad"));
175 EXPECT_EQ( 0, SmallString<10>("aab").compare("aab"));
176 EXPECT_EQ( 1, SmallString<10>("aab").compare("aaa"));
177 EXPECT_EQ(-1, SmallString<10>("aab").compare("aabb"));
178 EXPECT_EQ( 1, SmallString<10>("aab").compare("aa"));
179 EXPECT_EQ( 1, SmallString<10>("\xFF").compare("\1"));
180
181 EXPECT_EQ(-1, SmallString<10>("AaB").compare_lower("aAd"));
182 EXPECT_EQ( 0, SmallString<10>("AaB").compare_lower("aab"));
183 EXPECT_EQ( 1, SmallString<10>("AaB").compare_lower("AAA"));
184 EXPECT_EQ(-1, SmallString<10>("AaB").compare_lower("aaBb"));
185 EXPECT_EQ( 1, SmallString<10>("AaB").compare_lower("aA"));
186 EXPECT_EQ( 1, SmallString<10>("\xFF").compare_lower("\1"));
187
188 EXPECT_EQ(-1, SmallString<10>("aab").compare_numeric("aad"));
189 EXPECT_EQ( 0, SmallString<10>("aab").compare_numeric("aab"));
190 EXPECT_EQ( 1, SmallString<10>("aab").compare_numeric("aaa"));
191 EXPECT_EQ(-1, SmallString<10>("aab").compare_numeric("aabb"));
192 EXPECT_EQ( 1, SmallString<10>("aab").compare_numeric("aa"));
193 EXPECT_EQ(-1, SmallString<10>("1").compare_numeric("10"));
194 EXPECT_EQ( 0, SmallString<10>("10").compare_numeric("10"));
195 EXPECT_EQ( 0, SmallString<10>("10a").compare_numeric("10a"));
196 EXPECT_EQ( 1, SmallString<10>("2").compare_numeric("1"));
197 EXPECT_EQ( 0, SmallString<10>("llvm_v1i64_ty").compare_numeric("llvm_v1i64_ty"));
198 EXPECT_EQ( 1, SmallString<10>("\xFF").compare_numeric("\1"));
199 EXPECT_EQ( 1, SmallString<10>("V16").compare_numeric("V1_q0"));
200 EXPECT_EQ(-1, SmallString<10>("V1_q0").compare_numeric("V16"));
201 EXPECT_EQ(-1, SmallString<10>("V8_q0").compare_numeric("V16"));
202 EXPECT_EQ( 1, SmallString<10>("V16").compare_numeric("V8_q0"));
203 EXPECT_EQ(-1, SmallString<10>("V1_q0").compare_numeric("V8_q0"));
204 EXPECT_EQ( 1, SmallString<10>("V8_q0").compare_numeric("V1_q0"));
205 }
206
207 }
208