1 //===- StringExtrasTest.cpp - Unit tests for String extras ----------------===//
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 "llvm/ADT/StringExtras.h"
10 #include "llvm/Support/raw_ostream.h"
11 #include "gtest/gtest.h"
12
13 using namespace llvm;
14
TEST(StringExtrasTest,isPrint)15 TEST(StringExtrasTest, isPrint) {
16 EXPECT_FALSE(isPrint('\0'));
17 EXPECT_FALSE(isPrint('\t'));
18 EXPECT_TRUE(isPrint('0'));
19 EXPECT_TRUE(isPrint('a'));
20 EXPECT_TRUE(isPrint('A'));
21 EXPECT_TRUE(isPrint(' '));
22 EXPECT_TRUE(isPrint('~'));
23 EXPECT_TRUE(isPrint('?'));
24 }
25
TEST(StringExtrasTest,isSpace)26 TEST(StringExtrasTest, isSpace) {
27 EXPECT_TRUE(isSpace(' '));
28 EXPECT_TRUE(isSpace('\t'));
29 EXPECT_TRUE(isSpace('\n'));
30 EXPECT_TRUE(isSpace('\v'));
31 EXPECT_TRUE(isSpace('\f'));
32 EXPECT_TRUE(isSpace('\v'));
33 EXPECT_FALSE(isSpace('\0'));
34 EXPECT_FALSE(isSpace('_'));
35 }
36
TEST(StringExtrasTest,Join)37 TEST(StringExtrasTest, Join) {
38 std::vector<std::string> Items;
39 EXPECT_EQ("", join(Items.begin(), Items.end(), " <sep> "));
40
41 Items = {"foo"};
42 EXPECT_EQ("foo", join(Items.begin(), Items.end(), " <sep> "));
43
44 Items = {"foo", "bar"};
45 EXPECT_EQ("foo <sep> bar", join(Items.begin(), Items.end(), " <sep> "));
46
47 Items = {"foo", "bar", "baz"};
48 EXPECT_EQ("foo <sep> bar <sep> baz",
49 join(Items.begin(), Items.end(), " <sep> "));
50 }
51
TEST(StringExtrasTest,JoinItems)52 TEST(StringExtrasTest, JoinItems) {
53 const char *Foo = "foo";
54 std::string Bar = "bar";
55 llvm::StringRef Baz = "baz";
56 char X = 'x';
57
58 EXPECT_EQ("", join_items(" <sep> "));
59 EXPECT_EQ("", join_items('/'));
60
61 EXPECT_EQ("foo", join_items(" <sep> ", Foo));
62 EXPECT_EQ("foo", join_items('/', Foo));
63
64 EXPECT_EQ("foo <sep> bar", join_items(" <sep> ", Foo, Bar));
65 EXPECT_EQ("foo/bar", join_items('/', Foo, Bar));
66
67 EXPECT_EQ("foo <sep> bar <sep> baz", join_items(" <sep> ", Foo, Bar, Baz));
68 EXPECT_EQ("foo/bar/baz", join_items('/', Foo, Bar, Baz));
69
70 EXPECT_EQ("foo <sep> bar <sep> baz <sep> x",
71 join_items(" <sep> ", Foo, Bar, Baz, X));
72
73 EXPECT_EQ("foo/bar/baz/x", join_items('/', Foo, Bar, Baz, X));
74 }
75
TEST(StringExtrasTest,ToAndFromHex)76 TEST(StringExtrasTest, ToAndFromHex) {
77 std::vector<uint8_t> OddBytes = {0x5, 0xBD, 0x0D, 0x3E, 0xCD};
78 std::string OddStr = "05BD0D3ECD";
79 StringRef OddData(reinterpret_cast<const char *>(OddBytes.data()),
80 OddBytes.size());
81 EXPECT_EQ(OddStr, toHex(OddData));
82 EXPECT_EQ(OddData, fromHex(StringRef(OddStr).drop_front()));
83 EXPECT_EQ(StringRef(OddStr).lower(), toHex(OddData, true));
84
85 std::vector<uint8_t> EvenBytes = {0xA5, 0xBD, 0x0D, 0x3E, 0xCD};
86 std::string EvenStr = "A5BD0D3ECD";
87 StringRef EvenData(reinterpret_cast<const char *>(EvenBytes.data()),
88 EvenBytes.size());
89 EXPECT_EQ(EvenStr, toHex(EvenData));
90 EXPECT_EQ(EvenData, fromHex(EvenStr));
91 EXPECT_EQ(StringRef(EvenStr).lower(), toHex(EvenData, true));
92 }
93
TEST(StringExtrasTest,to_float)94 TEST(StringExtrasTest, to_float) {
95 float F;
96 EXPECT_TRUE(to_float("4.7", F));
97 EXPECT_FLOAT_EQ(4.7f, F);
98
99 double D;
100 EXPECT_TRUE(to_float("4.7", D));
101 EXPECT_DOUBLE_EQ(4.7, D);
102
103 long double LD;
104 EXPECT_TRUE(to_float("4.7", LD));
105 EXPECT_DOUBLE_EQ(4.7, LD);
106
107 EXPECT_FALSE(to_float("foo", F));
108 EXPECT_FALSE(to_float("7.4 foo", F));
109 EXPECT_FLOAT_EQ(4.7f, F); // F should be unchanged
110 }
111
TEST(StringExtrasTest,printLowerCase)112 TEST(StringExtrasTest, printLowerCase) {
113 std::string str;
114 raw_string_ostream OS(str);
115 printLowerCase("ABCdefg01234.,&!~`'}\"", OS);
116 EXPECT_EQ("abcdefg01234.,&!~`'}\"", OS.str());
117 }
118
TEST(StringExtrasTest,printEscapedString)119 TEST(StringExtrasTest, printEscapedString) {
120 std::string str;
121 raw_string_ostream OS(str);
122 printEscapedString("ABCdef123&<>\\\"'\t", OS);
123 EXPECT_EQ("ABCdef123&<>\\\\\\22'\\09", OS.str());
124 }
125
TEST(StringExtrasTest,printHTMLEscaped)126 TEST(StringExtrasTest, printHTMLEscaped) {
127 std::string str;
128 raw_string_ostream OS(str);
129 printHTMLEscaped("ABCdef123&<>\"'", OS);
130 EXPECT_EQ("ABCdef123&<>"'", OS.str());
131 }
132
TEST(StringExtras,ConvertToSnakeFromCamelCase)133 TEST(StringExtras, ConvertToSnakeFromCamelCase) {
134 auto testConvertToSnakeCase = [](llvm::StringRef input,
135 llvm::StringRef expected) {
136 EXPECT_EQ(convertToSnakeFromCamelCase(input), expected.str());
137 };
138
139 testConvertToSnakeCase("OpName", "op_name");
140 testConvertToSnakeCase("opName", "op_name");
141 testConvertToSnakeCase("_OpName", "_op_name");
142 testConvertToSnakeCase("Op_Name", "op_name");
143 testConvertToSnakeCase("", "");
144 testConvertToSnakeCase("A", "a");
145 testConvertToSnakeCase("_", "_");
146 testConvertToSnakeCase("a", "a");
147 testConvertToSnakeCase("op_name", "op_name");
148 testConvertToSnakeCase("_op_name", "_op_name");
149 testConvertToSnakeCase("__op_name", "__op_name");
150 testConvertToSnakeCase("op__name", "op__name");
151 }
152
TEST(StringExtras,ConvertToCamelFromSnakeCase)153 TEST(StringExtras, ConvertToCamelFromSnakeCase) {
154 auto testConvertToCamelCase = [](bool capitalizeFirst, llvm::StringRef input,
155 llvm::StringRef expected) {
156 EXPECT_EQ(convertToCamelFromSnakeCase(input, capitalizeFirst),
157 expected.str());
158 };
159
160 testConvertToCamelCase(false, "op_name", "opName");
161 testConvertToCamelCase(false, "_op_name", "_opName");
162 testConvertToCamelCase(false, "__op_name", "_OpName");
163 testConvertToCamelCase(false, "op__name", "op_Name");
164 testConvertToCamelCase(false, "", "");
165 testConvertToCamelCase(false, "A", "A");
166 testConvertToCamelCase(false, "_", "_");
167 testConvertToCamelCase(false, "a", "a");
168 testConvertToCamelCase(false, "OpName", "OpName");
169 testConvertToCamelCase(false, "opName", "opName");
170 testConvertToCamelCase(false, "_OpName", "_OpName");
171 testConvertToCamelCase(false, "Op_Name", "Op_Name");
172 testConvertToCamelCase(true, "op_name", "OpName");
173 testConvertToCamelCase(true, "_op_name", "_opName");
174 testConvertToCamelCase(true, "__op_name", "_OpName");
175 testConvertToCamelCase(true, "op__name", "Op_Name");
176 testConvertToCamelCase(true, "", "");
177 testConvertToCamelCase(true, "A", "A");
178 testConvertToCamelCase(true, "_", "_");
179 testConvertToCamelCase(true, "a", "A");
180 testConvertToCamelCase(true, "OpName", "OpName");
181 testConvertToCamelCase(true, "_OpName", "_OpName");
182 testConvertToCamelCase(true, "Op_Name", "Op_Name");
183 testConvertToCamelCase(true, "opName", "OpName");
184 }
185
186 constexpr uint64_t MaxUint64 = std::numeric_limits<uint64_t>::max();
187 constexpr int64_t MaxInt64 = std::numeric_limits<int64_t>::max();
188 constexpr int64_t MinInt64 = std::numeric_limits<int64_t>::min();
189
TEST(StringExtras,UToStr)190 TEST(StringExtras, UToStr) {
191 EXPECT_EQ("0", utostr(0));
192 EXPECT_EQ("0", utostr(0, /*isNeg=*/false));
193 EXPECT_EQ("1", utostr(1));
194 EXPECT_EQ("1", utostr(1, /*isNeg=*/false));
195 EXPECT_EQ(std::to_string(MaxUint64), utostr(MaxUint64));
196 EXPECT_EQ(std::to_string(MaxUint64), utostr(MaxUint64, /*isNeg=*/false));
197
198 EXPECT_EQ("-0", utostr(0, /*isNeg=*/true));
199 EXPECT_EQ("-1", utostr(1, /*isNeg=*/true));
200 EXPECT_EQ("-" + std::to_string(MaxInt64), utostr(MaxInt64, /*isNeg=*/true));
201 constexpr uint64_t MinusMinInt64 = -static_cast<uint64_t>(MinInt64);
202 EXPECT_EQ("-" + std::to_string(MinusMinInt64),
203 utostr(MinusMinInt64, /*isNeg=*/true));
204 EXPECT_EQ("-" + std::to_string(MaxUint64), utostr(MaxUint64, /*isNeg=*/true));
205 }
206
TEST(StringExtras,IToStr)207 TEST(StringExtras, IToStr) {
208 EXPECT_EQ("0", itostr(0));
209 EXPECT_EQ("1", itostr(1));
210 EXPECT_EQ("-1", itostr(-1));
211 EXPECT_EQ(std::to_string(MinInt64), itostr(MinInt64));
212 EXPECT_EQ(std::to_string(MaxInt64), itostr(MaxInt64));
213 }
214