| Index: chrome/browser/download/download_manager_unittest.cc
|
| diff --git a/chrome/browser/download/download_manager_unittest.cc b/chrome/browser/download/download_manager_unittest.cc
|
| index 271b1f98855c20ebb0c0bc89ef3a2c7a11f5e5cf..fe1376cbcbedb7dfb7b78b7fa609b097cab2a6d8 100644
|
| --- a/chrome/browser/download/download_manager_unittest.cc
|
| +++ b/chrome/browser/download/download_manager_unittest.cc
|
| @@ -10,6 +10,7 @@
|
| #include "base/stl_util-inl.h"
|
| #include "base/string_util.h"
|
| #include "build/build_config.h"
|
| +#include "chrome/browser/download/download_create_info.h"
|
| #include "chrome/browser/download/download_file.h"
|
| #include "chrome/browser/download/download_file_manager.h"
|
| #include "chrome/browser/download/download_item.h"
|
| @@ -18,7 +19,6 @@
|
| #include "chrome/browser/download/download_status_updater.h"
|
| #include "chrome/browser/download/download_util.h"
|
| #include "chrome/browser/download/mock_download_manager.h"
|
| -#include "chrome/browser/history/download_create_info.h"
|
| #include "chrome/browser/prefs/pref_service.h"
|
| #include "chrome/common/pref_names.h"
|
| #include "chrome/test/testing_profile.h"
|
| @@ -61,8 +61,8 @@ class DownloadManagerTest : public testing::Test {
|
| download_manager_->FileSelected(path, index, params);
|
| }
|
|
|
| - void AttachDownloadItem(DownloadCreateInfo* info) {
|
| - download_manager_->AttachDownloadItem(info);
|
| + void ContinueDownloadWithPath(DownloadItem* download, const FilePath& path) {
|
| + download_manager_->ContinueDownloadWithPath(download, path);
|
| }
|
|
|
| void OnDownloadError(int32 download_id, int64 size, int os_error) {
|
| @@ -312,11 +312,11 @@ TEST_F(DownloadManagerTest, StartDownload) {
|
| DownloadFile* download_file(new DownloadFile(info, download_manager_));
|
| AddDownloadToFileManager(info->download_id, download_file);
|
| download_file->Initialize(false);
|
| - download_manager_->StartDownload(info);
|
| + download_manager_->StartDownload(info->download_id);
|
| message_loop_.RunAllPending();
|
|
|
| - // NOTE: At this point, |AttachDownloadItem| will have been run if we don't
|
| - // need to prompt the user, so |info| could have been destructed.
|
| + // NOTE: At this point, |ContinueDownloadWithPath| will have been run if
|
| + // we don't need to prompt the user, so |info| could have been destructed.
|
| // This means that we can't check any of its values.
|
| // However, SelectFileObserver will have recorded any attempt to open the
|
| // select file dialog.
|
| @@ -324,7 +324,7 @@ TEST_F(DownloadManagerTest, StartDownload) {
|
| observer.ShowedFileDialogForId(i));
|
|
|
| // If the Save As dialog pops up, it never reached
|
| - // DownloadManager::AttachDownloadItem(), and never deleted info or
|
| + // DownloadManager::ContinueDownloadWithPath(), and never deleted info or
|
| // completed. This cleans up info.
|
| // Note that DownloadManager::FileSelectionCanceled() is never called.
|
| if (observer.ShowedFileDialogForId(i)) {
|
| @@ -347,7 +347,7 @@ TEST_F(DownloadManagerTest, DownloadRenameTest) {
|
| info->url_chain.push_back(GURL());
|
| info->is_dangerous_file = kDownloadRenameCases[i].is_dangerous_file;
|
| info->is_dangerous_url = kDownloadRenameCases[i].is_dangerous_url;
|
| - FilePath new_path(kDownloadRenameCases[i].suggested_path);
|
| + const FilePath new_path(kDownloadRenameCases[i].suggested_path);
|
|
|
| MockDownloadFile* download_file(
|
| new MockDownloadFile(info, download_manager_));
|
| @@ -372,12 +372,14 @@ TEST_F(DownloadManagerTest, DownloadRenameTest) {
|
| }
|
| download_manager_->CreateDownloadItem(info);
|
|
|
| + int32* id_ptr = new int32;
|
| + *id_ptr = i; // Deleted in FileSelected().
|
| if (kDownloadRenameCases[i].finish_before_rename) {
|
| OnAllDataSaved(i, 1024, std::string("fake_hash"));
|
| message_loop_.RunAllPending();
|
| - FileSelected(new_path, i, info);
|
| + FileSelected(new_path, i, id_ptr);
|
| } else {
|
| - FileSelected(new_path, i, info);
|
| + FileSelected(new_path, i, id_ptr);
|
| message_loop_.RunAllPending();
|
| OnAllDataSaved(i, 1024, std::string("fake_hash"));
|
| }
|
| @@ -425,8 +427,7 @@ TEST_F(DownloadManagerTest, DownloadInterruptTest) {
|
|
|
| download_file->AppendDataToFile(kTestData, kTestDataLen);
|
|
|
| - info->path = new_path;
|
| - AttachDownloadItem(info);
|
| + ContinueDownloadWithPath(download, new_path);
|
| message_loop_.RunAllPending();
|
| EXPECT_TRUE(GetActiveDownloadItem(0) != NULL);
|
|
|
| @@ -489,8 +490,7 @@ TEST_F(DownloadManagerTest, DownloadCancelTest) {
|
| EXPECT_EQ(DownloadItem::IN_PROGRESS, download->state());
|
| scoped_ptr<ItemObserver> observer(new ItemObserver(download));
|
|
|
| - info->path = new_path;
|
| - AttachDownloadItem(info);
|
| + ContinueDownloadWithPath(download, new_path);
|
| message_loop_.RunAllPending();
|
| EXPECT_TRUE(GetActiveDownloadItem(0) != NULL);
|
|
|
| @@ -568,8 +568,7 @@ TEST_F(DownloadManagerTest, DownloadOverwriteTest) {
|
| // |download_file| is owned by DownloadFileManager.
|
| AddDownloadToFileManager(info->download_id, download_file);
|
|
|
| - info->path = new_path;
|
| - AttachDownloadItem(info);
|
| + ContinueDownloadWithPath(download, new_path);
|
| message_loop_.RunAllPending();
|
| EXPECT_TRUE(GetActiveDownloadItem(0) != NULL);
|
|
|
|
|