Move Rename functionality from DownloadFileManager to DownloadFileImple.

content/browser/download/download_file_manager_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/browser/download/download_file_manager.h"
 
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/message_loop.h"
 #include "base/scoped_temp_dir.h"
@@ skipping 13 matching lines @@
 
 using content::BrowserThread;
 using content::BrowserThreadImpl;
 using content::DownloadId;
 using content::MockDownloadManager;
 
 using ::testing::_;
 using ::testing::AtLeast;
 using ::testing::Mock;
 using ::testing::Return;
+using ::testing::SaveArg;
 using ::testing::StrictMock;
 using ::testing::StrEq;
 
 namespace {
 
 // MockDownloadManager with the addition of a mock callback used for testing.
 class TestDownloadManager : public MockDownloadManager {
  public:
-  MOCK_METHOD2(OnDownloadRenamed,
-               void(int download_id, const FilePath& full_path));
+  MOCK_METHOD3(OnDownloadRenamed,
+               void(int download_id,
+                    content::DownloadInterruptReason reason,
+                    const FilePath& full_path));
  private:
   ~TestDownloadManager() {}
 };
 
 class MockDownloadFileFactory :
     public DownloadFileManager::DownloadFileFactory {
 
  public:
   MockDownloadFileFactory() {}
   virtual ~MockDownloadFileFactory() {}
@@ skipping 51 matching lines @@
   return manager_;
 }
 
 void NullCallback() { }
 
 }  // namespace
 
 class DownloadFileManagerTest : public testing::Test {
  public:
   // State of renamed file. Used with RenameFile().
   enum RenameFileState {

[asanka, 2012/07/10 18:11:45]

Remove.

     IN_PROGRESS,
     COMPLETE
   };
 
   // Whether to overwrite the target filename in RenameFile().
   enum RenameFileOverwrite {

[asanka, 2012/07/10 18:11:45]

Remove.

     OVERWRITE,
     DONT_OVERWRITE
   };
 
   static const char* kTestData1;
   static const char* kTestData2;
   static const char* kTestData3;
   static const char* kTestData4;
   static const char* kTestData5;
   static const char* kTestData6;
@@ skipping 73 matching lines @@
   //                renamed to. If there is an existing file at
   //                |new_path| and |replace| is false, then |new_path|
   //                will be uniquified.
   // |rename_error| is the error to inject.  For no error,
   //                use content::DOWNLOAD_INTERRUPT_REASON_NONE.
   // |state|        whether we are renaming an in-progress download or a
   //                completed download.
   // |should_overwrite| indicates whether to replace or uniquify the file.
   void RenameFile(const DownloadId& id,
                   const FilePath& new_path,
-                  const FilePath& unique_path,
-                  content::DownloadInterruptReason rename_error,
-                  RenameFileState state,
-                  RenameFileOverwrite should_overwrite) {
+                  bool should_overwrite) {
     MockDownloadFile* file = download_file_factory_->GetExistingFile(id);
     ASSERT_TRUE(file != NULL);
+    content::DownloadFile::RenameCompletionCallback rename_callback;
 
-    EXPECT_CALL(*file, Rename(unique_path))
-        .Times(1)
-        .WillOnce(Return(rename_error));
+    EXPECT_CALL(*file, Rename(new_path, should_overwrite, _))
+        .WillOnce(SaveArg<2>(&rename_callback));
 
-    if (rename_error != content::DOWNLOAD_INTERRUPT_REASON_NONE) {
-      EXPECT_CALL(*file, BytesSoFar())
-          .Times(AtLeast(1))
-          .WillRepeatedly(Return(byte_count_[id]));
-      EXPECT_CALL(*file, GetHashState())
-          .Times(AtLeast(1));
-      EXPECT_CALL(*file, GetDownloadManager())
-          .Times(AtLeast(1));
-    }
-
-    download_file_manager_->RenameDownloadFile(
-        id, new_path, (should_overwrite == OVERWRITE),
+    content::DownloadFile::RenameCompletionCallback passed_callback(
         base::Bind(&TestDownloadManager::OnDownloadRenamed,
                    download_manager_, id.local()));
 
-    if (rename_error != content::DOWNLOAD_INTERRUPT_REASON_NONE) {
-      EXPECT_CALL(*download_manager_,
-                  OnDownloadInterrupted(
-                      id.local(),
-                      byte_count_[id],
-                      "",
-                      rename_error));
-      EXPECT_CALL(*download_manager_,
-                  OnDownloadRenamed(id.local(), FilePath()));
-      ProcessAllPendingMessages();
-      ++error_count_[id];
-    } else {
-      EXPECT_CALL(*download_manager_,
-                  OnDownloadRenamed(id.local(), unique_path));
-      ProcessAllPendingMessages();
-    }
+    download_file_manager_->RenameDownloadFile(
+        id, new_path, should_overwrite, passed_callback);
+
+    EXPECT_TRUE(rename_callback.Equals(passed_callback));
   }
 
   void Complete(DownloadId id) {
     MockDownloadFile* file = download_file_factory_->GetExistingFile(id);
     ASSERT_TRUE(file != NULL);
 
     EXPECT_CALL(*file, AnnotateWithSourceInformation())
         .WillOnce(Return());
     EXPECT_CALL(*file, Detach())
         .WillOnce(Return());
@@ skipping 71 matching lines @@
 TEST_F(DownloadFileManagerTest, Complete) {
   scoped_ptr<DownloadCreateInfo> info(new DownloadCreateInfo);
   DownloadId dummy_id(download_manager_.get(), kDummyDownloadId);
   info->download_id = dummy_id;
 
   CreateDownloadFile(info.Pass());
 
   Complete(dummy_id);
 }
 
-TEST_F(DownloadFileManagerTest, RenameInProgress) {
+TEST_F(DownloadFileManagerTest, Rename) {
   scoped_ptr<DownloadCreateInfo> info(new DownloadCreateInfo);
   DownloadId dummy_id(download_manager_.get(), kDummyDownloadId);
   info->download_id = dummy_id;
-  ScopedTempDir download_dir;
-  ASSERT_TRUE(download_dir.CreateUniqueTempDir());
-
-  CreateDownloadFile(info.Pass());
-
-  FilePath foo(download_dir.path().Append(FILE_PATH_LITERAL("foo.txt")));
-  RenameFile(dummy_id, foo, foo, content::DOWNLOAD_INTERRUPT_REASON_NONE,
-             IN_PROGRESS, OVERWRITE);
-
-  CleanUp(dummy_id);
-}
-
-TEST_F(DownloadFileManagerTest, RenameInProgressWithUniquification) {
-  scoped_ptr<DownloadCreateInfo> info(new DownloadCreateInfo);
-  DownloadId dummy_id(download_manager_.get(), kDummyDownloadId);
-  info->download_id = dummy_id;
-  ScopedTempDir download_dir;
-  ASSERT_TRUE(download_dir.CreateUniqueTempDir());
-
-  CreateDownloadFile(info.Pass());
-
-  FilePath foo(download_dir.path().Append(FILE_PATH_LITERAL("foo.txt")));
-  FilePath unique_foo(foo.InsertBeforeExtension(FILE_PATH_LITERAL(" (1)")));
-  ASSERT_EQ(0, file_util::WriteFile(foo, "", 0));
-  RenameFile(dummy_id, foo, unique_foo,
-             content::DOWNLOAD_INTERRUPT_REASON_NONE, IN_PROGRESS,
-             DONT_OVERWRITE);
-
-  CleanUp(dummy_id);
-}
-
-TEST_F(DownloadFileManagerTest, RenameInProgressWithError) {
-  scoped_ptr<DownloadCreateInfo> info(new DownloadCreateInfo);
-  DownloadId dummy_id(download_manager_.get(), kDummyDownloadId);
-  info->download_id = dummy_id;
   ScopedTempDir download_dir;
   ASSERT_TRUE(download_dir.CreateUniqueTempDir());
 
   CreateDownloadFile(info.Pass());
 
   FilePath foo(download_dir.path().Append(FILE_PATH_LITERAL("foo.txt")));
-  RenameFile(dummy_id, foo, foo,
-             content::DOWNLOAD_INTERRUPT_REASON_FILE_NAME_TOO_LONG,
-             IN_PROGRESS, OVERWRITE);
+  RenameFile(dummy_id, foo, true);
 
   CleanUp(dummy_id);
 }
 
-TEST_F(DownloadFileManagerTest, RenameWithUniquification) {
+TEST_F(DownloadFileManagerTest, RenameNoOverwrite) {
   scoped_ptr<DownloadCreateInfo> info(new DownloadCreateInfo);
   DownloadId dummy_id(download_manager_.get(), kDummyDownloadId);
   info->download_id = dummy_id;
   ScopedTempDir download_dir;
   ASSERT_TRUE(download_dir.CreateUniqueTempDir());
 
   CreateDownloadFile(info.Pass());
 
   FilePath foo(download_dir.path().Append(FILE_PATH_LITERAL("foo.txt")));
-  FilePath unique_foo(foo.InsertBeforeExtension(FILE_PATH_LITERAL(" (1)")));
-  // Create a file at |foo|. Since we are specifying DONT_OVERWRITE,
-  // RenameDownloadFile() should pick "foo (1).txt" instead of
-  // overwriting this file.
-  ASSERT_EQ(0, file_util::WriteFile(foo, "", 0));
-  RenameFile(dummy_id, foo, unique_foo,
-             content::DOWNLOAD_INTERRUPT_REASON_NONE, COMPLETE, DONT_OVERWRITE);
+  RenameFile(dummy_id, foo, false);
 
   CleanUp(dummy_id);
 }
 
 TEST_F(DownloadFileManagerTest, RenameTwice) {
   scoped_ptr<DownloadCreateInfo> info(new DownloadCreateInfo);
   DownloadId dummy_id(download_manager_.get(), kDummyDownloadId);
   info->download_id = dummy_id;
   ScopedTempDir download_dir;
   ASSERT_TRUE(download_dir.CreateUniqueTempDir());
 
   CreateDownloadFile(info.Pass());
 
   FilePath crfoo(download_dir.path().Append(
       FILE_PATH_LITERAL("foo.txt.crdownload")));
-  RenameFile(dummy_id, crfoo, crfoo, content::DOWNLOAD_INTERRUPT_REASON_NONE,
-             IN_PROGRESS, OVERWRITE);
+  RenameFile(dummy_id, crfoo, true);
 
   FilePath foo(download_dir.path().Append(FILE_PATH_LITERAL("foo.txt")));
-  RenameFile(dummy_id, foo, foo, content::DOWNLOAD_INTERRUPT_REASON_NONE,
-             COMPLETE, OVERWRITE);
+  RenameFile(dummy_id, foo, true);
 
   CleanUp(dummy_id);
 }
 
 TEST_F(DownloadFileManagerTest, TwoDownloads) {
   // Same as StartDownload, at first.
   scoped_ptr<DownloadCreateInfo> info(new DownloadCreateInfo);
   DownloadId dummy_id(download_manager_.get(), kDummyDownloadId);
   info->download_id = dummy_id;
   scoped_ptr<DownloadCreateInfo> info2(new DownloadCreateInfo);
   DownloadId dummy_id2(download_manager_.get(), kDummyDownloadId2);
   info2->download_id = dummy_id2;
   ScopedTempDir download_dir;
   ASSERT_TRUE(download_dir.CreateUniqueTempDir());
 
   CreateDownloadFile(info.Pass());
   CreateDownloadFile(info2.Pass());
 
   FilePath crbar(download_dir.path().Append(
       FILE_PATH_LITERAL("bar.txt.crdownload")));
-  RenameFile(dummy_id2, crbar, crbar, content::DOWNLOAD_INTERRUPT_REASON_NONE,
-             IN_PROGRESS, OVERWRITE);
+  RenameFile(dummy_id2, crbar, true);
 
   FilePath crfoo(download_dir.path().Append(
       FILE_PATH_LITERAL("foo.txt.crdownload")));
-  RenameFile(dummy_id, crfoo, crfoo, content::DOWNLOAD_INTERRUPT_REASON_NONE,
-             IN_PROGRESS, OVERWRITE);
+  RenameFile(dummy_id, crfoo, true);
 
 
   FilePath bar(download_dir.path().Append(FILE_PATH_LITERAL("bar.txt")));
-  RenameFile(dummy_id2, bar, bar, content::DOWNLOAD_INTERRUPT_REASON_NONE,
-             COMPLETE, OVERWRITE);
+  RenameFile(dummy_id2, bar, true);
 
   CleanUp(dummy_id2);
 
   FilePath foo(download_dir.path().Append(FILE_PATH_LITERAL("foo.txt")));
-  RenameFile(dummy_id, foo, foo, content::DOWNLOAD_INTERRUPT_REASON_NONE,
-             COMPLETE, OVERWRITE);
+  RenameFile(dummy_id, foo, true);
 
   CleanUp(dummy_id);
 }
 
 // TODO(ahendrickson) -- A test for download manager shutdown.
 // Expected call sequence:
 //  OnDownloadManagerShutdown
 //    DownloadFile::GetDownloadManager
 //    DownloadFile::CancelDownloadRequest
 //    DownloadFile::~DownloadFile