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*) 0)); 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*) 0)); 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 } 147
