Adding QuotaManager logic to TestingProfile.

chrome/test/testing_profile.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 "chrome/test/testing_profile.h"
 
 #include "build/build_config.h"
 
 #include "base/base_paths.h"
 #include "base/command_line.h"
@@ skipping 33 matching lines @@
 #include "chrome/browser/ui/find_bar/find_bar_state.h"
 #include "chrome/browser/ui/webui/chrome_url_data_manager.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/bookmark_load_observer.h"
 #include "chrome/test/test_url_request_context_getter.h"
 #include "chrome/test/testing_pref_service.h"
 #include "chrome/test/ui_test_utils.h"
 #include "content/browser/browser_thread.h"
+#include "content/browser/file_system/browser_file_system_helper.h"
 #include "content/browser/in_process_webkit/webkit_context.h"
 #include "content/browser/mock_resource_context.h"
 #include "content/common/notification_service.h"
 #include "net/base/cookie_monster.h"
 #include "net/url_request/url_request_context.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "webkit/database/database_tracker.h"
 #include "webkit/fileapi/file_system_context.h"
@@ skipping 321 matching lines @@
 }
 
 bool TestingProfile::HasOffTheRecordProfile() {
   return incognito_profile_.get() != NULL;
 }
 
 Profile* TestingProfile::GetOriginalProfile() {
   return this;
 }
 
-void TestingProfile::SetAppCacheService(
-    ChromeAppCacheService* appcache_service) {
-  appcache_service_ = appcache_service;
-}
-
 ChromeAppCacheService* TestingProfile::GetAppCacheService() {
+  CreateQuotaManagerAndClients();
   return appcache_service_.get();
 }
 
 webkit_database::DatabaseTracker* TestingProfile::GetDatabaseTracker() {
-  if (!db_tracker_) {
-    db_tracker_ = new webkit_database::DatabaseTracker(
-        GetPath(), false, false, GetExtensionSpecialStoragePolicy(),
-        NULL, NULL);
-  }
+  CreateQuotaManagerAndClients();
   return db_tracker_;
 }
 
 VisitedLinkMaster* TestingProfile::GetVisitedLinkMaster() {
   return NULL;
 }
 
 ExtensionService* TestingProfile::GetExtensionService() {
   return extension_service_.get();
 }
@@ skipping 107 matching lines @@
 
 DownloadManager* TestingProfile::GetDownloadManager() {
   return NULL;
 }
 
 PersonalDataManager* TestingProfile::GetPersonalDataManager() {
   return NULL;
 }
 
 fileapi::FileSystemContext* TestingProfile::GetFileSystemContext() {
-  if (!file_system_context_) {
-    file_system_context_ = new fileapi::FileSystemContext(
-      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE),
-      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO),
-      GetExtensionSpecialStoragePolicy(),
-      NULL,
-      GetPath(),
-      IsOffTheRecord(),
-      true,  // Allow file access from files.
-      true,  // Unlimited quota.
-      NULL);
-  }
+  CreateQuotaManagerAndClients();
   return file_system_context_.get();
 }
 
 quota::QuotaManager* TestingProfile::GetQuotaManager() {
-  return NULL;
+  CreateQuotaManagerAndClients();
+  return quota_manager_.get();
+}
+
+void TestingProfile::CreateQuotaManagerAndClients() {
+  if (quota_manager_.get()) {
+    DCHECK(appcache_service_.get());
+    DCHECK(db_tracker_.get());
+    DCHECK(file_system_context_.get());
+    DCHECK(webkit_context_.get());
+    return;
+  }
+
+  // All the clients have to be created and registered with the
+  // quota manager prior to the manager being used. So, we do them
+  // all together here prior to handing our a reference to anything
+  // that uses the quota manager.
+  quota_manager_ = new quota::QuotaManager(
+      IsOffTheRecord(),
+      GetPath(),
+      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO),
+      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB),
+      GetExtensionSpecialStoragePolicy());
+
+  // Each consumer is responsible for registering its QuotaClient during
+  // its construction.
+  file_system_context_ = CreateFileSystemContext(
+      GetPath(), IsOffTheRecord(),
+      GetExtensionSpecialStoragePolicy(),
+      quota_manager_->proxy());
+  db_tracker_ = new webkit_database::DatabaseTracker(
+      GetPath(), IsOffTheRecord(), false,
+      GetExtensionSpecialStoragePolicy(),
+      quota_manager_->proxy(),
+      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE));
+  webkit_context_ = new WebKitContext(
+      IsOffTheRecord(), GetPath(), GetExtensionSpecialStoragePolicy(),
+      false, quota_manager_->proxy(),
+      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::WEBKIT));
+  appcache_service_ = new ChromeAppCacheService(quota_manager_->proxy());
+  BrowserThread::PostTask(
+      BrowserThread::IO, FROM_HERE,
+      NewRunnableMethod(
+          appcache_service_.get(),
+          &ChromeAppCacheService::InitializeOnIOThread,
+          IsOffTheRecord()
+              ? FilePath() : GetPath().Append(chrome::kAppCacheDirname),
+          &GetResourceContext(),
+          make_scoped_refptr(GetExtensionSpecialStoragePolicy()),
+          false));
+
+  // Block until the posted task is finished, otherwise the appcache service
+  // won't be properly initialized.
+  MessageLoop::current()->RunAllPending();
 }
 
 BrowserSignin* TestingProfile::GetBrowserSignin() {
   return NULL;
 }
 
 bool TestingProfile::HasCreatedDownloadManager() const {
   return false;
 }
 
@@ skipping 121 matching lines @@
 
 ProtocolHandlerRegistry* TestingProfile::GetProtocolHandlerRegistry() {
   return protocol_handler_registry_.get();
 }
 
 SpellCheckHost* TestingProfile::GetSpellCheckHost() {
   return NULL;
 }
 
 WebKitContext* TestingProfile::GetWebKitContext() {
-  if (webkit_context_ == NULL) {
-    webkit_context_ = new WebKitContext(
-          IsOffTheRecord(), GetPath(),
-          GetExtensionSpecialStoragePolicy(),
-          false, NULL, NULL);
-  }
+  CreateQuotaManagerAndClients();
   return webkit_context_;
 }
 
 WebKitContext* TestingProfile::GetOffTheRecordWebKitContext() {
   return NULL;
 }
 
 FilePath TestingProfile::last_selected_directory() {
   return last_selected_directory_;
 }
@@ skipping 82 matching lines @@
 quota::SpecialStoragePolicy* TestingProfile::GetSpecialStoragePolicy() {
   return GetExtensionSpecialStoragePolicy();
 }
 
 void TestingProfile::DestroyWebDataService() {
   if (!web_data_service_.get())
     return;
 
   web_data_service_->Shutdown();
 }