diff --git a/src/base/gtest/logging.cc b/src/base/gtest/logging.cc index d9cb9eb957..e1ef373dba 100644 --- a/src/base/gtest/logging.cc +++ b/src/base/gtest/logging.cc @@ -25,6 +25,8 @@ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ +#include "base/gtest/logging.hh" + #include #include @@ -49,7 +51,12 @@ class GTestLogger : public Logger using Logger::Logger; protected: - void log(const Loc &loc, std::string s) override { SUCCEED() << s; } + void + log(const Loc &loc, std::string s) override + { + gtestLogOutput << s; + SUCCEED() << s; + } }; class GTestExitLogger : public Logger @@ -61,6 +68,7 @@ class GTestExitLogger : public Logger void log(const Loc &loc, std::string s) override { + gtestLogOutput << s; std::cerr << loc.file << ":" << loc.line << ": " << s; } // Throw an exception to escape down to the gtest framework. @@ -75,6 +83,25 @@ GTestLogger hackLogger("hack: "); } // anonymous namespace +thread_local GTestLogOutput gtestLogOutput; + +GTestLogOutput::EventHook::EventHook(GTestLogOutput &_stream) : stream(_stream) +{ + ::testing::UnitTest::GetInstance()->listeners().Append(this); +} + +GTestLogOutput::EventHook::~EventHook() +{ + ::testing::UnitTest::GetInstance()->listeners().Release(this); +} + +void +GTestLogOutput::EventHook::OnTestStart(const ::testing::TestInfo &test_info) +{ + // Clear out the stream at the start of each test. + stream.str(""); +} + Logger &Logger::getPanic() { return panicLogger; } Logger &Logger::getFatal() { return fatalLogger; } Logger &Logger::getWarn() { return warnLogger; } diff --git a/src/base/gtest/logging.hh b/src/base/gtest/logging.hh new file mode 100644 index 0000000000..3889e30b7b --- /dev/null +++ b/src/base/gtest/logging.hh @@ -0,0 +1,56 @@ +/* + * Copyright 2021 Google Inc. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are + * met: redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer; + * redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution; + * neither the name of the copyright holders nor the names of its + * contributors may be used to endorse or promote products derived from + * this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include + +#include + +class GTestLogOutput : public std::ostringstream +{ + private: + class EventHook : public ::testing::EmptyTestEventListener + { + private: + GTestLogOutput &stream; + + public: + EventHook(GTestLogOutput &_stream); + ~EventHook(); + + void OnTestStart(const ::testing::TestInfo &test_info) override; + }; + + EventHook eventHook; + + public: + template + GTestLogOutput(Args &&...args) : + std::ostringstream(std::forward(args)...), eventHook(*this) + {} +}; + +extern thread_local GTestLogOutput gtestLogOutput;