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:

[danakj, 2013/01/18 23:16:20]

nit: 1 space before

[piman, 2013/01/18 23:29:14]

Done.

+    ScrollbarLayerLostContext() : commits_(0) {}
+
+    virtual void beginTest() OVERRIDE {
+        scoped_refptr<Layer> scroll_layer = Layer::create();

[danakj, 2013/01/18 23:16:20]

nit: 2space indents

[piman, 2013/01/18 23:29:14]

Done.

+        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;
+        }
+    }
+
+private:
+    int commits_;
+    scoped_refptr<FakeScrollbarLayer> scrollbar_layer_;
+};
+
+SINGLE_AND_MULTI_THREAD_TEST_F(ScrollbarLayerLostContext)
+
 }  // namespace
 }  // namespace cc