1 // Copyright 2014 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 "net/test/scoped_mock_log.h"
6
7 #include "base/logging.h"
8
9 namespace net {
10 namespace test {
11
12 // static
13 ScopedMockLog* ScopedMockLog::g_instance_ = NULL;
14
ScopedMockLog()15 ScopedMockLog::ScopedMockLog() : is_capturing_logs_(false) {}
16
~ScopedMockLog()17 ScopedMockLog::~ScopedMockLog() {
18 if (is_capturing_logs_) {
19 StopCapturingLogs();
20 }
21 }
22
StartCapturingLogs()23 void ScopedMockLog::StartCapturingLogs() {
24 // We don't use CHECK(), which can generate a new LOG message, and
25 // thus can confuse ScopedMockLog objects or other registered
26 // LogSinks.
27 RAW_CHECK(!is_capturing_logs_);
28 RAW_CHECK(!g_instance_);
29
30 is_capturing_logs_ = true;
31 g_instance_ = this;
32 previous_handler_ = logging::GetLogMessageHandler();
33 logging::SetLogMessageHandler(LogMessageHandler);
34 }
35
StopCapturingLogs()36 void ScopedMockLog::StopCapturingLogs() {
37 // We don't use CHECK(), which can generate a new LOG message, and
38 // thus can confuse ScopedMockLog objects or other registered
39 // LogSinks.
40 RAW_CHECK(is_capturing_logs_);
41 RAW_CHECK(g_instance_ == this);
42
43 is_capturing_logs_ = false;
44 logging::SetLogMessageHandler(previous_handler_);
45 g_instance_ = NULL;
46 }
47
48 // static
LogMessageHandler(int severity,const char * file,int line,size_t message_start,const std::string & str)49 bool ScopedMockLog::LogMessageHandler(int severity,
50 const char* file,
51 int line,
52 size_t message_start,
53 const std::string& str) {
54 return g_instance_->Log(severity, file, line, message_start, str);
55 }
56
57 } // namespace test
58 } // namespace net
59