| 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 "components/drive/chromeos/remove_stale_cache_files.h" | 5 #include "components/drive/chromeos/remove_stale_cache_files.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| (...skipping 14 matching lines...) Expand all Loading... |
| 25 namespace internal { | 25 namespace internal { |
| 26 | 26 |
| 27 class RemoveStaleCacheFilesTest : public testing::Test { | 27 class RemoveStaleCacheFilesTest : public testing::Test { |
| 28 protected: | 28 protected: |
| 29 void SetUp() override { | 29 void SetUp() override { |
| 30 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 30 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 31 | 31 |
| 32 fake_free_disk_space_getter_.reset(new FakeFreeDiskSpaceGetter); | 32 fake_free_disk_space_getter_.reset(new FakeFreeDiskSpaceGetter); |
| 33 | 33 |
| 34 metadata_storage_.reset(new ResourceMetadataStorage( | 34 metadata_storage_.reset(new ResourceMetadataStorage( |
| 35 temp_dir_.path(), base::ThreadTaskRunnerHandle::Get().get())); | 35 temp_dir_.GetPath(), base::ThreadTaskRunnerHandle::Get().get())); |
| 36 | 36 |
| 37 cache_.reset(new FileCache(metadata_storage_.get(), | 37 cache_.reset(new FileCache(metadata_storage_.get(), temp_dir_.GetPath(), |
| 38 temp_dir_.path(), | |
| 39 base::ThreadTaskRunnerHandle::Get().get(), | 38 base::ThreadTaskRunnerHandle::Get().get(), |
| 40 fake_free_disk_space_getter_.get())); | 39 fake_free_disk_space_getter_.get())); |
| 41 | 40 |
| 42 resource_metadata_.reset(new ResourceMetadata( | 41 resource_metadata_.reset(new ResourceMetadata( |
| 43 metadata_storage_.get(), cache_.get(), | 42 metadata_storage_.get(), cache_.get(), |
| 44 base::ThreadTaskRunnerHandle::Get())); | 43 base::ThreadTaskRunnerHandle::Get())); |
| 45 | 44 |
| 46 ASSERT_TRUE(metadata_storage_->Initialize()); | 45 ASSERT_TRUE(metadata_storage_->Initialize()); |
| 47 ASSERT_TRUE(cache_->Initialize()); | 46 ASSERT_TRUE(cache_->Initialize()); |
| 48 ASSERT_EQ(FILE_ERROR_OK, resource_metadata_->Initialize()); | 47 ASSERT_EQ(FILE_ERROR_OK, resource_metadata_->Initialize()); |
| 49 } | 48 } |
| 50 | 49 |
| 51 content::TestBrowserThreadBundle thread_bundle_; | 50 content::TestBrowserThreadBundle thread_bundle_; |
| 52 base::ScopedTempDir temp_dir_; | 51 base::ScopedTempDir temp_dir_; |
| 53 | 52 |
| 54 std::unique_ptr<ResourceMetadataStorage, test_util::DestroyHelperForTests> | 53 std::unique_ptr<ResourceMetadataStorage, test_util::DestroyHelperForTests> |
| 55 metadata_storage_; | 54 metadata_storage_; |
| 56 std::unique_ptr<FileCache, test_util::DestroyHelperForTests> cache_; | 55 std::unique_ptr<FileCache, test_util::DestroyHelperForTests> cache_; |
| 57 std::unique_ptr<ResourceMetadata, test_util::DestroyHelperForTests> | 56 std::unique_ptr<ResourceMetadata, test_util::DestroyHelperForTests> |
| 58 resource_metadata_; | 57 resource_metadata_; |
| 59 std::unique_ptr<FakeFreeDiskSpaceGetter> fake_free_disk_space_getter_; | 58 std::unique_ptr<FakeFreeDiskSpaceGetter> fake_free_disk_space_getter_; |
| 60 }; | 59 }; |
| 61 | 60 |
| 62 TEST_F(RemoveStaleCacheFilesTest, RemoveStaleCacheFiles) { | 61 TEST_F(RemoveStaleCacheFilesTest, RemoveStaleCacheFiles) { |
| 63 base::FilePath dummy_file; | 62 base::FilePath dummy_file; |
| 64 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &dummy_file)); | 63 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.GetPath(), &dummy_file)); |
| 65 std::string md5_metadata("abcdef0123456789"), md5_cache("ABCDEF9876543210"); | 64 std::string md5_metadata("abcdef0123456789"), md5_cache("ABCDEF9876543210"); |
| 66 | 65 |
| 67 // Create a stale cache file. | 66 // Create a stale cache file. |
| 68 ResourceEntry entry; | 67 ResourceEntry entry; |
| 69 std::string local_id; | 68 std::string local_id; |
| 70 entry.mutable_file_specific_info()->set_md5(md5_metadata); | 69 entry.mutable_file_specific_info()->set_md5(md5_metadata); |
| 71 entry.set_parent_local_id(util::kDriveGrandRootLocalId); | 70 entry.set_parent_local_id(util::kDriveGrandRootLocalId); |
| 72 entry.set_title("File.txt"); | 71 entry.set_title("File.txt"); |
| 73 EXPECT_EQ(FILE_ERROR_OK, resource_metadata_->AddEntry(entry, &local_id)); | 72 EXPECT_EQ(FILE_ERROR_OK, resource_metadata_->AddEntry(entry, &local_id)); |
| 74 | 73 |
| 75 EXPECT_EQ(FILE_ERROR_OK, | 74 EXPECT_EQ(FILE_ERROR_OK, |
| 76 cache_->Store(local_id, md5_cache, dummy_file, | 75 cache_->Store(local_id, md5_cache, dummy_file, |
| 77 FileCache::FILE_OPERATION_COPY)); | 76 FileCache::FILE_OPERATION_COPY)); |
| 78 | 77 |
| 79 // Remove stale cache files. | 78 // Remove stale cache files. |
| 80 RemoveStaleCacheFiles(cache_.get(), resource_metadata_.get()); | 79 RemoveStaleCacheFiles(cache_.get(), resource_metadata_.get()); |
| 81 | 80 |
| 82 // Verify that the cache is deleted. | 81 // Verify that the cache is deleted. |
| 83 EXPECT_EQ(FILE_ERROR_OK, | 82 EXPECT_EQ(FILE_ERROR_OK, |
| 84 resource_metadata_->GetResourceEntryById(local_id, &entry)); | 83 resource_metadata_->GetResourceEntryById(local_id, &entry)); |
| 85 EXPECT_FALSE(entry.file_specific_info().cache_state().is_present()); | 84 EXPECT_FALSE(entry.file_specific_info().cache_state().is_present()); |
| 86 } | 85 } |
| 87 | 86 |
| 88 TEST_F(RemoveStaleCacheFilesTest, DirtyCacheFiles) { | 87 TEST_F(RemoveStaleCacheFilesTest, DirtyCacheFiles) { |
| 89 base::FilePath dummy_file; | 88 base::FilePath dummy_file; |
| 90 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &dummy_file)); | 89 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.GetPath(), &dummy_file)); |
| 91 | 90 |
| 92 // Dirty entry. | 91 // Dirty entry. |
| 93 std::string md5_metadata("abcdef0123456789"); | 92 std::string md5_metadata("abcdef0123456789"); |
| 94 | 93 |
| 95 ResourceEntry entry; | 94 ResourceEntry entry; |
| 96 std::string local_id; | 95 std::string local_id; |
| 97 entry.mutable_file_specific_info()->set_md5(md5_metadata); | 96 entry.mutable_file_specific_info()->set_md5(md5_metadata); |
| 98 entry.set_parent_local_id(util::kDriveGrandRootLocalId); | 97 entry.set_parent_local_id(util::kDriveGrandRootLocalId); |
| 99 entry.set_title("file.txt"); | 98 entry.set_title("file.txt"); |
| 100 EXPECT_EQ(FILE_ERROR_OK, resource_metadata_->AddEntry(entry, &local_id)); | 99 EXPECT_EQ(FILE_ERROR_OK, resource_metadata_->AddEntry(entry, &local_id)); |
| 101 | 100 |
| 102 EXPECT_EQ(FILE_ERROR_OK, | 101 EXPECT_EQ(FILE_ERROR_OK, |
| 103 cache_->Store(local_id, std::string(), dummy_file, | 102 cache_->Store(local_id, std::string(), dummy_file, |
| 104 FileCache::FILE_OPERATION_COPY)); | 103 FileCache::FILE_OPERATION_COPY)); |
| 105 | 104 |
| 106 // Remove stale cache files. | 105 // Remove stale cache files. |
| 107 RemoveStaleCacheFiles(cache_.get(), resource_metadata_.get()); | 106 RemoveStaleCacheFiles(cache_.get(), resource_metadata_.get()); |
| 108 | 107 |
| 109 // Dirty cache should not be removed even though its MD5 doesn't match. | 108 // Dirty cache should not be removed even though its MD5 doesn't match. |
| 110 EXPECT_EQ(FILE_ERROR_OK, | 109 EXPECT_EQ(FILE_ERROR_OK, |
| 111 resource_metadata_->GetResourceEntryById(local_id, &entry)); | 110 resource_metadata_->GetResourceEntryById(local_id, &entry)); |
| 112 EXPECT_TRUE(entry.file_specific_info().cache_state().is_present()); | 111 EXPECT_TRUE(entry.file_specific_info().cache_state().is_present()); |
| 113 } | 112 } |
| 114 | 113 |
| 115 } // namespace internal | 114 } // namespace internal |
| 116 } // namespace drive | 115 } // namespace drive |
| OLD | NEW |