[content shell] add support for dumping frame load callbacks

content/shell/webkit_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/webkit_test_runner.h"
 
 #include <cmath>
 
 #include "base/base64.h"
 #include "base/md5.h"
@@ skipping 137 matching lines @@
   // Only the expected PNGs for Mac have a valid alpha channel.
   MakeBitmapOpaque(snapshot);
 #endif
 
 }
 
 }  // namespace
 
 WebKitTestRunner::WebKitTestRunner(RenderView* render_view)
     : RenderViewObserver(render_view),
-      dump_editing_callbacks_(false) {
+      dump_editing_callbacks_(false),
+      dump_frame_load_callbacks_(false),
+      dump_user_gesture_in_frame_load_callbacks_(false),
+      stop_provisional_frame_loads_(false),
+      dump_title_changes_(false),
+      test_is_running_(true),
+      wait_until_done_(false) {
 }
 
 WebKitTestRunner::~WebKitTestRunner() {
 }
 
 // WebTestDelegate  -----------------------------------------------------------
 
 void WebKitTestRunner::clearContextMenuData() {
   last_context_menu_data_.reset();
 }
@@ skipping 121 matching lines @@
   render_view()->SetWebkitPreferences(prefs);
   Send(new ShellViewHostMsg_OverridePreferences(routing_id(), prefs));
 }
 
 // WebTestRunner  -------------------------------------------------------------
 
 bool WebKitTestRunner::shouldDumpEditingCallbacks() const {
   return dump_editing_callbacks_;
 }
 
+bool WebKitTestRunner::shouldDumpFrameLoadCallbacks() const {
+  return test_is_running_ && dump_frame_load_callbacks_;
+}
+
+bool WebKitTestRunner::shouldDumpUserGestureInFrameLoadCallbacks() const {
+  return test_is_running_ && dump_user_gesture_in_frame_load_callbacks_;
+}
+
+bool WebKitTestRunner::stopProvisionalFrameLoads() const {
+  return stop_provisional_frame_loads_;
+}
+
+bool WebKitTestRunner::shouldDumpTitleChanges() const {
+  return dump_title_changes_;
+}
+
 // RenderViewObserver  --------------------------------------------------------
 
 void WebKitTestRunner::DidClearWindowObject(WebFrame* frame) {
   ShellRenderProcessObserver::GetInstance()->BindTestRunnersToWindow(frame);
 }
 
 void WebKitTestRunner::DidFinishLoad(WebFrame* frame) {
-  if (!frame->parent())
+  if (!frame->parent()) {
+    if (!wait_until_done_)
+      test_is_running_ = false;
     Send(new ShellViewHostMsg_DidFinishLoad(routing_id()));
+  }
 }
 
 void WebKitTestRunner::DidRequestShowContextMenu(
     WebFrame* frame,
     const WebContextMenuData& data) {
   last_context_menu_data_.reset(new WebContextMenuData(data));
 }
 
 bool WebKitTestRunner::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
@@ skipping 17 matching lines @@
   PaintInvalidatedRegion();
   DisplayRepaintMask();
 }
 
 void WebKitTestRunner::SetXSSAuditorEnabled(bool enabled) {
   prefs_.XSSAuditorEnabled = enabled;
   applyPreferences();
 }
 
 void WebKitTestRunner::NotifyDone() {
+  test_is_running_ = false;
   Send(new ShellViewHostMsg_NotifyDone(routing_id()));
 }
 
 void WebKitTestRunner::DumpAsText() {
   Send(new ShellViewHostMsg_DumpAsText(routing_id()));
 }
 
 void WebKitTestRunner::DumpChildFramesAsText() {
   Send(new ShellViewHostMsg_DumpChildFramesAsText(routing_id()));
 }
 
 void WebKitTestRunner::SetPrinting() {
   Send(new ShellViewHostMsg_SetPrinting(routing_id()));
 }
 
 void WebKitTestRunner::SetShouldStayOnPageAfterHandlingBeforeUnload(
     bool should_stay_on_page) {
   Send(new ShellViewHostMsg_SetShouldStayOnPageAfterHandlingBeforeUnload(
       routing_id(), should_stay_on_page));
 }
 
 void WebKitTestRunner::WaitUntilDone() {
+  wait_until_done_ = true;
   Send(new ShellViewHostMsg_WaitUntilDone(routing_id()));
 }
 
 void WebKitTestRunner::CanOpenWindows() {
   Send(new ShellViewHostMsg_CanOpenWindows(routing_id()));
 }
 
 void WebKitTestRunner::ShowWebInspector() {
   Send(new ShellViewHostMsg_ShowWebInspector(routing_id()));
 }
@@ skipping 66 matching lines @@
     std::string message("CONSOLE MESSAGE: Invalid name for preference: ");
     printMessage(message + key + "\n");
   }
   applyPreferences();
 }
 
 void WebKitTestRunner::DumpEditingCallbacks() {
   dump_editing_callbacks_ = true;
 }
 
+void WebKitTestRunner::DumpFrameLoadCallbacks() {
+  dump_frame_load_callbacks_ = true;
+}
+
+void WebKitTestRunner::DumpUserGestureInFrameLoadCallbacks() {
+  dump_user_gesture_in_frame_load_callbacks_ = true;
+}
+
+void WebKitTestRunner::StopProvisionalFrameLoads() {
+  stop_provisional_frame_loads_ = true;
+}
+
+void WebKitTestRunner::DumpTitleChanges() {
+  dump_title_changes_ = true;
+}
+
 void WebKitTestRunner::NotImplemented(const std::string& object,
                                       const std::string& method) {
   Send(new ShellViewHostMsg_NotImplemented(routing_id(), object, method));
 }
 
 void WebKitTestRunner::Reset() {
   prefs_.reset();
   webkit_glue::WebPreferences prefs = render_view()->GetWebkitPreferences();
   ExportLayoutTestSpecificPreferences(prefs_, &prefs);
   render_view()->SetWebkitPreferences(prefs);
   dump_editing_callbacks_ = false;
+  dump_frame_load_callbacks_ = false;
+  dump_user_gesture_in_frame_load_callbacks_ = false;
+  stop_provisional_frame_loads_ = false;
+  dump_title_changes_ = false;
+  test_is_running_ = true;
+  wait_until_done_ = false;
 }
 
 // Private methods  -----------------------------------------------------------
 
 void WebKitTestRunner::OnCaptureTextDump(bool as_text,
                                          bool printing,
                                          bool recursive) {
   WebFrame* frame = render_view()->GetWebView()->mainFrame();
   std::string dump;
   if (as_text) {
@@ skipping 109 matching lines @@
     PaintRect(rect);
   }
   CHECK(proxy_->paintRect().isEmpty());
 }
 
 void WebKitTestRunner::DisplayRepaintMask() {
   GetCanvas()->drawARGB(167, 0, 0, 0);
 }
 
 }  // namespace content