Make WebDataService no longer depend on ProfileKeyedService

chrome/browser/search_engines/template_url_service_test_util.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 "chrome/browser/search_engines/template_url_service_test_util.h"
 
 #include "base/bind.h"
-#include "base/files/scoped_temp_dir.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/threading/thread.h"
 #include "chrome/browser/google/google_url_tracker.h"
 #include "chrome/browser/search_engines/search_terms_data.h"
 #include "chrome/browser/search_engines/template_url_service.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/webdata/web_data_service_factory.h"
+#include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/automation/value_conversion_util.h"
 #include "chrome/test/base/testing_pref_service_syncable.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/test/test_browser_thread.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 
@@ skipping 18 matching lines @@
 static void WaitForThreadToProcessRequests(BrowserThread::ID identifier) {
   // Schedule a task on the thread that is processed after all
   // pending requests on the thread.
   BrowserThread::PostTask(identifier, FROM_HERE,
                           base::Bind(&QuitCallback, MessageLoop::current()));
   MessageLoop::current()->Run();
 }
 
 }  // namespace
 
-// Subclass the TestingProfile so that it can return a WebDataService.
-class TemplateURLServiceTestingProfile : public TestingProfile {
- public:
-  TemplateURLServiceTestingProfile()
-      : TestingProfile(),
-        db_thread_(BrowserThread::DB),
-        io_thread_(BrowserThread::IO) {
-  }
-
-  void SetUp();
-  void TearDown();
-
-  // Starts the I/O thread. This isn't done automatically because not every test
-  // needs this.
-  void StartIOThread() {
-    io_thread_.StartIOThread();
-  }
-
-  static scoped_refptr<RefcountedProfileKeyedService>
-      GetWebDataServiceForTemplateURLServiceTestingProfile(Profile* profile);
-
- private:
-  scoped_refptr<WebDataService> service_;
-  base::ScopedTempDir temp_dir_;
-  content::TestBrowserThread db_thread_;
-  content::TestBrowserThread io_thread_;
-};
-
-
 // Trivial subclass of TemplateURLService that records the last invocation of
 // SetKeywordSearchTermsForURL.
 class TestingTemplateURLService : public TemplateURLService {
  public:
   static ProfileKeyedService* Build(Profile* profile) {
     return new TestingTemplateURLService(profile);
   }
 
   explicit TestingTemplateURLService(Profile* profile)
       : TemplateURLService(profile) {
@@ skipping 11 matching lines @@
                                            const string16& term) OVERRIDE {
     search_term_ = term;
   }
 
  private:
   string16 search_term_;
 
   DISALLOW_COPY_AND_ASSIGN(TestingTemplateURLService);
 };
 
-void TemplateURLServiceTestingProfile::SetUp() {
-  db_thread_.Start();
-
-  // Make unique temp directory.
-  ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
-
-  base::FilePath path = temp_dir_.path().AppendASCII("TestDataService.db");
-  service_ = new WebDataService;
-  service_->Init(path);
-}
-
-void TemplateURLServiceTestingProfile::TearDown() {
-  // Clear the request context so it will get deleted. This should be done
-  // before shutting down the I/O thread to avoid memory leaks.
-  ResetRequestContext();
-
-  // Wait for the delete of the request context to happen.
-  if (io_thread_.IsRunning())
-    TemplateURLServiceTestUtil::BlockTillIOThreadProcessesRequests();
-
-  // The I/O thread must be shutdown before the DB thread.
-  io_thread_.Stop();
-
-  // Clean up the test directory.
-  if (service_.get()) {
-    service_->ShutdownOnUIThread();
-    service_ = NULL;
-  }
-  // Note that we must ensure the DB thread is stopped after WDS
-  // shutdown (so it can commit pending transactions) but before
-  // deleting the test profile directory, otherwise we may not be
-  // able to delete it due to an open transaction.
-  // Schedule another task on the DB thread to notify us that it's safe to
-  // carry on with the test.
-  base::WaitableEvent done(false, false);
-  BrowserThread::PostTask(BrowserThread::DB, FROM_HERE,
-      base::Bind(&base::WaitableEvent::Signal, base::Unretained(&done)));
-  done.Wait();
-  db_thread_.Stop();
-}
-
-scoped_refptr<RefcountedProfileKeyedService>
-TemplateURLServiceTestingProfile::
-    GetWebDataServiceForTemplateURLServiceTestingProfile(Profile* profile) {
-  TemplateURLServiceTestingProfile* test_profile =
-      reinterpret_cast<TemplateURLServiceTestingProfile*>(profile);
-  return test_profile->service_;
-}
-
 TemplateURLServiceTestUtil::TemplateURLServiceTestUtil()
     : ui_thread_(BrowserThread::UI, &message_loop_),
