Home | History | Annotate | Download | only in Support 
      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