1//===-- Unittests for string_view -----------------------------------------===//
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#include "src/__support/CPP/string_view.h"
10#include "test/UnitTest/Test.h"
11
12using LIBC_NAMESPACE::cpp::string_view;
13
14TEST(LlvmLibcStringViewTest, InitializeCheck) {
15 string_view v;
16 ASSERT_EQ(v.size(), size_t(0));
17 ASSERT_TRUE(v.data() == nullptr);
18
19 v = string_view("");
20 ASSERT_EQ(v.size(), size_t(0));
21 ASSERT_TRUE(v.data() != nullptr);
22
23 v = string_view("abc", 0);
24 ASSERT_EQ(v.size(), size_t(0));
25 ASSERT_TRUE(v.data() != nullptr);
26
27 v = string_view("123456789");
28 ASSERT_EQ(v.size(), size_t(9));
29}
30
31TEST(LlvmLibcStringViewTest, Equals) {
32 string_view v("abc");
33 ASSERT_EQ(v, string_view("abc"));
34 ASSERT_NE(v, string_view());
35 ASSERT_NE(v, string_view(""));
36 ASSERT_NE(v, string_view("123"));
37 ASSERT_NE(v, string_view("abd"));
38 ASSERT_NE(v, string_view("aaa"));
39 ASSERT_NE(v, string_view("abcde"));
40}
41
42TEST(LlvmLibcStringViewTest, startsWith) {
43 string_view v("abc");
44 ASSERT_TRUE(v.starts_with('a'));
45 ASSERT_TRUE(v.starts_with(string_view("a")));
46 ASSERT_TRUE(v.starts_with(string_view("ab")));
47 ASSERT_TRUE(v.starts_with(string_view("abc")));
48 ASSERT_TRUE(v.starts_with(string_view()));
49 ASSERT_TRUE(v.starts_with(string_view("")));
50 ASSERT_FALSE(v.starts_with('1'));
51 ASSERT_FALSE(v.starts_with(string_view("123")));
52 ASSERT_FALSE(v.starts_with(string_view("abd")));
53 ASSERT_FALSE(v.starts_with(string_view("aaa")));
54 ASSERT_FALSE(v.starts_with(string_view("abcde")));
55}
56
57TEST(LlvmLibcStringViewTest, endsWith) {
58 string_view v("abc");
59 ASSERT_TRUE(v.ends_with('c'));
60 ASSERT_TRUE(v.ends_with(string_view("c")));
61 ASSERT_TRUE(v.ends_with(string_view("bc")));
62 ASSERT_TRUE(v.ends_with(string_view("abc")));
63 ASSERT_TRUE(v.ends_with(string_view()));
64 ASSERT_TRUE(v.ends_with(string_view("")));
65 ASSERT_FALSE(v.ends_with('1'));
66 ASSERT_FALSE(v.ends_with(string_view("123")));
67 ASSERT_FALSE(v.ends_with(string_view("abd")));
68 ASSERT_FALSE(v.ends_with(string_view("aaa")));
69 ASSERT_FALSE(v.ends_with(string_view("abcde")));
70}
71
72TEST(LlvmLibcStringViewTest, RemovePrefix) {
73 string_view a("123456789");
74 a.remove_prefix(0);
75 ASSERT_EQ(a.size(), size_t(9));
76 ASSERT_TRUE(a == "123456789");
77
78 string_view b("123456789");
79 b.remove_prefix(4);
80 ASSERT_EQ(b.size(), size_t(5));
81 ASSERT_TRUE(b == "56789");
82
83 string_view c("123456789");
84 c.remove_prefix(9);
85 ASSERT_EQ(c.size(), size_t(0));
86}
87
88TEST(LlvmLibcStringViewTest, RemoveSuffix) {
89 string_view a("123456789");
90 a.remove_suffix(0);
91 ASSERT_EQ(a.size(), size_t(9));
92 ASSERT_TRUE(a == "123456789");
93
94 string_view b("123456789");
95 b.remove_suffix(4);
96 ASSERT_EQ(b.size(), size_t(5));
97 ASSERT_TRUE(b == "12345");
98
99 string_view c("123456789");
100 c.remove_suffix(9);
101 ASSERT_EQ(c.size(), size_t(0));
102}
103
104TEST(LlvmLibcStringViewTest, Observer) {
105 string_view ABC("abc");
106 ASSERT_EQ(ABC.size(), size_t(3));
107 ASSERT_FALSE(ABC.empty());
108 ASSERT_EQ(ABC.front(), 'a');
109 ASSERT_EQ(ABC.back(), 'c');
110}
111
112TEST(LlvmLibcStringViewTest, FindFirstOf) {
113 string_view Tmp("abca");
114 ASSERT_TRUE(Tmp.find_first_of('a') == 0);
115 ASSERT_TRUE(Tmp.find_first_of('d') == string_view::npos);
116 ASSERT_TRUE(Tmp.find_first_of('b') == 1);
117 ASSERT_TRUE(Tmp.find_first_of('a', 0) == 0);
118 ASSERT_TRUE(Tmp.find_first_of('b', 1) == 1);
119 ASSERT_TRUE(Tmp.find_first_of('a', 1) == 3);
120 ASSERT_TRUE(Tmp.find_first_of('a', 42) == string_view::npos);
121 ASSERT_FALSE(Tmp.find_first_of('c') == 1);
122 ASSERT_FALSE(Tmp.find_first_of('c', 0) == 1);
123 ASSERT_FALSE(Tmp.find_first_of('c', 1) == 1);
124}
125
126TEST(LlvmLibcStringViewTest, FindLastOf) {
127 string_view Tmp("abada");
128
129 ASSERT_EQ(Tmp.find_last_of('a'), size_t(4));
130 ASSERT_EQ(Tmp.find_last_of('a', 123), size_t(4));
131 ASSERT_EQ(Tmp.find_last_of('a', 5), size_t(4));
132 ASSERT_EQ(Tmp.find_last_of('a', 4), size_t(4));
133 ASSERT_EQ(Tmp.find_last_of('a', 3), size_t(2));
134 ASSERT_EQ(Tmp.find_last_of('a', 2), size_t(2));
135 ASSERT_EQ(Tmp.find_last_of('a', 1), size_t(0));
136 ASSERT_EQ(Tmp.find_last_of('a', 0), size_t(0));
137
138 ASSERT_EQ(Tmp.find_last_of('b'), size_t(1));
139 ASSERT_EQ(Tmp.find_last_of('b', 123), size_t(1));
140 ASSERT_EQ(Tmp.find_last_of('b', 5), size_t(1));
141 ASSERT_EQ(Tmp.find_last_of('b', 4), size_t(1));
142 ASSERT_EQ(Tmp.find_last_of('b', 3), size_t(1));
143 ASSERT_EQ(Tmp.find_last_of('b', 2), size_t(1));
144 ASSERT_EQ(Tmp.find_last_of('b', 1), size_t(1));
145 ASSERT_EQ(Tmp.find_last_of('b', 0), string_view::npos);
146
147 ASSERT_EQ(Tmp.find_last_of('d'), size_t(3));
148 ASSERT_EQ(Tmp.find_last_of('d', 123), size_t(3));
149 ASSERT_EQ(Tmp.find_last_of('d', 5), size_t(3));
150 ASSERT_EQ(Tmp.find_last_of('d', 4), size_t(3));
151 ASSERT_EQ(Tmp.find_last_of('d', 3), size_t(3));
152 ASSERT_EQ(Tmp.find_last_of('d', 2), string_view::npos);
153 ASSERT_EQ(Tmp.find_last_of('d', 1), string_view::npos);
154 ASSERT_EQ(Tmp.find_last_of('d', 0), string_view::npos);
155
156 ASSERT_EQ(Tmp.find_last_of('e'), string_view::npos);
157 ASSERT_EQ(Tmp.find_last_of('e', 123), string_view::npos);
158 ASSERT_EQ(Tmp.find_last_of('e', 5), string_view::npos);
159 ASSERT_EQ(Tmp.find_last_of('e', 4), string_view::npos);
160 ASSERT_EQ(Tmp.find_last_of('e', 3), string_view::npos);
161 ASSERT_EQ(Tmp.find_last_of('e', 2), string_view::npos);
162 ASSERT_EQ(Tmp.find_last_of('e', 1), string_view::npos);
163 ASSERT_EQ(Tmp.find_last_of('e', 0), string_view::npos);
164
165 string_view Empty;
166 ASSERT_EQ(Empty.find_last_of('a'), string_view::npos);
167 ASSERT_EQ(Empty.find_last_of('a', 0), string_view::npos);
168 ASSERT_EQ(Empty.find_last_of('a', 123), string_view::npos);
169
170 string_view Empty1("");
171 ASSERT_EQ(Empty1.find_last_of('a'), string_view::npos);
172 ASSERT_EQ(Empty1.find_last_of('a', 0), string_view::npos);
173 ASSERT_EQ(Empty1.find_last_of('a', 123), string_view::npos);
174}
175
176TEST(LlvmLibcStringViewTest, FindFirstNotOf) {
177 string_view Tmp("abada");
178
179 EXPECT_EQ(Tmp.find_first_not_of('a'), size_t(1));
180 EXPECT_EQ(Tmp.find_first_not_of('a', 123), string_view::npos);
181 EXPECT_EQ(Tmp.find_first_not_of('a', 5), string_view::npos);
182 EXPECT_EQ(Tmp.find_first_not_of('a', 4), string_view::npos);
183 EXPECT_EQ(Tmp.find_first_not_of('a', 3), size_t(3));
184 EXPECT_EQ(Tmp.find_first_not_of('a', 2), size_t(3));
185 EXPECT_EQ(Tmp.find_first_not_of('a', 1), size_t(1));
186 EXPECT_EQ(Tmp.find_first_not_of('a', 0), size_t(1));
187
188 EXPECT_EQ(Tmp.find_first_not_of('b'), size_t(0));
189 EXPECT_EQ(Tmp.find_first_not_of('b', 123), string_view::npos);
190 EXPECT_EQ(Tmp.find_first_not_of('b', 5), string_view::npos);
191 EXPECT_EQ(Tmp.find_first_not_of('b', 4), size_t(4));
192 EXPECT_EQ(Tmp.find_first_not_of('b', 3), size_t(3));
193 EXPECT_EQ(Tmp.find_first_not_of('b', 2), size_t(2));
194 EXPECT_EQ(Tmp.find_first_not_of('b', 1), size_t(2));
195 EXPECT_EQ(Tmp.find_first_not_of('b', 0), size_t(0));
196
197 EXPECT_EQ(Tmp.find_first_not_of('d'), size_t(0));
198 EXPECT_EQ(Tmp.find_first_not_of('d', 123), string_view::npos);
199 EXPECT_EQ(Tmp.find_first_not_of('d', 5), string_view::npos);
200 EXPECT_EQ(Tmp.find_first_not_of('d', 4), size_t(4));
201 EXPECT_EQ(Tmp.find_first_not_of('d', 3), size_t(4));
202 EXPECT_EQ(Tmp.find_first_not_of('d', 2), size_t(2));
203 EXPECT_EQ(Tmp.find_first_not_of('d', 1), size_t(1));
204 EXPECT_EQ(Tmp.find_first_not_of('d', 0), size_t(0));
205
206 EXPECT_EQ(Tmp.find_first_not_of('e'), size_t(0));
207 EXPECT_EQ(Tmp.find_first_not_of('e', 123), string_view::npos);
208 EXPECT_EQ(Tmp.find_first_not_of('e', 5), string_view::npos);
209 EXPECT_EQ(Tmp.find_first_not_of('e', 4), size_t(4));
210 EXPECT_EQ(Tmp.find_first_not_of('e', 3), size_t(3));
211 EXPECT_EQ(Tmp.find_first_not_of('e', 2), size_t(2));
212 EXPECT_EQ(Tmp.find_first_not_of('e', 1), size_t(1));
213 EXPECT_EQ(Tmp.find_first_not_of('e', 0), size_t(0));
214
215 string_view Empty;
216 EXPECT_EQ(Empty.find_first_not_of('a'), string_view::npos);
217 EXPECT_EQ(Empty.find_first_not_of('a', 0), string_view::npos);
218 EXPECT_EQ(Empty.find_first_not_of('a', 123), string_view::npos);
219
220 string_view Empty1("");
221 EXPECT_EQ(Empty1.find_first_not_of('a'), string_view::npos);
222 EXPECT_EQ(Empty1.find_first_not_of('a', 0), string_view::npos);
223 EXPECT_EQ(Empty1.find_first_not_of('a', 123), string_view::npos);
224
225 string_view Full("aaaaaaa");
226 EXPECT_EQ(Full.find_first_not_of('a'), string_view::npos);
227 EXPECT_EQ(Full.find_first_not_of('a', 0), string_view::npos);
228 EXPECT_EQ(Full.find_first_not_of('a', 123), string_view::npos);
229
230 EXPECT_EQ(Full.find_first_not_of('b'), size_t(0));
231 EXPECT_EQ(Full.find_first_not_of('b', 0), size_t(0));
232 EXPECT_EQ(Full.find_first_not_of('b', 123), string_view::npos);
233}
234
235TEST(LlvmLibcStringViewTest, Contains) {
236 string_view Empty;
237 static_assert(
238 'a' < 'z',
239 "This test only supports character encodings where 'a' is below 'z'");
240 for (char c = 'a'; c < 'z'; ++c)
241 EXPECT_FALSE(Empty.contains(c));
242
243 string_view Tmp("abada");
244 EXPECT_TRUE(Tmp.contains('a'));
245 EXPECT_TRUE(Tmp.contains('b'));
246 EXPECT_FALSE(Tmp.contains('c'));
247 EXPECT_TRUE(Tmp.contains('d'));
248 EXPECT_FALSE(Tmp.contains('e'));
249
250 EXPECT_TRUE(Tmp.substr(1).contains('a'));
251 EXPECT_TRUE(Tmp.substr(1).contains('b'));
252 EXPECT_FALSE(Tmp.substr(1).contains('c'));
253 EXPECT_TRUE(Tmp.substr(1).contains('d'));
254 EXPECT_FALSE(Tmp.substr(1).contains('e'));
255
256 EXPECT_TRUE(Tmp.substr(2).contains('a'));
257 EXPECT_FALSE(Tmp.substr(2).contains('b'));
258 EXPECT_FALSE(Tmp.substr(2).contains('c'));
259 EXPECT_TRUE(Tmp.substr(2).contains('d'));
260 EXPECT_FALSE(Tmp.substr(2).contains('e'));
261
262 EXPECT_TRUE(Tmp.substr(3).contains('a'));
263 EXPECT_FALSE(Tmp.substr(3).contains('b'));
264 EXPECT_FALSE(Tmp.substr(3).contains('c'));
265 EXPECT_TRUE(Tmp.substr(3).contains('d'));
266 EXPECT_FALSE(Tmp.substr(3).contains('e'));
267
268 EXPECT_TRUE(Tmp.substr(4).contains('a'));
269 EXPECT_FALSE(Tmp.substr(4).contains('b'));
270 EXPECT_FALSE(Tmp.substr(4).contains('c'));
271 EXPECT_FALSE(Tmp.substr(4).contains('d'));
272 EXPECT_FALSE(Tmp.substr(4).contains('e'));
273
274 EXPECT_FALSE(Tmp.substr(5).contains('a'));
275 EXPECT_FALSE(Tmp.substr(5).contains('b'));
276 EXPECT_FALSE(Tmp.substr(5).contains('c'));
277 EXPECT_FALSE(Tmp.substr(5).contains('d'));
278 EXPECT_FALSE(Tmp.substr(5).contains('e'));
279
280 EXPECT_FALSE(Tmp.substr(6).contains('a'));
281 EXPECT_FALSE(Tmp.substr(6).contains('b'));
282 EXPECT_FALSE(Tmp.substr(6).contains('c'));
283 EXPECT_FALSE(Tmp.substr(6).contains('d'));
284 EXPECT_FALSE(Tmp.substr(6).contains('e'));
285}
286

source code of libc/test/src/__support/CPP/stringview_test.cpp