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 <stddef.h> | 5 #include <stddef.h> |
6 #include <stdint.h> | 6 #include <stdint.h> |
7 | 7 |
8 #include <utility> | 8 #include <utility> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "base/files/file.h" | 11 #include "base/files/file.h" |
12 #include "base/files/file_util.h" | 12 #include "base/files/file_util.h" |
13 #include "base/location.h" | 13 #include "base/location.h" |
14 #include "base/run_loop.h" | 14 #include "base/run_loop.h" |
15 #include "base/single_thread_task_runner.h" | 15 #include "base/single_thread_task_runner.h" |
16 #include "base/strings/string_number_conversions.h" | 16 #include "base/strings/string_number_conversions.h" |
17 #include "base/test/test_file_util.h" | 17 #include "base/test/test_file_util.h" |
18 #include "base/threading/thread_task_runner_handle.h" | 18 #include "base/threading/thread_task_runner_handle.h" |
19 #include "build/build_config.h" | 19 #include "build/build_config.h" |
20 #include "content/browser/browser_thread_impl.h" | |
21 #include "content/browser/byte_stream.h" | 20 #include "content/browser/byte_stream.h" |
22 #include "content/browser/download/download_create_info.h" | 21 #include "content/browser/download/download_create_info.h" |
23 #include "content/browser/download/download_destination_observer.h" | 22 #include "content/browser/download/download_destination_observer.h" |
24 #include "content/browser/download/download_file_impl.h" | 23 #include "content/browser/download/download_file_impl.h" |
25 #include "content/browser/download/download_request_handle.h" | 24 #include "content/browser/download/download_request_handle.h" |
26 #include "content/public/browser/download_interrupt_reasons.h" | 25 #include "content/public/browser/download_interrupt_reasons.h" |
27 #include "content/public/browser/download_manager.h" | 26 #include "content/public/browser/download_manager.h" |
28 #include "content/public/test/mock_download_manager.h" | 27 #include "content/public/test/mock_download_manager.h" |
| 28 #include "content/public/test/test_browser_thread_bundle.h" |
29 #include "net/base/file_stream.h" | 29 #include "net/base/file_stream.h" |
30 #include "net/base/mock_file_stream.h" | 30 #include "net/base/mock_file_stream.h" |
31 #include "net/base/net_errors.h" | 31 #include "net/base/net_errors.h" |
32 #include "net/log/net_log_with_source.h" | 32 #include "net/log/net_log_with_source.h" |
33 #include "testing/gmock/include/gmock/gmock.h" | 33 #include "testing/gmock/include/gmock/gmock.h" |
34 #include "testing/gtest/include/gtest/gtest.h" | 34 #include "testing/gtest/include/gtest/gtest.h" |
35 | 35 |
36 using ::testing::_; | 36 using ::testing::_; |
37 using ::testing::AnyNumber; | 37 using ::testing::AnyNumber; |
38 using ::testing::DoAll; | 38 using ::testing::DoAll; |
(...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
131 public: | 131 public: |
132 static const char kTestData1[]; | 132 static const char kTestData1[]; |
133 static const char kTestData2[]; | 133 static const char kTestData2[]; |
134 static const char kTestData3[]; | 134 static const char kTestData3[]; |
135 static const char kDataHash[]; | 135 static const char kDataHash[]; |
136 static const char kEmptyHash[]; | 136 static const char kEmptyHash[]; |
137 static const uint32_t kDummyDownloadId; | 137 static const uint32_t kDummyDownloadId; |
138 static const int kDummyChildId; | 138 static const int kDummyChildId; |
139 static const int kDummyRequestId; | 139 static const int kDummyRequestId; |
140 | 140 |
141 DownloadFileTest() : | 141 DownloadFileTest() |
142 observer_(new StrictMock<MockDownloadDestinationObserver>), | 142 : observer_(new StrictMock<MockDownloadDestinationObserver>), |
143 observer_factory_(observer_.get()), | 143 observer_factory_(observer_.get()), |
144 input_stream_(NULL), | 144 input_stream_(NULL), |
145 bytes_(-1), | 145 bytes_(-1), |
146 bytes_per_sec_(-1), | 146 bytes_per_sec_(-1) {} |
147 ui_thread_(BrowserThread::UI, &loop_), | |
148 file_thread_(BrowserThread::FILE, &loop_) { | |
149 } | |
150 | 147 |
151 ~DownloadFileTest() override {} | 148 ~DownloadFileTest() override {} |
152 | 149 |
153 void SetUpdateDownloadInfo(int64_t bytes, int64_t bytes_per_sec) { | 150 void SetUpdateDownloadInfo(int64_t bytes, int64_t bytes_per_sec) { |
154 bytes_ = bytes; | 151 bytes_ = bytes; |
155 bytes_per_sec_ = bytes_per_sec; | 152 bytes_per_sec_ = bytes_per_sec; |
156 } | 153 } |
157 | 154 |
158 void ConfirmUpdateDownloadInfo() { | 155 void ConfirmUpdateDownloadInfo() { |
159 observer_->CurrentUpdateStatus(bytes_, bytes_per_sec_); | 156 observer_->CurrentUpdateStatus(bytes_, bytes_per_sec_); |
(...skipping 205 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
365 // Owned by download_file_; will be alive for lifetime of download_file_. | 362 // Owned by download_file_; will be alive for lifetime of download_file_. |
366 StrictMock<MockByteStreamReader>* input_stream_; | 363 StrictMock<MockByteStreamReader>* input_stream_; |
367 | 364 |
368 // Sink callback data for stream. | 365 // Sink callback data for stream. |
369 base::Closure sink_callback_; | 366 base::Closure sink_callback_; |
370 | 367 |
371 // Latest update sent to the observer. | 368 // Latest update sent to the observer. |
372 int64_t bytes_; | 369 int64_t bytes_; |
373 int64_t bytes_per_sec_; | 370 int64_t bytes_per_sec_; |
374 | 371 |
375 base::MessageLoop loop_; | |
376 | |
377 private: | 372 private: |
378 void SetRenameResult(const base::Closure& closure, | 373 void SetRenameResult(const base::Closure& closure, |
379 DownloadInterruptReason* reason_p, | 374 DownloadInterruptReason* reason_p, |
380 base::FilePath* result_path_p, | 375 base::FilePath* result_path_p, |
381 DownloadInterruptReason reason, | 376 DownloadInterruptReason reason, |
382 const base::FilePath& result_path) { | 377 const base::FilePath& result_path) { |
383 if (reason_p) | 378 if (reason_p) |
384 *reason_p = reason; | 379 *reason_p = reason; |
385 if (result_path_p) | 380 if (result_path_p) |
386 *result_path_p = result_path; | 381 *result_path_p = result_path; |
387 closure.Run(); | 382 closure.Run(); |
388 } | 383 } |
389 | 384 |
390 // UI thread. | 385 TestBrowserThreadBundle thread_bundle_; |
391 BrowserThreadImpl ui_thread_; | |
392 // File thread to satisfy debug checks in DownloadFile. | |
393 BrowserThreadImpl file_thread_; | |
394 | 386 |
395 // Keep track of what data should be saved to the disk file. | 387 // Keep track of what data should be saved to the disk file. |
396 std::string expected_data_; | 388 std::string expected_data_; |
397 }; | 389 }; |
398 | 390 |
399 // DownloadFile::RenameAndAnnotate and DownloadFile::RenameAndUniquify have a | 391 // DownloadFile::RenameAndAnnotate and DownloadFile::RenameAndUniquify have a |
400 // considerable amount of functional overlap. In order to re-use test logic, we | 392 // considerable amount of functional overlap. In order to re-use test logic, we |
401 // are going to introduce this value parameterized test fixture. It will take a | 393 // are going to introduce this value parameterized test fixture. It will take a |
402 // DownloadFileRenameMethodType value which can be either of the two rename | 394 // DownloadFileRenameMethodType value which can be either of the two rename |
403 // methods. | 395 // methods. |
(...skipping 386 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
790 EXPECT_CALL(*(observer_.get()), | 782 EXPECT_CALL(*(observer_.get()), |
791 CurrentUpdateStatus(strlen(kTestData1) + strlen(kTestData2), _)); | 783 CurrentUpdateStatus(strlen(kTestData1) + strlen(kTestData2), _)); |
792 | 784 |
793 sink_callback_.Run(); | 785 sink_callback_.Run(); |
794 base::RunLoop().RunUntilIdle(); | 786 base::RunLoop().RunUntilIdle(); |
795 VerifyStreamAndSize(); | 787 VerifyStreamAndSize(); |
796 DestroyDownloadFile(0); | 788 DestroyDownloadFile(0); |
797 } | 789 } |
798 | 790 |
799 } // namespace content | 791 } // namespace content |
OLD | NEW |