Index: cc/test/fake_layer_tree_host_impl.cc |
diff --git a/cc/test/fake_layer_tree_host_impl.cc b/cc/test/fake_layer_tree_host_impl.cc |
index d026120ab18c2afe658b2a9de77e94c51a219b3d..81216a0aa043c0373c79cce557fd6d2e1ae98cc7 100644 |
--- a/cc/test/fake_layer_tree_host_impl.cc |
+++ b/cc/test/fake_layer_tree_host_impl.cc |
@@ -61,14 +61,15 @@ void FakeLayerTreeHostImpl::CreatePendingTree() { |
} |
BeginFrameArgs FakeLayerTreeHostImpl::CurrentBeginFrameArgs() const { |
- if (!current_begin_frame_args_.IsValid()) |
- return LayerTreeHostImpl::CurrentBeginFrameArgs(); |
- return current_begin_frame_args_; |
+ return current_begin_frame_tracker_.DangerousMethodCurrentOrLast(); |
} |
void FakeLayerTreeHostImpl::SetCurrentBeginFrameArgs( |
const BeginFrameArgs& args) { |
- current_begin_frame_args_ = args; |
+ if (!current_begin_frame_tracker_.DangerousMethodHasFinished()) { |
+ current_begin_frame_tracker_.Finish(); |
+ } |
+ current_begin_frame_tracker_.Start(args); |
} |
int FakeLayerTreeHostImpl::RecursiveUpdateNumChildren(LayerImpl* layer) { |