blob: f63ccb29e5d1db491147d9d250373c7b3113ac1b [file] [log] [blame]
// Copyright 2013 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 "storage/browser/test/sandbox_file_system_test_helper.h"
#include <memory>
#include "base/files/file_util.h"
#include "base/run_loop.h"
#include "storage/browser/fileapi/file_system_context.h"
#include "storage/browser/fileapi/file_system_file_util.h"
#include "storage/browser/fileapi/file_system_operation_context.h"
#include "storage/browser/fileapi/file_system_operation_runner.h"
#include "storage/browser/fileapi/file_system_url.h"
#include "storage/browser/fileapi/file_system_usage_cache.h"
#include "storage/browser/fileapi/sandbox_file_system_backend.h"
#include "storage/browser/quota/quota_manager_proxy.h"
#include "storage/browser/test/mock_special_storage_policy.h"
#include "storage/browser/test/test_file_system_context.h"
#include "storage/common/fileapi/file_system_util.h"
#include "url/gurl.h"
using storage::FileSystemContext;
using storage::FileSystemOperationContext;
using storage::FileSystemOperationRunner;
using storage::FileSystemURL;
namespace content {
SandboxFileSystemTestHelper::SandboxFileSystemTestHelper(
const GURL& origin,
storage::FileSystemType type)
: origin_(origin), type_(type), file_util_(NULL) {
}
SandboxFileSystemTestHelper::SandboxFileSystemTestHelper()
: origin_(GURL("http://foo.com")),
type_(storage::kFileSystemTypeTemporary),
file_util_(NULL) {
}
SandboxFileSystemTestHelper::~SandboxFileSystemTestHelper() {
}
void SandboxFileSystemTestHelper::SetUp(const base::FilePath& base_dir) {
SetUp(base_dir, NULL);
}
void SandboxFileSystemTestHelper::SetUp(
FileSystemContext* file_system_context) {
file_system_context_ = file_system_context;
SetUpFileSystem();
}
void SandboxFileSystemTestHelper::SetUp(
const base::FilePath& base_dir,
storage::QuotaManagerProxy* quota_manager_proxy) {
file_system_context_ = CreateFileSystemContextForTesting(
quota_manager_proxy, base_dir);
SetUpFileSystem();
}
void SandboxFileSystemTestHelper::TearDown() {
file_system_context_ = NULL;
base::RunLoop().RunUntilIdle();
}
base::FilePath SandboxFileSystemTestHelper::GetOriginRootPath() {
return file_system_context_->sandbox_delegate()->
GetBaseDirectoryForOriginAndType(origin_, type_, false);
}
base::FilePath SandboxFileSystemTestHelper::GetLocalPath(
const base::FilePath& path) {
DCHECK(file_util_);
base::FilePath local_path;
std::unique_ptr<FileSystemOperationContext> context(NewOperationContext());
file_util_->GetLocalFilePath(context.get(), CreateURL(path), &local_path);
return local_path;
}
base::FilePath SandboxFileSystemTestHelper::GetLocalPathFromASCII(
const std::string& path) {
return GetLocalPath(base::FilePath().AppendASCII(path));
}
base::FilePath SandboxFileSystemTestHelper::GetUsageCachePath() const {
return file_system_context_->sandbox_delegate()->
GetUsageCachePathForOriginAndType(origin_, type_);
}
FileSystemURL SandboxFileSystemTestHelper::CreateURL(
const base::FilePath& path) const {
return file_system_context_->CreateCrackedFileSystemURL(origin_, type_, path);
}
int64_t SandboxFileSystemTestHelper::GetCachedOriginUsage() const {
return file_system_context_->GetQuotaUtil(type_)
->GetOriginUsageOnFileTaskRunner(
file_system_context_.get(), origin_, type_);
}
int64_t SandboxFileSystemTestHelper::ComputeCurrentOriginUsage() {
usage_cache()->CloseCacheFiles();
int64_t size = base::ComputeDirectorySize(GetOriginRootPath());
if (base::PathExists(GetUsageCachePath()))
size -= storage::FileSystemUsageCache::kUsageFileSize;
return size;
}
int64_t SandboxFileSystemTestHelper::ComputeCurrentDirectoryDatabaseUsage() {
return base::ComputeDirectorySize(
GetOriginRootPath().AppendASCII("Paths"));
}
FileSystemOperationRunner* SandboxFileSystemTestHelper::operation_runner() {
return file_system_context_->operation_runner();
}
FileSystemOperationContext*
SandboxFileSystemTestHelper::NewOperationContext() {
DCHECK(file_system_context_.get());
FileSystemOperationContext* context =
new FileSystemOperationContext(file_system_context_.get());
context->set_update_observers(
*file_system_context_->GetUpdateObservers(type_));
return context;
}
void SandboxFileSystemTestHelper::AddFileChangeObserver(
storage::FileChangeObserver* observer) {
file_system_context_->sandbox_delegate()->AddFileChangeObserver(
type_, observer, NULL);
}
void SandboxFileSystemTestHelper::AddFileUpdateObserver(
storage::FileUpdateObserver* observer) {
file_system_context_->sandbox_delegate()->AddFileUpdateObserver(
type_, observer, NULL);
}
storage::FileSystemUsageCache* SandboxFileSystemTestHelper::usage_cache() {
return file_system_context()->sandbox_delegate()->usage_cache();
}
void SandboxFileSystemTestHelper::SetUpFileSystem() {
DCHECK(file_system_context_.get());
DCHECK(file_system_context_->sandbox_backend()->CanHandleType(type_));
file_util_ = file_system_context_->sandbox_delegate()->sync_file_util();
DCHECK(file_util_);
// Prepare the origin's root directory.
file_system_context_->sandbox_delegate()->
GetBaseDirectoryForOriginAndType(origin_, type_, true /* create */);
// Initialize the usage cache file.
base::FilePath usage_cache_path = GetUsageCachePath();
if (!usage_cache_path.empty())
usage_cache()->UpdateUsage(usage_cache_path, 0);
}
} // namespace content