Browser tests for using the new SafeBrowsing protocol (v4)

chrome/browser/safe_browsing/test_safe_browsing_service.cc

Index: chrome/browser/safe_browsing/test_safe_browsing_service.cc
diff --git a/chrome/browser/safe_browsing/test_safe_browsing_service.cc b/chrome/browser/safe_browsing/test_safe_browsing_service.cc
index b836b18dec5c154330073c3c92b0fd9b87f05410..5cbf367571a56a16b38c234cc9f0863dc223437d 100644
--- a/chrome/browser/safe_browsing/test_safe_browsing_service.cc
+++ b/chrome/browser/safe_browsing/test_safe_browsing_service.cc
@@ -14,8 +14,10 @@
 namespace safe_browsing {
 
 // TestSafeBrowsingService functions:
-TestSafeBrowsingService::TestSafeBrowsingService()
-    : protocol_manager_delegate_disabled_(false),
+TestSafeBrowsingService::TestSafeBrowsingService(bool v4_enabled,
+                                                 bool use_v4_only)
+    : SafeBrowsingService(v4_enabled, use_v4_only),
+      protocol_manager_delegate_disabled_(false),
       serialized_download_report_(base::EmptyString()) {}
 
 TestSafeBrowsingService::~TestSafeBrowsingService() {}
@@ -90,15 +92,20 @@ void TestSafeBrowsingService::SetV4ProtocolConfig(
 }
 
 // TestSafeBrowsingServiceFactory functions:
-TestSafeBrowsingServiceFactory::TestSafeBrowsingServiceFactory()
-    : test_safe_browsing_service_(nullptr), test_protocol_config_(nullptr) {}
+TestSafeBrowsingServiceFactory::TestSafeBrowsingServiceFactory(bool v4_enabled,
+                                                               bool use_v4_only)
+    : test_safe_browsing_service_(nullptr),
+      test_protocol_config_(nullptr),
+      use_v4_only_(use_v4_only),
+      v4_enabled_(v4_enabled) {}
 
 TestSafeBrowsingServiceFactory::~TestSafeBrowsingServiceFactory() {}
 
 SafeBrowsingService*
 TestSafeBrowsingServiceFactory::CreateSafeBrowsingService() {
   // Instantiate TestSafeBrowsingService.
-  test_safe_browsing_service_ = new TestSafeBrowsingService();
+  test_safe_browsing_service_ =
+      new TestSafeBrowsingService(v4_enabled_, use_v4_only_);
   // Plug-in test member clases accordingly.
   if (test_ui_manager_)
     test_safe_browsing_service_->SetUIManager(test_ui_manager_.get());