Allow multiple logging::LogMessage{Handler,Listener}s

base/test/mock_log.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+#include "base/memory/ptr_util.h"
 #include "base/test/mock_log.h"
 
 namespace base {
 namespace test {
 
-// static
-MockLog* MockLog::g_instance_ = nullptr;
-Lock MockLog::g_lock;
+class MockLogMessageListener : logging::LogMessageListener {
+ public:
+  MockLogMessageListener(MockLog* mock_log) : mock_log_(mock_log) {}
+  void OnMessage(int severity,
+                 const char* file,
+                 int line,
+                 size_t message_start,
+                 const std::string& str) override {
+    return mock_log_->Log(severity, file, line, message_start, str);
+  }
+
+ private:
+  MockLog* mock_log_;
+};
 
 MockLog::MockLog() : is_capturing_logs_(false) {
 }
 
 MockLog::~MockLog() {
   if (is_capturing_logs_) {
     StopCapturingLogs();
   }
 }
 
 void MockLog::StartCapturingLogs() {
-  AutoLock scoped_lock(g_lock);
-
   // We don't use CHECK(), which can generate a new LOG message, and
   // thus can confuse MockLog objects or other registered
   // LogSinks.
   RAW_CHECK(!is_capturing_logs_);
-  RAW_CHECK(!g_instance_);
 
   is_capturing_logs_ = true;
-  g_instance_ = this;
-  previous_handler_ = logging::GetLogMessageHandler();
-  logging::SetLogMessageHandler(LogMessageHandler);
+  listener_ = base::MakeUnique<MockLogMessageListener>(this);
 }
 
 void MockLog::StopCapturingLogs() {
-  AutoLock scoped_lock(g_lock);
-
   // We don't use CHECK(), which can generate a new LOG message, and
   // thus can confuse MockLog objects or other registered
   // LogSinks.
   RAW_CHECK(is_capturing_logs_);
-  RAW_CHECK(g_instance_ == this);
 
   is_capturing_logs_ = false;
-  logging::SetLogMessageHandler(previous_handler_);
-  g_instance_ = nullptr;
-}
-
-// static
-bool MockLog::LogMessageHandler(int severity,
-                                const char* file,
-                                int line,
-                                size_t message_start,
-                                const std::string& str) {
-  // gMock guarantees thread-safety for calling a mocked method
-  // (https://github.com/google/googlemock/blob/master/googlemock/docs/CookBook.md#using-google-mock-and-threads)
-  // but we also need to make sure that Start/StopCapturingLogs are synchronized
-  // with LogMessageHandler.
-  AutoLock scoped_lock(g_lock);
-
-  return g_instance_->Log(severity, file, line, message_start, str);
+  listener_.reset();
 }
 
 }  // namespace test
 }  // namespace base