| OLD | NEW | 
 | (Empty) | 
|    1 // Copyright 2014 The Chromium Authors. All rights reserved. |  | 
|    2 // Use of this source code is governed by a BSD-style license that can be |  | 
|    3 // found in the LICENSE file. |  | 
|    4  |  | 
|    5 #include "base/bind.h" |  | 
|    6 #include "base/files/file_util.h" |  | 
|    7 #include "base/files/scoped_temp_dir.h" |  | 
|    8 #include "base/run_loop.h" |  | 
|    9 #include "content/public/test/async_file_test_helper.h" |  | 
|   10 #include "storage/browser/fileapi/file_system_backend.h" |  | 
|   11 #include "storage/browser/fileapi/file_system_context.h" |  | 
|   12 #include "storage/browser/fileapi/file_system_operation_runner.h" |  | 
|   13 #include "storage/browser/fileapi/file_system_url.h" |  | 
|   14 #include "storage/browser/quota/quota_manager.h" |  | 
|   15 #include "storage/common/fileapi/file_system_util.h" |  | 
|   16 #include "testing/gtest/include/gtest/gtest.h" |  | 
|   17  |  | 
|   18 namespace content { |  | 
|   19  |  | 
|   20 typedef storage::FileSystemOperation::FileEntryList FileEntryList; |  | 
|   21  |  | 
|   22 namespace { |  | 
|   23  |  | 
|   24 void AssignAndQuit(base::RunLoop* run_loop, |  | 
|   25                    base::File::Error* result_out, |  | 
|   26                    base::File::Error result) { |  | 
|   27   *result_out = result; |  | 
|   28   run_loop->Quit(); |  | 
|   29 } |  | 
|   30  |  | 
|   31 base::Callback<void(base::File::Error)> |  | 
|   32 AssignAndQuitCallback(base::RunLoop* run_loop, |  | 
|   33                       base::File::Error* result) { |  | 
|   34   return base::Bind(&AssignAndQuit, run_loop, base::Unretained(result)); |  | 
|   35 } |  | 
|   36  |  | 
|   37 void GetMetadataCallback(base::RunLoop* run_loop, |  | 
|   38                          base::File::Error* result_out, |  | 
|   39                          base::File::Info* file_info_out, |  | 
|   40                          base::File::Error result, |  | 
|   41                          const base::File::Info& file_info) { |  | 
|   42   *result_out = result; |  | 
|   43   if (file_info_out) |  | 
|   44     *file_info_out = file_info; |  | 
|   45   run_loop->Quit(); |  | 
|   46 } |  | 
|   47  |  | 
|   48 void CreateSnapshotFileCallback( |  | 
|   49     base::RunLoop* run_loop, |  | 
|   50     base::File::Error* result_out, |  | 
|   51     base::FilePath* platform_path_out, |  | 
|   52     base::File::Error result, |  | 
|   53     const base::File::Info& file_info, |  | 
|   54     const base::FilePath& platform_path, |  | 
|   55     const scoped_refptr<storage::ShareableFileReference>& file_ref) { |  | 
|   56   DCHECK(!file_ref.get()); |  | 
|   57   *result_out = result; |  | 
|   58   if (platform_path_out) |  | 
|   59     *platform_path_out = platform_path; |  | 
|   60   run_loop->Quit(); |  | 
|   61 } |  | 
|   62  |  | 
|   63 void ReadDirectoryCallback(base::RunLoop* run_loop, |  | 
|   64                            base::File::Error* result_out, |  | 
|   65                            FileEntryList* entries_out, |  | 
|   66                            base::File::Error result, |  | 
|   67                            const FileEntryList& entries, |  | 
|   68                            bool has_more) { |  | 
|   69   *result_out = result; |  | 
|   70   entries_out->insert(entries_out->end(), entries.begin(), entries.end()); |  | 
|   71   if (result != base::File::FILE_OK || !has_more) |  | 
|   72     run_loop->Quit(); |  | 
|   73 } |  | 
|   74  |  | 
|   75 void DidGetUsageAndQuota(storage::QuotaStatusCode* status_out, |  | 
|   76                          int64_t* usage_out, |  | 
|   77                          int64_t* quota_out, |  | 
|   78                          storage::QuotaStatusCode status, |  | 
|   79                          int64_t usage, |  | 
|   80                          int64_t quota) { |  | 
|   81   if (status_out) |  | 
|   82     *status_out = status; |  | 
|   83   if (usage_out) |  | 
|   84     *usage_out = usage; |  | 
|   85   if (quota_out) |  | 
|   86     *quota_out = quota; |  | 
|   87 } |  | 
|   88  |  | 
|   89 }  // namespace |  | 
|   90  |  | 
|   91 const int64_t AsyncFileTestHelper::kDontCheckSize = -1; |  | 
|   92  |  | 
|   93 base::File::Error AsyncFileTestHelper::Copy( |  | 
|   94     storage::FileSystemContext* context, |  | 
|   95     const storage::FileSystemURL& src, |  | 
|   96     const storage::FileSystemURL& dest) { |  | 
|   97   return CopyWithProgress(context, src, dest, CopyProgressCallback()); |  | 
|   98 } |  | 
|   99  |  | 
|  100 base::File::Error AsyncFileTestHelper::CopyWithProgress( |  | 
|  101     storage::FileSystemContext* context, |  | 
|  102     const storage::FileSystemURL& src, |  | 
|  103     const storage::FileSystemURL& dest, |  | 
|  104     const CopyProgressCallback& progress_callback) { |  | 
|  105   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  106   base::RunLoop run_loop; |  | 
|  107   context->operation_runner()->Copy( |  | 
|  108       src, dest, storage::FileSystemOperation::OPTION_NONE, |  | 
|  109       storage::FileSystemOperation::ERROR_BEHAVIOR_ABORT, progress_callback, |  | 
|  110       AssignAndQuitCallback(&run_loop, &result)); |  | 
|  111   run_loop.Run(); |  | 
|  112   return result; |  | 
|  113 } |  | 
|  114  |  | 
|  115 base::File::Error AsyncFileTestHelper::Move( |  | 
|  116     storage::FileSystemContext* context, |  | 
|  117     const storage::FileSystemURL& src, |  | 
|  118     const storage::FileSystemURL& dest) { |  | 
|  119   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  120   base::RunLoop run_loop; |  | 
|  121   context->operation_runner()->Move(src, |  | 
|  122                                     dest, |  | 
|  123                                     storage::FileSystemOperation::OPTION_NONE, |  | 
|  124                                     AssignAndQuitCallback(&run_loop, &result)); |  | 
|  125   run_loop.Run(); |  | 
|  126   return result; |  | 
|  127 } |  | 
|  128  |  | 
|  129 base::File::Error AsyncFileTestHelper::Remove( |  | 
|  130     storage::FileSystemContext* context, |  | 
|  131     const storage::FileSystemURL& url, |  | 
|  132     bool recursive) { |  | 
|  133   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  134   base::RunLoop run_loop; |  | 
|  135   context->operation_runner()->Remove( |  | 
|  136       url, recursive, AssignAndQuitCallback(&run_loop, &result)); |  | 
|  137   run_loop.Run(); |  | 
|  138   return result; |  | 
|  139 } |  | 
|  140  |  | 
|  141 base::File::Error AsyncFileTestHelper::ReadDirectory( |  | 
|  142     storage::FileSystemContext* context, |  | 
|  143     const storage::FileSystemURL& url, |  | 
|  144     FileEntryList* entries) { |  | 
|  145   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  146   DCHECK(entries); |  | 
|  147   entries->clear(); |  | 
|  148   base::RunLoop run_loop; |  | 
|  149   context->operation_runner()->ReadDirectory( |  | 
|  150       url, base::Bind(&ReadDirectoryCallback, &run_loop, &result, entries)); |  | 
|  151   run_loop.Run(); |  | 
|  152   return result; |  | 
|  153 } |  | 
|  154  |  | 
|  155 base::File::Error AsyncFileTestHelper::CreateDirectory( |  | 
|  156     storage::FileSystemContext* context, |  | 
|  157     const storage::FileSystemURL& url) { |  | 
|  158   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  159   base::RunLoop run_loop; |  | 
|  160   context->operation_runner()->CreateDirectory( |  | 
|  161       url, |  | 
|  162       false /* exclusive */, |  | 
|  163       false /* recursive */, |  | 
|  164       AssignAndQuitCallback(&run_loop, &result)); |  | 
|  165   run_loop.Run(); |  | 
|  166   return result; |  | 
|  167 } |  | 
|  168  |  | 
|  169 base::File::Error AsyncFileTestHelper::CreateFile( |  | 
|  170     storage::FileSystemContext* context, |  | 
|  171     const storage::FileSystemURL& url) { |  | 
|  172   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  173   base::RunLoop run_loop; |  | 
|  174   context->operation_runner()->CreateFile( |  | 
|  175       url, false /* exclusive */, |  | 
|  176       AssignAndQuitCallback(&run_loop, &result)); |  | 
|  177   run_loop.Run(); |  | 
|  178   return result; |  | 
|  179 } |  | 
|  180  |  | 
|  181 base::File::Error AsyncFileTestHelper::CreateFileWithData( |  | 
|  182     storage::FileSystemContext* context, |  | 
|  183     const storage::FileSystemURL& url, |  | 
|  184     const char* buf, |  | 
|  185     int buf_size) { |  | 
|  186   base::ScopedTempDir dir; |  | 
|  187   if (!dir.CreateUniqueTempDir()) |  | 
|  188     return base::File::FILE_ERROR_FAILED; |  | 
|  189   base::FilePath local_path = dir.GetPath().AppendASCII("tmp"); |  | 
|  190   if (buf_size != base::WriteFile(local_path, buf, buf_size)) |  | 
|  191     return base::File::FILE_ERROR_FAILED; |  | 
|  192   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  193   base::RunLoop run_loop; |  | 
|  194   context->operation_runner()->CopyInForeignFile( |  | 
|  195       local_path, url, AssignAndQuitCallback(&run_loop, &result)); |  | 
|  196   run_loop.Run(); |  | 
|  197   return result; |  | 
|  198 } |  | 
|  199  |  | 
|  200 base::File::Error AsyncFileTestHelper::TruncateFile( |  | 
|  201     storage::FileSystemContext* context, |  | 
|  202     const storage::FileSystemURL& url, |  | 
|  203     size_t size) { |  | 
|  204   base::RunLoop run_loop; |  | 
|  205   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  206   context->operation_runner()->Truncate( |  | 
|  207       url, size, AssignAndQuitCallback(&run_loop, &result)); |  | 
|  208   run_loop.Run(); |  | 
|  209   return result; |  | 
|  210 } |  | 
|  211  |  | 
|  212 base::File::Error AsyncFileTestHelper::GetMetadata( |  | 
|  213     storage::FileSystemContext* context, |  | 
|  214     const storage::FileSystemURL& url, |  | 
|  215     base::File::Info* file_info) { |  | 
|  216   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  217   base::RunLoop run_loop; |  | 
|  218   context->operation_runner()->GetMetadata( |  | 
|  219       url, storage::FileSystemOperation::GET_METADATA_FIELD_IS_DIRECTORY | |  | 
|  220                storage::FileSystemOperation::GET_METADATA_FIELD_SIZE | |  | 
|  221                storage::FileSystemOperation::GET_METADATA_FIELD_LAST_MODIFIED, |  | 
|  222       base::Bind(&GetMetadataCallback, &run_loop, &result, file_info)); |  | 
|  223   run_loop.Run(); |  | 
|  224   return result; |  | 
|  225 } |  | 
|  226  |  | 
|  227 base::File::Error AsyncFileTestHelper::GetPlatformPath( |  | 
|  228     storage::FileSystemContext* context, |  | 
|  229     const storage::FileSystemURL& url, |  | 
|  230     base::FilePath* platform_path) { |  | 
|  231   base::File::Error result = base::File::FILE_ERROR_FAILED; |  | 
|  232   base::RunLoop run_loop; |  | 
|  233   context->operation_runner()->CreateSnapshotFile( |  | 
|  234       url, base::Bind(&CreateSnapshotFileCallback, &run_loop, &result, |  | 
|  235                       platform_path)); |  | 
|  236   run_loop.Run(); |  | 
|  237   return result; |  | 
|  238 } |  | 
|  239  |  | 
|  240 bool AsyncFileTestHelper::FileExists(storage::FileSystemContext* context, |  | 
|  241                                      const storage::FileSystemURL& url, |  | 
|  242                                      int64_t expected_size) { |  | 
|  243   base::File::Info file_info; |  | 
|  244   base::File::Error result = GetMetadata(context, url, &file_info); |  | 
|  245   if (result != base::File::FILE_OK || file_info.is_directory) |  | 
|  246     return false; |  | 
|  247   return expected_size == kDontCheckSize || file_info.size == expected_size; |  | 
|  248 } |  | 
|  249  |  | 
|  250 bool AsyncFileTestHelper::DirectoryExists(storage::FileSystemContext* context, |  | 
|  251                                           const storage::FileSystemURL& url) { |  | 
|  252   base::File::Info file_info; |  | 
|  253   base::File::Error result = GetMetadata(context, url, &file_info); |  | 
|  254   return (result == base::File::FILE_OK) && file_info.is_directory; |  | 
|  255 } |  | 
|  256  |  | 
|  257 storage::QuotaStatusCode AsyncFileTestHelper::GetUsageAndQuota( |  | 
|  258     storage::QuotaManager* quota_manager, |  | 
|  259     const GURL& origin, |  | 
|  260     storage::FileSystemType type, |  | 
|  261     int64_t* usage, |  | 
|  262     int64_t* quota) { |  | 
|  263   storage::QuotaStatusCode status = storage::kQuotaStatusUnknown; |  | 
|  264   quota_manager->GetUsageAndQuota( |  | 
|  265       origin, |  | 
|  266       FileSystemTypeToQuotaStorageType(type), |  | 
|  267       base::Bind(&DidGetUsageAndQuota, &status, usage, quota)); |  | 
|  268   base::RunLoop().RunUntilIdle(); |  | 
|  269   return status; |  | 
|  270 } |  | 
|  271  |  | 
|  272 }  // namespace content |  | 
| OLD | NEW |