Add testRunner callbacks for better handling of view-source frames

content/shell/renderer/test_runner/test_runner.cc

 // Copyright 2014 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/test_runner/test_runner.h"
 
 #include <limits>
 
 #include "base/logging.h"
 #include "content/shell/common/test_runner/test_preferences.h"
@@ skipping 169 matching lines @@
   void SetPointerLockWillRespondAsynchronously();
   void SetPopupBlockingEnabled(bool block_popups);
   void SetJavaScriptCanAccessClipboard(bool can_access);
   void SetXSSAuditorEnabled(bool enabled);
   void SetAllowUniversalAccessFromFileURLs(bool allow);
   void SetAllowFileAccessFromFileURLs(bool allow);
   void OverridePreference(const std::string key, v8::Handle<v8::Value> value);
   void SetAcceptLanguages(const std::string& accept_languages);
   void SetPluginsEnabled(bool enabled);
   void DumpEditingCallbacks();
+  void DumpAsMarkup();
   void DumpAsText();
   void DumpAsTextWithPixelResults();
   void DumpChildFrameScrollPositions();
+  void DumpChildFramesAsMarkup();
   void DumpChildFramesAsText();
   void DumpIconChanges();
   void SetAudioData(const gin::ArrayBufferView& view);
   void DumpFrameLoadCallbacks();
   void DumpPingLoaderCallbacks();
   void DumpUserGestureInFrameLoadCallbacks();
   void DumpTitleChanges();
   void DumpCreateView();
   void SetCanOpenWindows();
   void DumpResourceLoadCallbacks();
@@ skipping 37 matching lines @@
   void AddMockSpeechRecognitionResult(const std::string& transcript,
                                       double confidence);
   void SetMockSpeechRecognitionError(const std::string& error,
                                      const std::string& message);
   bool WasMockSpeechRecognitionAborted();
   void AddWebPageOverlay();
   void RemoveWebPageOverlay();
   void DisplayAsync();
   void DisplayAsyncThen(v8::Handle<v8::Function> callback);
   void SetCustomTextOutput(std::string output);
+  void SetViewSourceForFrame(const std::string& name, bool enabled);
 
   bool GlobalFlag();
   void SetGlobalFlag(bool value);
   std::string PlatformName();
   std::string TooltipText();
   bool DisableNotifyDone();
   int WebHistoryItemCount();
   bool InterceptPostMessage();
   void SetInterceptPostMessage(bool value);
 
@@ skipping 121 matching lines @@
                  &TestRunnerBindings::SetXSSAuditorEnabled)
       .SetMethod("setAllowUniversalAccessFromFileURLs",
                  &TestRunnerBindings::SetAllowUniversalAccessFromFileURLs)
       .SetMethod("setAllowFileAccessFromFileURLs",
                  &TestRunnerBindings::SetAllowFileAccessFromFileURLs)
       .SetMethod("overridePreference", &TestRunnerBindings::OverridePreference)
       .SetMethod("setAcceptLanguages", &TestRunnerBindings::SetAcceptLanguages)
       .SetMethod("setPluginsEnabled", &TestRunnerBindings::SetPluginsEnabled)
       .SetMethod("dumpEditingCallbacks",
                  &TestRunnerBindings::DumpEditingCallbacks)
+      .SetMethod("dumpAsMarkup", &TestRunnerBindings::DumpAsMarkup)
       .SetMethod("dumpAsText", &TestRunnerBindings::DumpAsText)
       .SetMethod("dumpAsTextWithPixelResults",
                  &TestRunnerBindings::DumpAsTextWithPixelResults)
       .SetMethod("dumpChildFrameScrollPositions",
                  &TestRunnerBindings::DumpChildFrameScrollPositions)
       .SetMethod("dumpChildFramesAsText",
                  &TestRunnerBindings::DumpChildFramesAsText)
