Fixing flakiness of TextureLayerChangeInvisibleMailboxTest

cc/trees/proxy_main.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "cc/trees/proxy_main.h"
 
 #include <algorithm>
 #include <string>
 
 #include "base/memory/ptr_util.h"
@@ skipping 18 matching lines @@
                      TaskRunnerProvider* task_runner_provider)
     : layer_tree_host_(layer_tree_host),
       task_runner_provider_(task_runner_provider),
       layer_tree_host_id_(layer_tree_host->GetId()),
       max_requested_pipeline_stage_(NO_PIPELINE_STAGE),
       current_pipeline_stage_(NO_PIPELINE_STAGE),
       final_pipeline_stage_(NO_PIPELINE_STAGE),
       commit_waits_for_activation_(false),
       started_(false),
       defer_commits_(false),
+      compositor_frame_sink_weak_factory_(this),
       weak_factory_(this) {
   TRACE_EVENT0("cc", "ProxyMain::ProxyMain");
   DCHECK(task_runner_provider_);
   DCHECK(IsMainThread());
 }
 
 ProxyMain::~ProxyMain() {
   TRACE_EVENT0("cc", "ProxyMain::~ProxyMain");
   DCHECK(IsMainThread());
   DCHECK(!started_);
@@ skipping 32 matching lines @@
 
 void ProxyMain::SetAnimationEvents(std::unique_ptr<MutatorEvents> events) {
   TRACE_EVENT0("cc", "ProxyMain::SetAnimationEvents");
   DCHECK(IsMainThread());
   layer_tree_host_->SetAnimationEvents(std::move(events));
 }
 
 void ProxyMain::DidLoseCompositorFrameSink() {
   TRACE_EVENT0("cc", "ProxyMain::DidLoseCompositorFrameSink");
   DCHECK(IsMainThread());
+  compositor_frame_sink_weak_factory_.InvalidateWeakPtrs();
   layer_tree_host_->DidLoseCompositorFrameSink();
 }
 
 void ProxyMain::RequestNewCompositorFrameSink() {
   TRACE_EVENT0("cc", "ProxyMain::RequestNewCompositorFrameSink");
   DCHECK(IsMainThread());
   layer_tree_host_->RequestNewCompositorFrameSink();
 }
 
 void ProxyMain::DidInitializeCompositorFrameSink(bool success) {
@@ skipping 181 matching lines @@
   // With ProxyMain, we use a pending tree and activate it once it's ready to
   // draw to allow input to modify the active tree and draw during raster.
   return false;
 }
 
 void ProxyMain::SetCompositorFrameSink(
     CompositorFrameSink* compositor_frame_sink) {
   ImplThreadTaskRunner()->PostTask(
       FROM_HERE,
       base::Bind(&ProxyImpl::InitializeCompositorFrameSinkOnImpl,
-                 base::Unretained(proxy_impl_.get()), compositor_frame_sink));
+                 base::Unretained(proxy_impl_.get()), compositor_frame_sink,
+                 compositor_frame_sink_weak_factory_.GetWeakPtr()));
 }
 
 void ProxyMain::SetVisible(bool visible) {
   TRACE_EVENT1("cc", "ProxyMain::SetVisible", "visible", visible);
   ImplThreadTaskRunner()->PostTask(
       FROM_HERE, base::Bind(&ProxyImpl::SetVisibleOnImpl,
                             base::Unretained(proxy_impl_.get()), visible));
 }
 
 void ProxyMain::SetNeedsAnimate() {
@@ skipping 156 matching lines @@
         FROM_HERE, base::Bind(&ProxyImpl::MainFrameWillHappenOnImplForTesting,
                               base::Unretained(proxy_impl_.get()), &completion,
                               &main_frame_will_happen));
     completion.Wait();
   }
   return main_frame_will_happen;
 }
 
 void ProxyMain::ReleaseCompositorFrameSink() {
   DCHECK(IsMainThread());
+  compositor_frame_sink_weak_factory_.InvalidateWeakPtrs();
   DebugScopedSetMainThreadBlocked main_thread_blocked(task_runner_provider_);
   CompletionEvent completion;
   ImplThreadTaskRunner()->PostTask(
       FROM_HERE, base::Bind(&ProxyImpl::ReleaseCompositorFrameSinkOnImpl,
                             base::Unretained(proxy_impl_.get()), &completion));
   completion.Wait();
 }
 
 void ProxyMain::UpdateBrowserControlsState(BrowserControlsState constraints,
                                            BrowserControlsState current,
@@ skipping 26 matching lines @@
 
 bool ProxyMain::IsImplThread() const {
   return task_runner_provider_->IsImplThread();
 }
 
 base::SingleThreadTaskRunner* ProxyMain::ImplThreadTaskRunner() {
   return task_runner_provider_->ImplThreadTaskRunner();
 }
 
 }  // namespace cc