Index: chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc |
diff --git a/chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc b/chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc |
index 0cd880cb8962be4604e3b0974a971e142ac16ec1..32e3e691635ed4279fca8bb804e6f0caa897026c 100644 |
--- a/chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc |
+++ b/chrome/browser/safe_browsing/browser_feature_extractor_unittest.cc |
@@ -18,6 +18,7 @@ |
#include "chrome/browser/safe_browsing/client_side_detection_host.h" |
#include "chrome/browser/safe_browsing/database_manager.h" |
#include "chrome/browser/safe_browsing/safe_browsing_service.h" |
+#include "chrome/browser/safe_browsing/test_database_manager.h" |
#include "chrome/browser/safe_browsing/ui_manager.h" |
#include "chrome/common/safe_browsing/csd.pb.h" |
#include "chrome/test/base/chrome_render_view_host_test_harness.h" |
@@ -47,11 +48,9 @@ namespace safe_browsing { |
namespace { |
-class MockSafeBrowsingDatabaseManager : public SafeBrowsingDatabaseManager { |
+class MockSafeBrowsingDatabaseManager : public TestSafeBrowsingDatabaseManager { |
public: |
- explicit MockSafeBrowsingDatabaseManager( |
- const scoped_refptr<SafeBrowsingService>& service) |
- : SafeBrowsingDatabaseManager(service) { } |
+ MockSafeBrowsingDatabaseManager() {} |
MOCK_METHOD1(MatchMalwareIP, bool(const std::string& ip_address)); |
@@ -84,8 +83,7 @@ class BrowserFeatureExtractorTest : public ChromeRenderViewHostTestHarness { |
ASSERT_TRUE(profile()->CreateHistoryService( |
true /* delete_file */, false /* no_db */)); |
- db_manager_ = new StrictMock<MockSafeBrowsingDatabaseManager>( |
- SafeBrowsingService::CreateSafeBrowsingService()); |
+ db_manager_ = new StrictMock<MockSafeBrowsingDatabaseManager>(); |
host_.reset(new StrictMock<MockClientSideDetectionHost>( |
web_contents(), db_manager_.get())); |
extractor_.reset( |