Make safe browsing work in a multi-profile environment.

chrome/browser/download/download_safe_browsing_client_unittest.cc

 // Copyright (c) 2011 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 "base/message_loop.h"
 #include "chrome/browser/download/download_safe_browsing_client.h"
 #include "chrome/browser/safe_browsing/safe_browsing_service.h"
 #include "content/browser/browser_thread.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 const char* kUrl1 = "http://127.0.0.1/";
 const char* kUrl2 = "http://127.0.0.1/two";
 const char* kUrl3 = "http://127.0.0.1/three";
 const char* kRefUrl = "http://127.0.0.1/referrer";
 
 class MockSafeBrowsingService : public SafeBrowsingService {
  public:
   MockSafeBrowsingService() {}
   virtual ~MockSafeBrowsingService() {}
 
-  MOCK_METHOD6(ReportSafeBrowsingHit,
+  MOCK_METHOD7(ReportSafeBrowsingHit,
                void(const GURL&, const GURL&, const GURL&, bool, UrlCheckResult,
-                    const std::string&));
+                    const std::string&, bool));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockSafeBrowsingService);
 };
 
 }  // namespace
 
 class DownloadSBClientTest : public testing::Test {
  public:
   DownloadSBClientTest() :
       ui_thread_(BrowserThread::UI, &loop_) {
   }
 
  private:
   MessageLoop loop_;
 
   // UI thread to satisfy debug checks in DownloadSBClient.
   BrowserThread ui_thread_;
 };
 
 TEST_F(DownloadSBClientTest, UrlHit) {
   std::string expected_post =
       std::string(kUrl1) + "\n" + kUrl2 + "\n" + kUrl3 + "\n";
   scoped_refptr<MockSafeBrowsingService> sb_service(
       new MockSafeBrowsingService);
   EXPECT_CALL(*sb_service.get(),
       ReportSafeBrowsingHit(GURL(kUrl3), GURL(kUrl1), GURL(kRefUrl), true,
                             SafeBrowsingService::BINARY_MALWARE_URL,
-                            expected_post));
+                            expected_post, true));
   std::vector<GURL> url_chain;
   url_chain.push_back(GURL(kUrl1));
   url_chain.push_back(GURL(kUrl2));
   url_chain.push_back(GURL(kUrl3));
 
   scoped_refptr<DownloadSBClient> client(
-      new DownloadSBClient(1, url_chain, GURL(kRefUrl)));
+      new DownloadSBClient(1, url_chain, GURL(kRefUrl), true));
   client->SetSBService(sb_service.get());
 
   client->ReportMalware(SafeBrowsingService::BINARY_MALWARE_URL, "");
 }
 
 TEST_F(DownloadSBClientTest, DigestHit) {
   std::string hash_data = "\xDE\xAD\xBE\xEF";
   std::string hash_string = "DEADBEEF";
   std::string expected_post =
       hash_string + "\n" + kUrl1 + "\n" + kUrl2 + "\n" + kUrl3 + "\n";
   scoped_refptr<MockSafeBrowsingService> sb_service(
       new MockSafeBrowsingService);
   EXPECT_CALL(*sb_service.get(),
       ReportSafeBrowsingHit(GURL(kUrl3), GURL(kUrl1), GURL(kRefUrl), true,
                             SafeBrowsingService::BINARY_MALWARE_HASH,
-                            expected_post));
+                            expected_post, true));
   std::vector<GURL> url_chain;
   url_chain.push_back(GURL(kUrl1));
   url_chain.push_back(GURL(kUrl2));
   url_chain.push_back(GURL(kUrl3));
 
   scoped_refptr<DownloadSBClient> client(
-      new DownloadSBClient(1, url_chain, GURL(kRefUrl)));
+      new DownloadSBClient(1, url_chain, GURL(kRefUrl), true));
   client->SetSBService(sb_service.get());
 
   client->ReportMalware(SafeBrowsingService::BINARY_MALWARE_HASH, hash_data);
 }