Fix scrollbars missing after lost context

cc/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 "cc/layer_tree_host.h"
 
 #include "base/basictypes.h"
 #include "cc/content_layer.h"
 #include "cc/delegated_renderer_layer.h"
 #include "cc/delegated_renderer_layer_impl.h"
 #include "cc/heads_up_display_layer.h"
 #include "cc/io_surface_layer.h"
 #include "cc/layer_impl.h"
 #include "cc/layer_tree_host_impl.h"
 #include "cc/layer_tree_impl.h"
 #include "cc/picture_layer.h"
 #include "cc/scrollbar_layer.h"
 #include "cc/single_thread_proxy.h"
 #include "cc/test/fake_content_layer.h"
 #include "cc/test/fake_content_layer_client.h"
 #include "cc/test/fake_content_layer_impl.h"
 #include "cc/test/fake_output_surface.h"
+#include "cc/test/fake_scrollbar_layer.h"
 #include "cc/test/fake_scrollbar_theme_painter.h"
 #include "cc/test/fake_video_frame_provider.h"
 #include "cc/test/fake_web_graphics_context_3d.h"
 #include "cc/test/fake_web_scrollbar.h"
 #include "cc/test/fake_web_scrollbar_theme_geometry.h"
 #include "cc/test/layer_tree_test_common.h"
 #include "cc/test/render_pass_test_common.h"
 #include "cc/texture_layer.h"
 #include "cc/video_layer.h"
 #include "cc/video_layer_impl.h"
+#include "gpu/GLES2/gl2extchromium.h"
 #include "media/base/media.h"
 
 using media::VideoFrame;
 using WebKit::WebGraphicsContext3D;
 
 namespace cc {
 namespace {
 
 // These tests deal with losing the 3d graphics context.
 class LayerTreeHostContextTest : public ThreadedTest {
  public:
   LayerTreeHostContextTest()
       : ThreadedTest(),
         context3d_(NULL),
         times_to_fail_create_(0),
         times_to_create_and_lose_(0),
         times_to_lose_during_commit_(0),
         times_to_repeat_loss_(0),
         times_to_fail_recreate_(0) {
     media::InitializeMediaLibraryForTesting();
   }
 
   void LoseContext() {
-    context3d_->loseContextCHROMIUM();
+    context3d_->loseContextCHROMIUM(GL_GUILTY_CONTEXT_RESET_ARB,
+                                    GL_INNOCENT_CONTEXT_RESET_ARB);
     context3d_ = NULL;
   }
 
   virtual scoped_ptr<FakeWebGraphicsContext3D> CreateContext3d() {
     return FakeWebGraphicsContext3D::Create();
   }
 
   virtual scoped_ptr<OutputSurface> createOutputSurface() OVERRIDE {
     if (times_to_fail_create_) {
       --times_to_fail_create_;
@@ skipping 729 matching lines @@
   virtual void beginTest() OVERRIDE {
     postSetNeedsCommitToMainThread();
   }
 
   virtual void afterTest() OVERRIDE {
   }
 
   virtual scoped_ptr<FakeWebGraphicsContext3D> CreateContext3d() OVERRIDE {
     scoped_ptr<FakeWebGraphicsContext3D> context =
         LayerTreeHostContextTest::CreateContext3d();
-    context->loseContextCHROMIUM();
+    context->loseContextCHROMIUM(GL_GUILTY_CONTEXT_RESET_ARB,
+                                 GL_INNOCENT_CONTEXT_RESET_ARB);
     return context.Pass();
   }
 
   virtual void didRecreateOutputSurface(bool succeeded) OVERRIDE {
     EXPECT_FALSE(succeeded);
     // If we make it this far without crashing, we pass!
     endTest();
   }
 };
 
@@ skipping 37 matching lines @@
     EXPECT_TRUE(succeeded);
     endTest();
   }
 
  private:
   FakeContentLayerClient client_;
 };
 
 MULTI_THREAD_TEST_F(LayerTreeHostContextTestImplSidePainting)
 
+class ScrollbarLayerLostContext : public LayerTreeHostContextTest {
+ public:
+  ScrollbarLayerLostContext() : commits_(0) {}
+
+  virtual void beginTest() OVERRIDE {
+    scoped_refptr<Layer> scroll_layer = Layer::create();
+    scrollbar_layer_ = FakeScrollbarLayer::Create(
+        false, true, scroll_layer->id());
+    scrollbar_layer_->setBounds(gfx::Size(10, 100));
+    m_layerTreeHost->rootLayer()->addChild(scrollbar_layer_);
+    m_layerTreeHost->rootLayer()->addChild(scroll_layer);
+    postSetNeedsCommitToMainThread();
+  }
+
+  virtual void afterTest() OVERRIDE {
+  }
+
+  virtual void commitCompleteOnThread(LayerTreeHostImpl* impl) OVERRIDE {
+    ++commits_;
+    size_t upload_count = scrollbar_layer_->last_update_full_upload_size() +
+        scrollbar_layer_->last_update_partial_upload_size();
+    switch(commits_) {
+      case 1:
+        // First (regular) update, we should upload 2 resources (thumb, and
+        // backtrack).
+        EXPECT_EQ(1, scrollbar_layer_->update_count());
+        EXPECT_EQ(2, upload_count);
+        LoseContext();
+        break;
+      case 2:
+        // Second update, after the lost context, we should still upload 2
+        // resources even if the contents haven't changed.
+        EXPECT_EQ(2, scrollbar_layer_->update_count());
+        EXPECT_EQ(2, upload_count);
+        endTest();
+        break;
+      default:
+        NOTREACHED();
+    }
+  }
+
+ private:
+  int commits_;
+  scoped_refptr<FakeScrollbarLayer> scrollbar_layer_;
+};
+
+SINGLE_AND_MULTI_THREAD_TEST_F(ScrollbarLayerLostContext)
+
 }  // namespace
 }  // namespace cc