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