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 | *ss_ << val; |
149 | return *this; |
150 | } |
151 | |
152 | #ifdef GTEST_HAS_ABSL |
153 | // Streams a non-pointer value with an AbslStringify definition to this |
154 | // object. |
155 | template <typename T, |
156 | typename std::enable_if< |
157 | absl::strings_internal::HasAbslStringify<T>::value, // NOLINT |
158 | int>::type = 0> |
159 | inline Message& operator<<(const T& val) { |
160 | // ::operator<< is needed here for a similar reason as with the non-Abseil |
161 | // version above |
162 | using ::operator<<; |
163 | *ss_ << absl::StrCat(val); |
164 | return *this; |
165 | } |
166 | #endif // GTEST_HAS_ABSL |
167 | |
168 | // Streams a pointer value to this object. |
169 | // |
170 | // This function is an overload of the previous one. When you |
171 | // stream a pointer to a Message, this definition will be used as it |
172 | // is more specialized. (The C++ Standard, section |
173 | // [temp.func.order].) If you stream a non-pointer, then the |
174 | // previous definition will be used. |
175 | // |
176 | // The reason for this overload is that streaming a NULL pointer to |
177 | // ostream is undefined behavior. Depending on the compiler, you |
178 | // may get "0", "(nil)", "(null)", or an access violation. To |
179 | // ensure consistent result across compilers, we always treat NULL |
180 | // as "(null)". |
181 | template <typename T> |
182 | inline Message& operator<<(T* const& pointer) { // NOLINT |
183 | if (pointer == nullptr) { |
184 | *ss_ << "(null)" ; |
185 | } else { |
186 | *ss_ << pointer; |
187 | } |
188 | return *this; |
189 | } |
190 | |
191 | // Since the basic IO manipulators are overloaded for both narrow |
192 | // and wide streams, we have to provide this specialized definition |
193 | // of operator <<, even though its body is the same as the |
194 | // templatized version above. Without this definition, streaming |
195 | // endl or other basic IO manipulators to Message will confuse the |
196 | // compiler. |
197 | Message& operator<<(BasicNarrowIoManip val) { |
198 | *ss_ << val; |
199 | return *this; |
200 | } |
201 | |
202 | // Instead of 1/0, we want to see true/false for bool values. |
203 | Message& operator<<(bool b) { return *this << (b ? "true" : "false" ); } |
204 | |
205 | // These two overloads allow streaming a wide C string to a Message |
206 | // using the UTF-8 encoding. |
207 | Message& operator<<(const wchar_t* wide_c_str); |
208 | Message& operator<<(wchar_t* wide_c_str); |
209 | |
210 | #if GTEST_HAS_STD_WSTRING |
211 | // Converts the given wide string to a narrow string using the UTF-8 |
212 | // encoding, and streams the result to this Message object. |
213 | Message& operator<<(const ::std::wstring& wstr); |
214 | #endif // GTEST_HAS_STD_WSTRING |
215 | |
216 | // Gets the text streamed to this object so far as an std::string. |
217 | // Each '\0' character in the buffer is replaced with "\\0". |
218 | // |
219 | // INTERNAL IMPLEMENTATION - DO NOT USE IN A USER PROGRAM. |
220 | std::string GetString() const; |
221 | |
222 | private: |
223 | // We'll hold the text streamed to this object here. |
224 | const std::unique_ptr< ::std::stringstream> ss_; |
225 | |
226 | // We declare (but don't implement) this to prevent the compiler |
227 | // from implementing the assignment operator. |
228 | void operator=(const Message&); |
229 | }; |
230 | |
231 | // Streams a Message to an ostream. |
232 | inline std::ostream& operator<<(std::ostream& os, const Message& sb) { |
233 | return os << sb.GetString(); |
234 | } |
235 | |
236 | namespace internal { |
237 | |
238 | // Converts a streamable value to an std::string. A NULL pointer is |
239 | // converted to "(null)". When the input value is a ::string, |
240 | // ::std::string, ::wstring, or ::std::wstring object, each NUL |
241 | // character in it is replaced with "\\0". |
242 | template <typename T> |
243 | std::string StreamableToString(const T& streamable) { |
244 | return (Message() << streamable).GetString(); |
245 | } |
246 | |
247 | } // namespace internal |
248 | } // namespace testing |
249 | |
250 | GTEST_DISABLE_MSC_WARNINGS_POP_() // 4251 |
251 | |
252 | #endif // GOOGLETEST_INCLUDE_GTEST_GTEST_MESSAGE_H_ |
253 | |