Small modifications to safebrowsing code to make it simpler to add the extension

chrome/browser/prerender/prerender_browsertest.cc

 // 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.
 
 #include <deque>
 
 #include "base/command_line.h"
 #include "base/path_service.h"
 #include "base/string_util.h"
 #include "base/stringprintf.h"
@@ skipping 484 matching lines @@
 
   void SetThreatTypeForUrl(const GURL& url, SBThreatType threat_type) {
     url_ = url;
     threat_type_ = threat_type;
   }
 
  private:
   virtual ~FakeSafeBrowsingDatabaseManager() {}
 
   void OnCheckBrowseURLDone(const GURL& gurl, Client* client) {
-    SafeBrowsingDatabaseManager::SafeBrowsingCheck check;
+    SafeBrowsingDatabaseManager::SafeBrowsingCheck check(
+        safe_browsing_util::MALWARE);
     check.urls.push_back(gurl);
     check.client = client;
-    check.threat_type = threat_type_;
+    check.url_threats[gurl] = threat_type_;
     client->OnSafeBrowsingResult(check);
   }
 
   GURL url_;
   SBThreatType threat_type_;
   DISALLOW_COPY_AND_ASSIGN(FakeSafeBrowsingDatabaseManager);
 };
 
 class FakeSafeBrowsingService : public SafeBrowsingService {
  public:
@@ skipping 1988 matching lines @@
   channel_close_watcher.WatchChannel(
       chrome::GetActiveWebContents(browser())->GetRenderProcessHost());
   NavigateToDestURL();
   channel_close_watcher.WaitForChannelClose();
 
   ASSERT_TRUE(IsEmptyPrerenderLinkManager());
   ASSERT_TRUE(catcher.GetNextResult()) << catcher.message();
 }
 
 }  // namespace prerender