| 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 |