Serenity Operating System
1/*
2 * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include <LibTest/TestCase.h>
8
9#include <AK/DeprecatedString.h>
10#include <AK/Vector.h>
11
12TEST_CASE(construct_empty)
13{
14 EXPECT(StringView().is_null());
15 EXPECT(StringView().is_empty());
16 EXPECT(!StringView().characters_without_null_termination());
17 EXPECT_EQ(StringView().length(), 0u);
18}
19
20TEST_CASE(view_literal)
21{
22 char const* truth = "cats rule dogs drool";
23 StringView view { truth, strlen(truth) };
24 EXPECT_EQ(view.is_null(), false);
25 EXPECT_EQ(view.characters_without_null_termination(), truth);
26 EXPECT_EQ(view, view);
27 EXPECT_EQ(view, truth);
28}
29
30TEST_CASE(compare_views)
31{
32 DeprecatedString foo1 = "foo";
33 DeprecatedString foo2 = "foo";
34 auto view1 = foo1.view();
35 auto view2 = foo2.view();
36
37 EXPECT_EQ(view1, view2);
38 EXPECT_EQ(view1, foo1);
39 EXPECT_EQ(view1, foo2);
40 EXPECT_EQ(view1, "foo");
41}
42
43TEST_CASE(string_view_literal_operator)
44{
45 StringView literal_view = "foo"sv;
46 DeprecatedString test_string = "foo";
47
48 EXPECT_EQ(literal_view.length(), test_string.length());
49 EXPECT_EQ(literal_view, test_string);
50}
51
52TEST_CASE(starts_with)
53{
54 DeprecatedString test_string = "ABCDEF";
55 StringView test_string_view = test_string.view();
56 EXPECT(test_string_view.starts_with('A'));
57 EXPECT(!test_string_view.starts_with('B'));
58 EXPECT(test_string_view.starts_with("AB"sv));
59 EXPECT(test_string_view.starts_with("ABCDEF"sv));
60 EXPECT(!test_string_view.starts_with("DEF"sv));
61 EXPECT(test_string_view.starts_with("abc"sv, CaseSensitivity::CaseInsensitive));
62 EXPECT(!test_string_view.starts_with("abc"sv, CaseSensitivity::CaseSensitive));
63}
64
65TEST_CASE(ends_with)
66{
67 DeprecatedString test_string = "ABCDEF";
68 StringView test_string_view = test_string.view();
69 EXPECT(test_string_view.ends_with("DEF"sv));
70 EXPECT(test_string_view.ends_with('F'));
71 EXPECT(!test_string_view.ends_with('E'));
72 EXPECT(test_string_view.ends_with("ABCDEF"sv));
73 EXPECT(!test_string_view.ends_with("ABCDE"sv));
74 EXPECT(!test_string_view.ends_with("ABCDEFG"sv));
75 EXPECT(test_string_view.ends_with("def"sv, CaseSensitivity::CaseInsensitive));
76 EXPECT(!test_string_view.ends_with("def"sv, CaseSensitivity::CaseSensitive));
77}
78
79TEST_CASE(lines)
80{
81 DeprecatedString test_string = "a\rb\nc\r\nd";
82 StringView test_string_view = test_string.view();
83 Vector<StringView> test_string_vector = test_string_view.lines();
84 EXPECT_EQ(test_string_vector.size(), 4u);
85 EXPECT(test_string_vector.at(0) == DeprecatedString("a"));
86 EXPECT(test_string_vector.at(1) == DeprecatedString("b"));
87 EXPECT(test_string_vector.at(2) == DeprecatedString("c"));
88 EXPECT(test_string_vector.at(3) == DeprecatedString("d"));
89
90 test_string = "```\nHello there\r\nHello there\n```";
91 test_string_view = test_string.view();
92 test_string_vector = test_string_view.lines();
93 EXPECT_EQ(test_string_vector.size(), 4u);
94 EXPECT(test_string_vector.at(0) == DeprecatedString("```"));
95 EXPECT(test_string_vector.at(1) == DeprecatedString("Hello there"));
96 EXPECT(test_string_vector.at(2) == DeprecatedString("Hello there"));
97 EXPECT(test_string_vector.at(3) == DeprecatedString("```"));
98
99 test_string = "\n\n\n";
100 test_string_view = test_string.view();
101 test_string_vector = test_string_view.lines();
102 EXPECT_EQ(test_string_vector.size(), 3u);
103 EXPECT_EQ(test_string_vector.at(0).is_empty(), true);
104 EXPECT_EQ(test_string_vector.at(1).is_empty(), true);
105 EXPECT_EQ(test_string_vector.at(2).is_empty(), true);
106}
107
108TEST_CASE(find)
109{
110 auto test_string_view = "aabbcc_xy_ccbbaa"sv;
111 EXPECT_EQ(test_string_view.find('b'), 2U);
112 EXPECT_EQ(test_string_view.find('_'), 6U);
113 EXPECT_EQ(test_string_view.find('n').has_value(), false);
114}
115
116TEST_CASE(find_last)
117{
118 auto test_string_view = "aabbcc_xy_ccbbaa"sv;
119 EXPECT_EQ(test_string_view.find_last('b'), 13U);
120 EXPECT_EQ(test_string_view.find_last('_'), 9U);
121 EXPECT_EQ(test_string_view.find_last('3').has_value(), false);
122
123 test_string_view = "/"sv;
124 EXPECT_EQ(test_string_view.find_last('/'), 0U);
125}
126
127TEST_CASE(find_any_of)
128{
129 auto test_string_view = "aabbcc_xy_ccbbaa"sv;
130 EXPECT_EQ(test_string_view.find_any_of("bc"sv, StringView::SearchDirection::Forward), 2U);
131 EXPECT_EQ(test_string_view.find_any_of("yx"sv, StringView::SearchDirection::Forward), 7U);
132 EXPECT_EQ(test_string_view.find_any_of("defg"sv, StringView::SearchDirection::Forward).has_value(), false);
133 EXPECT_EQ(test_string_view.find_any_of("bc"sv, StringView::SearchDirection::Backward), 13U);
134 EXPECT_EQ(test_string_view.find_any_of("yx"sv, StringView::SearchDirection::Backward), 8U);
135 EXPECT_EQ(test_string_view.find_any_of("fghi"sv, StringView::SearchDirection::Backward).has_value(), false);
136
137 test_string_view = "/"sv;
138 EXPECT_EQ(test_string_view.find_any_of("/"sv, StringView::SearchDirection::Forward), 0U);
139 EXPECT_EQ(test_string_view.find_any_of("/"sv, StringView::SearchDirection::Backward), 0U);
140}
141
142TEST_CASE(split_view)
143{
144 StringView test_string_view = "axxbxcxd"sv;
145 EXPECT_EQ(test_string_view.split_view('x'), Vector({ "a"sv, "b"sv, "c"sv, "d"sv }));
146 EXPECT_EQ(test_string_view.split_view('x', SplitBehavior::KeepEmpty), Vector({ "a"sv, ""sv, "b"sv, "c"sv, "d"sv }));
147 EXPECT_EQ(test_string_view.split_view("x"sv), Vector({ "a"sv, "b"sv, "c"sv, "d"sv }));
148 EXPECT_EQ(test_string_view.split_view("x"sv, SplitBehavior::KeepEmpty), Vector({ "a"sv, ""sv, "b"sv, "c"sv, "d"sv }));
149
150 test_string_view = "axxbx"sv;
151 EXPECT_EQ(test_string_view.split_view('x'), Vector({ "a"sv, "b"sv }));
152 EXPECT_EQ(test_string_view.split_view('x', SplitBehavior::KeepEmpty), Vector({ "a"sv, ""sv, "b"sv, ""sv }));
153 EXPECT_EQ(test_string_view.split_view("x"sv), Vector({ "a"sv, "b"sv }));
154 EXPECT_EQ(test_string_view.split_view("x"sv, SplitBehavior::KeepEmpty), Vector({ "a"sv, ""sv, "b"sv, ""sv }));
155
156 test_string_view = "axxbcxxdxx"sv;
157 EXPECT_EQ(test_string_view.split_view("xx"sv), Vector({ "a"sv, "bc"sv, "d"sv }));
158 EXPECT_EQ(test_string_view.split_view("xx"sv, SplitBehavior::KeepEmpty), Vector({ "a"sv, "bc"sv, "d"sv, ""sv }));
159
160 test_string_view = "ax_b_cxd"sv;
161 Function<bool(char)> predicate = [](char ch) { return ch == 'x' || ch == '_'; };
162 EXPECT_EQ(test_string_view.split_view_if(predicate), Vector({ "a"sv, "b"sv, "c"sv, "d"sv }));
163 EXPECT_EQ(test_string_view.split_view_if(predicate, SplitBehavior::KeepEmpty), Vector({ "a"sv, ""sv, "b"sv, "c"sv, "d"sv }));
164 EXPECT_EQ(test_string_view.split_view_if(predicate), Vector({ "a"sv, "b"sv, "c"sv, "d"sv }));
165 EXPECT_EQ(test_string_view.split_view_if(predicate, SplitBehavior::KeepEmpty), Vector({ "a"sv, ""sv, "b"sv, "c"sv, "d"sv }));
166
167 test_string_view = "a,,,b"sv;
168 EXPECT_EQ(test_string_view.split_view(","sv, SplitBehavior::KeepEmpty), Vector({ "a"sv, ""sv, ""sv, "b"sv }));
169 EXPECT_EQ(test_string_view.split_view(","sv, SplitBehavior::KeepTrailingSeparator), Vector({ "a,"sv, "b"sv }));
170 EXPECT_EQ(test_string_view.split_view(","sv, SplitBehavior::KeepTrailingSeparator | SplitBehavior::KeepEmpty), Vector({ "a,"sv, ","sv, ","sv, "b"sv }));
171}
172
173TEST_CASE(constexpr_stuff)
174{
175#define do_test() \
176 static_assert(test_constexpr.length() == 3); \
177 static_assert(!test_constexpr.is_empty()); \
178 static_assert(test_constexpr.is_one_of("foo", "bar", "baz")); \
179 static_assert(test_constexpr.is_one_of("foo"sv, "bar"sv, "baz"sv)); \
180 static_assert(test_constexpr != "fob"sv); \
181 static_assert(test_constexpr != "fob"); \
182 static_assert(test_constexpr.substring_view(1).is_one_of("oo"sv));
183
184 {
185 // Can initialize from ""sv.
186 constexpr StringView test_constexpr { "foo"sv };
187 do_test();
188 }
189#undef do_test
190}
191
192TEST_CASE(case_insensitive_hash)
193{
194 auto string1 = "abcdef"sv;
195 auto string2 = "ABCDEF"sv;
196 auto string3 = "aBcDeF"sv;
197 auto string4 = "foo"sv;
198
199 EXPECT_EQ(CaseInsensitiveStringViewTraits::hash(string1), CaseInsensitiveStringViewTraits::hash(string2));
200 EXPECT_EQ(CaseInsensitiveStringViewTraits::hash(string1), CaseInsensitiveStringViewTraits::hash(string3));
201 EXPECT_NE(CaseInsensitiveStringViewTraits::hash(string1), CaseInsensitiveStringViewTraits::hash(string4));
202}