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..a26d738f796212d8ee31e40661d88771c60730aa 100644 |
--- a/cc/test/layer_tree_test.h |
+++ b/cc/test/layer_tree_test.h |
@@ -253,9 +253,10 @@ 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()); |
+ return task_runner_provider()->ImplThreadTaskRunner() |
vmpstr
2015/11/10 21:59:34
save it into a temporary impl_thread_task_runner,
Khushal
2015/11/10 23:34:03
Done.
|
+ ? task_runner_provider()->ImplThreadTaskRunner() |
+ : main_task_runner_.get(); |
} |
base::SingleThreadTaskRunner* MainThreadTaskRunner() { |
return main_task_runner_.get(); |
@@ -263,6 +264,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(); |
} |