blob: 7386dbad3335f4b599213ed5bbb8843e65225c9a [file] [log] [blame]
// Copyright 2014 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/async_file_test_helper.h"
#include "base/bind.h"
#include "base/files/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "base/run_loop.h"
#include "storage/browser/fileapi/file_system_backend.h"
#include "storage/browser/fileapi/file_system_context.h"
#include "storage/browser/fileapi/file_system_operation_runner.h"
#include "storage/browser/fileapi/file_system_url.h"
#include "storage/browser/quota/quota_manager.h"
#include "storage/common/fileapi/file_system_util.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace content {
typedef storage::FileSystemOperation::FileEntryList FileEntryList;
namespace {
void AssignAndQuit(base::RunLoop* run_loop,
base::File::Error* result_out,
base::File::Error result) {
*result_out = result;
run_loop->Quit();
}
base::Callback<void(base::File::Error)> AssignAndQuitCallback(
base::RunLoop* run_loop,
base::File::Error* result) {
return base::Bind(&AssignAndQuit, run_loop, base::Unretained(result));
}
void GetMetadataCallback(base::RunLoop* run_loop,
base::File::Error* result_out,
base::File::Info* file_info_out,
base::File::Error result,
const base::File::Info& file_info) {
*result_out = result;
if (file_info_out)
*file_info_out = file_info;
run_loop->Quit();
}
void CreateSnapshotFileCallback(
base::RunLoop* run_loop,
base::File::Error* result_out,
base::FilePath* platform_path_out,
base::File::Error result,
const base::File::Info& file_info,
const base::FilePath& platform_path,
const scoped_refptr<storage::ShareableFileReference>& file_ref) {
DCHECK(!file_ref.get());
*result_out = result;
if (platform_path_out)
*platform_path_out = platform_path;
run_loop->Quit();
}
void ReadDirectoryCallback(base::RunLoop* run_loop,
base::File::Error* result_out,
FileEntryList* entries_out,
base::File::Error result,
const FileEntryList& entries,
bool has_more) {
*result_out = result;
entries_out->insert(entries_out->end(), entries.begin(), entries.end());
if (result != base::File::FILE_OK || !has_more)
run_loop->Quit();
}
void DidGetUsageAndQuota(storage::QuotaStatusCode* status_out,
int64_t* usage_out,
int64_t* quota_out,
storage::QuotaStatusCode status,
int64_t usage,
int64_t quota) {
if (status_out)
*status_out = status;
if (usage_out)
*usage_out = usage;
if (quota_out)
*quota_out = quota;
}
} // namespace
const int64_t AsyncFileTestHelper::kDontCheckSize = -1;
base::File::Error AsyncFileTestHelper::Copy(
storage::FileSystemContext* context,
const storage::FileSystemURL& src,
const storage::FileSystemURL& dest) {
return CopyWithProgress(context, src, dest, CopyProgressCallback());
}
base::File::Error AsyncFileTestHelper::CopyWithProgress(
storage::FileSystemContext* context,
const storage::FileSystemURL& src,
const storage::FileSystemURL& dest,
const CopyProgressCallback& progress_callback) {
base::File::Error result = base::File::FILE_ERROR_FAILED;
base::RunLoop run_loop;
context->operation_runner()->Copy(
src, dest, storage::FileSystemOperation::OPTION_NONE,
storage::FileSystemOperation::ERROR_BEHAVIOR_ABORT, progress_callback,
AssignAndQuitCallback(&run_loop, &result));
run_loop.Run();
return result;
}
base::File::Error AsyncFileTestHelper::Move(
storage::FileSystemContext* context,
const storage::FileSystemURL& src,
const storage::FileSystemURL& dest) {
base::File::Error result = base::File::FILE_ERROR_FAILED;
base::RunLoop run_loop;
context->operation_runner()->Move(src, dest,
storage::FileSystemOperation::OPTION_NONE,
AssignAndQuitCallback(&run_loop, &result));
run_loop.Run();
return result;
}
base::File::Error AsyncFileTestHelper::Remove(
storage::FileSystemContext* context,
const storage::FileSystemURL& url,
bool recursive) {
base::File::Error result = base::File::FILE_ERROR_FAILED;
base::RunLoop run_loop;
context->operation_runner()->Remove(
url, recursive, AssignAndQuitCallback(&run_loop, &result));
run_loop.Run();
return result;
}
base::File::Error AsyncFileTestHelper::ReadDirectory(
storage::FileSystemContext* context,
const storage::FileSystemURL& url,
FileEntryList* entries) {
base::File::Error result = base::File::FILE_ERROR_FAILED;
DCHECK(entries);
entries->clear();
base::RunLoop run_loop;
context->operation_runner()->ReadDirectory(
url, base::Bind(&ReadDirectoryCallback, &run_loop, &result, entries));
run_loop.Run();
return result;
}
base::File::Error AsyncFileTestHelper::CreateDirectory(
storage::FileSystemContext* context,
const storage::FileSystemURL& url) {
base::File::Error result = base::File::FILE_ERROR_FAILED;
base::RunLoop run_loop;
context->operation_runner()->CreateDirectory(
url, false /* exclusive */, false /* recursive */,
AssignAndQuitCallback(&run_loop, &result));
run_loop.Run();
return result;
}
base::File::Error AsyncFileTestHelper::CreateFile(
storage::FileSystemContext* context,
const storage::FileSystemURL& url) {
base::File::Error result = base::File::FILE_ERROR_FAILED;
base::RunLoop run_loop;
context->operation_runner()->CreateFile(
url, false /* exclusive */, AssignAndQuitCallback(&run_loop, &result));
run_loop.Run();
return result;
}
base::File::Error AsyncFileTestHelper::CreateFileWithData(
storage::FileSystemContext* context,
const storage::FileSystemURL& url,
const char* buf,
int buf_size) {
base::ScopedTempDir dir;
if (!dir.CreateUniqueTempDir())
return base::File::FILE_ERROR_FAILED;
base::FilePath local_path = dir.GetPath().AppendASCII("tmp");
if (buf_size != base::WriteFile(local_path, buf, buf_size))
return base::File::FILE_ERROR_FAILED;
base::File::Error result = base::File::FILE_ERROR_FAILED;
base::RunLoop run_loop;
context->operation_runner()->CopyInForeignFile(
local_path, url, AssignAndQuitCallback(&run_loop, &result));
run_loop.Run();
return result;
}
base::File::Error AsyncFileTestHelper::TruncateFile(
storage::FileSystemContext* context,
const storage::FileSystemURL& url,
size_t size) {
base::RunLoop run_loop;
base::File::Error result = base::File::FILE_ERROR_FAILED;
context->operation_runner()->Truncate(
url, size, AssignAndQuitCallback(&run_loop, &result));
run_loop.Run();
return result;
}
base::File::Error AsyncFileTestHelper::GetMetadata(
storage::FileSystemContext* context,
const storage::FileSystemURL& url,
base::File::Info* file_info) {
base::File::Error result = base::File::FILE_ERROR_FAILED;
base::RunLoop run_loop;
context->operation_runner()->GetMetadata(
url,
storage::FileSystemOperation::GET_METADATA_FIELD_IS_DIRECTORY |
storage::FileSystemOperation::GET_METADATA_FIELD_SIZE |
storage::FileSystemOperation::GET_METADATA_FIELD_LAST_MODIFIED,
base::Bind(&GetMetadataCallback, &run_loop, &result, file_info));
run_loop.Run();
return result;
}
base::File::Error AsyncFileTestHelper::GetPlatformPath(
storage::FileSystemContext* context,
const storage::FileSystemURL& url,
base::FilePath* platform_path) {
base::File::Error result = base::File::FILE_ERROR_FAILED;
base::RunLoop run_loop;
context->operation_runner()->CreateSnapshotFile(
url, base::Bind(&CreateSnapshotFileCallback, &run_loop, &result,
platform_path));
run_loop.Run();
return result;
}
bool AsyncFileTestHelper::FileExists(storage::FileSystemContext* context,
const storage::FileSystemURL& url,
int64_t expected_size) {
base::File::Info file_info;
base::File::Error result = GetMetadata(context, url, &file_info);
if (result != base::File::FILE_OK || file_info.is_directory)
return false;
return expected_size == kDontCheckSize || file_info.size == expected_size;
}
bool AsyncFileTestHelper::DirectoryExists(storage::FileSystemContext* context,
const storage::FileSystemURL& url) {
base::File::Info file_info;
base::File::Error result = GetMetadata(context, url, &file_info);
return (result == base::File::FILE_OK) && file_info.is_directory;
}
storage::QuotaStatusCode AsyncFileTestHelper::GetUsageAndQuota(
storage::QuotaManager* quota_manager,
const GURL& origin,
storage::FileSystemType type,
int64_t* usage,
int64_t* quota) {
storage::QuotaStatusCode status = storage::kQuotaStatusUnknown;
quota_manager->GetUsageAndQuota(
origin, FileSystemTypeToQuotaStorageType(type),
base::Bind(&DidGetUsageAndQuota, &status, usage, quota));
base::RunLoop().RunUntilIdle();
return status;
}
} // namespace content