| Index: cc/test/layer_tree_test.h
|
| diff --git a/cc/test/layer_tree_test.h b/cc/test/layer_tree_test.h
|
| index 33593aae97bdcb855cac031f381eaf2cfbb43a93..0c2dfb103a4485bf0765a5fac70019956811dc8d 100644
|
| --- a/cc/test/layer_tree_test.h
|
| +++ b/cc/test/layer_tree_test.h
|
| @@ -253,9 +253,11 @@ class LayerTreeTest : public testing::Test, public TestHooks {
|
|
|
| bool HasImplThread() { return !!impl_thread_; }
|
| base::SingleThreadTaskRunner* ImplThreadTaskRunner() {
|
| - DCHECK(proxy());
|
| - return proxy()->ImplThreadTaskRunner() ? proxy()->ImplThreadTaskRunner()
|
| - : main_task_runner_.get();
|
| + DCHECK(task_runner_provider());
|
| + base::SingleThreadTaskRunner* impl_thread_task_runner =
|
| + task_runner_provider()->ImplThreadTaskRunner();
|
| + return impl_thread_task_runner ? impl_thread_task_runner
|
| + : main_task_runner_.get();
|
| }
|
| base::SingleThreadTaskRunner* MainThreadTaskRunner() {
|
| return main_task_runner_.get();
|
| @@ -263,6 +265,10 @@ class LayerTreeTest : public testing::Test, public TestHooks {
|
| Proxy* proxy() const {
|
| return layer_tree_host_ ? layer_tree_host_->proxy() : NULL;
|
| }
|
| + TaskRunnerProvider* task_runner_provider() const {
|
| + return layer_tree_host_ ? layer_tree_host_->task_runner_provider()
|
| + : nullptr;
|
| + }
|
| TaskGraphRunner* task_graph_runner() const {
|
| return task_graph_runner_.get();
|
| }
|
|
|