Chromium Code Reviews| Index: base/trace_event/trace_event_unittest.cc |
| diff --git a/base/trace_event/trace_event_unittest.cc b/base/trace_event/trace_event_unittest.cc |
| index c22f660b43c183af60f21e055ffa975b11ee0f9b..1170d7a27a8475e817bae3c259f4d72ef8949918 100644 |
| --- a/base/trace_event/trace_event_unittest.cc |
| +++ b/base/trace_event/trace_event_unittest.cc |
| @@ -2824,18 +2824,21 @@ TEST_F(TraceEventTestFixture, ThreadOnceBlocking) { |
| } |
| std::string* g_log_buffer = NULL; |
| -bool MockLogMessageHandler(int, const char*, int, size_t, |
| - const std::string& str) { |
| - if (!g_log_buffer) |
| - g_log_buffer = new std::string(); |
| - g_log_buffer->append(str); |
| - return false; |
| -} |
| +class MockLogMessageListener : logging::LogMessageListener { |
|
grt (UTC plus 2)
2017/01/04 09:12:19
public
|
| + public: |
| + void OnMessage(int severity, |
| + const char* file, |
| + int line, |
| + size_t message_start, |
| + const std::string& str) override { |
| + if (!g_log_buffer) |
| + g_log_buffer = new std::string(); |
| + g_log_buffer->append(str); |
| + } |
| +}; |
|
grt (UTC plus 2)
2017/01/04 09:12:19
DISALLOW
|
| TEST_F(TraceEventTestFixture, EchoToConsole) { |
| - logging::LogMessageHandlerFunction old_log_message_handler = |
| - logging::GetLogMessageHandler(); |
| - logging::SetLogMessageHandler(MockLogMessageHandler); |
| + MockLogMessageListener l; |
|
grt (UTC plus 2)
2017/01/04 09:12:19
https://google.github.io/styleguide/cppguide.html#
|
| TraceLog::GetInstance()->SetEnabled( |
| TraceConfig(kRecordAllCategoryFilter, ECHO_TO_CONSOLE), |
| @@ -2858,20 +2861,22 @@ TEST_F(TraceEventTestFixture, EchoToConsole) { |
| EndTraceAndFlush(); |
| delete g_log_buffer; |
| - logging::SetLogMessageHandler(old_log_message_handler); |
| g_log_buffer = NULL; |
| } |
| -bool LogMessageHandlerWithTraceEvent(int, const char*, int, size_t, |
| - const std::string&) { |
| - TRACE_EVENT0("log", "trace_event"); |
| - return false; |
| -} |
| +class LogMessageListenerWithTraceEvent : logging::LogMessageListener { |
|
grt (UTC plus 2)
2017/01/04 09:12:19
public
|
| + public: |
| + void OnMessage(int severity, |
| + const char* file, |
| + int line, |
| + size_t message_start, |
| + const std::string& str) override { |
| + TRACE_EVENT0("log", "trace_event"); |
| + } |
| +}; |
|
grt (UTC plus 2)
2017/01/04 09:12:19
private:
DISALLOW...
|
| TEST_F(TraceEventTestFixture, EchoToConsoleTraceEventRecursion) { |
| - logging::LogMessageHandlerFunction old_log_message_handler = |
| - logging::GetLogMessageHandler(); |
| - logging::SetLogMessageHandler(LogMessageHandlerWithTraceEvent); |
| + LogMessageListenerWithTraceEvent l; |
| TraceLog::GetInstance()->SetEnabled( |
| TraceConfig(kRecordAllCategoryFilter, ECHO_TO_CONSOLE), |
| @@ -2882,7 +2887,6 @@ TEST_F(TraceEventTestFixture, EchoToConsoleTraceEventRecursion) { |
| } |
| EndTraceAndFlush(); |
| - logging::SetLogMessageHandler(old_log_message_handler); |
| } |
| TEST_F(TraceEventTestFixture, TimeOffset) { |