Skeleton of the Safe Browsing Subresource Filter.

components/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 "components/test_runner/test_runner.h"
 
 #include <stddef.h>
 #include <limits>
 #include <utility>
 
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
 #include "build/build_config.h"
 #include "components/test_runner/app_banner_client.h"
 #include "components/test_runner/layout_and_paint_async_then.h"
 #include "components/test_runner/layout_dump.h"
 #include "components/test_runner/mock_content_settings_client.h"
 #include "components/test_runner/mock_credential_manager_client.h"
 #include "components/test_runner/mock_screen_orientation_client.h"
+#include "components/test_runner/mock_web_document_subresource_filter.h"
 #include "components/test_runner/mock_web_speech_recognizer.h"
 #include "components/test_runner/mock_web_user_media_client.h"
 #include "components/test_runner/pixel_dump.h"
 #include "components/test_runner/spell_check_client.h"
 #include "components/test_runner/test_common.h"
 #include "components/test_runner/test_interfaces.h"
 #include "components/test_runner/test_preferences.h"
 #include "components/test_runner/test_runner_for_specific_view.h"
 #include "components/test_runner/web_task.h"
 #include "components/test_runner/web_test_delegate.h"
@@ skipping 170 matching lines @@
   void SetBluetoothFakeAdapter(const std::string& adapter_name,
                                v8::Local<v8::Function> callback);
   void SetBluetoothManualChooser(bool enable);
   void SetCanOpenWindows();
   void SetCloseRemainingWindowsWhenComplete(gin::Arguments* args);
   void SetColorProfile(const std::string& name,
                        v8::Local<v8::Function> callback);
   void SetCustomPolicyDelegate(gin::Arguments* args);
   void SetCustomTextOutput(const std::string& output);
   void SetDatabaseQuota(int quota);
+  void SetDisallowedSubresourcePathSuffixes(
+      const std::vector<std::string>& suffixes);
   void SetDomainRelaxationForbiddenForURLScheme(bool forbidden,
                                                 const std::string& scheme);
   void SetGeofencingMockPosition(double latitude, double longitude);
   void SetGeofencingMockProvider(bool service_available);
   void SetImagesAllowed(bool allowed);
   void SetInterceptPostMessage(bool value);
   void SetIsolatedWorldContentSecurityPolicy(int world_id,
                                              const std::string& policy);
   void SetIsolatedWorldSecurityOrigin(int world_id,
                                       v8::Local<v8::Value> origin);
@@ skipping 140 matching lines @@
                  &TestRunnerBindings::CopyImageAtAndCapturePixelsAsyncThen)
       .SetMethod("didAcquirePointerLock",
                  &TestRunnerBindings::DidAcquirePointerLock)
       .SetMethod("didLosePointerLock", &TestRunnerBindings::DidLosePointerLock)
       .SetMethod("didNotAcquirePointerLock",
                  &TestRunnerBindings::DidNotAcquirePointerLock)
       .SetMethod("disableAutoResizeMode",
                  &TestRunnerBindings::DisableAutoResizeMode)
       .SetMethod("disableMockScreenOrientation",
                  &TestRunnerBindings::DisableMockScreenOrientation)
+      .SetMethod("setDisallowedSubresourcePathSuffixes",
+                 &TestRunnerBindings::SetDisallowedSubresourcePathSuffixes)
       .SetMethod("dispatchBeforeInstallPromptEvent",
                  &TestRunnerBindings::DispatchBeforeInstallPromptEvent)
       .SetMethod("dumpAsMarkup", &TestRunnerBindings::DumpAsMarkup)
       .SetMethod("dumpAsText", &TestRunnerBindings::DumpAsText)
       .SetMethod("dumpAsTextWithPixelResults",
                  &TestRunnerBindings::DumpAsTextWithPixelResults)
       .SetMethod("dumpBackForwardList",
                  &TestRunnerBindings::DumpBackForwardList)
       .SetMethod("dumpChildFrameScrollPositions",
                  &TestRunnerBindings::DumpChildFrameScrollPositions)
@@ skipping 90 matching lines @@
       .SetMethod("queueReload", &TestRunnerBindings::QueueReload)
       .SetMethod("removeOriginAccessWhitelistEntry",
                  &TestRunnerBindings::RemoveOriginAccessWhitelistEntry)
       .SetMethod("removeWebPageOverlay",
                  &TestRunnerBindings::RemoveWebPageOverlay)
       .SetMethod("resetDeviceLight", &TestRunnerBindings::ResetDeviceLight)
       .SetMethod("resetTestHelperControllers",
                  &TestRunnerBindings::ResetTestHelperControllers)
       .SetMethod("resolveBeforeInstallPromptPromise",
                  &TestRunnerBindings::ResolveBeforeInstallPromptPromise)
-      .SetMethod("runIdleTasks",
-                 &TestRunnerBindings::RunIdleTasks)
+      .SetMethod("runIdleTasks", &TestRunnerBindings::RunIdleTasks)
       .SetMethod("selectionAsMarkup", &TestRunnerBindings::SelectionAsMarkup)
 
       // The Bluetooth functions are specified at
       // https://webbluetoothcg.github.io/web-bluetooth/tests/.
       .SetMethod("sendBluetoothManualChooserEvent",
                  &TestRunnerBindings::SendBluetoothManualChooserEvent)
       .SetMethod("setAcceptLanguages", &TestRunnerBindings::SetAcceptLanguages)
       .SetMethod("setAllowDisplayOfInsecureContent",
                  &TestRunnerBindings::SetAllowDisplayOfInsecureContent)
       .SetMethod("setAllowFileAccessFromFileURLs",
                  &TestRunnerBindings::SetAllowFileAccessFromFileURLs)
       .SetMethod("setAllowRunningOfInsecureContent",
                  &TestRunnerBindings::SetAllowRunningOfInsecureContent)
-      .SetMethod("setAutoplayAllowed",
-                 &TestRunnerBindings::SetAutoplayAllowed)
+      .SetMethod("setAutoplayAllowed", &TestRunnerBindings::SetAutoplayAllowed)
       .SetMethod("setAllowUniversalAccessFromFileURLs",
                  &TestRunnerBindings::SetAllowUniversalAccessFromFileURLs)
       .SetMethod("setAlwaysAcceptCookies",
                  &TestRunnerBindings::SetAlwaysAcceptCookies)
       .SetMethod("setAudioData", &TestRunnerBindings::SetAudioData)
       .SetMethod("setBackingScaleFactor",
                  &TestRunnerBindings::SetBackingScaleFactor)
       // The Bluetooth functions are specified at
       // https://webbluetoothcg.github.io/web-bluetooth/tests/.
       .SetMethod("setBluetoothFakeAdapter",
@@ skipping 428 matching lines @@
     return;
 
   runner_->SetMockScreenOrientation(orientation);
 }
 
 void TestRunnerBindings::DisableMockScreenOrientation() {
   if (runner_)
     runner_->DisableMockScreenOrientation();
 }
 
+void TestRunnerBindings::SetDisallowedSubresourcePathSuffixes(
+    const std::vector<std::string>& suffixes) {
+  if (runner_)
+    runner_->SetDisallowedSubresourcePathSuffixes(suffixes);
+}
+
 void TestRunnerBindings::DidAcquirePointerLock() {
   if (view_runner_)
     view_runner_->DidAcquirePointerLock();
 }
 
 void TestRunnerBindings::DidNotAcquirePointerLock() {
   if (view_runner_)
     view_runner_->DidNotAcquirePointerLock();
 }
 
@@ skipping 1579 matching lines @@
   layout_test_runtime_flags_.set_autoplay_allowed(allowed);
   OnLayoutTestRuntimeFlagsChanged();
 }
 
 void TestRunner::DumpPermissionClientCallbacks() {
   layout_test_runtime_flags_.set_dump_web_content_settings_client_callbacks(
       true);
   OnLayoutTestRuntimeFlagsChanged();
 }
 
+void TestRunner::SetDisallowedSubresourcePathSuffixes(
+    const std::vector<std::string>& suffixes) {
+  DCHECK(main_view_);
+  WebDataSource* data_source = main_view_->mainFrame()->dataSource();
+  data_source->setSubresourceFilter(

[Mike West, 2016/06/01 13:43:17]

Nit: You don't really need |data_source| here.

[engedy, 2016/06/02 23:07:19]

Done.

+      new MockWebDocumentSubresourceFilter(suffixes));
+}
+
 void TestRunner::DumpWindowStatusChanges() {
   dump_window_status_changes_ = true;
 }
 
 void TestRunner::DumpSpellCheckCallbacks() {
   dump_spell_check_callbacks_ = true;
 }
 
 void TestRunner::DumpBackForwardList() {
   dump_back_forward_list_ = true;
@@ skipping 221 matching lines @@
 
 void TestRunner::CompleteNotifyDone() {
   if (layout_test_runtime_flags_.wait_until_done() && !topLoadingFrame() &&
       work_queue_.is_empty())
     delegate_->TestFinished();
   layout_test_runtime_flags_.set_wait_until_done(false);
   OnLayoutTestRuntimeFlagsChanged();
 }
 
 }  // namespace test_runner