Create test fixture for SafeBrowsingService

chrome/test/ppapi/ppapi_filechooser_browsertest.cc

 // Copyright 2015 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/bind.h"
 #include "base/command_line.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/thread_task_runner_handle.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/ppapi/ppapi_test.h"
 #include "ppapi/shared_impl/test_utils.h"
 #include "ui/shell_dialogs/select_file_dialog.h"
 #include "ui/shell_dialogs/select_file_dialog_factory.h"
 #include "ui/shell_dialogs/selected_file_info.h"
 
 #if defined(FULL_SAFE_BROWSING)
-#include "chrome/browser/safe_browsing/safe_browsing_service.h"
+#include "chrome/browser/safe_browsing/test_safe_browsing_service.h"
 #include "components/safe_browsing_db/test_database_manager.h"
 #endif
 
 using safe_browsing::SafeBrowsingService;
 
 namespace {
 
 class TestSelectFileDialogFactory final : public ui::SelectFileDialogFactory {
  public:
   using SelectedFileInfoList = std::vector<ui::SelectedFileInfo>;
@@ skipping 102 matching lines @@
   bool MatchDownloadWhitelistUrl(const GURL& url) override {
     // This matches the behavior in RunTestViaHTTP().
     return url.SchemeIsHTTPOrHTTPS() && url.has_path() &&
            url.path().find("/test_case.html") == 0;
   }
 
  protected:
   ~FakeDatabaseManager() override {}
 };
 
-class TestSafeBrowsingService : public SafeBrowsingService {
- public:
-  safe_browsing::SafeBrowsingDatabaseManager* CreateDatabaseManager() override {
-    return new FakeDatabaseManager();
-  }
-
- protected:
-  ~TestSafeBrowsingService() override {}
-
-  safe_browsing::SafeBrowsingProtocolManagerDelegate*
-      GetProtocolManagerDelegate() override {
-    // Our FakeDatabaseManager doesn't implement this delegate.
-    return NULL;
-  }
-};
-
-class TestSafeBrowsingServiceFactory
-    : public safe_browsing::SafeBrowsingServiceFactory {
- public:
-  SafeBrowsingService* CreateSafeBrowsingService() override {
-    SafeBrowsingService* service = new TestSafeBrowsingService();
-    return service;
-  }
-};
-
 class PPAPIFileChooserTest : public OutOfProcessPPAPITest {};
 
 class PPAPIFileChooserTestWithSBService : public PPAPIFileChooserTest {
  public:
   void SetUp() override {
+    safe_browsing_service_factory_.SetTestDatabaseManager(
+        new FakeDatabaseManager());
     SafeBrowsingService::RegisterFactory(&safe_browsing_service_factory_);
     PPAPIFileChooserTest::SetUp();
   }
+
   void TearDown() override {
     PPAPIFileChooserTest::TearDown();
     SafeBrowsingService::RegisterFactory(nullptr);
   }
 
- private:
-  TestSafeBrowsingServiceFactory safe_browsing_service_factory_;
+ protected:
+  safe_browsing::TestSafeBrowsingServiceFactory safe_browsing_service_factory_;
 };
 
 }  // namespace
 
 IN_PROC_BROWSER_TEST_F(PPAPIFileChooserTest, FileChooser_Open_Success) {
   const char kContents[] = "Hello from browser";
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
 
   base::FilePath existing_filename = temp_dir.path().AppendASCII("foo");
@@ skipping 150 matching lines @@
   RunTestViaHTTP("FileChooser_SaveAsDangerousExecutableAllowed");
   base::FilePath actual_filename = temp_dir.path().AppendASCII("dangerous.exe");
 
   ASSERT_TRUE(base::PathExists(actual_filename));
   std::string file_contents;
   ASSERT_TRUE(base::ReadFileToString(actual_filename, &file_contents));
   EXPECT_EQ("Hello from PPAPI", file_contents);
 }
 
 #endif  // FULL_SAFE_BROWSING