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 b426847bc0808342c3bf9a84abfbbb80762d96a6..6c9377cccdc1da5139ed28c2a9da56aaa1c03424 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,16 +63,26 @@ void ContentSubresourceFilterDriverFactory::
         safe_browsing::ThreatPatternType threat_type) {
   if (threat_type != safe_browsing::ThreatPatternType::SOCIAL_ENGINEERING_ADS)
     return;
-  activate_on_origins_.insert(url.host());
+  AddToSocEngList(url);
   for (const auto& url : redirect_urls) {
-    activate_on_origins_.insert(url.host());
+    AddToSocEngList(url);
   }
 }
 
+void ContentSubresourceFilterDriverFactory::AddToSocEngList(const GURL& url) {
+  activate_on_origins_.insert(url.host());
+}
+
 bool ContentSubresourceFilterDriverFactory::IsURLInSocEngList(const GURL& url) {
   return GetSocEngList().find(url.host()) != GetSocEngList().end();
 }
 
+void ContentSubresourceFilterDriverFactory::TestingSetDriverForFrame(
+    content::RenderFrameHost* render_frame_host,
+    std::unique_ptr<ContentSubresourceFilterDriver> driver) {
+  frame_drivers_.insert(std::make_pair(render_frame_host, std::move(driver)));
+}
+
 ContentSubresourceFilterDriver*
 ContentSubresourceFilterDriverFactory::DriverFromFrameHost(
     content::RenderFrameHost* render_frame_host) {