1 files changed, 14 insertions, 0 deletions
diff --git a/cc/test/layer_tree_test.cc b/cc/test/layer_tree_test.cc
index d70d797..de0b907 100644
--- a/cc/test/layer_tree_test.cc
+++ b/cc/test/layer_tree_test.cc
@@ -409,6 +409,13 @@ void LayerTreeTest::PostSetNeedsCommitToMainThread() {
                                          main_thread_weak_ptr_));
 }
 
+void LayerTreeTest::PostSetNeedsUpdateLayersToMainThread() {
+  main_task_runner_->PostTask(
+      FROM_HERE,
+      base::Bind(&LayerTreeTest::DispatchSetNeedsUpdateLayers,
+                 main_thread_weak_ptr_));
+}
+
 void LayerTreeTest::PostReadbackToMainThread() {
   main_task_runner_->PostTask(
       FROM_HERE,
@@ -538,6 +545,13 @@ void LayerTreeTest::DispatchSetNeedsCommit() {
     layer_tree_host_->SetNeedsCommit();
 }
 
+void LayerTreeTest::DispatchSetNeedsUpdateLayers() {
+  DCHECK(!proxy() || proxy()->IsMainThread());
+
+  if (layer_tree_host_)
+    layer_tree_host_->SetNeedsUpdateLayers();
+}
+
 void LayerTreeTest::DispatchReadback() {
   DCHECK(!proxy() || proxy()->IsMainThread());