| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/public/test/test_file_error_injector.h" | 5 #include "content/public/test/test_file_error_injector.h" |
| 6 | 6 |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/compiler_specific.h" | 9 #include "base/compiler_specific.h" |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| (...skipping 12 matching lines...) Expand all Loading... |
| 23 namespace { | 23 namespace { |
| 24 | 24 |
| 25 // A class that performs file operations and injects errors. | 25 // A class that performs file operations and injects errors. |
| 26 class DownloadFileWithErrors: public DownloadFileImpl { | 26 class DownloadFileWithErrors: public DownloadFileImpl { |
| 27 public: | 27 public: |
| 28 typedef base::Callback<void(const GURL& url)> ConstructionCallback; | 28 typedef base::Callback<void(const GURL& url)> ConstructionCallback; |
| 29 typedef base::Callback<void(const GURL& url)> DestructionCallback; | 29 typedef base::Callback<void(const GURL& url)> DestructionCallback; |
| 30 | 30 |
| 31 DownloadFileWithErrors( | 31 DownloadFileWithErrors( |
| 32 scoped_ptr<DownloadSaveInfo> save_info, | 32 scoped_ptr<DownloadSaveInfo> save_info, |
| 33 const FilePath& default_download_directory, | 33 const base::FilePath& default_download_directory, |
| 34 const GURL& url, | 34 const GURL& url, |
| 35 const GURL& referrer_url, | 35 const GURL& referrer_url, |
| 36 bool calculate_hash, | 36 bool calculate_hash, |
| 37 scoped_ptr<ByteStreamReader> stream, | 37 scoped_ptr<ByteStreamReader> stream, |
| 38 const net::BoundNetLog& bound_net_log, | 38 const net::BoundNetLog& bound_net_log, |
| 39 scoped_ptr<PowerSaveBlocker> power_save_blocker, | 39 scoped_ptr<PowerSaveBlocker> power_save_blocker, |
| 40 base::WeakPtr<DownloadDestinationObserver> observer, | 40 base::WeakPtr<DownloadDestinationObserver> observer, |
| 41 const TestFileErrorInjector::FileErrorInfo& error_info, | 41 const TestFileErrorInjector::FileErrorInfo& error_info, |
| 42 const ConstructionCallback& ctor_callback, | 42 const ConstructionCallback& ctor_callback, |
| 43 const DestructionCallback& dtor_callback); | 43 const DestructionCallback& dtor_callback); |
| 44 | 44 |
| 45 virtual ~DownloadFileWithErrors(); | 45 virtual ~DownloadFileWithErrors(); |
| 46 | 46 |
| 47 virtual void Initialize(const InitializeCallback& callback) OVERRIDE; | 47 virtual void Initialize(const InitializeCallback& callback) OVERRIDE; |
| 48 | 48 |
| 49 // DownloadFile interface. | 49 // DownloadFile interface. |
| 50 virtual DownloadInterruptReason AppendDataToFile( | 50 virtual DownloadInterruptReason AppendDataToFile( |
| 51 const char* data, size_t data_len) OVERRIDE; | 51 const char* data, size_t data_len) OVERRIDE; |
| 52 virtual void RenameAndUniquify( | 52 virtual void RenameAndUniquify( |
| 53 const FilePath& full_path, | 53 const base::FilePath& full_path, |
| 54 const RenameCompletionCallback& callback) OVERRIDE; | 54 const RenameCompletionCallback& callback) OVERRIDE; |
| 55 virtual void RenameAndAnnotate( | 55 virtual void RenameAndAnnotate( |
| 56 const FilePath& full_path, | 56 const base::FilePath& full_path, |
| 57 const RenameCompletionCallback& callback) OVERRIDE; | 57 const RenameCompletionCallback& callback) OVERRIDE; |
| 58 | 58 |
| 59 private: | 59 private: |
| 60 // Error generating helper. | 60 // Error generating helper. |
| 61 DownloadInterruptReason ShouldReturnError( | 61 DownloadInterruptReason ShouldReturnError( |
| 62 TestFileErrorInjector::FileOperationCode code, | 62 TestFileErrorInjector::FileOperationCode code, |
| 63 DownloadInterruptReason original_error); | 63 DownloadInterruptReason original_error); |
| 64 | 64 |
| 65 // Determine whether to overwrite an operation with the given code | 65 // Determine whether to overwrite an operation with the given code |
| 66 // with a substitute error; if returns true, |*original_error| is | 66 // with a substitute error; if returns true, |*original_error| is |
| (...skipping 22 matching lines...) Expand all Loading... |
| 89 const DownloadFile::InitializeCallback original_callback, | 89 const DownloadFile::InitializeCallback original_callback, |
| 90 DownloadInterruptReason overwrite_error, | 90 DownloadInterruptReason overwrite_error, |
| 91 DownloadInterruptReason original_error) { | 91 DownloadInterruptReason original_error) { |
| 92 original_callback.Run(overwrite_error); | 92 original_callback.Run(overwrite_error); |
| 93 } | 93 } |
| 94 | 94 |
| 95 static void RenameErrorCallback( | 95 static void RenameErrorCallback( |
| 96 const DownloadFile::RenameCompletionCallback original_callback, | 96 const DownloadFile::RenameCompletionCallback original_callback, |
| 97 DownloadInterruptReason overwrite_error, | 97 DownloadInterruptReason overwrite_error, |
| 98 DownloadInterruptReason original_error, | 98 DownloadInterruptReason original_error, |
| 99 const FilePath& path_result) { | 99 const base::FilePath& path_result) { |
| 100 original_callback.Run( | 100 original_callback.Run( |
| 101 overwrite_error, | 101 overwrite_error, |
| 102 overwrite_error == DOWNLOAD_INTERRUPT_REASON_NONE ? | 102 overwrite_error == DOWNLOAD_INTERRUPT_REASON_NONE ? |
| 103 path_result : FilePath()); | 103 path_result : base::FilePath()); |
| 104 } | 104 } |
| 105 | 105 |
| 106 DownloadFileWithErrors::DownloadFileWithErrors( | 106 DownloadFileWithErrors::DownloadFileWithErrors( |
| 107 scoped_ptr<DownloadSaveInfo> save_info, | 107 scoped_ptr<DownloadSaveInfo> save_info, |
| 108 const FilePath& default_download_directory, | 108 const base::FilePath& default_download_directory, |
| 109 const GURL& url, | 109 const GURL& url, |
| 110 const GURL& referrer_url, | 110 const GURL& referrer_url, |
| 111 bool calculate_hash, | 111 bool calculate_hash, |
| 112 scoped_ptr<ByteStreamReader> stream, | 112 scoped_ptr<ByteStreamReader> stream, |
| 113 const net::BoundNetLog& bound_net_log, | 113 const net::BoundNetLog& bound_net_log, |
| 114 scoped_ptr<PowerSaveBlocker> power_save_blocker, | 114 scoped_ptr<PowerSaveBlocker> power_save_blocker, |
| 115 base::WeakPtr<DownloadDestinationObserver> observer, | 115 base::WeakPtr<DownloadDestinationObserver> observer, |
| 116 const TestFileErrorInjector::FileErrorInfo& error_info, | 116 const TestFileErrorInjector::FileErrorInfo& error_info, |
| 117 const ConstructionCallback& ctor_callback, | 117 const ConstructionCallback& ctor_callback, |
| 118 const DestructionCallback& dtor_callback) | 118 const DestructionCallback& dtor_callback) |
| (...skipping 28 matching lines...) Expand all Loading... |
| 147 } | 147 } |
| 148 | 148 |
| 149 DownloadInterruptReason DownloadFileWithErrors::AppendDataToFile( | 149 DownloadInterruptReason DownloadFileWithErrors::AppendDataToFile( |
| 150 const char* data, size_t data_len) { | 150 const char* data, size_t data_len) { |
| 151 return ShouldReturnError( | 151 return ShouldReturnError( |
| 152 TestFileErrorInjector::FILE_OPERATION_WRITE, | 152 TestFileErrorInjector::FILE_OPERATION_WRITE, |
| 153 DownloadFileImpl::AppendDataToFile(data, data_len)); | 153 DownloadFileImpl::AppendDataToFile(data, data_len)); |
| 154 } | 154 } |
| 155 | 155 |
| 156 void DownloadFileWithErrors::RenameAndUniquify( | 156 void DownloadFileWithErrors::RenameAndUniquify( |
| 157 const FilePath& full_path, | 157 const base::FilePath& full_path, |
| 158 const RenameCompletionCallback& callback) { | 158 const RenameCompletionCallback& callback) { |
| 159 DownloadInterruptReason error_to_return = DOWNLOAD_INTERRUPT_REASON_NONE; | 159 DownloadInterruptReason error_to_return = DOWNLOAD_INTERRUPT_REASON_NONE; |
| 160 RenameCompletionCallback callback_to_use = callback; | 160 RenameCompletionCallback callback_to_use = callback; |
| 161 | 161 |
| 162 // Replace callback if the error needs to be overwritten. | 162 // Replace callback if the error needs to be overwritten. |
| 163 if (OverwriteError( | 163 if (OverwriteError( |
| 164 TestFileErrorInjector::FILE_OPERATION_RENAME_UNIQUIFY, | 164 TestFileErrorInjector::FILE_OPERATION_RENAME_UNIQUIFY, |
| 165 &error_to_return)) { | 165 &error_to_return)) { |
| 166 callback_to_use = base::Bind(&RenameErrorCallback, callback, | 166 callback_to_use = base::Bind(&RenameErrorCallback, callback, |
| 167 error_to_return); | 167 error_to_return); |
| 168 } | 168 } |
| 169 | 169 |
| 170 DownloadFileImpl::RenameAndUniquify(full_path, callback_to_use); | 170 DownloadFileImpl::RenameAndUniquify(full_path, callback_to_use); |
| 171 } | 171 } |
| 172 | 172 |
| 173 void DownloadFileWithErrors::RenameAndAnnotate( | 173 void DownloadFileWithErrors::RenameAndAnnotate( |
| 174 const FilePath& full_path, | 174 const base::FilePath& full_path, |
| 175 const RenameCompletionCallback& callback) { | 175 const RenameCompletionCallback& callback) { |
| 176 DownloadInterruptReason error_to_return = DOWNLOAD_INTERRUPT_REASON_NONE; | 176 DownloadInterruptReason error_to_return = DOWNLOAD_INTERRUPT_REASON_NONE; |
| 177 RenameCompletionCallback callback_to_use = callback; | 177 RenameCompletionCallback callback_to_use = callback; |
| 178 | 178 |
| 179 // Replace callback if the error needs to be overwritten. | 179 // Replace callback if the error needs to be overwritten. |
| 180 if (OverwriteError( | 180 if (OverwriteError( |
| 181 TestFileErrorInjector::FILE_OPERATION_RENAME_ANNOTATE, | 181 TestFileErrorInjector::FILE_OPERATION_RENAME_ANNOTATE, |
| 182 &error_to_return)) { | 182 &error_to_return)) { |
| 183 callback_to_use = base::Bind(&RenameErrorCallback, callback, | 183 callback_to_use = base::Bind(&RenameErrorCallback, callback, |
| 184 error_to_return); | 184 error_to_return); |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 216 class DownloadFileWithErrorsFactory : public DownloadFileFactory { | 216 class DownloadFileWithErrorsFactory : public DownloadFileFactory { |
| 217 public: | 217 public: |
| 218 DownloadFileWithErrorsFactory( | 218 DownloadFileWithErrorsFactory( |
| 219 const DownloadFileWithErrors::ConstructionCallback& ctor_callback, | 219 const DownloadFileWithErrors::ConstructionCallback& ctor_callback, |
| 220 const DownloadFileWithErrors::DestructionCallback& dtor_callback); | 220 const DownloadFileWithErrors::DestructionCallback& dtor_callback); |
| 221 virtual ~DownloadFileWithErrorsFactory(); | 221 virtual ~DownloadFileWithErrorsFactory(); |
| 222 | 222 |
| 223 // DownloadFileFactory interface. | 223 // DownloadFileFactory interface. |
| 224 virtual DownloadFile* CreateFile( | 224 virtual DownloadFile* CreateFile( |
| 225 scoped_ptr<DownloadSaveInfo> save_info, | 225 scoped_ptr<DownloadSaveInfo> save_info, |
| 226 const FilePath& default_download_directory, | 226 const base::FilePath& default_download_directory, |
| 227 const GURL& url, | 227 const GURL& url, |
| 228 const GURL& referrer_url, | 228 const GURL& referrer_url, |
| 229 bool calculate_hash, | 229 bool calculate_hash, |
| 230 scoped_ptr<ByteStreamReader> stream, | 230 scoped_ptr<ByteStreamReader> stream, |
| 231 const net::BoundNetLog& bound_net_log, | 231 const net::BoundNetLog& bound_net_log, |
| 232 base::WeakPtr<DownloadDestinationObserver> observer) OVERRIDE; | 232 base::WeakPtr<DownloadDestinationObserver> observer) OVERRIDE; |
| 233 | 233 |
| 234 bool AddError( | 234 bool AddError( |
| 235 const TestFileErrorInjector::FileErrorInfo& error_info); | 235 const TestFileErrorInjector::FileErrorInfo& error_info); |
| 236 | 236 |
| (...skipping 13 matching lines...) Expand all Loading... |
| 250 const DownloadFileWithErrors::DestructionCallback& dtor_callback) | 250 const DownloadFileWithErrors::DestructionCallback& dtor_callback) |
| 251 : construction_callback_(ctor_callback), | 251 : construction_callback_(ctor_callback), |
| 252 destruction_callback_(dtor_callback) { | 252 destruction_callback_(dtor_callback) { |
| 253 } | 253 } |
| 254 | 254 |
| 255 DownloadFileWithErrorsFactory::~DownloadFileWithErrorsFactory() { | 255 DownloadFileWithErrorsFactory::~DownloadFileWithErrorsFactory() { |
| 256 } | 256 } |
| 257 | 257 |
| 258 DownloadFile* DownloadFileWithErrorsFactory::CreateFile( | 258 DownloadFile* DownloadFileWithErrorsFactory::CreateFile( |
| 259 scoped_ptr<DownloadSaveInfo> save_info, | 259 scoped_ptr<DownloadSaveInfo> save_info, |
| 260 const FilePath& default_download_directory, | 260 const base::FilePath& default_download_directory, |
| 261 const GURL& url, | 261 const GURL& url, |
| 262 const GURL& referrer_url, | 262 const GURL& referrer_url, |
| 263 bool calculate_hash, | 263 bool calculate_hash, |
| 264 scoped_ptr<ByteStreamReader> stream, | 264 scoped_ptr<ByteStreamReader> stream, |
| 265 const net::BoundNetLog& bound_net_log, | 265 const net::BoundNetLog& bound_net_log, |
| 266 base::WeakPtr<DownloadDestinationObserver> observer) { | 266 base::WeakPtr<DownloadDestinationObserver> observer) { |
| 267 if (injected_errors_.find(url.spec()) == injected_errors_.end()) { | 267 if (injected_errors_.find(url.spec()) == injected_errors_.end()) { |
| 268 // Have to create entry, because FileErrorInfo is not a POD type. | 268 // Have to create entry, because FileErrorInfo is not a POD type. |
| 269 TestFileErrorInjector::FileErrorInfo err_info = { | 269 TestFileErrorInjector::FileErrorInfo err_info = { |
| 270 url.spec(), | 270 url.spec(), |
| (...skipping 168 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 439 case FILE_OPERATION_RENAME_ANNOTATE: | 439 case FILE_OPERATION_RENAME_ANNOTATE: |
| 440 return "RENAME_ANNOTATE"; | 440 return "RENAME_ANNOTATE"; |
| 441 default: | 441 default: |
| 442 break; | 442 break; |
| 443 } | 443 } |
| 444 | 444 |
| 445 return "Unknown"; | 445 return "Unknown"; |
| 446 } | 446 } |
| 447 | 447 |
| 448 } // namespace content | 448 } // namespace content |
| OLD | NEW |