Index: chrome/browser/download/download_file_unittest.cc |
=================================================================== |
--- chrome/browser/download/download_file_unittest.cc (revision 96793) |
+++ chrome/browser/download/download_file_unittest.cc (working copy) |
@@ -1,196 +0,0 @@ |
-// Copyright (c) 2011 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 "base/file_util.h" |
-#include "base/message_loop.h" |
-#include "base/scoped_temp_dir.h" |
-#include "base/string_number_conversions.h" |
-#include "chrome/browser/download/download_create_info.h" |
-#include "chrome/browser/download/download_file.h" |
-#include "chrome/browser/download/download_manager.h" |
-#include "chrome/browser/download/download_request_handle.h" |
-#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/download/mock_download_manager_delegate.h" |
-#include "content/browser/browser_thread.h" |
-#include "net/base/file_stream.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-class DownloadFileTest : public testing::Test { |
- public: |
- |
- static const char* kTestData1; |
- static const char* kTestData2; |
- static const char* kTestData3; |
- static const char* kDataHash; |
- static const int32 kDummyDownloadId; |
- static const int kDummyChildId; |
- static const int kDummyRequestId; |
- |
- // We need a UI |BrowserThread| in order to destruct |download_manager_|, |
- // which has trait |BrowserThread::DeleteOnUIThread|. Without this, |
- // calling Release() on |download_manager_| won't ever result in its |
- // destructor being called and we get a leak. |
- DownloadFileTest() : |
- ui_thread_(BrowserThread::UI, &loop_), |
- file_thread_(BrowserThread::FILE, &loop_) { |
- } |
- |
- ~DownloadFileTest() { |
- } |
- |
- virtual void SetUp() { |
- ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
- download_manager_delegate_.reset(new MockDownloadManagerDelegate()); |
- download_manager_ = new MockDownloadManager( |
- download_manager_delegate_.get(), &download_status_updater_); |
- } |
- |
- virtual void TearDown() { |
- // When a DownloadManager's reference count drops to 0, it is not |
- // deleted immediately. Instead, a task is posted to the UI thread's |
- // message loop to delete it. |
- // So, drop the reference count to 0 and run the message loop once |
- // to ensure that all resources are cleaned up before the test exits. |
- download_manager_ = NULL; |
- ui_thread_.message_loop()->RunAllPending(); |
- } |
- |
- virtual void CreateDownloadFile(scoped_ptr<DownloadFile>* file, int offset) { |
- DownloadCreateInfo info; |
- info.download_id = kDummyDownloadId + offset; |
- // info.request_handle default constructed to null. |
- info.save_info.file_stream = file_stream_; |
- file->reset(new DownloadFile(&info, download_manager_)); |
- } |
- |
- virtual void DestroyDownloadFile(scoped_ptr<DownloadFile>* file, int offset) { |
- EXPECT_EQ(kDummyDownloadId + offset, (*file)->id()); |
- EXPECT_EQ(download_manager_, (*file)->GetDownloadManager()); |
- EXPECT_FALSE((*file)->in_progress()); |
- EXPECT_EQ(static_cast<int64>(expected_data_.size()), |
- (*file)->bytes_so_far()); |
- |
- // Make sure the data has been properly written to disk. |
- std::string disk_data; |
- EXPECT_TRUE(file_util::ReadFileToString((*file)->full_path(), |
- &disk_data)); |
- EXPECT_EQ(expected_data_, disk_data); |
- |
- // Make sure the mock BrowserThread outlives the DownloadFile to satisfy |
- // thread checks inside it. |
- file->reset(); |
- } |
- |
- void AppendDataToFile(scoped_ptr<DownloadFile>* file, |
- const std::string& data) { |
- EXPECT_TRUE((*file)->in_progress()); |
- (*file)->AppendDataToFile(data.data(), data.size()); |
- expected_data_ += data; |
- EXPECT_EQ(static_cast<int64>(expected_data_.size()), |
- (*file)->bytes_so_far()); |
- } |
- |
- protected: |
- // Temporary directory for renamed downloads. |
- ScopedTempDir temp_dir_; |
- |
- DownloadStatusUpdater download_status_updater_; |
- scoped_ptr<MockDownloadManagerDelegate> download_manager_delegate_; |
- scoped_refptr<DownloadManager> download_manager_; |
- |
- linked_ptr<net::FileStream> file_stream_; |
- |
- // DownloadFile instance we are testing. |
- scoped_ptr<DownloadFile> download_file_; |
- |
- private: |
- MessageLoop loop_; |
- // UI thread. |
- BrowserThread ui_thread_; |
- // File thread to satisfy debug checks in DownloadFile. |
- BrowserThread file_thread_; |
- |
- // Keep track of what data should be saved to the disk file. |
- std::string expected_data_; |
-}; |
- |
-const char* DownloadFileTest::kTestData1 = |
- "Let's write some data to the file!\n"; |
-const char* DownloadFileTest::kTestData2 = "Writing more data.\n"; |
-const char* DownloadFileTest::kTestData3 = "Final line."; |
-const char* DownloadFileTest::kDataHash = |
- "CBF68BF10F8003DB86B31343AFAC8C7175BD03FB5FC905650F8C80AF087443A8"; |
- |
-const int32 DownloadFileTest::kDummyDownloadId = 23; |
-const int DownloadFileTest::kDummyChildId = 3; |
-const int DownloadFileTest::kDummyRequestId = 67; |
- |
-// Rename the file before any data is downloaded, after some has, after it all |
-// has, and after it's closed. |
-TEST_F(DownloadFileTest, RenameFileFinal) { |
- CreateDownloadFile(&download_file_, 0); |
- ASSERT_TRUE(download_file_->Initialize(true)); |
- FilePath initial_path(download_file_->full_path()); |
- EXPECT_TRUE(file_util::PathExists(initial_path)); |
- FilePath path_1(initial_path.InsertBeforeExtensionASCII("_1")); |
- FilePath path_2(initial_path.InsertBeforeExtensionASCII("_2")); |
- FilePath path_3(initial_path.InsertBeforeExtensionASCII("_3")); |
- FilePath path_4(initial_path.InsertBeforeExtensionASCII("_4")); |
- |
- // Rename the file before downloading any data. |
- EXPECT_TRUE(download_file_->Rename(path_1)); |
- FilePath renamed_path = download_file_->full_path(); |
- EXPECT_EQ(path_1, renamed_path); |
- |
- // Check the files. |
- EXPECT_FALSE(file_util::PathExists(initial_path)); |
- EXPECT_TRUE(file_util::PathExists(path_1)); |
- |
- // Download the data. |
- AppendDataToFile(&download_file_, kTestData1); |
- AppendDataToFile(&download_file_, kTestData2); |
- |
- // Rename the file after downloading some data. |
- EXPECT_TRUE(download_file_->Rename(path_2)); |
- renamed_path = download_file_->full_path(); |
- EXPECT_EQ(path_2, renamed_path); |
- |
- // Check the files. |
- EXPECT_FALSE(file_util::PathExists(path_1)); |
- EXPECT_TRUE(file_util::PathExists(path_2)); |
- |
- AppendDataToFile(&download_file_, kTestData3); |
- |
- // Rename the file after downloading all the data. |
- EXPECT_TRUE(download_file_->Rename(path_3)); |
- renamed_path = download_file_->full_path(); |
- EXPECT_EQ(path_3, renamed_path); |
- |
- // Check the files. |
- EXPECT_FALSE(file_util::PathExists(path_2)); |
- EXPECT_TRUE(file_util::PathExists(path_3)); |
- |
- // Should not be able to get the hash until the file is closed. |
- std::string hash; |
- EXPECT_FALSE(download_file_->GetSha256Hash(&hash)); |
- |
- download_file_->Finish(); |
- |
- // Rename the file after downloading all the data and closing the file. |
- EXPECT_TRUE(download_file_->Rename(path_4)); |
- renamed_path = download_file_->full_path(); |
- EXPECT_EQ(path_4, renamed_path); |
- |
- // Check the files. |
- EXPECT_FALSE(file_util::PathExists(path_3)); |
- EXPECT_TRUE(file_util::PathExists(path_4)); |
- |
- // Check the hash. |
- EXPECT_TRUE(download_file_->GetSha256Hash(&hash)); |
- EXPECT_EQ(kDataHash, base::HexEncode(hash.data(), hash.size())); |
- |
- DestroyDownloadFile(&download_file_, 0); |
-} |