| // Copyright 2019 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/test/scoped_run_loop_timeout.h" |
| |
| #include "base/bind.h" |
| #include "base/callback_helpers.h" |
| #include "base/location.h" |
| #include "base/logging.h" |
| #include "base/strings/strcat.h" |
| #include "base/time/time.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace base { |
| namespace test { |
| |
| namespace { |
| |
| bool g_add_gtest_failure_on_timeout = false; |
| |
| std::string TimeoutMessage(const RepeatingCallback<std::string()>& get_log) { |
| std::string message = "RunLoop::Run() timed out."; |
| if (get_log) |
| StrAppend(&message, {"\n", get_log.Run()}); |
| return message; |
| } |
| |
| } // namespace |
| |
| ScopedRunLoopTimeout::ScopedRunLoopTimeout(const Location& from_here, |
| TimeDelta timeout) |
| : ScopedRunLoopTimeout(from_here, timeout, NullCallback()) {} |
| |
| ScopedRunLoopTimeout::~ScopedRunLoopTimeout() { |
| RunLoop::SetTimeoutForCurrentThread(nested_timeout_); |
| } |
| |
| ScopedRunLoopTimeout::ScopedRunLoopTimeout( |
| const Location& from_here, |
| TimeDelta timeout, |
| RepeatingCallback<std::string()> on_timeout_log) |
| : nested_timeout_(RunLoop::GetTimeoutForCurrentThread()) { |
| DCHECK_GT(timeout, TimeDelta()); |
| run_timeout_.timeout = timeout; |
| |
| if (g_add_gtest_failure_on_timeout) { |
| run_timeout_.on_timeout = BindRepeating( |
| [](const Location& from_here, |
| RepeatingCallback<std::string()> on_timeout_log) { |
| GTEST_FAIL_AT(from_here.file_name(), from_here.line_number()) |
| << TimeoutMessage(on_timeout_log); |
| }, |
| from_here, std::move(on_timeout_log)); |
| } else { |
| run_timeout_.on_timeout = BindRepeating( |
| [](const Location& from_here, |
| RepeatingCallback<std::string()> on_timeout_log) { |
| std::string message = TimeoutMessage(on_timeout_log); |
| logging::LogMessage(from_here.file_name(), from_here.line_number(), |
| message.data()); |
| }, |
| from_here, std::move(on_timeout_log)); |
| } |
| |
| RunLoop::SetTimeoutForCurrentThread(&run_timeout_); |
| } |
| |
| // static |
| bool ScopedRunLoopTimeout::ExistsForCurrentThread() { |
| return RunLoop::GetTimeoutForCurrentThread() != nullptr; |
| } |
| |
| // static |
| void ScopedRunLoopTimeout::SetAddGTestFailureOnTimeout() { |
| g_add_gtest_failure_on_timeout = true; |
| } |
| |
| // static |
| const RunLoop::RunLoopTimeout* |
| ScopedRunLoopTimeout::GetTimeoutForCurrentThread() { |
| return RunLoop::GetTimeoutForCurrentThread(); |
| } |
| |
| ScopedDisableRunLoopTimeout::ScopedDisableRunLoopTimeout() |
| : nested_timeout_(RunLoop::GetTimeoutForCurrentThread()) { |
| RunLoop::SetTimeoutForCurrentThread(nullptr); |
| } |
| |
| ScopedDisableRunLoopTimeout::~ScopedDisableRunLoopTimeout() { |
| RunLoop::SetTimeoutForCurrentThread(nested_timeout_); |
| } |
| |
| } // namespace test |
| } // namespace base |