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