xref: /llvm-project/libc/test/src/__support/CPP/stringview_test.cpp (revision a0c4f854cad2b97e44a1b58dc1fd982e1c4d60f3)
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 
12 using LIBC_NAMESPACE::cpp::string_view;
13 
14 TEST(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 
31 TEST(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 
42 TEST(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 
57 TEST(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 
72 TEST(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 
88 TEST(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 
104 TEST(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 
112 TEST(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 
126 TEST(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 
176 TEST(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 
235 TEST(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