Define frame subscription interface and implementation on Mac

content/browser/renderer_host/render_widget_host_view_browsertest.cc

 // Copyright (c) 2013 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 "base/command_line.h"
 #include "base/path_service.h"
 #include "base/run_loop.h"
+#include "content/port/browser/render_widget_host_view_frame_subscriber.h"
 #include "content/port/browser/render_widget_host_view_port.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/content_paths.h"
 #include "content/shell/shell.h"
 #include "content/test/content_browser_test.h"
 #include "content/test/content_browser_test_utils.h"
+#include "media/base/video_frame.h"
 #include "net/base/net_util.h"
 #include "skia/ext/platform_canvas.h"
 #include "ui/compositor/compositor_setup.h"
 #if defined(OS_MACOSX)
 #include "ui/surface/io_surface_support_mac.h"
 #endif
 
 namespace content {
 
 class RenderWidgetHostViewBrowserTest : public ContentBrowserTest {
@@ skipping 43 matching lines @@
       ASSERT_FALSE(bitmap.empty());
     finish_called_ = true;
   }
 
  protected:
   base::FilePath test_dir_;
   bool finish_called_;
   gfx::Size size_;
 };
 
+class FakeFrameSubscriber : public RenderWidgetHostViewFrameSubscriber {
+ public:
+  FakeFrameSubscriber(
+    RenderWidgetHostViewFrameSubscriber::DeliverFrameCallback callback)
+      : callback_(callback) {
+  }
+
+  virtual bool ShouldCaptureFrame(
+      scoped_refptr<media::VideoFrame>* storage,
+      DeliverFrameCallback* callback) OVERRIDE {
+    *storage = media::VideoFrame::CreateBlackFrame(gfx::Size(100, 100));
+    *callback = callback_;
+    return true;
+  }
+
+ private:
+  DeliverFrameCallback callback_;
+};
+
 #if defined(OS_MACOSX)
 // Tests that the callback passed to CopyFromBackingStore is always called, even
 // when the RenderWidgetHost is deleting in the middle of an async copy.
 IN_PROC_BROWSER_TEST_F(RenderWidgetHostViewBrowserTest,
                        MacAsyncCopyFromBackingStoreCallbackTest) {
   if (!IOSurfaceSupport::Initialize())
     return;
 
   SetupCompositingSurface();
 
@@ skipping 30 matching lines @@
   base::RunLoop run_loop;
   shell()->web_contents()->GetRenderViewHost()->CopyFromBackingStore(
       gfx::Rect(),
       size_,
       base::Bind(&RenderWidgetHostViewBrowserTest::FinishCopyFromBackingStore,
                  base::Unretained(this), true, run_loop.QuitClosure()));
   run_loop.Run();
 
   ASSERT_TRUE(finish_called_);
 }
+
+static void DeliverFrameFunc(base::Closure quit_closure,
+                             bool* frame_captured_out,
+                             base::Time timestamp,
+                             bool frame_captured) {
+  *frame_captured_out = frame_captured;
+  quit_closure.Run();
+}
+
+IN_PROC_BROWSER_TEST_F(RenderWidgetHostViewBrowserTest,
+                       MacFrameSubscriberTest) {
+  if (!IOSurfaceSupport::Initialize())
+    return;
+
+  SetupCompositingSurface();
+
+  base::RunLoop run_loop;
+  RenderWidgetHostViewPort* view = RenderWidgetHostViewPort::FromRWHV(
+      shell()->web_contents()->GetRenderViewHost()->GetView());
+  ASSERT_TRUE(view);
+
+  EXPECT_TRUE(view->CanSubscribeFrame());
+
+  bool frame_captured = false;
+  view->BeginFrameSubscription(
+      new FakeFrameSubscriber(base::Bind(&DeliverFrameFunc,
+                                         run_loop.QuitClosure(),
+                                         &frame_captured)));
+
+  // Do a resize of the window to trigger a repaint and present.
+  SetWindowBounds(shell()->window(), gfx::Rect(size_));
+  run_loop.Run();
+  view->EndFrameSubscription();
+
+  EXPECT_TRUE(frame_captured);
+}
+
 #endif
 
 }  // namespace content