| 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 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 48 const content::TestFileErrorInjector::FileErrorInfo& error_info, | 48 const content::TestFileErrorInjector::FileErrorInfo& error_info, |
| 49 const ConstructionCallback& ctor_callback, | 49 const ConstructionCallback& ctor_callback, |
| 50 const DestructionCallback& dtor_callback); | 50 const DestructionCallback& dtor_callback); |
| 51 | 51 |
| 52 ~DownloadFileWithErrors(); | 52 ~DownloadFileWithErrors(); |
| 53 | 53 |
| 54 // DownloadFile interface. | 54 // DownloadFile interface. |
| 55 virtual content::DownloadInterruptReason Initialize() OVERRIDE; | 55 virtual content::DownloadInterruptReason Initialize() OVERRIDE; |
| 56 virtual content::DownloadInterruptReason AppendDataToFile( | 56 virtual content::DownloadInterruptReason AppendDataToFile( |
| 57 const char* data, size_t data_len) OVERRIDE; | 57 const char* data, size_t data_len) OVERRIDE; |
| 58 virtual content::DownloadInterruptReason Rename( | 58 virtual void Rename(const FilePath& full_path, |
| 59 const FilePath& full_path) OVERRIDE; | 59 bool overwrite_existing_file, |
| 60 const RenameCompletionCallback& callback) OVERRIDE; |
| 60 | 61 |
| 61 private: | 62 private: |
| 62 // Error generating helper. | 63 // Error generating helper. |
| 63 content::DownloadInterruptReason ShouldReturnError( | 64 content::DownloadInterruptReason ShouldReturnError( |
| 64 content::TestFileErrorInjector::FileOperationCode code, | 65 content::TestFileErrorInjector::FileOperationCode code, |
| 65 content::DownloadInterruptReason original_error); | 66 content::DownloadInterruptReason original_error); |
| 66 | 67 |
| 68 // Used in place of original rename callback to intercept with |
| 69 // ShouldReturnError. |
| 70 void RenameErrorCallback( |
| 71 const RenameCompletionCallback& original_callback, |
| 72 content::DownloadInterruptReason original_error, |
| 73 const FilePath& path_result); |
| 74 |
| 67 // Source URL for the file being downloaded. | 75 // Source URL for the file being downloaded. |
| 68 GURL source_url_; | 76 GURL source_url_; |
| 69 | 77 |
| 70 // Our injected error. Only one per file. | 78 // Our injected error. Only one per file. |
| 71 content::TestFileErrorInjector::FileErrorInfo error_info_; | 79 content::TestFileErrorInjector::FileErrorInfo error_info_; |
| 72 | 80 |
| 73 // Count per operation. 0-based. | 81 // Count per operation. 0-based. |
| 74 std::map<content::TestFileErrorInjector::FileOperationCode, int> | 82 std::map<content::TestFileErrorInjector::FileOperationCode, int> |
| 75 operation_counter_; | 83 operation_counter_; |
| 76 | 84 |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 111 DownloadFileImpl::Initialize()); | 119 DownloadFileImpl::Initialize()); |
| 112 } | 120 } |
| 113 | 121 |
| 114 content::DownloadInterruptReason DownloadFileWithErrors::AppendDataToFile( | 122 content::DownloadInterruptReason DownloadFileWithErrors::AppendDataToFile( |
| 115 const char* data, size_t data_len) { | 123 const char* data, size_t data_len) { |
| 116 return ShouldReturnError( | 124 return ShouldReturnError( |
| 117 content::TestFileErrorInjector::FILE_OPERATION_WRITE, | 125 content::TestFileErrorInjector::FILE_OPERATION_WRITE, |
| 118 DownloadFileImpl::AppendDataToFile(data, data_len)); | 126 DownloadFileImpl::AppendDataToFile(data, data_len)); |
| 119 } | 127 } |
| 120 | 128 |
| 121 content::DownloadInterruptReason DownloadFileWithErrors::Rename( | 129 void DownloadFileWithErrors::Rename( |
| 122 const FilePath& full_path) { | 130 const FilePath& full_path, |
| 123 return ShouldReturnError( | 131 bool overwrite_existing_file, |
| 124 content::TestFileErrorInjector::FILE_OPERATION_RENAME, | 132 const RenameCompletionCallback& callback) { |
| 125 DownloadFileImpl::Rename(full_path)); | 133 DownloadFileImpl::Rename( |
| 134 full_path, overwrite_existing_file, |
| 135 base::Bind(&DownloadFileWithErrors::RenameErrorCallback, |
| 136 // Unretained since this'll only be called from |
| 137 // the DownloadFileImpl slice of the same object. |
| 138 base::Unretained(this), callback)); |
| 126 } | 139 } |
| 127 | 140 |
| 128 content::DownloadInterruptReason DownloadFileWithErrors::ShouldReturnError( | 141 content::DownloadInterruptReason DownloadFileWithErrors::ShouldReturnError( |
| 129 content::TestFileErrorInjector::FileOperationCode code, | 142 content::TestFileErrorInjector::FileOperationCode code, |
| 130 content::DownloadInterruptReason original_error) { | 143 content::DownloadInterruptReason original_error) { |
| 131 int counter = operation_counter_[code]; | 144 int counter = operation_counter_[code]; |
| 132 ++operation_counter_[code]; | 145 ++operation_counter_[code]; |
| 133 | 146 |
| 134 if (code != error_info_.code) | 147 if (code != error_info_.code) |
| 135 return original_error; | 148 return original_error; |
| 136 | 149 |
| 137 if (counter != error_info_.operation_instance) | 150 if (counter != error_info_.operation_instance) |
| 138 return original_error; | 151 return original_error; |
| 139 | 152 |
| 140 VLOG(1) << " " << __FUNCTION__ << "()" | 153 VLOG(1) << " " << __FUNCTION__ << "()" |
| 141 << " url = '" << source_url_.spec() << "'" | 154 << " url = '" << source_url_.spec() << "'" |
| 142 << " code = " << content::TestFileErrorInjector::DebugString(code) | 155 << " code = " << content::TestFileErrorInjector::DebugString(code) |
| 143 << " (" << code << ")" | 156 << " (" << code << ")" |
| 144 << " counter = " << counter | 157 << " counter = " << counter |
| 145 << " original_error = " | 158 << " original_error = " |
| 146 << content::InterruptReasonDebugString(original_error) | 159 << content::InterruptReasonDebugString(original_error) |
| 147 << " (" << original_error << ")" | 160 << " (" << original_error << ")" |
| 148 << " new error = " | 161 << " new error = " |
| 149 << content::InterruptReasonDebugString(error_info_.error) | 162 << content::InterruptReasonDebugString(error_info_.error) |
| 150 << " (" << error_info_.error << ")"; | 163 << " (" << error_info_.error << ")"; |
| 151 | 164 |
| 152 return error_info_.error; | 165 return error_info_.error; |
| 153 } | 166 } |
| 154 | 167 |
| 168 void DownloadFileWithErrors::RenameErrorCallback( |
| 169 const RenameCompletionCallback& original_callback, |
| 170 content::DownloadInterruptReason original_error, |
| 171 const FilePath& path_result) { |
| 172 original_callback.Run(ShouldReturnError( |
| 173 content::TestFileErrorInjector::FILE_OPERATION_RENAME, |
| 174 original_error), path_result); |
| 175 } |
| 176 |
| 155 } // namespace | 177 } // namespace |
| 156 | 178 |
| 157 namespace content { | 179 namespace content { |
| 158 | 180 |
| 159 // A factory for constructing DownloadFiles that inject errors. | 181 // A factory for constructing DownloadFiles that inject errors. |
| 160 class DownloadFileWithErrorsFactory | 182 class DownloadFileWithErrorsFactory |
| 161 : public DownloadFileManager::DownloadFileFactory { | 183 : public DownloadFileManager::DownloadFileFactory { |
| 162 public: | 184 public: |
| 163 | 185 |
| 164 DownloadFileWithErrorsFactory( | 186 DownloadFileWithErrorsFactory( |
| 165 const DownloadFileWithErrors::ConstructionCallback& ctor_callback, | 187 const DownloadFileWithErrors::ConstructionCallback& ctor_callback, |
| 166 const DownloadFileWithErrors::DestructionCallback& dtor_callback); | 188 const DownloadFileWithErrors::DestructionCallback& dtor_callback); |
| 167 virtual ~DownloadFileWithErrorsFactory(); | 189 virtual ~DownloadFileWithErrorsFactory(); |
| 168 | 190 |
| 169 // DownloadFileFactory interface. | 191 // DownloadFileFactory interface. |
| 170 virtual content::DownloadFile* CreateFile( | 192 virtual DownloadFile* CreateFile( |
| 171 DownloadCreateInfo* info, | 193 DownloadCreateInfo* info, |
| 172 scoped_ptr<content::ByteStreamReader> stream, | 194 scoped_ptr<content::ByteStreamReader> stream, |
| 173 content::DownloadManager* download_manager, | 195 content::DownloadManager* download_manager, |
| 174 bool calculate_hash, | 196 bool calculate_hash, |
| 175 const net::BoundNetLog& bound_net_log); | 197 const net::BoundNetLog& bound_net_log); |
| 176 | 198 |
| 177 bool AddError( | 199 bool AddError( |
| 178 const TestFileErrorInjector::FileErrorInfo& error_info); | 200 const TestFileErrorInjector::FileErrorInfo& error_info); |
| 179 | 201 |
| 180 void ClearErrors(); | 202 void ClearErrors(); |
| (...skipping 245 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 426 case FILE_OPERATION_RENAME: | 448 case FILE_OPERATION_RENAME: |
| 427 return "RENAME"; | 449 return "RENAME"; |
| 428 default: | 450 default: |
| 429 break; | 451 break; |
| 430 } | 452 } |
| 431 | 453 |
| 432 return "Unknown"; | 454 return "Unknown"; |
| 433 } | 455 } |
| 434 | 456 |
| 435 } // namespace content | 457 } // namespace content |
| OLD | NEW |