1 // Copyright 2015 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/test/mock_log.h"
6
7 namespace base {
8 namespace test {
9
10 // static
11 MockLog* MockLog::g_instance_ = nullptr;
12 Lock MockLog::g_lock;
13
MockLog()14 MockLog::MockLog() : is_capturing_logs_(false) {
15 }
16
~MockLog()17 MockLog::~MockLog() {
18 if (is_capturing_logs_) {
19 StopCapturingLogs();
20 }
21 }
22
StartCapturingLogs()23 void MockLog::StartCapturingLogs() {
24 AutoLock scoped_lock(g_lock);
25
26 // We don't use CHECK(), which can generate a new LOG message, and
27 // thus can confuse MockLog objects or other registered
28 // LogSinks.
29 RAW_CHECK(!is_capturing_logs_);
30 RAW_CHECK(!g_instance_);
31
32 is_capturing_logs_ = true;
33 g_instance_ = this;
34 previous_handler_ = logging::GetLogMessageHandler();
35 logging::SetLogMessageHandler(LogMessageHandler);
36 }
37
StopCapturingLogs()38 void MockLog::StopCapturingLogs() {
39 AutoLock scoped_lock(g_lock);
40
41 // We don't use CHECK(), which can generate a new LOG message, and
42 // thus can confuse MockLog objects or other registered
43 // LogSinks.
44 RAW_CHECK(is_capturing_logs_);
45 RAW_CHECK(g_instance_ == this);
46
47 is_capturing_logs_ = false;
48 logging::SetLogMessageHandler(previous_handler_);
49 g_instance_ = nullptr;
50 }
51
52 // static
LogMessageHandler(int severity,const char * file,int line,size_t message_start,const std::string & str)53 bool MockLog::LogMessageHandler(int severity,
54 const char* file,
55 int line,
56 size_t message_start,
57 const std::string& str) {
58 // gMock guarantees thread-safety for calling a mocked method
59 // (https://github.com/google/googlemock/blob/master/googlemock/docs/CookBook.md#using-google-mock-and-threads)
60 // but we also need to make sure that Start/StopCapturingLogs are synchronized
61 // with LogMessageHandler.
62 AutoLock scoped_lock(g_lock);
63
64 return g_instance_->Log(severity, file, line, message_start, str);
65 }
66
67 } // namespace test
68 } // namespace base
69