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 <stdint.h> | 5 #include <stdint.h> |
6 | 6 |
7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
8 #include "base/files/file_util.h" | 8 #include "base/files/file_util.h" |
9 #include "base/metrics/field_trial.h" | 9 #include "base/metrics/field_trial.h" |
10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
(...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
82 // will be filled with times, used by DoomEntriesSince and DoomEntriesBetween. | 82 // will be filled with times, used by DoomEntriesSince and DoomEntriesBetween. |
83 // There are 4 entries after doomed_start and 2 after doomed_end. | 83 // There are 4 entries after doomed_start and 2 after doomed_end. |
84 void InitSparseCache(base::Time* doomed_start, base::Time* doomed_end); | 84 void InitSparseCache(base::Time* doomed_start, base::Time* doomed_end); |
85 | 85 |
86 bool CreateSetOfRandomEntries(std::set<std::string>* key_pool); | 86 bool CreateSetOfRandomEntries(std::set<std::string>* key_pool); |
87 bool EnumerateAndMatchKeys(int max_to_open, | 87 bool EnumerateAndMatchKeys(int max_to_open, |
88 TestIterator* iter, | 88 TestIterator* iter, |
89 std::set<std::string>* keys_to_match, | 89 std::set<std::string>* keys_to_match, |
90 size_t* count); | 90 size_t* count); |
91 | 91 |
92 // Computes the expected size of entry metadata, i.e. the total size without | |
93 // the actual data stored. This depends only on the entry's |key| size. | |
94 int GetEntryMetadataSize(std::string key); | |
95 | |
96 // Actual tests: | 92 // Actual tests: |
97 void BackendBasics(); | 93 void BackendBasics(); |
98 void BackendKeying(); | 94 void BackendKeying(); |
99 void BackendShutdownWithPendingFileIO(bool fast); | 95 void BackendShutdownWithPendingFileIO(bool fast); |
100 void BackendShutdownWithPendingIO(bool fast); | 96 void BackendShutdownWithPendingIO(bool fast); |
101 void BackendShutdownWithPendingCreate(bool fast); | 97 void BackendShutdownWithPendingCreate(bool fast); |
102 void BackendSetSize(); | 98 void BackendSetSize(); |
103 void BackendLoad(); | 99 void BackendLoad(); |
104 void BackendChain(); | 100 void BackendChain(); |
105 void BackendValidEntry(); | 101 void BackendValidEntry(); |
(...skipping 173 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
279 EXPECT_EQ(1U, keys_to_match->erase(entry->GetKey())); | 275 EXPECT_EQ(1U, keys_to_match->erase(entry->GetKey())); |
280 entry->Close(); | 276 entry->Close(); |
281 ++(*count); | 277 ++(*count); |
282 if (max_to_open >= 0 && static_cast<int>(*count) >= max_to_open) | 278 if (max_to_open >= 0 && static_cast<int>(*count) >= max_to_open) |
283 break; | 279 break; |
284 }; | 280 }; |
285 | 281 |
286 return true; | 282 return true; |
287 } | 283 } |
288 | 284 |
289 int DiskCacheBackendTest::GetEntryMetadataSize(std::string key) { | |
290 // For blockfile and memory backends, it is just the key size. | |
291 if (!simple_cache_mode_) | |
292 return key.size(); | |
293 | |
294 // For the simple cache, we must add the file header and EOF, and that for | |
295 // every stream. | |
296 return disk_cache::kSimpleEntryStreamCount * | |
297 (sizeof(disk_cache::SimpleFileHeader) + | |
298 sizeof(disk_cache::SimpleFileEOF) + key.size()); | |
299 } | |
300 | |
301 void DiskCacheBackendTest::BackendBasics() { | 285 void DiskCacheBackendTest::BackendBasics() { |
302 InitCache(); | 286 InitCache(); |
303 disk_cache::Entry *entry1 = NULL, *entry2 = NULL; | 287 disk_cache::Entry *entry1 = NULL, *entry2 = NULL; |
304 EXPECT_NE(net::OK, OpenEntry("the first key", &entry1)); | 288 EXPECT_NE(net::OK, OpenEntry("the first key", &entry1)); |
305 ASSERT_EQ(net::OK, CreateEntry("the first key", &entry1)); | 289 ASSERT_EQ(net::OK, CreateEntry("the first key", &entry1)); |
306 ASSERT_TRUE(NULL != entry1); | 290 ASSERT_TRUE(NULL != entry1); |
307 entry1->Close(); | 291 entry1->Close(); |
308 entry1 = NULL; | 292 entry1 = NULL; |
309 | 293 |
310 ASSERT_EQ(net::OK, OpenEntry("the first key", &entry1)); | 294 ASSERT_EQ(net::OK, OpenEntry("the first key", &entry1)); |
(...skipping 1377 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1688 start = end; | 1672 start = end; |
1689 end = base::Time::Now(); | 1673 end = base::Time::Now(); |
1690 DoomEntriesBetween(start, end); | 1674 DoomEntriesBetween(start, end); |
1691 EXPECT_EQ(3, cache_->GetEntryCount()); | 1675 EXPECT_EQ(3, cache_->GetEntryCount()); |
1692 } | 1676 } |
1693 | 1677 |
1694 void DiskCacheBackendTest::BackendCalculateSizeOfAllEntries() { | 1678 void DiskCacheBackendTest::BackendCalculateSizeOfAllEntries() { |
1695 InitCache(); | 1679 InitCache(); |
1696 | 1680 |
1697 // The cache is initially empty. | 1681 // The cache is initially empty. |
1698 EXPECT_EQ(0, CalculateSizeOfAllEntries()); | 1682 if (memory_only_ || simple_cache_mode_) { |
| 1683 // TODO(msramek): Implement. |
| 1684 EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, CalculateSizeOfAllEntries()); |
| 1685 } else { |
| 1686 EXPECT_EQ(0, CalculateSizeOfAllEntries()); |
| 1687 } |
1699 | 1688 |
1700 // Generate random entries and populate them with data of respective | 1689 // Generate random entries and populate them with data of respective |
1701 // sizes 0, 1, ..., count - 1 bytes. | 1690 // sizes 0, 1, ..., count - 1 bytes. |
1702 std::set<std::string> key_pool; | 1691 std::set<std::string> key_pool; |
1703 CreateSetOfRandomEntries(&key_pool); | 1692 CreateSetOfRandomEntries(&key_pool); |
1704 | 1693 |
1705 int count = 0; | 1694 int count = 0; |
1706 for (std::string key : key_pool) { | 1695 for (std::string key : key_pool) { |
1707 std::string data(count, ' '); | 1696 std::string data(count, ' '); |
1708 scoped_refptr<net::StringIOBuffer> buffer = new net::StringIOBuffer(data); | 1697 scoped_refptr<net::StringIOBuffer> buffer = new net::StringIOBuffer(data); |
1709 | 1698 |
1710 // Alternate between writing to the first and second stream to test that | 1699 // Alternate between writing to the first and second stream to test that |
1711 // we are not taking just the first stream into account. For convenience, | 1700 // we are not taking just the first stream into account. |
1712 // the last written stream should be 0. This is because writing to | |
1713 // the stream 1 in simple cache triggers a write to the stream 0 as well. | |
1714 // This will happen asynchronously and possibly later than our call to | |
1715 // |CalculateSizeOfAllEntries|. | |
1716 disk_cache::Entry* entry; | 1701 disk_cache::Entry* entry; |
1717 ASSERT_EQ(net::OK, OpenEntry(key, &entry)); | 1702 ASSERT_EQ(net::OK, OpenEntry(key, &entry)); |
1718 ASSERT_EQ(count, | 1703 ASSERT_EQ(count, WriteData(entry, count % 2, 0, buffer.get(), count, true)); |
1719 WriteData(entry, (count + 1) % 2, 0, buffer.get(), count, true)); | |
1720 entry->Close(); | 1704 entry->Close(); |
1721 | 1705 |
1722 ++count; | 1706 ++count; |
1723 } | 1707 } |
1724 | 1708 |
1725 // The resulting size should be (0 + 1 + ... + count - 1) plus keys. | 1709 // The resulting size should be (0 + 1 + ... + count - 1) plus keys. |
1726 int result = CalculateSizeOfAllEntries(); | 1710 int result = CalculateSizeOfAllEntries(); |
1727 int total_metadata_size = 0; | 1711 if (memory_only_ || simple_cache_mode_) { |
1728 for (std::string key : key_pool) | 1712 // TODO(msramek): Implement. |
1729 total_metadata_size += GetEntryMetadataSize(key); | 1713 EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, result); |
1730 EXPECT_EQ((count - 1) * count / 2 + total_metadata_size, result); | 1714 } else { |
| 1715 int total_key_size = 0; |
| 1716 for (std::string key : key_pool) |
| 1717 total_key_size += key.size(); |
| 1718 |
| 1719 EXPECT_EQ((count - 1) * count / 2 + total_key_size, result); |
| 1720 } |
1731 | 1721 |
1732 // Add another entry and test if the size is updated. Then remove it and test | 1722 // Add another entry and test if the size is updated. Then remove it and test |
1733 // if the size is back to original value. | 1723 // if the size is back to original value. |
1734 { | 1724 { |
1735 const int last_entry_size = 47; | 1725 const int last_entry_size = 47; |
1736 std::string data(last_entry_size, ' '); | 1726 std::string data(last_entry_size, ' '); |
1737 scoped_refptr<net::StringIOBuffer> buffer = new net::StringIOBuffer(data); | 1727 scoped_refptr<net::StringIOBuffer> buffer = new net::StringIOBuffer(data); |
1738 | 1728 |
1739 disk_cache::Entry* entry; | 1729 disk_cache::Entry* entry; |
1740 std::string key = GenerateKey(true); | 1730 std::string key = GenerateKey(true); |
1741 ASSERT_EQ(net::OK, CreateEntry(key, &entry)); | 1731 ASSERT_EQ(net::OK, CreateEntry(key, &entry)); |
1742 ASSERT_EQ(last_entry_size, | 1732 ASSERT_EQ(last_entry_size, |
1743 WriteData(entry, 0, 0, buffer.get(), last_entry_size, true)); | 1733 WriteData(entry, 0, 0, buffer.get(), last_entry_size, true)); |
1744 entry->Close(); | 1734 entry->Close(); |
1745 | 1735 |
1746 int new_result = CalculateSizeOfAllEntries(); | 1736 int new_result = CalculateSizeOfAllEntries(); |
1747 EXPECT_EQ(result + last_entry_size + GetEntryMetadataSize(key), new_result); | 1737 if (memory_only_ || simple_cache_mode_) { |
| 1738 // TODO(msramek): Implement. |
| 1739 EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, new_result); |
| 1740 } else { |
| 1741 EXPECT_EQ(result + last_entry_size + static_cast<int>(key.size()), |
| 1742 new_result); |
| 1743 } |
1748 | 1744 |
1749 DoomEntry(key); | 1745 DoomEntry(key); |
1750 new_result = CalculateSizeOfAllEntries(); | 1746 new_result = CalculateSizeOfAllEntries(); |
1751 EXPECT_EQ(result, new_result); | 1747 if (memory_only_ || simple_cache_mode_) { |
| 1748 // TODO(msramek): Implement. |
| 1749 EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, new_result); |
| 1750 } else { |
| 1751 EXPECT_EQ(result, new_result); |
| 1752 } |
1752 } | 1753 } |
1753 | 1754 |
1754 // After dooming the entries, the size should be back to zero. | 1755 // After dooming the entries, the size should be back to zero. |
1755 ASSERT_EQ(net::OK, DoomAllEntries()); | 1756 ASSERT_EQ(net::OK, DoomAllEntries()); |
1756 EXPECT_EQ(0, CalculateSizeOfAllEntries()); | 1757 if (memory_only_ || simple_cache_mode_) { |
| 1758 // TODO(msramek): Implement. |
| 1759 EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, CalculateSizeOfAllEntries()); |
| 1760 } else { |
| 1761 EXPECT_EQ(0, CalculateSizeOfAllEntries()); |
| 1762 } |
1757 } | 1763 } |
1758 | 1764 |
1759 TEST_F(DiskCacheBackendTest, CalculateSizeOfAllEntries) { | 1765 TEST_F(DiskCacheBackendTest, CalculateSizeOfAllEntries) { |
1760 BackendCalculateSizeOfAllEntries(); | 1766 BackendCalculateSizeOfAllEntries(); |
1761 } | 1767 } |
1762 | 1768 |
1763 TEST_F(DiskCacheBackendTest, MemoryOnlyCalculateSizeOfAllEntries) { | 1769 TEST_F(DiskCacheBackendTest, MemoryOnlyCalculateSizeOfAllEntries) { |
1764 SetMemoryOnlyMode(); | 1770 SetMemoryOnlyMode(); |
1765 BackendCalculateSizeOfAllEntries(); | 1771 BackendCalculateSizeOfAllEntries(); |
1766 } | 1772 } |
(...skipping 1852 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
3619 // after closing. | 3625 // after closing. |
3620 // NOTE: IF THIS TEST IS FLAKY THEN IT IS FAILING. See https://crbug.com/416940 | 3626 // NOTE: IF THIS TEST IS FLAKY THEN IT IS FAILING. See https://crbug.com/416940 |
3621 TEST_F(DiskCacheBackendTest, SimpleCacheDeleteQuickly) { | 3627 TEST_F(DiskCacheBackendTest, SimpleCacheDeleteQuickly) { |
3622 SetSimpleCacheMode(); | 3628 SetSimpleCacheMode(); |
3623 for (int i = 0; i < 100; ++i) { | 3629 for (int i = 0; i < 100; ++i) { |
3624 InitCache(); | 3630 InitCache(); |
3625 cache_.reset(); | 3631 cache_.reset(); |
3626 EXPECT_TRUE(CleanupCacheDir()); | 3632 EXPECT_TRUE(CleanupCacheDir()); |
3627 } | 3633 } |
3628 } | 3634 } |
OLD | NEW |