Fixing Canvas2DLayerBridge to handle lost graphics contexts

Source/WebKit/chromium/tests/Canvas2DLayerManagerTest.cpp

 /*
  * Copyright (C) 2012 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1.  Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
  * 2.  Redistributions in binary form must reproduce the above copyright
  *     notice, this list of conditions and the following disclaimer in the
@@ skipping 30 matching lines @@
 
 
 class FakeCanvas2DLayerBridge : public Canvas2DLayerBridge {
 public:
     FakeCanvas2DLayerBridge(PassRefPtr<GraphicsContext3D> context, SkDeferredCanvas* canvas)
         : Canvas2DLayerBridge(context, canvas, NonOpaque, SingleThread)
         , m_freeableBytes(0)
         , m_freeMemoryIfPossibleCount(0)
         , m_flushCount(0)
     {
+        m_fakeSharedContext = m_context;
     }
 
     virtual size_t storageAllocatedForRecording() OVERRIDE
     {
         // Because the fake layer has no canvas to query, just
         // return status quo. Allocation changes that would normally be
         // initiated by the canvas can be faked by invoking
         // storageAllocatedForRecordingChanged directly from the test code.
         return m_bytesAllocated;
     }
@@ skipping 13 matching lines @@
         m_bytesAllocated -= bytesFreed;
         return bytesFreed;
     }
 
     virtual void flush() OVERRIDE
     {
         flushedDrawCommands();
         m_flushCount++;
     }
 
+protected:
+    virtual PassRefPtr<GraphicsContext3D> getSharedContext() const OVERRIDE
+    {
+        return m_fakeSharedContext;
+    }
 public:
     size_t m_freeableBytes;
     int m_freeMemoryIfPossibleCount;
     int m_flushCount;
+private:
+    RefPtr<GraphicsContext3D> m_fakeSharedContext;
 };
 
 static PassOwnPtr<SkDeferredCanvas> createCanvas(GraphicsContext3D* context) {
     SkAutoTUnref<SkDevice> device(new SkDevice(SkBitmap::kARGB_8888_Config, 1, 1));
     return adoptPtr(new SkDeferredCanvas(device.get()));
 }
 
 class Canvas2DLayerManagerTest : public Test {
 protected:
     void storageAllocationTrackingTest()
@@ skipping 158 matching lines @@
     flushEvictionTest();
 }
 
 TEST_F(Canvas2DLayerManagerTest, testDeferredFrame)
 {
     deferredFrameTest();
 }
 
 } // namespace