Add tests for redirect responses from SafeBrowsingProtocolManager.

chrome/browser/safe_browsing/safe_browsing_service.h

 // Copyright (c) 2012 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.
 //
 // The Safe Browsing service is responsible for downloading anti-phishing and
 // anti-malware tables and checking urls against them.
 
 #ifndef CHROME_BROWSER_SAFE_BROWSING_SAFE_BROWSING_SERVICE_H_
 #define CHROME_BROWSER_SAFE_BROWSING_SAFE_BROWSING_SERVICE_H_
 
@@ skipping 351 matching lines @@
 
   // Called on the database thread to retrieve chunks.
   void GetAllChunksFromDatabase(GetChunksCallback callback);
 
   // Called on the IO thread with the results of all chunks.
   void OnGetAllChunksFromDatabase(const std::vector<SBListChunkRanges>& lists,
                                   bool database_error,
                                   GetChunksCallback callback);
 
   // Called on the IO thread after the database reports that it added a chunk.
-  void OnChunkInserted();
+  void OnAddChunksComplete(AddChunksCallback callback);
 
   // Notification that the database is done loading its bloom filter.  We may
   // have had to queue checks until the database is ready, and if so, this
   // checks them.
   void DatabaseLoadComplete();
 
   // Called on the database thread to add/remove chunks and host keys.
   // Callee will free the data when it's done.
   void HandleChunkForDatabase(const std::string& list,
-                              SBChunkList* chunks);
+                              SBChunkList* chunks,
+                              AddChunksCallback callback);
 
   void DeleteDatabaseChunks(std::vector<SBChunkDelete>* chunk_deletes);
 
   static SBThreatType GetThreatTypeFromListname(const std::string& list_name);
 
   void NotifyClientBlockingComplete(Client* client, bool proceed);
 
   void DatabaseUpdateFinished(bool update_succeeded);
 
   void NotifyDatabaseUpdateFinished(bool update_succeeded);
@@ skipping 73 matching lines @@
 
   // PrefObserver override
   virtual void OnPreferenceChanged(PrefServiceBase* service,
                                    const std::string& pref_name) OVERRIDE;
 
   // SafeBrowsingProtocolManageDelegate override
   virtual void ResetDatabase() OVERRIDE;
   virtual void UpdateStarted() OVERRIDE;
   virtual void UpdateFinished(bool success) OVERRIDE;
   virtual void GetChunks(GetChunksCallback callback) OVERRIDE;
-  virtual void AddChunks(const std::string& list, SBChunkList* chunks) OVERRIDE;
+  virtual void AddChunks(const std::string& list, SBChunkList* chunks,
+                         AddChunksCallback) OVERRIDE;
   virtual void DeleteChunks(
       std::vector<SBChunkDelete>* delete_chunks) OVERRIDE;
 
   // Starts following the safe browsing preference on |pref_service|.
   void AddPrefService(PrefService* pref_service);
 
   // Stop following the safe browsing preference on |pref_service|.
   void RemovePrefService(PrefService* pref_service);
 
   // Checks if any profile is currently using the safe browsing service, and
@@ skipping 104 matching lines @@
 class SafeBrowsingServiceFactory {
  public:
   SafeBrowsingServiceFactory() { }
   virtual ~SafeBrowsingServiceFactory() { }
   virtual SafeBrowsingService* CreateSafeBrowsingService() = 0;
  private:
   DISALLOW_COPY_AND_ASSIGN(SafeBrowsingServiceFactory);
 };
 
 #endif  // CHROME_BROWSER_SAFE_BROWSING_SAFE_BROWSING_SERVICE_H_