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(); |
} |