1 //===- StringExtrasTest.cpp - Unit tests for String extras ----------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/ADT/StringExtras.h"
11 #include "llvm/Support/raw_ostream.h"
12 #include "gtest/gtest.h"
13 
14 using namespace llvm;
15 
16 TEST(StringExtrasTest, isPrint) {
17   EXPECT_FALSE(isPrint('\0'));
18   EXPECT_FALSE(isPrint('\t'));
19   EXPECT_TRUE(isPrint('0'));
20   EXPECT_TRUE(isPrint('a'));
21   EXPECT_TRUE(isPrint('A'));
22   EXPECT_TRUE(isPrint(' '));
23   EXPECT_TRUE(isPrint('~'));
24   EXPECT_TRUE(isPrint('?'));
25 }
26 
27 TEST(StringExtrasTest, Join) {
28   std::vector<std::string> Items;
29   EXPECT_EQ("", join(Items.begin(), Items.end(), " <sep> "));
30 
31   Items = {"foo"};
32   EXPECT_EQ("foo", join(Items.begin(), Items.end(), " <sep> "));
33 
34   Items = {"foo", "bar"};
35   EXPECT_EQ("foo <sep> bar", join(Items.begin(), Items.end(), " <sep> "));
36 
37   Items = {"foo", "bar", "baz"};
38   EXPECT_EQ("foo <sep> bar <sep> baz",
39             join(Items.begin(), Items.end(), " <sep> "));
40 }
41 
42 TEST(StringExtrasTest, JoinItems) {
43   const char *Foo = "foo";
44   std::string Bar = "bar";
45   llvm::StringRef Baz = "baz";
46   char X = 'x';
47 
48   EXPECT_EQ("", join_items(" <sep> "));
49   EXPECT_EQ("", join_items('/'));
50 
51   EXPECT_EQ("foo", join_items(" <sep> ", Foo));
52   EXPECT_EQ("foo", join_items('/', Foo));
53 
54   EXPECT_EQ("foo <sep> bar", join_items(" <sep> ", Foo, Bar));
55   EXPECT_EQ("foo/bar", join_items('/', Foo, Bar));
56 
57   EXPECT_EQ("foo <sep> bar <sep> baz", join_items(" <sep> ", Foo, Bar, Baz));
58   EXPECT_EQ("foo/bar/baz", join_items('/', Foo, Bar, Baz));
59 
60   EXPECT_EQ("foo <sep> bar <sep> baz <sep> x",
61             join_items(" <sep> ", Foo, Bar, Baz, X));
62 
63   EXPECT_EQ("foo/bar/baz/x", join_items('/', Foo, Bar, Baz, X));
64 }
65 
66 TEST(StringExtrasTest, ToAndFromHex) {
67   std::vector<uint8_t> OddBytes = {0x5, 0xBD, 0x0D, 0x3E, 0xCD};
68   std::string OddStr = "05BD0D3ECD";
69   StringRef OddData(reinterpret_cast<const char *>(OddBytes.data()),
70                     OddBytes.size());
71   EXPECT_EQ(OddStr, toHex(OddData));
72   EXPECT_EQ(OddData, fromHex(StringRef(OddStr).drop_front()));
73   EXPECT_EQ(StringRef(OddStr).lower(), toHex(OddData, true));
74 
75   std::vector<uint8_t> EvenBytes = {0xA5, 0xBD, 0x0D, 0x3E, 0xCD};
76   std::string EvenStr = "A5BD0D3ECD";
77   StringRef EvenData(reinterpret_cast<const char *>(EvenBytes.data()),
78                      EvenBytes.size());
79   EXPECT_EQ(EvenStr, toHex(EvenData));
80   EXPECT_EQ(EvenData, fromHex(EvenStr));
81   EXPECT_EQ(StringRef(EvenStr).lower(), toHex(EvenData, true));
82 }
83 
84 TEST(StringExtrasTest, to_float) {
85   float F;
86   EXPECT_TRUE(to_float("4.7", F));
87   EXPECT_FLOAT_EQ(4.7f, F);
88 
89   double D;
90   EXPECT_TRUE(to_float("4.7", D));
91   EXPECT_DOUBLE_EQ(4.7, D);
92 
93   long double LD;
94   EXPECT_TRUE(to_float("4.7", LD));
95   EXPECT_DOUBLE_EQ(4.7, LD);
96 
97   EXPECT_FALSE(to_float("foo", F));
98   EXPECT_FALSE(to_float("7.4 foo", F));
99   EXPECT_FLOAT_EQ(4.7f, F); // F should be unchanged
100 }
101 
102 TEST(StringExtrasTest, printLowerCase) {
103   std::string str;
104   raw_string_ostream OS(str);
105   printLowerCase("ABCdefg01234.,&!~`'}\"", OS);
106   EXPECT_EQ("abcdefg01234.,&!~`'}\"", OS.str());
107 }
108 
109 TEST(StringExtrasTest, printEscapedString) {
110   std::string str;
111   raw_string_ostream OS(str);
112   printEscapedString("ABCdef123&<>\\\"'\t", OS);
113   EXPECT_EQ("ABCdef123&<>\\5C\\22'\\09", OS.str());
114 }
115 
116 TEST(StringExtrasTest, printHTMLEscaped) {
117   std::string str;
118   raw_string_ostream OS(str);
119   printHTMLEscaped("ABCdef123&<>\"'", OS);
120   EXPECT_EQ("ABCdef123&amp;&lt;&gt;&quot;&apos;", OS.str());
121 }
122