OOPIF support for layout test pixel dumps.

content/shell/renderer/layout_test/layout_test_render_frame_observer.cc

 // Copyright 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 "content/shell/renderer/layout_test/layout_test_render_frame_observer.h"
 
 #include <string>
 
+#include "base/callback.h"
 #include "content/public/common/associated_interface_registry.h"
 #include "content/public/renderer/render_frame.h"
 #include "content/shell/renderer/layout_test/blink_test_runner.h"
 #include "content/shell/renderer/layout_test/layout_test_render_thread_observer.h"
 #include "content/shell/test_runner/web_test_interfaces.h"
 #include "content/shell/test_runner/web_test_runner.h"
+#include "third_party/WebKit/public/platform/WebCompositeAndReadbackAsyncCallback.h"
+#include "third_party/WebKit/public/web/WebFrameWidget.h"
 #include "third_party/WebKit/public/web/WebLocalFrame.h"
+#include "third_party/WebKit/public/web/WebView.h"
 
 namespace content {
 
+namespace {
+
+class CompositeAndReadbackCallback
+    : public blink::WebCompositeAndReadbackAsyncCallback {
+ public:
+  explicit CompositeAndReadbackCallback(base::OnceClosure callback)
+      : callback_(std::move(callback)) {}
+
+  virtual ~CompositeAndReadbackCallback() = default;
+
+  // WebCompositeAndReadbackAsyncCallback implementation.
+  void DidCompositeAndReadback(const SkBitmap& bitmap) override {
+    LOG(ERROR) << "DidCompositeAndReadback";
+    std::move(callback_).Run();
+    delete this;
+  }
+
+ private:
+  base::OnceClosure callback_;
+};
+
+}  // namespace
+
 LayoutTestRenderFrameObserver::LayoutTestRenderFrameObserver(
     RenderFrame* render_frame)
     : RenderFrameObserver(render_frame), binding_(this) {
   test_runner::WebTestRunner* test_runner =
       LayoutTestRenderThreadObserver::GetInstance()
           ->test_interfaces()
           ->TestRunner();
   render_frame->GetWebFrame()->SetContentSettingsClient(
       test_runner->GetWebContentSettings());
   render_frame->GetWebFrame()->SetTextCheckClient(
@@ skipping 16 matching lines @@
 void LayoutTestRenderFrameObserver::DumpFrameLayout(
     DumpFrameLayoutCallback callback) {
   std::string dump =
       LayoutTestRenderThreadObserver::GetInstance()
           ->test_interfaces()
           ->TestRunner()
           ->DumpLayout(render_frame()->GetWebFrame());
   std::move(callback).Run(dump);
 }
 
+void LayoutTestRenderFrameObserver::TriggerCompositing(
+    TriggerCompositingCallback callback) {
+  blink::WebWidget* widget =
+      render_frame()->IsMainFrame()
+          ? render_frame()->GetWebFrame()->View()->GetWidget()
+          : render_frame()->GetWebFrame()->FrameWidget();
+
+  // DO NOT SUBMIT.
+  LOG(ERROR) << "LayoutTestRenderFrameObserver::TriggerCompositing"
+             << "; IsMainFrame = " << render_frame()->IsMainFrame()
+             << "; widget = " << widget;
+
+  if (widget) {
+    widget->CompositeAndReadbackAsync(
+        new CompositeAndReadbackCallback(std::move(callback)));
+  } else {
+    std::move(callback).Run();
+  }
+}
+
 void LayoutTestRenderFrameObserver::ReplicateTestConfiguration(
     mojom::ShellTestConfigurationPtr config) {
   BlinkTestRunner::Get(render_frame()->GetRenderView())
       ->OnReplicateTestConfiguration(std::move(config));
 }
 
 void LayoutTestRenderFrameObserver::SetTestConfiguration(
     mojom::ShellTestConfigurationPtr config) {
   BlinkTestRunner::Get(render_frame()->GetRenderView())
       ->OnSetTestConfiguration(std::move(config));
 }
 
 void LayoutTestRenderFrameObserver::SetupSecondaryRenderer() {
   BlinkTestRunner::Get(render_frame()->GetRenderView())
       ->OnSetupSecondaryRenderer();
 }
 
 }  // namespace content