OOPIF support for testRunner.dumpAsText and similar layout dumps.

content/shell/renderer/layout_test/blink_test_runner.cc

 // Copyright (c) 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 "content/shell/renderer/layout_test/blink_test_runner.h"
 
 #include <stddef.h>
 #include <algorithm>
 #include <clocale>
 #include <cmath>
@@ skipping 12 matching lines @@
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/thread_task_runner_handle.h"
 #include "base/time/time.h"
 #include "build/build_config.h"
 #include "components/plugins/renderer/plugin_placeholder.h"
 #include "components/test_runner/app_banner_client.h"
 #include "components/test_runner/gamepad_controller.h"
+#include "components/test_runner/layout_dump.h"
+#include "components/test_runner/layout_dump_flags.h"
 #include "components/test_runner/mock_screen_orientation_client.h"
 #include "components/test_runner/test_interfaces.h"
 #include "components/test_runner/web_task.h"
 #include "components/test_runner/web_test_interfaces.h"
 #include "components/test_runner/web_test_proxy.h"
 #include "components/test_runner/web_test_runner.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/url_constants.h"
 #include "content/public/common/web_preferences.h"
 #include "content/public/renderer/media_stream_api.h"
@@ skipping 769 matching lines @@
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(BlinkTestRunner, message)
     IPC_MESSAGE_HANDLER(ShellViewMsg_SetTestConfiguration,
                         OnSetTestConfiguration)
     IPC_MESSAGE_HANDLER(ShellViewMsg_SessionHistory, OnSessionHistory)
     IPC_MESSAGE_HANDLER(ShellViewMsg_Reset, OnReset)
     IPC_MESSAGE_HANDLER(ShellViewMsg_NotifyDone, OnNotifyDone)
     IPC_MESSAGE_HANDLER(ShellViewMsg_TryLeakDetection, OnTryLeakDetection)
     IPC_MESSAGE_HANDLER(ShellViewMsg_ReplyBluetoothManualChooserEvents,
                         OnReplyBluetoothManualChooserEvents)
