chrome/browser/chromeos/drive/remove_stale_cache_files_unittest.cc

   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