fix leak in a test, make the sanitizer bot green
[oota-llvm.git] / unittests / Support / raw_ostream_test.cpp
1 //===- llvm/unittest/Support/raw_ostream_test.cpp - raw_ostream tests -----===//
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 "gtest/gtest.h"
11 #include "llvm/ADT/SmallString.h"
12 #include "llvm/Support/Format.h"
13 #include "llvm/Support/raw_ostream.h"
14
15 using namespace llvm;
16
17 namespace {
18
19 template<typename T> std::string printToString(const T &Value) {
20   std::string res;
21   llvm::raw_string_ostream(res) << Value;
22   return res;    
23 }
24
25 /// printToString - Print the given value to a stream which only has \arg
26 /// BytesLeftInBuffer bytes left in the buffer. This is useful for testing edge
27 /// cases in the buffer handling logic.
28 template<typename T> std::string printToString(const T &Value,
29                                                unsigned BytesLeftInBuffer) {
30   // FIXME: This is relying on internal knowledge of how raw_ostream works to
31   // get the buffer position right.
32   SmallString<256> SVec;
33   assert(BytesLeftInBuffer < 256 && "Invalid buffer count!");
34   llvm::raw_svector_ostream OS(SVec);
35   unsigned StartIndex = 256 - BytesLeftInBuffer;
36   for (unsigned i = 0; i != StartIndex; ++i)
37     OS << '?';
38   OS << Value;
39   return OS.str().substr(StartIndex);
40 }
41
42 template<typename T> std::string printToStringUnbuffered(const T &Value) {
43   std::string res;
44   llvm::raw_string_ostream OS(res);
45   OS.SetUnbuffered();
46   OS << Value;
47   return res;
48 }
49
50 TEST(raw_ostreamTest, Types_Buffered) {
51   // Char
52   EXPECT_EQ("c", printToString('c'));
53
54   // String
55   EXPECT_EQ("hello", printToString("hello"));
56   EXPECT_EQ("hello", printToString(std::string("hello")));
57
58   // Int
59   EXPECT_EQ("0", printToString(0));
60   EXPECT_EQ("2425", printToString(2425));
61   EXPECT_EQ("-2425", printToString(-2425));
62
63   // Long long
64   EXPECT_EQ("0", printToString(0LL));
65   EXPECT_EQ("257257257235709", printToString(257257257235709LL));
66   EXPECT_EQ("-257257257235709", printToString(-257257257235709LL));
67
68   // Double
69   EXPECT_EQ("1.100000e+00", printToString(1.1));
70
71   // void*
72   EXPECT_EQ("0x0", printToString((void*) nullptr));
73   EXPECT_EQ("0xbeef", printToString((void*) 0xbeef));
74   EXPECT_EQ("0xdeadbeef", printToString((void*) 0xdeadbeef));
75
76   // Min and max.
77   EXPECT_EQ("18446744073709551615", printToString(UINT64_MAX));
78   EXPECT_EQ("-9223372036854775808", printToString(INT64_MIN));
79 }
80
81 TEST(raw_ostreamTest, Types_Unbuffered) {  
82   // Char
83   EXPECT_EQ("c", printToStringUnbuffered('c'));
84
85   // String
86   EXPECT_EQ("hello", printToStringUnbuffered("hello"));
87   EXPECT_EQ("hello", printToStringUnbuffered(std::string("hello")));
88
89   // Int
90   EXPECT_EQ("0", printToStringUnbuffered(0));
91   EXPECT_EQ("2425", printToStringUnbuffered(2425));
92   EXPECT_EQ("-2425", printToStringUnbuffered(-2425));
93
94   // Long long
95   EXPECT_EQ("0", printToStringUnbuffered(0LL));
96   EXPECT_EQ("257257257235709", printToStringUnbuffered(257257257235709LL));
97   EXPECT_EQ("-257257257235709", printToStringUnbuffered(-257257257235709LL));
98
99   // Double
100   EXPECT_EQ("1.100000e+00", printToStringUnbuffered(1.1));
101
102   // void*
103   EXPECT_EQ("0x0", printToStringUnbuffered((void*) nullptr));
104   EXPECT_EQ("0xbeef", printToStringUnbuffered((void*) 0xbeef));
105   EXPECT_EQ("0xdeadbeef", printToStringUnbuffered((void*) 0xdeadbeef));
106
107   // Min and max.
108   EXPECT_EQ("18446744073709551615", printToStringUnbuffered(UINT64_MAX));
109   EXPECT_EQ("-9223372036854775808", printToStringUnbuffered(INT64_MIN));
110 }
111
112 TEST(raw_ostreamTest, BufferEdge) {  
113   EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 1));
114   EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 2));
115   EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 3));
116   EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 4));
117   EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 10));
118 }
119
120 TEST(raw_ostreamTest, TinyBuffer) {
121   std::string Str;
122   raw_string_ostream OS(Str);
123   OS.SetBufferSize(1);
124   OS << "hello";
125   OS << 1;
126   OS << 'w' << 'o' << 'r' << 'l' << 'd';
127   EXPECT_EQ("hello1world", OS.str());
128 }
129
130 TEST(raw_ostreamTest, WriteEscaped) {
131   std::string Str;
132
133   Str = "";
134   raw_string_ostream(Str).write_escaped("hi");
135   EXPECT_EQ("hi", Str);
136
137   Str = "";
138   raw_string_ostream(Str).write_escaped("\\\t\n\"");
139   EXPECT_EQ("\\\\\\t\\n\\\"", Str);
140
141   Str = "";
142   raw_string_ostream(Str).write_escaped("\1\10\200");
143   EXPECT_EQ("\\001\\010\\200", Str);
144 }
145
146 TEST(raw_ostreamTest, Justify) {  
147   EXPECT_EQ("xyz   ", printToString(left_justify("xyz", 6), 6));
148   EXPECT_EQ("abc",    printToString(left_justify("abc", 3), 3));
149   EXPECT_EQ("big",    printToString(left_justify("big", 1), 3));
150   EXPECT_EQ("   xyz", printToString(right_justify("xyz", 6), 6));
151   EXPECT_EQ("abc",    printToString(right_justify("abc", 3), 3));
152   EXPECT_EQ("big",    printToString(right_justify("big", 1), 3));
153 }
154
155 TEST(raw_ostreamTest, FormatHex) {  
156   EXPECT_EQ("0x1234",     printToString(format_hex(0x1234, 6), 6));
157   EXPECT_EQ("0x001234",   printToString(format_hex(0x1234, 8), 8));
158   EXPECT_EQ("0x00001234", printToString(format_hex(0x1234, 10), 10));
159   EXPECT_EQ("0x1234",     printToString(format_hex(0x1234, 4), 6));
160   EXPECT_EQ("0xff",       printToString(format_hex(255, 4), 4));
161   EXPECT_EQ("0xFF",       printToString(format_hex(255, 4, true), 4));
162   EXPECT_EQ("0x1",        printToString(format_hex(1, 3), 3));
163   EXPECT_EQ("0x12",       printToString(format_hex(0x12, 3), 4));
164   EXPECT_EQ("0x123",      printToString(format_hex(0x123, 3), 5));
165   EXPECT_EQ("FF",         printToString(format_hex_no_prefix(0xFF, 2, true), 4));
166   EXPECT_EQ("ABCD",       printToString(format_hex_no_prefix(0xABCD, 2, true), 4));
167   EXPECT_EQ("0xffffffffffffffff",     
168                           printToString(format_hex(UINT64_MAX, 18), 18));
169   EXPECT_EQ("0x8000000000000000",     
170                           printToString(format_hex((INT64_MIN), 18), 18));
171 }
172
173 TEST(raw_ostreamTest, FormatDecimal) {  
174   EXPECT_EQ("   0",        printToString(format_decimal(0, 4), 4));
175   EXPECT_EQ("  -1",        printToString(format_decimal(-1, 4), 4));
176   EXPECT_EQ("    -1",      printToString(format_decimal(-1, 6), 6));
177   EXPECT_EQ("1234567890",  printToString(format_decimal(1234567890, 10), 10));
178   EXPECT_EQ("  9223372036854775807", 
179                           printToString(format_decimal(INT64_MAX, 21), 21));
180   EXPECT_EQ(" -9223372036854775808", 
181                           printToString(format_decimal(INT64_MIN, 21), 21));
182 }
183
184
185 }