cc: HUD handle context lost in Gpu raster.

cc/trees/layer_tree_host_unittest_context.cc

 // Copyright 2012 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 <stddef.h>
 #include <stdint.h>
 
 #include "base/memory/ptr_util.h"
 #include "cc/base/filter_operations.h"
 #include "cc/layers/heads_up_display_layer.h"
@@ skipping 985 matching lines @@
       // video frames should not be reused by the VideoFrameProvider, but
       // software frames can be.
       hw_frame_provider_.set_frame(NULL);
       scaled_hw_frame_provider_.set_frame(NULL);
     }
   }
 
   DrawResult PrepareToDrawOnThread(LayerTreeHostImpl* host_impl,
                                    LayerTreeHostImpl::FrameData* frame,
                                    DrawResult draw_result) override {
+    if (host_impl->active_tree()->source_frame_number() == 2) {
+      // Lose the context after draw on the second commit. This will cause
+      // a third commit to recover.
+      context3d_->set_times_bind_texture_succeeds(0);
+    }
     return draw_result;
   }
 
   void RequestNewLayerTreeFrameSink() override {
     // This will get called twice:
     // First when we create the initial LayerTreeFrameSink...
     if (layer_tree_host()->SourceFrameNumber() > 0) {
       // ... and then again after we forced the context to be lost.
       lost_context_ = true;
     }
     LayerTreeHostContextTest::RequestNewLayerTreeFrameSink();
   }
 
   void DidCommitAndDrawFrame() override {
     ASSERT_TRUE(layer_tree_host()->hud_layer());
-
-    if (layer_tree_host()->SourceFrameNumber() == 2) {
-      // Lose the context after draw on the second commit. This will cause
-      // a third commit to recover.
-      context3d_->set_times_bind_texture_succeeds(0);
-    }
-
     // End the test once we know the 3nd frame drew.
     if (layer_tree_host()->SourceFrameNumber() < 5) {
       layer_tree_host()->root_layer()->SetNeedsDisplay();
       layer_tree_host()->SetNeedsCommit();
     } else {
       EndTest();
     }
   }
 
   void AfterTest() override { EXPECT_TRUE(lost_context_); }
@@ skipping 606 matching lines @@
 
   FakeContentLayerClient client_;
   int num_draws_ = 0;
 };
 
 MULTI_THREAD_TEST_F(
     LayerTreeHostContextTestLoseWorkerContextDuringPrepareTiles);
 
 }  // namespace
 }  // namespace cc