Add RegisterMountPointProvider and TestMountPointProvider for testing

webkit/fileapi/file_system_context.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 "webkit/fileapi/file_system_context.h"
 
 #include "base/bind.h"
 #include "base/file_util.h"
+#include "base/stl_util.h"
 #include "base/single_thread_task_runner.h"
 #include "googleurl/src/gurl.h"
 #include "webkit/fileapi/file_system_file_util.h"
 #include "webkit/fileapi/file_system_operation_interface.h"
 #include "webkit/fileapi/file_system_options.h"
 #include "webkit/fileapi/file_system_quota_client.h"
 #include "webkit/fileapi/file_system_util.h"
 #include "webkit/fileapi/isolated_mount_point_provider.h"
 #include "webkit/fileapi/sandbox_mount_point_provider.h"
 #include "webkit/quota/quota_manager.h"
@@ skipping 70 matching lines @@
       type == fileapi::kFileSystemTypePersistent) {
     DCHECK(sandbox_provider());
     return sandbox_provider()->DeleteOriginDataOnFileThread(
         quota_manager_proxy(), origin_url, type);
   }
   return false;
 }
 
 FileSystemQuotaUtil*
 FileSystemContext::GetQuotaUtil(FileSystemType type) const {
-  if (type == fileapi::kFileSystemTypeTemporary ||
-      type == fileapi::kFileSystemTypePersistent) {
-    DCHECK(sandbox_provider());
-    return sandbox_provider()->quota_util();
-  }
-  return NULL;
+  FileSystemMountPointProvider* mount_point_provider =
+      GetMountPointProvider(type);
+  if (!mount_point_provider)
+    return NULL;
+  return mount_point_provider->GetQuotaUtil();
 }
 
 FileSystemFileUtil* FileSystemContext::GetFileUtil(
     FileSystemType type) const {
   FileSystemMountPointProvider* mount_point_provider =
       GetMountPointProvider(type);
   if (!mount_point_provider)
     return NULL;
   return mount_point_provider->GetFileUtil();
 }
 
 FileSystemMountPointProvider* FileSystemContext::GetMountPointProvider(
     FileSystemType type) const {
   switch (type) {
     case kFileSystemTypeTemporary:
     case kFileSystemTypePersistent:
       return sandbox_provider_.get();
     case kFileSystemTypeExternal:
       return external_provider_.get();
     case kFileSystemTypeIsolated:
       return isolated_provider_.get();
+    default:
+      if (provider_map_.find(type) != provider_map_.end())
+        return provider_map_.find(type)->second;
+      // Fall through.
     case kFileSystemTypeUnknown:
-    default:
       NOTREACHED();
       return NULL;
   }
 }
 
 SandboxMountPointProvider*
 FileSystemContext::sandbox_provider() const {
   return sandbox_provider_.get();
 }
 
@@ skipping 47 matching lines @@
   FilePath file_path;
   if (!CrackFileSystemURL(url, &origin_url, &file_system_type, &file_path))
     return NULL;
   FileSystemMountPointProvider* mount_point_provider =
       GetMountPointProvider(file_system_type);
   if (!mount_point_provider)
     return NULL;
   return mount_point_provider->CreateFileReader(url, offset, this);
 }
 
+void FileSystemContext::RegisterMountPointProvider(
+    FileSystemType type,
+    FileSystemMountPointProvider* provider) {
+  DCHECK(provider);
+  DCHECK(provider_map_.find(type) == provider_map_.end());
+  provider_map_[type] = provider;
+}
+
 FileSystemContext::~FileSystemContext() {}
 
 void FileSystemContext::DeleteOnCorrectThread() const {
   if (!io_task_runner_->RunsTasksOnCurrentThread() &&
       io_task_runner_->DeleteSoon(FROM_HERE, this)) {
     return;
   }
+  STLDeleteContainerPairSecondPointers(provider_map_.begin(),
+                                       provider_map_.end());
   delete this;
 }
 
 }  // namespace fileapi