[content shell] add support for a testRunner.setXSSAuditorEnabled

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/md5.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 14 matching lines @@
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebContextMenuData.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebDocument.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebElement.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebKit.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebView.h"
 #include "third_party/WebKit/Tools/DumpRenderTree/chromium/TestRunner/public/WebTask.h"
 #include "third_party/WebKit/Tools/DumpRenderTree/chromium/TestRunner/public/WebTestProxy.h"
 #include "webkit/glue/webkit_glue.h"
+#include "webkit/glue/webpreferences.h"
 
 using WebKit::WebContextMenuData;
 using WebKit::WebElement;
 using WebKit::WebFrame;
 using WebKit::WebGamepads;
 using WebKit::WebRect;
 using WebKit::WebSize;
 using WebKit::WebString;
 using WebKit::WebVector;
 using WebKit::WebView;
@@ skipping 220 matching lines @@
 // Public methods - -----------------------------------------------------------
 
 void WebKitTestRunner::Display() {
   const WebSize& size = render_view()->GetWebView()->size();
   WebRect rect(0, 0, size.width, size.height);
   proxy_->setPaintRect(rect);
   PaintInvalidatedRegion();
   DisplayRepaintMask();
 }
 
+void WebKitTestRunner::SetXSSAuditorEnabled(bool enabled) {
+  prefs_.xss_auditor_enabled = enabled;
+  webkit_glue::WebPreferences prefs = render_view()->GetWebkitPreferences();
+  prefs_.Apply(&prefs);
+  render_view()->SetWebkitPreferences(prefs);
+  Send(new ShellViewHostMsg_OverridePreferences(routing_id(), prefs_));
+}
+
+void WebKitTestRunner::Reset() {
+  prefs_ = ShellWebPreferences();
+  webkit_glue::WebPreferences prefs = render_view()->GetWebkitPreferences();
+  prefs_.Apply(&prefs);
+  render_view()->SetWebkitPreferences(prefs);
+}
+
 // Private methods  -----------------------------------------------------------
 
 void WebKitTestRunner::OnCaptureTextDump(bool as_text,
                                          bool printing,
                                          bool recursive) {
   WebFrame* frame = render_view()->GetWebView()->mainFrame();
   std::string dump;
   if (as_text) {
     dump = DumpFramesAsText(frame, printing, recursive);
   } else {
@@ skipping 107 matching lines @@
     PaintRect(rect);
   }
   CHECK(proxy_->paintRect().isEmpty());
 }
 
 void WebKitTestRunner::DisplayRepaintMask() {
   GetCanvas()->drawARGB(167, 0, 0, 0);
 }
 
 }  // namespace content