| 1 | // Copyright 2005, Google Inc. |
| 2 | // All rights reserved. |
| 3 | // |
| 4 | // Redistribution and use in source and binary forms, with or without |
| 5 | // modification, are permitted provided that the following conditions are |
| 6 | // met: |
| 7 | // |
| 8 | // * Redistributions of source code must retain the above copyright |
| 9 | // notice, this list of conditions and the following disclaimer. |
| 10 | // * Redistributions in binary form must reproduce the above |
| 11 | // copyright notice, this list of conditions and the following disclaimer |
| 12 | // in the documentation and/or other materials provided with the |
| 13 | // distribution. |
| 14 | // * Neither the name of Google Inc. nor the names of its |
| 15 | // contributors may be used to endorse or promote products derived from |
| 16 | // this software without specific prior written permission. |
| 17 | // |
| 18 | // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 19 | // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 20 | // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 21 | // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 22 | // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 23 | // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 24 | // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 25 | // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 26 | // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 27 | // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 28 | // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 29 | |
| 30 | // The Google C++ Testing and Mocking Framework (Google Test) |
| 31 | // |
| 32 | // This header file defines the Message class. |
| 33 | // |
| 34 | // IMPORTANT NOTE: Due to limitation of the C++ language, we have to |
| 35 | // leave some internal implementation details in this header file. |
| 36 | // They are clearly marked by comments like this: |
| 37 | // |
| 38 | // // INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM. |
| 39 | // |
| 40 | // Such code is NOT meant to be used by a user directly, and is subject |
| 41 | // to CHANGE WITHOUT NOTICE. Therefore DO NOT DEPEND ON IT in a user |
| 42 | // program! |
| 43 | |
| 44 | // IWYU pragma: private, include "gtest/gtest.h" |
| 45 | // IWYU pragma: friend gtest/.* |
| 46 | // IWYU pragma: friend gmock/.* |
| 47 | |
| 48 | #ifndef GOOGLETEST_INCLUDE_GTEST_GTEST_MESSAGE_H_ |
| 49 | #define GOOGLETEST_INCLUDE_GTEST_GTEST_MESSAGE_H_ |
| 50 | |
| 51 | #include <limits> |
| 52 | #include <memory> |
| 53 | #include <ostream> |
| 54 | #include <sstream> |
| 55 | #include <string> |
| 56 | |
| 57 | #include "gtest/internal/gtest-port.h" |
| 58 | |
| 59 | #ifdef GTEST_HAS_ABSL |
| 60 | #include <type_traits> |
| 61 | |
| 62 | #include "absl/strings/internal/has_absl_stringify.h" |
| 63 | #include "absl/strings/str_cat.h" |
| 64 | #endif // GTEST_HAS_ABSL |
| 65 | |
| 66 | GTEST_DISABLE_MSC_WARNINGS_PUSH_(4251 \ |
| 67 | /* class A needs to have dll-interface to be used by clients of class B */) |
| 68 | |
| 69 | // Ensures that there is at least one operator<< in the global namespace. |
| 70 | // See Message& operator<<(...) below for why. |
| 71 | void operator<<(const testing::internal::Secret&, int); |
| 72 | |
| 73 | namespace testing { |
| 74 | |
| 75 | // The Message class works like an ostream repeater. |
| 76 | // |
| 77 | // Typical usage: |
| 78 | // |
| 79 | // 1. You stream a bunch of values to a Message object. |
| 80 | // It will remember the text in a stringstream. |
| 81 | // 2. Then you stream the Message object to an ostream. |
| 82 | // This causes the text in the Message to be streamed |
| 83 | // to the ostream. |
| 84 | // |
| 85 | // For example; |
| 86 | // |
| 87 | // testing::Message foo; |
| 88 | // foo << 1 << " != " << 2; |
| 89 | // std::cout << foo; |
| 90 | // |
| 91 | // will print "1 != 2". |
| 92 | // |
| 93 | // Message is not intended to be inherited from. In particular, its |
| 94 | // destructor is not virtual. |
| 95 | // |
| 96 | // Note that stringstream behaves differently in gcc and in MSVC. You |
| 97 | // can stream a NULL char pointer to it in the former, but not in the |
| 98 | // latter (it causes an access violation if you do). The Message |
| 99 | // class hides this difference by treating a NULL char pointer as |
| 100 | // "(null)". |
| 101 | class GTEST_API_ Message { |
| 102 | private: |
| 103 | // The type of basic IO manipulators (endl, ends, and flush) for |
| 104 | // narrow streams. |
| 105 | typedef std::ostream& (*BasicNarrowIoManip)(std::ostream&); |
| 106 | |
| 107 | public: |
| 108 | // Constructs an empty Message. |
| 109 | Message(); |
| 110 | |
| 111 | // Copy constructor. |
| 112 | Message(const Message& msg) : ss_(new ::std::stringstream) { // NOLINT |
| 113 | *ss_ << msg.GetString(); |
| 114 | } |
| 115 | |
| 116 | // Constructs a Message from a C-string. |
| 117 | explicit Message(const char* str) : ss_(new ::std::stringstream) { |
| 118 | *ss_ << str; |
| 119 | } |
| 120 | |
| 121 | // Streams a non-pointer value to this object. If building a version of |
| 122 | // GoogleTest with ABSL, this overload is only enabled if the value does not |
| 123 | // have an AbslStringify definition. |
| 124 | template <typename T |
| 125 | #ifdef GTEST_HAS_ABSL |
| 126 | , |
| 127 | typename std::enable_if< |
| 128 | !absl::strings_internal::HasAbslStringify<T>::value, // NOLINT |
| 129 | int>::type = 0 |
| 130 | #endif // GTEST_HAS_ABSL |
| 131 | > |
| 132 | inline Message& operator<<(const T& val) { |
| 133 | // Some libraries overload << for STL containers. These |
| 134 | // overloads are defined in the global namespace instead of ::std. |
| 135 | // |
| 136 | // C++'s symbol lookup rule (i.e. Koenig lookup) says that these |
| 137 | // overloads are visible in either the std namespace or the global |
| 138 | // namespace, but not other namespaces, including the testing |
| 139 | // namespace which Google Test's Message class is in. |
| 140 | // |
| 141 | // To allow STL containers (and other types that has a << operator |
| 142 | // defined in the global namespace) to be used in Google Test |
| 143 | // assertions, testing::Message must access the custom << operator |
| 144 | // from the global namespace. With this using declaration, |
| 145 | // overloads of << defined in the global namespace and those |
| 146 | // visible via Koenig lookup are both exposed in this function. |
| 147 | using ::operator<<; |
| 148 | // LLVM local change to support llvm printables. |
| 149 | // |
| 150 | // *ss_ << val; |
| 151 | *ss_ << llvm_gtest::printable(val); |
| 152 | // LLVM local change end. |
| 153 | return *this; |
| 154 | } |
| 155 | |
| 156 | #ifdef GTEST_HAS_ABSL |
| 157 | // Streams a non-pointer value with an AbslStringify definition to this |
| 158 | // object. |
| 159 | template <typename T, |
| 160 | typename std::enable_if< |
| 161 | absl::strings_internal::HasAbslStringify<T>::value, // NOLINT |
| 162 | int>::type = 0> |
| 163 | inline Message& operator<<(const T& val) { |
| 164 | // ::operator<< is needed here for a similar reason as with the non-Abseil |
| 165 | // version above |
| 166 | using ::operator<<; |
| 167 | *ss_ << absl::StrCat(val); |
| 168 | return *this; |
| 169 | } |
| 170 | #endif // GTEST_HAS_ABSL |
| 171 | |
| 172 | // Streams a pointer value to this object. |
| 173 | // |
| 174 | // This function is an overload of the previous one. When you |
| 175 | // stream a pointer to a Message, this definition will be used as it |
| 176 | // is more specialized. (The C++ Standard, section |
| 177 | // [temp.func.order].) If you stream a non-pointer, then the |
| 178 | // previous definition will be used. |
| 179 | // |
| 180 | // The reason for this overload is that streaming a NULL pointer to |
| 181 | // ostream is undefined behavior. Depending on the compiler, you |
| 182 | // may get "0", "(nil)", "(null)", or an access violation. To |
| 183 | // ensure consistent result across compilers, we always treat NULL |
| 184 | // as "(null)". |
| 185 | template <typename T> |
| 186 | inline Message& operator<<(T* const& pointer) { // NOLINT |
| 187 | if (pointer == nullptr) { |
| 188 | *ss_ << "(null)" ; |
| 189 | } else { |
| 190 | // LLVM local change to support llvm printables. |
| 191 | // |
| 192 | // *ss_ << pointer; |
| 193 | *ss_ << llvm_gtest::printable(pointer); |
| 194 | // LLVM local change end. |
| 195 | } |
| 196 | return *this; |
| 197 | } |
| 198 | |
| 199 | // Since the basic IO manipulators are overloaded for both narrow |
| 200 | // and wide streams, we have to provide this specialized definition |
| 201 | // of operator <<, even though its body is the same as the |
| 202 | // templatized version above. Without this definition, streaming |
| 203 | // endl or other basic IO manipulators to Message will confuse the |
| 204 | // compiler. |
| 205 | Message& operator<<(BasicNarrowIoManip val) { |
| 206 | *ss_ << val; |
| 207 | return *this; |
| 208 | } |
| 209 | |
| 210 | // Instead of 1/0, we want to see true/false for bool values. |
| 211 | Message& operator<<(bool b) { return *this << (b ? "true" : "false" ); } |
| 212 | |
| 213 | // These two overloads allow streaming a wide C string to a Message |
| 214 | // using the UTF-8 encoding. |
| 215 | Message& operator<<(const wchar_t* wide_c_str); |
| 216 | Message& operator<<(wchar_t* wide_c_str); |
| 217 | |
| 218 | #if GTEST_HAS_STD_WSTRING |
| 219 | // Converts the given wide string to a narrow string using the UTF-8 |
| 220 | // encoding, and streams the result to this Message object. |
| 221 | Message& operator<<(const ::std::wstring& wstr); |
| 222 | #endif // GTEST_HAS_STD_WSTRING |
| 223 | |
| 224 | // Gets the text streamed to this object so far as an std::string. |
| 225 | // Each '\0' character in the buffer is replaced with "\\0". |
| 226 | // |
| 227 | // INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM. |
| 228 | std::string GetString() const; |
| 229 | |
| 230 | private: |
| 231 | // We'll hold the text streamed to this object here. |
| 232 | const std::unique_ptr< ::std::stringstream> ss_; |
| 233 | |
| 234 | // We declare (but don't implement) this to prevent the compiler |
| 235 | // from implementing the assignment operator. |
| 236 | void operator=(const Message&); |
| 237 | }; |
| 238 | |
| 239 | // Streams a Message to an ostream. |
| 240 | inline std::ostream& operator<<(std::ostream& os, const Message& sb) { |
| 241 | return os << sb.GetString(); |
| 242 | } |
| 243 | |
| 244 | namespace internal { |
| 245 | |
| 246 | // Converts a streamable value to an std::string. A NULL pointer is |
| 247 | // converted to "(null)". When the input value is a ::string, |
| 248 | // ::std::string, ::wstring, or ::std::wstring object, each NUL |
| 249 | // character in it is replaced with "\\0". |
| 250 | template <typename T> |
| 251 | std::string StreamableToString(const T& streamable) { |
| 252 | return (Message() << streamable).GetString(); |
| 253 | } |
| 254 | |
| 255 | } // namespace internal |
| 256 | } // namespace testing |
| 257 | |
| 258 | GTEST_DISABLE_MSC_WARNINGS_POP_() // 4251 |
| 259 | |
| 260 | #endif // GOOGLETEST_INCLUDE_GTEST_GTEST_MESSAGE_H_ |
| 261 | |