+    IPC_MESSAGE_HANDLER(ShellViewMsg_LayoutDumpCompleted, OnLayoutDumpCompleted)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   return handled;
 }
 
 void BlinkTestRunner::Navigate(const GURL& url) {
   focus_on_next_commit_ = true;
   if (!is_main_window_ &&
       LayoutTestRenderProcessObserver::GetInstance()->main_test_runner() ==
@@ skipping 48 matching lines @@
 
 void BlinkTestRunner::CaptureDump() {
   test_runner::WebTestInterfaces* interfaces =
       LayoutTestRenderProcessObserver::GetInstance()->test_interfaces();
   TRACE_EVENT0("shell", "BlinkTestRunner::CaptureDump");
 
   if (interfaces->TestRunner()->ShouldDumpAsAudio()) {
     std::vector<unsigned char> vector_data;
     interfaces->TestRunner()->GetAudioData(&vector_data);
     Send(new ShellViewHostMsg_AudioDump(routing_id(), vector_data));
-  } else {
-      const base::CommandLine& command_line =
-          *base::CommandLine::ForCurrentProcess();
-    Send(new ShellViewHostMsg_TextDump(
-        routing_id(), proxy()->CaptureTree(
-            false, command_line.HasSwitch(switches::kDumpLineBoxTrees))));
+    CaptureDumpContinued();
+    return;
+  }
 
-    if (test_config_.enable_pixel_dumping &&
-        interfaces->TestRunner()->ShouldGeneratePixelResults()) {
-      CHECK(render_view()->GetWebView()->isAcceleratedCompositingActive());
-      proxy()->CapturePixelsAsync(base::Bind(
-          &BlinkTestRunner::CaptureDumpPixels, base::Unretained(this)));
-      return;
-    }
+  std::string custom_text_dump;
+  if (interfaces->TestRunner()->HasCustomTextDump(&custom_text_dump)) {
+    Send(new ShellViewHostMsg_TextDump(routing_id(), custom_text_dump + "\n"));
+    CaptureDumpContinued();
+    return;
   }
+
+  test_runner::LayoutDumpFlags layout_dump_flags =
+      interfaces->TestRunner()->GetLayoutDumpFlags();
+  layout_dump_flags.dump_line_box_trees =
+      base::CommandLine::ForCurrentProcess()->HasSwitch(
+          switches::kDumpLineBoxTrees);
+
+  if (!layout_dump_flags.dump_child_frames) {
+    std::string layout_dump = DumpLayout(
+        render_view()->GetMainRenderFrame()->GetWebFrame(), layout_dump_flags);
+    OnLayoutDumpCompleted(layout_dump);
+    return;
+  }
+
+  Send(
+      new ShellViewHostMsg_InitiateLayoutDump(routing_id(), layout_dump_flags));
+  // OnLayoutDumpCompleted will be eventually called by an IPC from the browser.
+}
+
+void BlinkTestRunner::OnLayoutDumpCompleted(std::string completed_layout_dump) {
+  test_runner::WebTestInterfaces* interfaces =
+      LayoutTestRenderProcessObserver::GetInstance()->test_interfaces();
+  if (interfaces->TestRunner()->ShouldDumpBackForwardList()) {
+    completed_layout_dump.append(proxy()->DumpBackForwardLists());
+  }
+
+  Send(new ShellViewHostMsg_TextDump(routing_id(), completed_layout_dump));
+
+  CaptureDumpContinued();
+}
+
+void BlinkTestRunner::CaptureDumpContinued() {
+  test_runner::WebTestInterfaces* interfaces =
+      LayoutTestRenderProcessObserver::GetInstance()->test_interfaces();
+  if (test_config_.enable_pixel_dumping &&
+      interfaces->TestRunner()->ShouldGeneratePixelResults() &&
+      !interfaces->TestRunner()->ShouldDumpAsAudio()) {
+    CHECK(render_view()->GetWebView()->isAcceleratedCompositingActive());
+    proxy()->CapturePixelsAsync(base::Bind(
+        &BlinkTestRunner::OnPixelsDumpCompleted, base::Unretained(this)));
+    return;
+  }
+
 #ifndef NDEBUG
-    // Force a layout/paint by the end of the test to ensure test coverage of
-    // incremental painting.
-    proxy()->LayoutAndPaintAsyncThen(base::Bind(
-        &BlinkTestRunner::CaptureDumpComplete, base::Unretained(this)));
-    return;
+  // Force a layout/paint by the end of the test to ensure test coverage of
+  // incremental painting.
+  proxy()->LayoutAndPaintAsyncThen(base::Bind(
+      &BlinkTestRunner::CaptureDumpComplete, base::Unretained(this)));
 #else
   CaptureDumpComplete();
 #endif
 }
 
-void BlinkTestRunner::CaptureDumpPixels(const SkBitmap& snapshot) {
+void BlinkTestRunner::OnPixelsDumpCompleted(const SkBitmap& snapshot) {
   DCHECK_NE(0, snapshot.info().width());
   DCHECK_NE(0, snapshot.info().height());
 
   SkAutoLockPixels snapshot_lock(snapshot);
   // The snapshot arrives from the GPU process via shared memory. Because MSan
   // can't track initializedness across processes, we must assure it that the
   // pixels are in fact initialized.
   MSAN_UNPOISON(snapshot.getPixels(), snapshot.getSize());
   base::MD5Digest digest;
   base::MD5Sum(snapshot.getPixels(), snapshot.getSize(), &digest);
@@ skipping 79 matching lines @@
   get_bluetooth_events_callbacks_.pop_front();
   callback.Run(events);
 }
 
 void BlinkTestRunner::ReportLeakDetectionResult(
     const LeakDetectionResult& report) {
   Send(new ShellViewHostMsg_LeakDetectionDone(routing_id(), report));
 }
 
 }  // namespace content