+      .SetMethod("dumpChildFramesAsMarkup",
+                 &TestRunnerBindings::DumpChildFramesAsMarkup)
       .SetMethod("dumpIconChanges", &TestRunnerBindings::DumpIconChanges)
       .SetMethod("setAudioData", &TestRunnerBindings::SetAudioData)
       .SetMethod("dumpFrameLoadCallbacks",
                  &TestRunnerBindings::DumpFrameLoadCallbacks)
       .SetMethod("dumpPingLoaderCallbacks",
                  &TestRunnerBindings::DumpPingLoaderCallbacks)
       .SetMethod("dumpUserGestureInFrameLoadCallbacks",
                  &TestRunnerBindings::DumpUserGestureInFrameLoadCallbacks)
       .SetMethod("dumpTitleChanges", &TestRunnerBindings::DumpTitleChanges)
       .SetMethod("dumpCreateView", &TestRunnerBindings::DumpCreateView)
@@ skipping 65 matching lines @@
                  &TestRunnerBindings::SetMockSpeechRecognitionError)
       .SetMethod("wasMockSpeechRecognitionAborted",
                  &TestRunnerBindings::WasMockSpeechRecognitionAborted)
       .SetMethod("addWebPageOverlay", &TestRunnerBindings::AddWebPageOverlay)
       .SetMethod("removeWebPageOverlay",
                  &TestRunnerBindings::RemoveWebPageOverlay)
       .SetMethod("displayAsync", &TestRunnerBindings::DisplayAsync)
       .SetMethod("displayAsyncThen", &TestRunnerBindings::DisplayAsyncThen)
       .SetMethod("setCustomTextOutput",
                  &TestRunnerBindings::SetCustomTextOutput)
+      .SetMethod("setViewSourceForFrame",
+                 &TestRunnerBindings::SetViewSourceForFrame)
 
       // Properties.
       .SetProperty("globalFlag",
                    &TestRunnerBindings::GlobalFlag,
                    &TestRunnerBindings::SetGlobalFlag)
       .SetProperty("platformName", &TestRunnerBindings::PlatformName)
       .SetProperty("tooltipText", &TestRunnerBindings::TooltipText)
       .SetProperty("disableNotifyDone", &TestRunnerBindings::DisableNotifyDone)
       // webHistoryItemCount is used by tests in LayoutTests\http\tests\history
       .SetProperty("webHistoryItemCount",
@@ skipping 440 matching lines @@
 void TestRunnerBindings::SetPluginsEnabled(bool enabled) {
   if (runner_)
     runner_->SetPluginsEnabled(enabled);
 }
 
 void TestRunnerBindings::DumpEditingCallbacks() {
   if (runner_)
     runner_->DumpEditingCallbacks();
 }
 
+void TestRunnerBindings::DumpAsMarkup() {
+  if (runner_)
+    runner_->DumpAsMarkup();
+}
+
 void TestRunnerBindings::DumpAsText() {
   if (runner_)
     runner_->DumpAsText();
 }
 
 void TestRunnerBindings::DumpAsTextWithPixelResults() {
   if (runner_)
     runner_->DumpAsTextWithPixelResults();
 }
 
 void TestRunnerBindings::DumpChildFrameScrollPositions() {
   if (runner_)
     runner_->DumpChildFrameScrollPositions();
 }
 
 void TestRunnerBindings::DumpChildFramesAsText() {
   if (runner_)
     runner_->DumpChildFramesAsText();
 }
 
+void TestRunnerBindings::DumpChildFramesAsMarkup() {
+  if (runner_)
+    runner_->DumpChildFramesAsMarkup();
+}
+
 void TestRunnerBindings::DumpIconChanges() {
   if (runner_)
     runner_->DumpIconChanges();
 }
 
 void TestRunnerBindings::SetAudioData(const gin::ArrayBufferView& view) {
   if (runner_)
     runner_->SetAudioData(view);
 }
 
@@ skipping 266 matching lines @@
 
 void TestRunnerBindings::DisplayAsyncThen(v8::Handle<v8::Function> callback) {
   if (runner_)
     runner_->DisplayAsyncThen(callback);
 }
 
 void TestRunnerBindings::SetCustomTextOutput(std::string output) {
   runner_->setCustomTextOutput(output);
 }
 
+void TestRunnerBindings::SetViewSourceForFrame(const std::string& name,
+                                               bool enabled) {
+  if (runner_ && runner_->web_view_) {
+    WebFrame* target_frame =
+        runner_->web_view_->findFrameByName(WebString::fromUTF8(name));
+    if (target_frame)
+      target_frame->enableViewSourceMode(enabled);
+  }
+}
+
 bool TestRunnerBindings::GlobalFlag() {
   if (runner_)
     return runner_->global_flag_;
   return false;
 }
 
 void TestRunnerBindings::SetGlobalFlag(bool value) {
   if (runner_)
     runner_->global_flag_ = value;
 }
@@ skipping 188 matching lines @@
     delegate_->disableAutoResizeMode(WebSize());
     delegate_->deleteAllCookies();
     delegate_->resetScreenOrientation();
   }
 
   dump_editting_callbacks_ = false;
   dump_as_text_ = false;
   dump_as_markup_ = false;
   generate_pixel_results_ = true;
   dump_child_frame_scroll_positions_ = false;
