Un-refcount AutofillWebData and TokenWebData

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/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_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 "components/webdata/common/web_data_service_test_util.h"
+#include "components/webdata/common/web_database_service.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/test/test_browser_thread.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/google/google_util_chromeos.h"
 #endif
 
 using content::BrowserThread;
@@ skipping 15 matching lines @@
   // pending requests on the thread.
   BrowserThread::PostTask(
       identifier,
       FROM_HERE,
       base::Bind(&QuitCallback, base::MessageLoop::current()));
   base::MessageLoop::current()->Run();
 }
 
 }  // namespace
 
+class MockWebDataServiceWrapperDB : public MockWebDataServiceWrapper {
+ public:
+  static BrowserContextKeyedService* Build(content::BrowserContext* profile) {
+    return new MockWebDataServiceWrapperDB(static_cast<Profile*>(profile));
+  }
+
+  explicit MockWebDataServiceWrapperDB(Profile* profile)
+      : MockWebDataServiceWrapper(NULL, NULL, NULL) {
+    base::FilePath profile_path = profile->GetPath();
+    base::FilePath path = profile_path.AppendASCII("TestWebDB");
+    web_database_ = new WebDatabaseService(path);
+    web_database_->AddTable(scoped_ptr<WebDatabaseTable>(new KeywordTable()));
+    fake_web_data_ = new WebDataService(
+        web_database_, WebDataServiceBase::ProfileErrorCallback());
+    fake_web_data_->Init();
+  }
+
+  virtual void Shutdown() OVERRIDE {
+    fake_web_data_->ShutdownDatabase();
+  }
+
+ private:
+  scoped_refptr<WebDatabaseService> web_database_;
+  DISALLOW_COPY_AND_ASSIGN(MockWebDataServiceWrapperDB);
+};
+
 // Trivial subclass of TemplateURLService that records the last invocation of
 // SetKeywordSearchTermsForURL.
 class TestingTemplateURLService : public TemplateURLService {
  public:
   static BrowserContextKeyedService* Build(content::BrowserContext* profile) {
     return new TestingTemplateURLService(static_cast<Profile*>(profile));
   }
 
   explicit TestingTemplateURLService(Profile* profile)
       : TemplateURLService(profile) {
@@ skipping 26 matching lines @@
 }
 
 TemplateURLServiceTestUtil::~TemplateURLServiceTestUtil() {
 }
 
 void TemplateURLServiceTestUtil::SetUp() {
   // Make unique temp directory.
   ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
   profile_.reset(new TestingProfile(temp_dir_.path()));
   db_thread_.Start();
-  profile_->CreateWebDataService();
+
+  WebDataServiceFactory::GetInstance()->SetTestingFactory(
+      profile_.get(), MockWebDataServiceWrapperDB::Build);
 
   TemplateURLService* service = static_cast<TemplateURLService*>(
       TemplateURLServiceFactory::GetInstance()->SetTestingFactoryAndUse(
           profile_.get(), TestingTemplateURLService::Build));
   service->AddObserver(this);
 
 #if defined(OS_CHROMEOS)
   google_util::chromeos::ClearBrandForCurrentSession();
 #endif
 }
@@ skipping 165 matching lines @@
   return profile_.get();
 }
 
 void TemplateURLServiceTestUtil::StartIOThread() {
   io_thread_.StartIOThread();
 }
 
 void TemplateURLServiceTestUtil::PumpLoop() {
   message_loop_.RunUntilIdle();
 }