1 /*
2 * Copyright (C) 2018 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "perfetto/ext/base/string_view.h"
18
19 #include <forward_list>
20 #include <unordered_map>
21 #include <unordered_set>
22
23 #include "test/gtest_and_gmock.h"
24
25 namespace perfetto {
26 namespace base {
27 namespace {
28
TEST(StringViewTest,BasicCases)29 TEST(StringViewTest, BasicCases) {
30 EXPECT_EQ(StringView(), StringView(""));
31 EXPECT_EQ(StringView(""), StringView(""));
32 EXPECT_EQ(StringView(""), StringView("", 0));
33 EXPECT_EQ(StringView("ab"), StringView("ab", 2));
34 EXPECT_EQ(StringView("ax", 1), StringView("ay", 1));
35 EXPECT_EQ(StringView("ax", 1), StringView("a"));
36 EXPECT_EQ(StringView("ax", 1), "a");
37 EXPECT_EQ(StringView(reinterpret_cast<const char*>(0x100), 0).ToStdString(),
38 std::string(""));
39 EXPECT_EQ(StringView("foo|", 3).ToStdString(), std::string("foo"));
40 EXPECT_TRUE(StringView("x") != StringView(""));
41 EXPECT_TRUE(StringView("") != StringView("y"));
42 EXPECT_TRUE(StringView("a") != StringView("b"));
43 EXPECT_EQ(StringView().size(), 0ul);
44 EXPECT_EQ(StringView().data(), nullptr);
45 EXPECT_EQ(StringView("").size(), 0ul);
46 EXPECT_NE(StringView("").data(), nullptr);
47 EXPECT_TRUE(StringView("").empty());
48 EXPECT_FALSE(StringView("x").empty());
49
50 {
51 StringView x("abc");
52 EXPECT_EQ(x.size(), 3u);
53 EXPECT_EQ(x.data()[0], 'a');
54 EXPECT_EQ(x.data()[2], 'c');
55 EXPECT_TRUE(x == "abc");
56 EXPECT_TRUE(x == StringView("abc"));
57 EXPECT_TRUE(x != StringView("abcd"));
58 EXPECT_FALSE(x == StringView("aBc"));
59 EXPECT_TRUE(x.CaseInsensitiveEq("aBc"));
60 EXPECT_TRUE(x.CaseInsensitiveEq("AbC"));
61 EXPECT_FALSE(x.CaseInsensitiveEq("AbCd"));
62 EXPECT_FALSE(x.CaseInsensitiveEq("ab"));
63 EXPECT_FALSE(x.CaseInsensitiveEq("abcd"));
64 EXPECT_FALSE(x.CaseInsensitiveEq(""));
65 EXPECT_FALSE(x.CaseInsensitiveOneOf({}));
66 EXPECT_FALSE(x.CaseInsensitiveOneOf({"AbCd", "ab", "abcd"}));
67 EXPECT_TRUE(x.CaseInsensitiveOneOf({"aBc", "AbC"}));
68 EXPECT_TRUE(x.CaseInsensitiveOneOf({"AbCd", "aBc"}));
69 }
70
71 // Test find(char).
72 EXPECT_EQ(StringView().find('x'), StringView::npos);
73 EXPECT_EQ(StringView("").find('x'), StringView::npos);
74 EXPECT_EQ(StringView("foo").find('x'), StringView::npos);
75 EXPECT_EQ(StringView("foo").find('f'), 0u);
76 EXPECT_EQ(StringView("foo").find('o'), 1u);
77
78 // Test rfind(char).
79 EXPECT_EQ(StringView().rfind('x'), StringView::npos);
80 EXPECT_EQ(StringView("").rfind('x'), StringView::npos);
81 EXPECT_EQ(StringView("foo").rfind('x'), StringView::npos);
82 EXPECT_EQ(StringView("foo").rfind('f'), 0u);
83 EXPECT_EQ(StringView("foo").rfind('o'), 2u);
84
85 // Test find(const char*).
86 EXPECT_EQ(StringView().find("x"), StringView::npos);
87 EXPECT_EQ(StringView("").find("x"), StringView::npos);
88 EXPECT_EQ(StringView("foo").find("x"), StringView::npos);
89 EXPECT_EQ(StringView("foo").find("foobar"), StringView::npos);
90 EXPECT_EQ(StringView("foo").find("f", 1), StringView::npos);
91 EXPECT_EQ(StringView("foo").find("f"), 0u);
92 EXPECT_EQ(StringView("foo").find("fo"), 0u);
93 EXPECT_EQ(StringView("foo").find("oo"), 1u);
94 EXPECT_EQ(StringView("foo").find("o"), 1u);
95 EXPECT_EQ(StringView("foo").find("o", 2), 2u);
96 EXPECT_EQ(StringView("foo").find("o", 3), StringView::npos);
97 EXPECT_EQ(StringView("foo").find("o", 10), StringView::npos);
98 EXPECT_EQ(StringView("foobar").find("bar", 3), 3u);
99 EXPECT_EQ(StringView("foobar").find("bartender"), StringView::npos);
100 EXPECT_EQ(StringView("foobar").find("bartender", 3), StringView::npos);
101 EXPECT_EQ(StringView("foo").find(""), 0u); // std::string behaves the same.
102
103 // Test substr().
104 EXPECT_EQ(StringView().substr(0, 0).ToStdString(), "");
105 EXPECT_EQ(StringView().substr(3, 1).ToStdString(), "");
106 EXPECT_EQ(StringView("foo").substr(3, 1).ToStdString(), "");
107 EXPECT_EQ(StringView("foo").substr(4, 0).ToStdString(), "");
108 EXPECT_EQ(StringView("foo").substr(4, 1).ToStdString(), "");
109 EXPECT_EQ(StringView("foo").substr(0, 1).ToStdString(), "f");
110 EXPECT_EQ(StringView("foo").substr(0, 3).ToStdString(), "foo");
111 EXPECT_EQ(StringView("foo").substr(0, 99).ToStdString(), "foo");
112 EXPECT_EQ(StringView("foo").substr(1, 2).ToStdString(), "oo");
113 EXPECT_EQ(StringView("foo").substr(1, 3).ToStdString(), "oo");
114 EXPECT_EQ(StringView("foo").substr(1, 99).ToStdString(), "oo");
115 EXPECT_EQ(StringView("xyz").substr(0).ToStdString(), "xyz");
116 EXPECT_EQ(StringView("xyz").substr(2).ToStdString(), "z");
117 EXPECT_EQ(StringView("xyz").substr(3).ToStdString(), "");
118
119 // Test the < operator.
120 EXPECT_FALSE(StringView() < StringView());
121 EXPECT_FALSE(StringView() < StringView(""));
122 EXPECT_TRUE(StringView() < StringView("foo"));
123 EXPECT_TRUE(StringView("") < StringView("foo"));
124 EXPECT_FALSE(StringView() < StringView("foo", 0));
125 EXPECT_FALSE(StringView("foo") < StringView("foo"));
126 EXPECT_TRUE(StringView("foo") < StringView("fooo"));
127 EXPECT_FALSE(StringView("fooo") < StringView("foo"));
128 EXPECT_TRUE(StringView("bar") < StringView("foo"));
129
130 // Test the <= operator.
131 EXPECT_TRUE(StringView() <= StringView());
132 EXPECT_TRUE(StringView() <= StringView(""));
133 EXPECT_TRUE(StringView() <= StringView("foo"));
134 EXPECT_TRUE(StringView("") <= StringView("foo"));
135 EXPECT_TRUE(StringView() <= StringView("foo", 0));
136 EXPECT_TRUE(StringView("foo") <= StringView("foo"));
137 EXPECT_TRUE(StringView("foo") <= StringView("fooo"));
138 EXPECT_FALSE(StringView("fooo") <= StringView("foo"));
139 EXPECT_TRUE(StringView("bar") <= StringView("foo"));
140
141 // Test the > operator.
142 EXPECT_FALSE(StringView() > StringView());
143 EXPECT_FALSE(StringView() > StringView(""));
144 EXPECT_FALSE(StringView() > StringView("foo"));
145 EXPECT_FALSE(StringView("") > StringView("foo"));
146 EXPECT_FALSE(StringView() > StringView("foo", 0));
147 EXPECT_FALSE(StringView("foo") > StringView("foo"));
148 EXPECT_FALSE(StringView("foo") > StringView("fooo"));
149 EXPECT_TRUE(StringView("fooo") > StringView("foo"));
150 EXPECT_FALSE(StringView("bar") > StringView("foo"));
151
152 // Test the >= operator.
153 EXPECT_TRUE(StringView() >= StringView());
154 EXPECT_TRUE(StringView() >= StringView(""));
155 EXPECT_FALSE(StringView() >= StringView("foo"));
156 EXPECT_FALSE(StringView("") >= StringView("foo"));
157 EXPECT_TRUE(StringView() >= StringView("foo", 0));
158 EXPECT_TRUE(StringView("foo") >= StringView("foo"));
159 EXPECT_FALSE(StringView("foo") >= StringView("fooo"));
160 EXPECT_TRUE(StringView("fooo") >= StringView("foo"));
161 EXPECT_FALSE(StringView("bar") >= StringView("foo"));
162
163 // Test StartsWith.
164 EXPECT_TRUE(StringView().StartsWith(StringView()));
165 EXPECT_TRUE(StringView().StartsWith(StringView("")));
166 EXPECT_TRUE(StringView("").StartsWith(StringView("")));
167 EXPECT_TRUE(StringView("").StartsWith(StringView()));
168 EXPECT_TRUE(StringView("foo").StartsWith(StringView()));
169 EXPECT_TRUE(StringView("foo").StartsWith(StringView("")));
170 EXPECT_FALSE(StringView().StartsWith("foo"));
171 EXPECT_FALSE(StringView("").StartsWith("foo"));
172 EXPECT_TRUE(StringView("foo").StartsWith("foo"));
173 EXPECT_TRUE(StringView("foorbar").StartsWith("foo"));
174 EXPECT_FALSE(StringView("foorbar").StartsWith("bar"));
175 EXPECT_FALSE(StringView("foo").StartsWith("fooo"));
176
177 // Test EndsWith.
178 EXPECT_TRUE(StringView().EndsWith(StringView()));
179 EXPECT_TRUE(StringView().EndsWith(StringView("")));
180 EXPECT_TRUE(StringView("").EndsWith(StringView("")));
181 EXPECT_TRUE(StringView("").EndsWith(StringView()));
182 EXPECT_TRUE(StringView("foo").EndsWith(StringView()));
183 EXPECT_TRUE(StringView("foo").EndsWith(StringView("")));
184 EXPECT_FALSE(StringView().EndsWith("foo"));
185 EXPECT_FALSE(StringView("").EndsWith("foo"));
186 EXPECT_TRUE(StringView("foo").EndsWith("foo"));
187 EXPECT_FALSE(StringView("foorbar").EndsWith("foo"));
188 EXPECT_TRUE(StringView("foorbar").EndsWith("bar"));
189 EXPECT_FALSE(StringView("foo").EndsWith("fooo"));
190 }
191
TEST(StringViewTest,HashCollisions)192 TEST(StringViewTest, HashCollisions) {
193 std::unordered_map<uint64_t, StringView> hashes;
194 std::unordered_set<StringView> sv_set;
195 auto insert_sv = [&hashes, &sv_set](StringView sv) {
196 hashes.emplace(sv.Hash(), sv);
197 size_t prev_set_size = sv_set.size();
198 sv_set.insert(sv);
199 ASSERT_EQ(sv_set.size(), prev_set_size + 1);
200 };
201
202 insert_sv("");
203 EXPECT_EQ(hashes.size(), 1u);
204 size_t last_size = 1;
205 std::forward_list<std::string> strings;
206 for (uint8_t c = 0; c < 0x80; c++) {
207 char buf[500];
208 memset(buf, static_cast<char>(c), sizeof(buf));
209 for (size_t i = 1; i <= sizeof(buf); i++) {
210 strings.emplace_front(buf, i);
211 StringView sv(strings.front());
212 auto other = hashes.find(sv.Hash());
213 if (other == hashes.end()) {
214 insert_sv(sv);
215 ++last_size;
216 ASSERT_EQ(hashes.size(), last_size);
217 continue;
218 }
219 EXPECT_TRUE(false) << "H(" << sv.ToStdString() << ") = "
220 << "H(" << other->second.ToStdString() << ")";
221 }
222 }
223 }
224
225 } // namespace
226 } // namespace base
227 } // namespace perfetto
228