OOPIF support for testRunner.dumpAsText and similar layout dumps.

content/shell/browser/blink_test_controller.h

 // 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.
 
 #ifndef CONTENT_SHELL_BROWSER_BLINK_TEST_CONTROLLER_H_
 #define CONTENT_SHELL_BROWSER_BLINK_TEST_CONTROLLER_H_
 
+#include <map>
 #include <ostream>
 #include <string>
 
 #include "base/cancelable_callback.h"
 #include "base/files/file_path.h"
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/synchronization/lock.h"
 #include "base/threading/non_thread_safe.h"
 #include "build/build_config.h"
+#include "components/test_runner/layout_dump_flags.h"
 #include "content/public/browser/bluetooth_chooser.h"
 #include "content/public/browser/gpu_data_manager_observer.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/common/web_preferences.h"
 #include "content/shell/common/leak_detection_result.h"
 #include "ui/gfx/geometry/size.h"
 
 #if defined(OS_ANDROID)
 #include "base/threading/thread_restrictions.h"
 #endif
 
 class SkBitmap;
 
 namespace content {
 
 class LayoutTestBluetoothChooserFactory;
 class LayoutTestDevToolsFrontend;
+class RenderFrameHost;
 class Shell;
 
 #if defined(OS_ANDROID)
 // Android uses a nested message loop for running layout tests because the
 // default message loop, provided by the system, does not offer a blocking
 // Run() method. The loop itself, implemented as NestedMessagePumpAndroid,
 // uses a base::WaitableEvent allowing it to sleep until more events arrive.
 class ScopedAllowWaitForAndroidLayoutTests {
  private:
   base::ThreadRestrictions::ScopedAllowWait wait;
@@ skipping 86 matching lines @@
       const BluetoothChooser::EventHandler& event_handler,
       const GURL& origin);
 
   BlinkTestResultPrinter* printer() { return printer_.get(); }
   void set_printer(BlinkTestResultPrinter* printer) { printer_.reset(printer); }
 
   void DevToolsProcessCrashed();
 
   // WebContentsObserver implementation.
   bool OnMessageReceived(const IPC::Message& message) override;
+  bool OnMessageReceived(const IPC::Message& message,
+                         RenderFrameHost* render_frame_host) override;
   void PluginCrashed(const base::FilePath& plugin_path,
                      base::ProcessId plugin_pid) override;
   void RenderViewCreated(RenderViewHost* render_view_host) override;
   void RenderProcessGone(base::TerminationStatus status) override;
   void WebContentsDestroyed() override;
 
   // NotificationObserver implementation.
   void Observe(int type,
                const NotificationSource& source,
                const NotificationDetails& details) override;
@@ skipping 10 matching lines @@
 
   static BlinkTestController* instance_;
 
   void DiscardMainWindow();
   void SendTestConfiguration();
 
   // Message handlers.
   void OnAudioDump(const std::vector<unsigned char>& audio_dump);
   void OnImageDump(const std::string& actual_pixel_hash, const SkBitmap& image);
   void OnTextDump(const std::string& dump);
+  void OnInitiateLayoutDump(test_runner::LayoutDumpFlags layout_dump_flags);
+  void OnLayoutDumpResponse(RenderFrameHost* sender, const std::string& dump);
   void OnPrintMessage(const std::string& message);
   void OnOverridePreferences(const WebPreferences& prefs);
   void OnTestFinished();
   void OnClearDevToolsLocalStorage();
   void OnShowDevTools(const std::string& settings,
                       const std::string& frontend_url);
   void OnCloseDevTools();
   void OnGoToOffset(int offset);
   void OnReload();
   void OnLoadURLForFrame(const GURL& url, const std::string& frame_name);
@@ skipping 39 matching lines @@
 
   NotificationRegistrar registrar_;
 
   const bool is_leak_detection_enabled_;
   bool crash_when_leak_found_;
 
   LayoutTestDevToolsFrontend* devtools_frontend_;
 
   scoped_ptr<LayoutTestBluetoothChooserFactory> bluetooth_chooser_factory_;
 
+  // Map from frame_tree_node_id into frame-specific dumps.
+  std::map<int, std::string> frame_to_layout_dump_map_;
+  // Number of ShellViewHostMsg_LayoutDumpResponse messages we are waiting for.
+  int pending_layout_dumps_;
+
 #if defined(OS_ANDROID)
   // Because of the nested message pump implementation, Android needs to allow
   // waiting on the UI thread while layout tests are being ran.
   ScopedAllowWaitForAndroidLayoutTests reduced_restrictions_;
 #endif
 
   DISALLOW_COPY_AND_ASSIGN(BlinkTestController);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_SHELL_BROWSER_BLINK_TEST_CONTROLLER_H_