/3rd_party/llvm/unittests/Support/raw_ostream_test.cpp

https://code.google.com/p/softart/ · C++ · 146 lines · 92 code · 27 blank · 27 comment · 3 complexity · 03e43c95896eb97d494a06290dfeca7e MD5 · raw file

  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. #include "gtest/gtest.h"
  10. #include "llvm/ADT/SmallString.h"
  11. #include "llvm/Support/Format.h"
  12. #include "llvm/Support/raw_ostream.h"
  13. using namespace llvm;
  14. namespace {
  15. template<typename T> std::string printToString(const T &Value) {
  16. std::string res;
  17. llvm::raw_string_ostream(res) << Value;
  18. return res;
  19. }
  20. /// printToString - Print the given value to a stream which only has \arg
  21. /// BytesLeftInBuffer bytes left in the buffer. This is useful for testing edge
  22. /// cases in the buffer handling logic.
  23. template<typename T> std::string printToString(const T &Value,
  24. unsigned BytesLeftInBuffer) {
  25. // FIXME: This is relying on internal knowledge of how raw_ostream works to
  26. // get the buffer position right.
  27. SmallString<256> SVec;
  28. assert(BytesLeftInBuffer < 256 && "Invalid buffer count!");
  29. llvm::raw_svector_ostream OS(SVec);
  30. unsigned StartIndex = 256 - BytesLeftInBuffer;
  31. for (unsigned i = 0; i != StartIndex; ++i)
  32. OS << '?';
  33. OS << Value;
  34. return OS.str().substr(StartIndex);
  35. }
  36. template<typename T> std::string printToStringUnbuffered(const T &Value) {
  37. std::string res;
  38. llvm::raw_string_ostream OS(res);
  39. OS.SetUnbuffered();
  40. OS << Value;
  41. return res;
  42. }
  43. TEST(raw_ostreamTest, Types_Buffered) {
  44. // Char
  45. EXPECT_EQ("c", printToString('c'));
  46. // String
  47. EXPECT_EQ("hello", printToString("hello"));
  48. EXPECT_EQ("hello", printToString(std::string("hello")));
  49. // Int
  50. EXPECT_EQ("0", printToString(0));
  51. EXPECT_EQ("2425", printToString(2425));
  52. EXPECT_EQ("-2425", printToString(-2425));
  53. // Long long
  54. EXPECT_EQ("0", printToString(0LL));
  55. EXPECT_EQ("257257257235709", printToString(257257257235709LL));
  56. EXPECT_EQ("-257257257235709", printToString(-257257257235709LL));
  57. // Double
  58. EXPECT_EQ("1.100000e+00", printToString(1.1));
  59. // void*
  60. EXPECT_EQ("0x0", printToString((void*) 0));
  61. EXPECT_EQ("0xbeef", printToString((void*) 0xbeef));
  62. EXPECT_EQ("0xdeadbeef", printToString((void*) 0xdeadbeef));
  63. // Min and max.
  64. EXPECT_EQ("18446744073709551615", printToString(UINT64_MAX));
  65. EXPECT_EQ("-9223372036854775808", printToString(INT64_MIN));
  66. }
  67. TEST(raw_ostreamTest, Types_Unbuffered) {
  68. // Char
  69. EXPECT_EQ("c", printToStringUnbuffered('c'));
  70. // String
  71. EXPECT_EQ("hello", printToStringUnbuffered("hello"));
  72. EXPECT_EQ("hello", printToStringUnbuffered(std::string("hello")));
  73. // Int
  74. EXPECT_EQ("0", printToStringUnbuffered(0));
  75. EXPECT_EQ("2425", printToStringUnbuffered(2425));
  76. EXPECT_EQ("-2425", printToStringUnbuffered(-2425));
  77. // Long long
  78. EXPECT_EQ("0", printToStringUnbuffered(0LL));
  79. EXPECT_EQ("257257257235709", printToStringUnbuffered(257257257235709LL));
  80. EXPECT_EQ("-257257257235709", printToStringUnbuffered(-257257257235709LL));
  81. // Double
  82. EXPECT_EQ("1.100000e+00", printToStringUnbuffered(1.1));
  83. // void*
  84. EXPECT_EQ("0x0", printToStringUnbuffered((void*) 0));
  85. EXPECT_EQ("0xbeef", printToStringUnbuffered((void*) 0xbeef));
  86. EXPECT_EQ("0xdeadbeef", printToStringUnbuffered((void*) 0xdeadbeef));
  87. // Min and max.
  88. EXPECT_EQ("18446744073709551615", printToStringUnbuffered(UINT64_MAX));
  89. EXPECT_EQ("-9223372036854775808", printToStringUnbuffered(INT64_MIN));
  90. }
  91. TEST(raw_ostreamTest, BufferEdge) {
  92. EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 1));
  93. EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 2));
  94. EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 3));
  95. EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 4));
  96. EXPECT_EQ("1.20", printToString(format("%.2f", 1.2), 10));
  97. }
  98. TEST(raw_ostreamTest, TinyBuffer) {
  99. std::string Str;
  100. raw_string_ostream OS(Str);
  101. OS.SetBufferSize(1);
  102. OS << "hello";
  103. OS << 1;
  104. OS << 'w' << 'o' << 'r' << 'l' << 'd';
  105. EXPECT_EQ("hello1world", OS.str());
  106. }
  107. TEST(raw_ostreamTest, WriteEscaped) {
  108. std::string Str;
  109. Str = "";
  110. raw_string_ostream(Str).write_escaped("hi");
  111. EXPECT_EQ("hi", Str);
  112. Str = "";
  113. raw_string_ostream(Str).write_escaped("\\\t\n\"");
  114. EXPECT_EQ("\\\\\\t\\n\\\"", Str);
  115. Str = "";
  116. raw_string_ostream(Str).write_escaped("\1\10\200");
  117. EXPECT_EQ("\\001\\010\\200", Str);
  118. }
  119. }