file_system_usage_cache_unittest.cc 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. // Copyright 2014 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. #include "storage/browser/file_system/file_system_usage_cache.h"
  5. #include <stdint.h>
  6. #include <limits>
  7. #include "base/files/file_util.h"
  8. #include "base/files/scoped_temp_dir.h"
  9. #include "base/test/task_environment.h"
  10. #include "testing/gtest/include/gtest/gtest.h"
  11. namespace storage {
  12. class FileSystemUsageCacheTest : public testing::Test,
  13. public ::testing::WithParamInterface<bool> {
  14. public:
  15. FileSystemUsageCacheTest() : usage_cache_(is_incognito()) {}
  16. FileSystemUsageCacheTest(const FileSystemUsageCacheTest&) = delete;
  17. FileSystemUsageCacheTest& operator=(const FileSystemUsageCacheTest&) = delete;
  18. void SetUp() override { ASSERT_TRUE(data_dir_.CreateUniqueTempDir()); }
  19. bool is_incognito() { return GetParam(); }
  20. protected:
  21. base::FilePath GetUsageFilePath() {
  22. return data_dir_.GetPath().Append(FileSystemUsageCache::kUsageFileName);
  23. }
  24. FileSystemUsageCache* usage_cache() { return &usage_cache_; }
  25. private:
  26. base::test::SingleThreadTaskEnvironment task_environment_;
  27. base::ScopedTempDir data_dir_;
  28. FileSystemUsageCache usage_cache_;
  29. };
  30. INSTANTIATE_TEST_SUITE_P(All, FileSystemUsageCacheTest, testing::Bool());
  31. TEST_P(FileSystemUsageCacheTest, CreateTest) {
  32. base::FilePath usage_file_path = GetUsageFilePath();
  33. EXPECT_TRUE(usage_cache()->UpdateUsage(usage_file_path, 0));
  34. }
  35. TEST_P(FileSystemUsageCacheTest, SetSizeTest) {
  36. static const int64_t size = 240122;
  37. base::FilePath usage_file_path = GetUsageFilePath();
  38. int64_t usage = 0;
  39. ASSERT_TRUE(usage_cache()->UpdateUsage(usage_file_path, size));
  40. EXPECT_TRUE(usage_cache()->GetUsage(usage_file_path, &usage));
  41. EXPECT_EQ(size, usage);
  42. }
  43. TEST_P(FileSystemUsageCacheTest, SetLargeSizeTest) {
  44. static const int64_t size = std::numeric_limits<int64_t>::max();
  45. base::FilePath usage_file_path = GetUsageFilePath();
  46. int64_t usage = 0;
  47. ASSERT_TRUE(usage_cache()->UpdateUsage(usage_file_path, size));
  48. EXPECT_TRUE(usage_cache()->GetUsage(usage_file_path, &usage));
  49. EXPECT_EQ(size, usage);
  50. }
  51. TEST_P(FileSystemUsageCacheTest, IncAndGetSizeTest) {
  52. base::FilePath usage_file_path = GetUsageFilePath();
  53. uint32_t dirty = 0;
  54. int64_t usage = 0;
  55. ASSERT_TRUE(usage_cache()->UpdateUsage(usage_file_path, 98214));
  56. ASSERT_TRUE(usage_cache()->IncrementDirty(usage_file_path));
  57. EXPECT_TRUE(usage_cache()->GetDirty(usage_file_path, &dirty));
  58. EXPECT_EQ(1u, dirty);
  59. EXPECT_TRUE(usage_cache()->GetUsage(usage_file_path, &usage));
  60. EXPECT_EQ(98214, usage);
  61. }
  62. TEST_P(FileSystemUsageCacheTest, DecAndGetSizeTest) {
  63. static const int64_t size = 71839;
  64. base::FilePath usage_file_path = GetUsageFilePath();
  65. int64_t usage = 0;
  66. ASSERT_TRUE(usage_cache()->UpdateUsage(usage_file_path, size));
  67. // DecrementDirty for dirty = 0 is invalid. It returns false.
  68. ASSERT_FALSE(usage_cache()->DecrementDirty(usage_file_path));
  69. EXPECT_TRUE(usage_cache()->GetUsage(usage_file_path, &usage));
  70. EXPECT_EQ(size, usage);
  71. }
  72. TEST_P(FileSystemUsageCacheTest, IncDecAndGetSizeTest) {
  73. static const int64_t size = 198491;
  74. base::FilePath usage_file_path = GetUsageFilePath();
  75. int64_t usage = 0;
  76. ASSERT_TRUE(usage_cache()->UpdateUsage(usage_file_path, size));
  77. ASSERT_TRUE(usage_cache()->IncrementDirty(usage_file_path));
  78. ASSERT_TRUE(usage_cache()->DecrementDirty(usage_file_path));
  79. EXPECT_TRUE(usage_cache()->GetUsage(usage_file_path, &usage));
  80. EXPECT_EQ(size, usage);
  81. }
  82. TEST_P(FileSystemUsageCacheTest, DecIncAndGetSizeTest) {
  83. base::FilePath usage_file_path = GetUsageFilePath();
  84. uint32_t dirty = 0;
  85. int64_t usage = 0;
  86. ASSERT_TRUE(usage_cache()->UpdateUsage(usage_file_path, 854238));
  87. // DecrementDirty for dirty = 0 is invalid. It returns false.
  88. ASSERT_FALSE(usage_cache()->DecrementDirty(usage_file_path));
  89. ASSERT_TRUE(usage_cache()->IncrementDirty(usage_file_path));
  90. // It tests DecrementDirty (which returns false) has no effect, i.e
  91. // does not make dirty = -1 after DecrementDirty.
  92. EXPECT_TRUE(usage_cache()->GetDirty(usage_file_path, &dirty));
  93. EXPECT_EQ(1u, dirty);
  94. EXPECT_TRUE(usage_cache()->GetUsage(usage_file_path, &usage));
  95. EXPECT_EQ(854238, usage);
  96. }
  97. TEST_P(FileSystemUsageCacheTest, ManyIncsSameDecsAndGetSizeTest) {
  98. static const int64_t size = 82412;
  99. base::FilePath usage_file_path = GetUsageFilePath();
  100. int64_t usage = 0;
  101. ASSERT_TRUE(usage_cache()->UpdateUsage(usage_file_path, size));
  102. for (int i = 0; i < 20; i++)
  103. ASSERT_TRUE(usage_cache()->IncrementDirty(usage_file_path));
  104. for (int i = 0; i < 20; i++)
  105. ASSERT_TRUE(usage_cache()->DecrementDirty(usage_file_path));
  106. EXPECT_TRUE(usage_cache()->GetUsage(usage_file_path, &usage));
  107. EXPECT_EQ(size, usage);
  108. }
  109. TEST_P(FileSystemUsageCacheTest, ManyIncsLessDecsAndGetSizeTest) {
  110. uint32_t dirty = 0;
  111. int64_t usage = 0;
  112. base::FilePath usage_file_path = GetUsageFilePath();
  113. ASSERT_TRUE(usage_cache()->UpdateUsage(usage_file_path, 19319));
  114. for (int i = 0; i < 20; i++)
  115. ASSERT_TRUE(usage_cache()->IncrementDirty(usage_file_path));
  116. for (int i = 0; i < 19; i++)
  117. ASSERT_TRUE(usage_cache()->DecrementDirty(usage_file_path));
  118. EXPECT_TRUE(usage_cache()->GetDirty(usage_file_path, &dirty));
  119. EXPECT_EQ(1u, dirty);
  120. EXPECT_TRUE(usage_cache()->GetUsage(usage_file_path, &usage));
  121. EXPECT_EQ(19319, usage);
  122. }
  123. TEST_P(FileSystemUsageCacheTest, GetSizeWithoutCacheFileTest) {
  124. int64_t usage = 0;
  125. base::FilePath usage_file_path = GetUsageFilePath();
  126. EXPECT_FALSE(usage_cache()->GetUsage(usage_file_path, &usage));
  127. }
  128. TEST_P(FileSystemUsageCacheTest, IncrementDirtyWithoutCacheFileTest) {
  129. base::FilePath usage_file_path = GetUsageFilePath();
  130. EXPECT_FALSE(usage_cache()->IncrementDirty(usage_file_path));
  131. }
  132. TEST_P(FileSystemUsageCacheTest, DecrementDirtyWithoutCacheFileTest) {
  133. base::FilePath usage_file_path = GetUsageFilePath();
  134. EXPECT_FALSE(usage_cache()->IncrementDirty(usage_file_path));
  135. }
  136. } // namespace storage