Commit e2b632b5 authored by Gabriel Charette's avatar Gabriel Charette Committed by Commit Bot

Make RunLoop 100% standalone.

MessageLoop is no longer a friend of RunLoop :)!

The only minor dependency remaining shows up in tests as allowing
nestable tasks in a scope (different from allowing nested loops on a
thread in general) is still a MessageLoop concept. I plan to move that
to RunLoop as well in a follow-up.

R=danakj@chromium.org

Bug: 703346
Change-Id: I0a6269c94de179d97c14855499edd70be9e1d1af
Reviewed-on: https://chromium-review.googlesource.com/594352Reviewed-by: default avatardanakj <danakj@chromium.org>
Commit-Queue: Gabriel Charette <gab@chromium.org>
Cr-Commit-Position: refs/heads/master@{#491258}
parent a2cd2257
...@@ -105,8 +105,7 @@ MessageLoop::~MessageLoop() { ...@@ -105,8 +105,7 @@ MessageLoop::~MessageLoop() {
// There should be no active RunLoops on this thread, unless this MessageLoop // There should be no active RunLoops on this thread, unless this MessageLoop
// isn't bound to the current thread (see other condition at the top of this // isn't bound to the current thread (see other condition at the top of this
// method). // method).
DCHECK((!pump_ && current() != this) || DCHECK((!pump_ && current() != this) || !RunLoop::IsRunningOnCurrentThread());
!run_loop_client_->GetTopMostRunLoop());
#endif #endif
#if defined(OS_WIN) #if defined(OS_WIN)
...@@ -566,7 +565,7 @@ bool MessageLoop::DoIdleWork() { ...@@ -566,7 +565,7 @@ bool MessageLoop::DoIdleWork() {
if (ProcessNextDelayedNonNestableTask()) if (ProcessNextDelayedNonNestableTask())
return true; return true;
if (run_loop_client_->GetTopMostRunLoop()->quit_when_idle_received_) if (run_loop_client_->ShouldQuitWhenIdle())
pump_->Quit(); pump_->Quit();
// When we return we will do a kernel wait for more tasks. // When we return we will do a kernel wait for more tasks.
......
...@@ -48,11 +48,10 @@ RunLoop::Delegate::~Delegate() { ...@@ -48,11 +48,10 @@ RunLoop::Delegate::~Delegate() {
tls_delegate.Get().Set(nullptr); tls_delegate.Get().Set(nullptr);
} }
RunLoop* RunLoop::Delegate::Client::GetTopMostRunLoop() const { bool RunLoop::Delegate::Client::ShouldQuitWhenIdle() const {
DCHECK_CALLED_ON_VALID_THREAD(outer_->bound_thread_checker_); DCHECK_CALLED_ON_VALID_THREAD(outer_->bound_thread_checker_);
DCHECK(outer_->bound_); DCHECK(outer_->bound_);
return outer_->active_run_loops_.empty() ? nullptr return outer_->active_run_loops_.top()->quit_when_idle_received_;
: outer_->active_run_loops_.top();
} }
bool RunLoop::Delegate::Client::IsNested() const { bool RunLoop::Delegate::Client::IsNested() const {
...@@ -82,8 +81,8 @@ RunLoop::RunLoop() ...@@ -82,8 +81,8 @@ RunLoop::RunLoop()
: delegate_(tls_delegate.Get().Get()), : delegate_(tls_delegate.Get().Get()),
origin_task_runner_(ThreadTaskRunnerHandle::Get()), origin_task_runner_(ThreadTaskRunnerHandle::Get()),
weak_factory_(this) { weak_factory_(this) {
// A RunLoop::Delegate must be bound to this thread prior to using RunLoop. DCHECK(delegate_) << "A RunLoop::Delegate must be bound to this thread prior "
DCHECK(delegate_); "to using RunLoop.";
DCHECK(origin_task_runner_); DCHECK(origin_task_runner_);
} }
...@@ -272,7 +271,7 @@ void RunLoop::AfterRun() { ...@@ -272,7 +271,7 @@ void RunLoop::AfterRun() {
RunLoop* previous_run_loop = RunLoop* previous_run_loop =
active_run_loops_.empty() ? nullptr : active_run_loops_.top(); active_run_loops_.empty() ? nullptr : active_run_loops_.top();
// Execute deferred QuitNow, if any: // Execute deferred Quit, if any:
if (previous_run_loop && previous_run_loop->quit_called_) if (previous_run_loop && previous_run_loop->quit_called_)
delegate_->Quit(); delegate_->Quit();
} }
......
...@@ -134,12 +134,12 @@ class BASE_EXPORT RunLoop { ...@@ -134,12 +134,12 @@ class BASE_EXPORT RunLoop {
// The client interface provided back to the caller who registers this // The client interface provided back to the caller who registers this
// Delegate via RegisterDelegateForCurrentThread. // Delegate via RegisterDelegateForCurrentThread.
class Client { class BASE_EXPORT Client {
public: public:
// Returns the RunLoop with the topmost active Run() call on the stack. // Returns true if the Delegate should return from the topmost Run() when
// TODO(gab): Break the inter-dependency between MessageLoop and RunLoop // it becomes idle. The Delegate is responsible for probing this when it
// further. http://crbug.com/703346 // becomes idle.
RunLoop* GetTopMostRunLoop() const; bool ShouldQuitWhenIdle() const;
// Returns true if this |outer_| is currently in nested runs. This is a // Returns true if this |outer_| is currently in nested runs. This is a
// shortcut for RunLoop::IsNestedOnCurrentThread() for the owner of this // shortcut for RunLoop::IsNestedOnCurrentThread() for the owner of this
...@@ -166,7 +166,9 @@ class BASE_EXPORT RunLoop { ...@@ -166,7 +166,9 @@ class BASE_EXPORT RunLoop {
// eventual matching Quit() call. Upon receiving a Quit() call it should // eventual matching Quit() call. Upon receiving a Quit() call it should
// return from the Run() call as soon as possible without executing // return from the Run() call as soon as possible without executing
// remaining tasks/messages. Run() calls can nest in which case each Quit() // remaining tasks/messages. Run() calls can nest in which case each Quit()
// call should result in the topmost active Run() call returning. // call should result in the topmost active Run() call returning. The
// only other trigger for Run() to return is Client::ShouldQuitWhenIdle()
// which the Delegate should probe before sleeping when it becomes idle.
virtual void Run() = 0; virtual void Run() = 0;
virtual void Quit() = 0; virtual void Quit() = 0;
...@@ -207,17 +209,14 @@ class BASE_EXPORT RunLoop { ...@@ -207,17 +209,14 @@ class BASE_EXPORT RunLoop {
static void QuitCurrentWhenIdleDeprecated(); static void QuitCurrentWhenIdleDeprecated();
private: private:
// TODO(gab): Break the inter-dependency between MessageLoop and RunLoop
// further. http://crbug.com/703346
friend class MessageLoop;
#if defined(OS_ANDROID) #if defined(OS_ANDROID)
// Android doesn't support the blocking MessageLoop::Run, so it calls // Android doesn't support the blocking RunLoop::Run, so it calls
// BeforeRun and AfterRun directly. // BeforeRun and AfterRun directly.
friend class base::MessagePumpForUI; friend class base::MessagePumpForUI;
#endif #endif
#if defined(OS_IOS) #if defined(OS_IOS)
// iOS doesn't support the blocking MessageLoop::Run, so it calls // iOS doesn't support the blocking RunLoop::Run, so it calls
// BeforeRun directly. // BeforeRun directly.
friend class base::MessagePumpUIApplication; friend class base::MessagePumpUIApplication;
#endif #endif
...@@ -237,8 +236,11 @@ class BASE_EXPORT RunLoop { ...@@ -237,8 +236,11 @@ class BASE_EXPORT RunLoop {
bool quit_called_ = false; bool quit_called_ = false;
bool running_ = false; bool running_ = false;
// Used to record that QuitWhenIdle() was called on the MessageLoop, meaning // Used to record that QuitWhenIdle() was called on this RunLoop, meaning that
// that we should quit Run once it becomes idle. // the Delegate should quit Run() once it becomes idle (it's responsible for
// probing this state via Client::ShouldQuitWhenIdle()). This state is stored
// here rather than pushed to Delegate via, e.g., Delegate::QuitWhenIdle() to
// support nested RunLoops.
bool quit_when_idle_received_ = false; bool quit_when_idle_received_ = false;
// RunLoop is not thread-safe. Its state/methods, unless marked as such, may // RunLoop is not thread-safe. Its state/methods, unless marked as such, may
......
...@@ -4,15 +4,25 @@ ...@@ -4,15 +4,25 @@
#include "base/run_loop.h" #include "base/run_loop.h"
#include <queue>
#include <utility>
#include "base/bind.h" #include "base/bind.h"
#include "base/bind_helpers.h" #include "base/bind_helpers.h"
#include "base/location.h" #include "base/location.h"
#include "base/macros.h" #include "base/macros.h"
#include "base/memory/ptr_util.h"
#include "base/memory/ref_counted.h"
#include "base/single_thread_task_runner.h" #include "base/single_thread_task_runner.h"
#include "base/synchronization/lock.h"
#include "base/synchronization/waitable_event.h" #include "base/synchronization/waitable_event.h"
#include "base/task_scheduler/post_task.h" #include "base/test/gtest_util.h"
#include "base/test/scoped_task_environment.h" #include "base/test/scoped_task_environment.h"
#include "base/threading/platform_thread.h"
#include "base/threading/thread.h"
#include "base/threading/thread_checker_impl.h"
#include "base/threading/thread_task_runner_handle.h" #include "base/threading/thread_task_runner_handle.h"
#include "build/build_config.h"
#include "testing/gmock/include/gmock/gmock.h" #include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h" #include "testing/gtest/include/gtest/gtest.h"
...@@ -44,17 +54,148 @@ void RunNestedLoopTask(int* counter) { ...@@ -44,17 +54,148 @@ void RunNestedLoopTask(int* counter) {
ThreadTaskRunnerHandle::Get()->PostDelayedTask( ThreadTaskRunnerHandle::Get()->PostDelayedTask(
FROM_HERE, BindOnce(&ShouldNotRunTask), TimeDelta::FromDays(1)); FROM_HERE, BindOnce(&ShouldNotRunTask), TimeDelta::FromDays(1));
MessageLoop::ScopedNestableTaskAllower allower(MessageLoop::current()); std::unique_ptr<MessageLoop::ScopedNestableTaskAllower> allower;
if (MessageLoop::current()) {
// Need to allow nestable tasks in MessageLoop driven environments.
// TODO(gab): Move nestable task allowance concept to RunLoop.
allower = base::MakeUnique<MessageLoop::ScopedNestableTaskAllower>(
MessageLoop::current());
}
nested_run_loop.Run(); nested_run_loop.Run();
++(*counter); ++(*counter);
} }
class RunLoopTest : public testing::Test { // A simple SingleThreadTaskRunner that just queues undelayed tasks (and ignores
// delayed tasks). Tasks can then be processed one by one by ProcessTask() which
// will return true if it processed a task and false otherwise.
class SimpleSingleThreadTaskRunner : public SingleThreadTaskRunner {
public:
SimpleSingleThreadTaskRunner() = default;
bool PostDelayedTask(const tracked_objects::Location& from_here,
OnceClosure task,
base::TimeDelta delay) override {
if (delay > base::TimeDelta())
return false;
AutoLock auto_lock(tasks_lock_);
pending_tasks_.push(std::move(task));
return true;
}
bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here,
OnceClosure task,
base::TimeDelta delay) override {
return PostDelayedTask(from_here, std::move(task), delay);
}
bool RunsTasksInCurrentSequence() const override {
return origin_thread_checker_.CalledOnValidThread();
}
bool ProcessTask() {
OnceClosure task;
{
AutoLock auto_lock(tasks_lock_);
if (pending_tasks_.empty())
return false;
task = std::move(pending_tasks_.front());
pending_tasks_.pop();
}
// It's important to Run() after pop() and outside the lock as |task| may
// run a nested loop which will re-enter ProcessTask().
std::move(task).Run();
return true;
}
private:
~SimpleSingleThreadTaskRunner() override = default;
Lock tasks_lock_;
std::queue<OnceClosure> pending_tasks_;
// RunLoop relies on RunsTasksInCurrentSequence() signal. Use a
// ThreadCheckerImpl to be able to reliably provide that signal even in
// non-dcheck builds.
ThreadCheckerImpl origin_thread_checker_;
DISALLOW_COPY_AND_ASSIGN(SimpleSingleThreadTaskRunner);
};
// A simple test RunLoop::Delegate to exercise Runloop logic independent of any
// other base constructs.
class TestDelegate : public RunLoop::Delegate {
public:
TestDelegate() = default;
void BindToCurrentThread() {
thread_task_runner_handle_ =
MakeUnique<ThreadTaskRunnerHandle>(simple_task_runner_);
run_loop_client_ = RunLoop::RegisterDelegateForCurrentThread(this);
}
private:
void Run() override {
while (!should_quit_) {
if (simple_task_runner_->ProcessTask())
continue;
if (run_loop_client_->ShouldQuitWhenIdle())
break;
PlatformThread::YieldCurrentThread();
}
should_quit_ = false;
}
void Quit() override { should_quit_ = true; }
scoped_refptr<SimpleSingleThreadTaskRunner> simple_task_runner_ =
MakeRefCounted<SimpleSingleThreadTaskRunner>();
std::unique_ptr<ThreadTaskRunnerHandle> thread_task_runner_handle_;
bool should_quit_ = false;
RunLoop::Delegate::Client* run_loop_client_ = nullptr;
};
enum class RunLoopTestType {
// Runs all RunLoopTests under a ScopedTaskEnvironment to make sure real world
// scenarios work.
kRealEnvironment,
// Runs all RunLoopTests under a test RunLoop::Delegate to make sure the
// delegate interface fully works standalone.
kTestDelegate,
};
// The task environment for the RunLoopTest of a given type. A separate class
// so it can be instantiated on the stack in the RunLoopTest fixture.
class RunLoopTestEnvironment {
public:
RunLoopTestEnvironment(RunLoopTestType type) {
switch (type) {
case RunLoopTestType::kRealEnvironment:
task_environment_ = base::MakeUnique<test::ScopedTaskEnvironment>();
break;
case RunLoopTestType::kTestDelegate:
test_delegate_ = base::MakeUnique<TestDelegate>();
test_delegate_->BindToCurrentThread();
break;
}
}
private:
// Instantiates one or the other based on the RunLoopTestType.
std::unique_ptr<test::ScopedTaskEnvironment> task_environment_;
std::unique_ptr<TestDelegate> test_delegate_;
};
class RunLoopTest : public testing::TestWithParam<RunLoopTestType> {
protected: protected:
RunLoopTest() = default; RunLoopTest() : test_environment_(GetParam()) {}
test::ScopedTaskEnvironment task_environment_; RunLoopTestEnvironment test_environment_;
RunLoop run_loop_; RunLoop run_loop_;
int counter_ = 0; int counter_ = 0;
...@@ -64,7 +205,7 @@ class RunLoopTest : public testing::Test { ...@@ -64,7 +205,7 @@ class RunLoopTest : public testing::Test {
} // namespace } // namespace
TEST_F(RunLoopTest, QuitWhenIdle) { TEST_P(RunLoopTest, QuitWhenIdle) {
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
FROM_HERE, BindOnce(&QuitWhenIdleTask, Unretained(&run_loop_), FROM_HERE, BindOnce(&QuitWhenIdleTask, Unretained(&run_loop_),
Unretained(&counter_))); Unretained(&counter_)));
...@@ -77,7 +218,7 @@ TEST_F(RunLoopTest, QuitWhenIdle) { ...@@ -77,7 +218,7 @@ TEST_F(RunLoopTest, QuitWhenIdle) {
EXPECT_EQ(2, counter_); EXPECT_EQ(2, counter_);
} }
TEST_F(RunLoopTest, QuitWhenIdleNestedLoop) { TEST_P(RunLoopTest, QuitWhenIdleNestedLoop) {
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
FROM_HERE, BindOnce(&RunNestedLoopTask, Unretained(&counter_))); FROM_HERE, BindOnce(&RunNestedLoopTask, Unretained(&counter_)));
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
...@@ -92,7 +233,7 @@ TEST_F(RunLoopTest, QuitWhenIdleNestedLoop) { ...@@ -92,7 +233,7 @@ TEST_F(RunLoopTest, QuitWhenIdleNestedLoop) {
EXPECT_EQ(4, counter_); EXPECT_EQ(4, counter_);
} }
TEST_F(RunLoopTest, QuitWhenIdleClosure) { TEST_P(RunLoopTest, QuitWhenIdleClosure) {
ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
run_loop_.QuitWhenIdleClosure()); run_loop_.QuitWhenIdleClosure());
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
...@@ -106,7 +247,7 @@ TEST_F(RunLoopTest, QuitWhenIdleClosure) { ...@@ -106,7 +247,7 @@ TEST_F(RunLoopTest, QuitWhenIdleClosure) {
// Verify that the QuitWhenIdleClosure() can run after the RunLoop has been // Verify that the QuitWhenIdleClosure() can run after the RunLoop has been
// deleted. It should have no effect. // deleted. It should have no effect.
TEST_F(RunLoopTest, QuitWhenIdleClosureAfterRunLoopScope) { TEST_P(RunLoopTest, QuitWhenIdleClosureAfterRunLoopScope) {
Closure quit_when_idle_closure; Closure quit_when_idle_closure;
{ {
RunLoop run_loop; RunLoop run_loop;
...@@ -117,9 +258,11 @@ TEST_F(RunLoopTest, QuitWhenIdleClosureAfterRunLoopScope) { ...@@ -117,9 +258,11 @@ TEST_F(RunLoopTest, QuitWhenIdleClosureAfterRunLoopScope) {
} }
// Verify that Quit can be executed from another sequence. // Verify that Quit can be executed from another sequence.
TEST_F(RunLoopTest, QuitFromOtherSequence) { TEST_P(RunLoopTest, QuitFromOtherSequence) {
Thread other_thread("test");
other_thread.Start();
scoped_refptr<SequencedTaskRunner> other_sequence = scoped_refptr<SequencedTaskRunner> other_sequence =
CreateSequencedTaskRunnerWithTraits({}); other_thread.task_runner();
// Always expected to run before asynchronous Quit() kicks in. // Always expected to run before asynchronous Quit() kicks in.
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
...@@ -137,8 +280,8 @@ TEST_F(RunLoopTest, QuitFromOtherSequence) { ...@@ -137,8 +280,8 @@ TEST_F(RunLoopTest, QuitFromOtherSequence) {
// Anything that's posted after the Quit closure was posted back to this // Anything that's posted after the Quit closure was posted back to this
// sequence shouldn't get a chance to run. // sequence shouldn't get a chance to run.
loop_was_quit.Wait(); loop_was_quit.Wait();
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
FROM_HERE, base::BindOnce(&ShouldRunTask, Unretained(&counter_))); base::BindOnce(&ShouldNotRunTask));
run_loop_.Run(); run_loop_.Run();
...@@ -146,9 +289,11 @@ TEST_F(RunLoopTest, QuitFromOtherSequence) { ...@@ -146,9 +289,11 @@ TEST_F(RunLoopTest, QuitFromOtherSequence) {
} }
// Verify that QuitClosure can be executed from another sequence. // Verify that QuitClosure can be executed from another sequence.
TEST_F(RunLoopTest, QuitFromOtherSequenceWithClosure) { TEST_P(RunLoopTest, QuitFromOtherSequenceWithClosure) {
Thread other_thread("test");
other_thread.Start();
scoped_refptr<SequencedTaskRunner> other_sequence = scoped_refptr<SequencedTaskRunner> other_sequence =
CreateSequencedTaskRunnerWithTraits({}); other_thread.task_runner();
// Always expected to run before asynchronous Quit() kicks in. // Always expected to run before asynchronous Quit() kicks in.
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
...@@ -164,8 +309,8 @@ TEST_F(RunLoopTest, QuitFromOtherSequenceWithClosure) { ...@@ -164,8 +309,8 @@ TEST_F(RunLoopTest, QuitFromOtherSequenceWithClosure) {
// Anything that's posted after the Quit closure was posted back to this // Anything that's posted after the Quit closure was posted back to this
// sequence shouldn't get a chance to run. // sequence shouldn't get a chance to run.
loop_was_quit.Wait(); loop_was_quit.Wait();
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
FROM_HERE, base::BindOnce(&ShouldRunTask, Unretained(&counter_))); base::BindOnce(&ShouldNotRunTask));
run_loop_.Run(); run_loop_.Run();
...@@ -174,9 +319,11 @@ TEST_F(RunLoopTest, QuitFromOtherSequenceWithClosure) { ...@@ -174,9 +319,11 @@ TEST_F(RunLoopTest, QuitFromOtherSequenceWithClosure) {
// Verify that Quit can be executed from another sequence even when the // Verify that Quit can be executed from another sequence even when the
// Quit is racing with Run() -- i.e. forgo the WaitableEvent used above. // Quit is racing with Run() -- i.e. forgo the WaitableEvent used above.
TEST_F(RunLoopTest, QuitFromOtherSequenceRacy) { TEST_P(RunLoopTest, QuitFromOtherSequenceRacy) {
Thread other_thread("test");
other_thread.Start();
scoped_refptr<SequencedTaskRunner> other_sequence = scoped_refptr<SequencedTaskRunner> other_sequence =
CreateSequencedTaskRunnerWithTraits({}); other_thread.task_runner();
// Always expected to run before asynchronous Quit() kicks in. // Always expected to run before asynchronous Quit() kicks in.
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
...@@ -193,9 +340,11 @@ TEST_F(RunLoopTest, QuitFromOtherSequenceRacy) { ...@@ -193,9 +340,11 @@ TEST_F(RunLoopTest, QuitFromOtherSequenceRacy) {
// Verify that QuitClosure can be executed from another sequence even when the // Verify that QuitClosure can be executed from another sequence even when the
// Quit is racing with Run() -- i.e. forgo the WaitableEvent used above. // Quit is racing with Run() -- i.e. forgo the WaitableEvent used above.
TEST_F(RunLoopTest, QuitFromOtherSequenceRacyWithClosure) { TEST_P(RunLoopTest, QuitFromOtherSequenceRacyWithClosure) {
Thread other_thread("test");
other_thread.Start();
scoped_refptr<SequencedTaskRunner> other_sequence = scoped_refptr<SequencedTaskRunner> other_sequence =
CreateSequencedTaskRunnerWithTraits({}); other_thread.task_runner();
// Always expected to run before asynchronous Quit() kicks in. // Always expected to run before asynchronous Quit() kicks in.
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
...@@ -209,9 +358,11 @@ TEST_F(RunLoopTest, QuitFromOtherSequenceRacyWithClosure) { ...@@ -209,9 +358,11 @@ TEST_F(RunLoopTest, QuitFromOtherSequenceRacyWithClosure) {
} }
// Verify that QuitWhenIdle can be executed from another sequence. // Verify that QuitWhenIdle can be executed from another sequence.
TEST_F(RunLoopTest, QuitWhenIdleFromOtherSequence) { TEST_P(RunLoopTest, QuitWhenIdleFromOtherSequence) {
Thread other_thread("test");
other_thread.Start();
scoped_refptr<SequencedTaskRunner> other_sequence = scoped_refptr<SequencedTaskRunner> other_sequence =
CreateSequencedTaskRunnerWithTraits({}); other_thread.task_runner();
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
FROM_HERE, base::BindOnce(&ShouldRunTask, Unretained(&counter_))); FROM_HERE, base::BindOnce(&ShouldRunTask, Unretained(&counter_)));
...@@ -232,9 +383,11 @@ TEST_F(RunLoopTest, QuitWhenIdleFromOtherSequence) { ...@@ -232,9 +383,11 @@ TEST_F(RunLoopTest, QuitWhenIdleFromOtherSequence) {
} }
// Verify that QuitWhenIdleClosure can be executed from another sequence. // Verify that QuitWhenIdleClosure can be executed from another sequence.
TEST_F(RunLoopTest, QuitWhenIdleFromOtherSequenceWithClosure) { TEST_P(RunLoopTest, QuitWhenIdleFromOtherSequenceWithClosure) {
Thread other_thread("test");
other_thread.Start();
scoped_refptr<SequencedTaskRunner> other_sequence = scoped_refptr<SequencedTaskRunner> other_sequence =
CreateSequencedTaskRunnerWithTraits({}); other_thread.task_runner();
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
FROM_HERE, base::BindOnce(&ShouldRunTask, Unretained(&counter_))); FROM_HERE, base::BindOnce(&ShouldRunTask, Unretained(&counter_)));
...@@ -251,7 +404,7 @@ TEST_F(RunLoopTest, QuitWhenIdleFromOtherSequenceWithClosure) { ...@@ -251,7 +404,7 @@ TEST_F(RunLoopTest, QuitWhenIdleFromOtherSequenceWithClosure) {
EXPECT_EQ(2, counter_); EXPECT_EQ(2, counter_);
} }
TEST_F(RunLoopTest, IsRunningOnCurrentThread) { TEST_P(RunLoopTest, IsRunningOnCurrentThread) {
EXPECT_FALSE(RunLoop::IsRunningOnCurrentThread()); EXPECT_FALSE(RunLoop::IsRunningOnCurrentThread());
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
FROM_HERE, FROM_HERE,
...@@ -260,7 +413,7 @@ TEST_F(RunLoopTest, IsRunningOnCurrentThread) { ...@@ -260,7 +413,7 @@ TEST_F(RunLoopTest, IsRunningOnCurrentThread) {
run_loop_.Run(); run_loop_.Run();
} }
TEST_F(RunLoopTest, IsNestedOnCurrentThread) { TEST_P(RunLoopTest, IsNestedOnCurrentThread) {
EXPECT_FALSE(RunLoop::IsNestedOnCurrentThread()); EXPECT_FALSE(RunLoop::IsNestedOnCurrentThread());
ThreadTaskRunnerHandle::Get()->PostTask( ThreadTaskRunnerHandle::Get()->PostTask(
...@@ -277,7 +430,13 @@ TEST_F(RunLoopTest, IsNestedOnCurrentThread) { ...@@ -277,7 +430,13 @@ TEST_F(RunLoopTest, IsNestedOnCurrentThread) {
nested_run_loop.QuitClosure()); nested_run_loop.QuitClosure());
EXPECT_FALSE(RunLoop::IsNestedOnCurrentThread()); EXPECT_FALSE(RunLoop::IsNestedOnCurrentThread());
MessageLoop::ScopedNestableTaskAllower allower(MessageLoop::current()); std::unique_ptr<MessageLoop::ScopedNestableTaskAllower> allower;
if (MessageLoop::current()) {
// Need to allow nestable tasks in MessageLoop driven environments.
// TODO(gab): Move nestable task allowance concept to RunLoop.
allower = base::MakeUnique<MessageLoop::ScopedNestableTaskAllower>(
MessageLoop::current());
}
nested_run_loop.Run(); nested_run_loop.Run();
EXPECT_FALSE(RunLoop::IsNestedOnCurrentThread()); EXPECT_FALSE(RunLoop::IsNestedOnCurrentThread());
})); }));
...@@ -297,7 +456,7 @@ class MockNestingObserver : public RunLoop::NestingObserver { ...@@ -297,7 +456,7 @@ class MockNestingObserver : public RunLoop::NestingObserver {
DISALLOW_COPY_AND_ASSIGN(MockNestingObserver); DISALLOW_COPY_AND_ASSIGN(MockNestingObserver);
}; };
TEST_F(RunLoopTest, NestingObservers) { TEST_P(RunLoopTest, NestingObservers) {
EXPECT_TRUE(RunLoop::IsNestingAllowedOnCurrentThread()); EXPECT_TRUE(RunLoop::IsNestingAllowedOnCurrentThread());
testing::StrictMock<MockNestingObserver> nesting_observer; testing::StrictMock<MockNestingObserver> nesting_observer;
...@@ -312,7 +471,13 @@ TEST_F(RunLoopTest, NestingObservers) { ...@@ -312,7 +471,13 @@ TEST_F(RunLoopTest, NestingObservers) {
})); }));
ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
nested_run_loop.QuitClosure()); nested_run_loop.QuitClosure());
MessageLoop::ScopedNestableTaskAllower allower(MessageLoop::current()); std::unique_ptr<MessageLoop::ScopedNestableTaskAllower> allower;
if (MessageLoop::current()) {
// Need to allow nestable tasks in MessageLoop driven environments.
// TODO(gab): Move nestable task allowance concept to RunLoop.
allower = base::MakeUnique<MessageLoop::ScopedNestableTaskAllower>(
MessageLoop::current());
}
nested_run_loop.Run(); nested_run_loop.Run();
}); });
...@@ -330,7 +495,7 @@ TEST_F(RunLoopTest, NestingObservers) { ...@@ -330,7 +495,7 @@ TEST_F(RunLoopTest, NestingObservers) {
// Disabled on Android per http://crbug.com/643760. // Disabled on Android per http://crbug.com/643760.
#if defined(GTEST_HAS_DEATH_TEST) && !defined(OS_ANDROID) #if defined(GTEST_HAS_DEATH_TEST) && !defined(OS_ANDROID)
TEST_F(RunLoopTest, DisallowWaitingDeathTest) { TEST_P(RunLoopTest, DisallowNestingDeathTest) {
EXPECT_TRUE(RunLoop::IsNestingAllowedOnCurrentThread()); EXPECT_TRUE(RunLoop::IsNestingAllowedOnCurrentThread());
RunLoop::DisallowNestingOnCurrentThread(); RunLoop::DisallowNestingOnCurrentThread();
EXPECT_FALSE(RunLoop::IsNestingAllowedOnCurrentThread()); EXPECT_FALSE(RunLoop::IsNestingAllowedOnCurrentThread());
...@@ -343,4 +508,17 @@ TEST_F(RunLoopTest, DisallowWaitingDeathTest) { ...@@ -343,4 +508,17 @@ TEST_F(RunLoopTest, DisallowWaitingDeathTest) {
} }
#endif // defined(GTEST_HAS_DEATH_TEST) && !defined(OS_ANDROID) #endif // defined(GTEST_HAS_DEATH_TEST) && !defined(OS_ANDROID)
INSTANTIATE_TEST_CASE_P(Real,
RunLoopTest,
testing::Values(RunLoopTestType::kRealEnvironment));
INSTANTIATE_TEST_CASE_P(Mock,
RunLoopTest,
testing::Values(RunLoopTestType::kTestDelegate));
TEST(RunLoopDeathTest, MustRegisterBeforeInstantiating) {
TestDelegate unbound_test_delegate_;
// Exercise the DCHECK in RunLoop::RunLoop().
EXPECT_DCHECK_DEATH({ RunLoop(); });
}
} // namespace base } // namespace base
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment