Navigation throttle for the Safe Browsing Subresource Filter.

components/subresource_filter/content/browser/content_subresource_filter_driver_factory.cc

Index: components/subresource_filter/content/browser/content_subresource_filter_driver_factory.cc
diff --git a/components/subresource_filter/content/browser/content_subresource_filter_driver_factory.cc b/components/subresource_filter/content/browser/content_subresource_filter_driver_factory.cc
index aec0b81d45c34b6a46b2514b98a008285002e338..882218804a2091509b05e81a2db8b0d3eaa1db9e 100644
--- a/components/subresource_filter/content/browser/content_subresource_filter_driver_factory.cc
+++ b/components/subresource_filter/content/browser/content_subresource_filter_driver_factory.cc
@@ -63,17 +63,31 @@ void ContentSubresourceFilterDriverFactory::
         safe_browsing::ThreatPatternType threat_type) {
   if (threat_type != safe_browsing::ThreatPatternType::SOCIAL_ENGINEERING_ADS)
     return;
-  activate_on_origins_.insert(url.host());
-  for (const auto& url : redirect_urls) {
-    activate_on_origins_.insert(url.host());
-  }
+  AddOriginOfURLToActivationSet(url);
+  for (const auto& url : redirect_urls)
+    AddOriginOfURLToActivationSet(url);
 }
 
 bool ContentSubresourceFilterDriverFactory::ShouldActivateForURL(
-    const GURL& url) {
+    const GURL& url) const {
   return activation_set().find(url.host()) != activation_set().end();
 }
 
+void ContentSubresourceFilterDriverFactory::AddOriginOfURLToActivationSet(
+    const GURL& url) {
+  activate_on_origins_.insert(url.host());
+}
+
+void ContentSubresourceFilterDriverFactory::SetDriverForFrameHostForTesting(
+    content::RenderFrameHost* render_frame_host,
+    std::unique_ptr<ContentSubresourceFilterDriver> driver) {
+  auto iterator_and_inserted =
+      frame_drivers_.insert(std::make_pair(render_frame_host, nullptr));
+  if (iterator_and_inserted.second) {

[engedy, 2016/07/19 11:27:02]

nit: Wouldn't it be better to have overwrite semantics here and drop the
conditional?

+    iterator_and_inserted.first->second = std::move(driver);
+  }
+}
+
 ContentSubresourceFilterDriver*
 ContentSubresourceFilterDriverFactory::DriverFromFrameHost(
     content::RenderFrameHost* render_frame_host) {