+      db_thread_(BrowserThread::DB),
+      io_thread_(BrowserThread::IO),
       changed_count_(0) {
 }
 
 TemplateURLServiceTestUtil::~TemplateURLServiceTestUtil() {
 }
 
 void TemplateURLServiceTestUtil::SetUp() {
-  profile_.reset(new TemplateURLServiceTestingProfile());
-  WebDataServiceFactory::GetInstance()->SetTestingFactory(
-      profile_.get(), TemplateURLServiceTestingProfile::
-          GetWebDataServiceForTemplateURLServiceTestingProfile);
+  ASSERT_TRUE(db_thread_.Start());
+  // Make unique temp directory.
+  ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
 
-  profile_->SetUp();
+  profile_.reset(new TestingProfile(temp_dir_.path()));
+  profile_->CreateWebDataService();
+
   TemplateURLService* service = static_cast<TemplateURLService*>(
       TemplateURLServiceFactory::GetInstance()->SetTestingFactoryAndUse(
           profile_.get(), TestingTemplateURLService::Build));
   service->AddObserver(this);
 
 #if defined(OS_CHROMEOS)
   google_util::chromeos::ClearBrandForCurrentSession();
 #endif
 }
 
 void TemplateURLServiceTestUtil::TearDown() {
+  // Clear the request context so it will get deleted. This should be done
+  // before shutting down the I/O thread to avoid memory leaks.
+  profile_->ResetRequestContext();
+
+  // Wait for the delete of the request context to happen.
+  if (io_thread_.IsRunning())
+    TemplateURLServiceTestUtil::BlockTillIOThreadProcessesRequests();
+
+  // The I/O thread must be shutdown before the DB thread.
+  io_thread_.Stop();
+
   if (profile_.get()) {
-    profile_->TearDown();
     profile_.reset();
   }
+  base::WaitableEvent done(false, false);
+  BrowserThread::PostTask(BrowserThread::DB, FROM_HERE,
+      base::Bind(&base::WaitableEvent::Signal, base::Unretained(&done)));
+  done.Wait();
+  db_thread_.Stop();
   UIThreadSearchTermsData::SetGoogleBaseURL(std::string());
 
   // Flush the message loop to make application verifiers happy.
   message_loop_.RunUntilIdle();
 }
 
 void TemplateURLServiceTestUtil::OnTemplateURLServiceChanged() {
   changed_count_++;
 }
 
@@ skipping 18 matching lines @@
   model()->Load();
   BlockTillServiceProcessesRequests();
   EXPECT_EQ(1, GetObserverCount());
   ResetObserverCount();
 }
 
 void TemplateURLServiceTestUtil::ChangeModelToLoadState() {
   model()->ChangeToLoadedState();
   // Initialize the web data service so that the database gets updated with
   // any changes made.
-
-  model()->service_ = WebDataServiceFactory::GetForProfile(
-      profile_.get(), Profile::EXPLICIT_ACCESS);
+  model()->service_ = WebDataService::FromBrowserContext(profile_.get());
+  BlockTillServiceProcessesRequests();
 }
 
 void TemplateURLServiceTestUtil::ClearModel() {
   TemplateURLServiceFactory::GetInstance()->SetTestingFactory(
       profile_.get(), NULL);
 }
 
 void TemplateURLServiceTestUtil::ResetModel(bool verify_load) {
   TemplateURLServiceFactory::GetInstance()->SetTestingFactoryAndUse(
       profile_.get(), TestingTemplateURLService::Build);
@@ skipping 77 matching lines @@
 
 TemplateURLService* TemplateURLServiceTestUtil::model() const {
   return TemplateURLServiceFactory::GetForProfile(profile_.get());
 }
 
 TestingProfile* TemplateURLServiceTestUtil::profile() const {
   return profile_.get();
 }
 
 void TemplateURLServiceTestUtil::StartIOThread() {
-  profile_->StartIOThread();
+  io_thread_.StartIOThread();
 }
 
 void TemplateURLServiceTestUtil::PumpLoop() {
   message_loop_.RunUntilIdle();
 }