+  dump_child_frames_as_markup_ = false;
   dump_child_frames_as_text_ = false;
   dump_icon_changes_ = false;
   dump_as_audio_ = false;
   dump_frame_load_callbacks_ = false;
   dump_ping_loader_callbacks_ = false;
   dump_user_gesture_in_frame_load_callbacks_ = false;
   dump_title_changes_ = false;
   dump_create_view_ = false;
   can_open_windows_ = false;
   dump_resource_load_callbacks_ = false;
@@ skipping 86 matching lines @@
 }
 
 void TestRunner::setShouldGeneratePixelResults(bool value) {
   generate_pixel_results_ = value;
 }
 
 bool TestRunner::shouldDumpChildFrameScrollPositions() const {
   return dump_child_frame_scroll_positions_;
 }
 
+bool TestRunner::shouldDumpChildFramesAsMarkup() const {
+  return dump_child_frames_as_markup_;
+}
+
 bool TestRunner::shouldDumpChildFramesAsText() const {
   return dump_child_frames_as_text_;
 }
 
 bool TestRunner::ShouldDumpAsAudio() const {
   return dump_as_audio_;
 }
 
 void TestRunner::GetAudioData(std::vector<unsigned char>* buffer_view) const {
   *buffer_view = audio_data_;
@@ skipping 765 matching lines @@
 
 void TestRunner::SetPluginsEnabled(bool enabled) {
   delegate_->preferences()->plugins_enabled = enabled;
   delegate_->applyPreferences();
 }
 
 void TestRunner::DumpEditingCallbacks() {
   dump_editting_callbacks_ = true;
 }
 
+void TestRunner::DumpAsMarkup() {
+  dump_as_markup_ = true;
+  generate_pixel_results_ = false;
+}
+
 void TestRunner::DumpAsText() {
   dump_as_text_ = true;
   generate_pixel_results_ = false;
 }
 
 void TestRunner::DumpAsTextWithPixelResults() {
   dump_as_text_ = true;
   generate_pixel_results_ = true;
 }
 
 void TestRunner::DumpChildFrameScrollPositions() {
   dump_child_frame_scroll_positions_ = true;
 }
 
+void TestRunner::DumpChildFramesAsMarkup() {
+  dump_child_frames_as_markup_ = true;
+}
+
 void TestRunner::DumpChildFramesAsText() {
   dump_child_frames_as_text_ = true;
 }
 
 void TestRunner::DumpIconChanges() {
   dump_icon_changes_ = true;
 }
 
 void TestRunner::SetAudioData(const gin::ArrayBufferView& view) {
   unsigned char* bytes = static_cast<unsigned char*>(view.bytes());
@@ skipping 275 matching lines @@
 }
 
 void TestRunner::DidLosePointerLockInternal() {
   bool was_locked = pointer_locked_;
   pointer_locked_ = false;
   if (was_locked)
     web_view_->didLosePointerLock();
 }
 
 }  // namespace content