Browser tests for using the new SafeBrowsing protocol (v4)

chrome/browser/safe_browsing/test_safe_browsing_service.h

Index: chrome/browser/safe_browsing/test_safe_browsing_service.h
diff --git a/chrome/browser/safe_browsing/test_safe_browsing_service.h b/chrome/browser/safe_browsing/test_safe_browsing_service.h
index facf791b83e2b0acad4529814245db6c0eeeb089..e5fbdc4336b7182c229c75957e2852c44aba3785 100644
--- a/chrome/browser/safe_browsing/test_safe_browsing_service.h
+++ b/chrome/browser/safe_browsing/test_safe_browsing_service.h
@@ -40,7 +40,8 @@ class TestSafeBrowsingUIManager;
 //   test_sb_factory_.
 class TestSafeBrowsingService : public SafeBrowsingService {
  public:
-  TestSafeBrowsingService();
+  explicit TestSafeBrowsingService(bool v4_enabled = false,
+                                   bool use_v4_only = false);
   // SafeBrowsingService overrides
   SafeBrowsingProtocolConfig GetProtocolConfig() const override;
   V4ProtocolConfig GetV4ProtocolConfig() const override;
@@ -80,7 +81,8 @@ class TestSafeBrowsingService : public SafeBrowsingService {
 
 class TestSafeBrowsingServiceFactory : public SafeBrowsingServiceFactory {
  public:
-  TestSafeBrowsingServiceFactory();
+  TestSafeBrowsingServiceFactory(bool v4_enabled = false,
+                                 bool use_v4_only = false);
   ~TestSafeBrowsingServiceFactory() override;
 
   // Creates test safe browsing service, and configures test UI manager,
@@ -101,6 +103,8 @@ class TestSafeBrowsingServiceFactory : public SafeBrowsingServiceFactory {
   scoped_refptr<TestSafeBrowsingDatabaseManager> test_database_manager_;
   scoped_refptr<TestSafeBrowsingUIManager> test_ui_manager_;
   SafeBrowsingProtocolConfig* test_protocol_config_;
+  bool use_v4_only_;
+  bool v4_enabled_;
 
   DISALLOW_COPY_AND_ASSIGN(TestSafeBrowsingServiceFactory);
 };