Index: net/disk_cache/backend_unittest.cc |
diff --git a/net/disk_cache/backend_unittest.cc b/net/disk_cache/backend_unittest.cc |
index 8833729f6965ce3a78f837a9e08d97f08105d3a5..7bb6046509ce876734210bbe5180dacbb5d0d29e 100644 |
--- a/net/disk_cache/backend_unittest.cc |
+++ b/net/disk_cache/backend_unittest.cc |
@@ -89,6 +89,10 @@ class DiskCacheBackendTest : public DiskCacheTestWithCache { |
std::set<std::string>* keys_to_match, |
size_t* count); |
+ // Computes the expected size of entry metadata, i.e. the total size without |
+ // the actual data stored. This depends only on the entry's |key| size. |
+ int GetEntryMetadataSize(std::string key); |
+ |
// Actual tests: |
void BackendBasics(); |
void BackendKeying(); |
@@ -282,6 +286,18 @@ bool DiskCacheBackendTest::EnumerateAndMatchKeys( |
return true; |
} |
+int DiskCacheBackendTest::GetEntryMetadataSize(std::string key) { |
+ // For blockfile and memory backends, it is just the key size. |
+ if (!simple_cache_mode_) |
+ return key.size(); |
+ |
+ // For the simple cache, we must add the file header and EOF, and that for |
+ // every stream. |
+ return disk_cache::kSimpleEntryStreamCount * |
+ (sizeof(disk_cache::SimpleFileHeader) + |
+ sizeof(disk_cache::SimpleFileEOF) + key.size()); |
+} |
+ |
void DiskCacheBackendTest::BackendBasics() { |
InitCache(); |
disk_cache::Entry *entry1 = NULL, *entry2 = NULL; |
@@ -1679,12 +1695,7 @@ void DiskCacheBackendTest::BackendCalculateSizeOfAllEntries() { |
InitCache(); |
// The cache is initially empty. |
- if (memory_only_ || simple_cache_mode_) { |
- // TODO(msramek): Implement. |
- EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, CalculateSizeOfAllEntries()); |
- } else { |
- EXPECT_EQ(0, CalculateSizeOfAllEntries()); |
- } |
+ EXPECT_EQ(0, CalculateSizeOfAllEntries()); |
// Generate random entries and populate them with data of respective |
// sizes 0, 1, ..., count - 1 bytes. |
@@ -1696,8 +1707,8 @@ void DiskCacheBackendTest::BackendCalculateSizeOfAllEntries() { |
std::string data(count, ' '); |
scoped_refptr<net::StringIOBuffer> buffer = new net::StringIOBuffer(data); |
- // Alternate between writing to the first and second stream to test that |
- // we are not taking just the first stream into account. |
+ // Alternate between writing to first two streams to test that we do not |
+ // take only one stream into account. |
disk_cache::Entry* entry; |
ASSERT_EQ(net::OK, OpenEntry(key, &entry)); |
ASSERT_EQ(count, WriteData(entry, count % 2, 0, buffer.get(), count, true)); |
@@ -1708,16 +1719,10 @@ void DiskCacheBackendTest::BackendCalculateSizeOfAllEntries() { |
// The resulting size should be (0 + 1 + ... + count - 1) plus keys. |
int result = CalculateSizeOfAllEntries(); |
- if (memory_only_ || simple_cache_mode_) { |
- // TODO(msramek): Implement. |
- EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, result); |
- } else { |
- int total_key_size = 0; |
- for (std::string key : key_pool) |
- total_key_size += key.size(); |
- |
- EXPECT_EQ((count - 1) * count / 2 + total_key_size, result); |
- } |
+ int total_metadata_size = 0; |
+ for (std::string key : key_pool) |
+ total_metadata_size += GetEntryMetadataSize(key); |
+ EXPECT_EQ((count - 1) * count / 2 + total_metadata_size, result); |
// Add another entry and test if the size is updated. Then remove it and test |
// if the size is back to original value. |
@@ -1734,32 +1739,16 @@ void DiskCacheBackendTest::BackendCalculateSizeOfAllEntries() { |
entry->Close(); |
int new_result = CalculateSizeOfAllEntries(); |
- if (memory_only_ || simple_cache_mode_) { |
- // TODO(msramek): Implement. |
- EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, new_result); |
- } else { |
- EXPECT_EQ(result + last_entry_size + static_cast<int>(key.size()), |
- new_result); |
- } |
+ EXPECT_EQ(result + last_entry_size + GetEntryMetadataSize(key), new_result); |
DoomEntry(key); |
new_result = CalculateSizeOfAllEntries(); |
- if (memory_only_ || simple_cache_mode_) { |
- // TODO(msramek): Implement. |
- EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, new_result); |
- } else { |
- EXPECT_EQ(result, new_result); |
- } |
+ EXPECT_EQ(result, new_result); |
} |
// After dooming the entries, the size should be back to zero. |
ASSERT_EQ(net::OK, DoomAllEntries()); |
- if (memory_only_ || simple_cache_mode_) { |
- // TODO(msramek): Implement. |
- EXPECT_EQ(net::ERR_NOT_IMPLEMENTED, CalculateSizeOfAllEntries()); |
- } else { |
- EXPECT_EQ(0, CalculateSizeOfAllEntries()); |
- } |
+ EXPECT_EQ(0, CalculateSizeOfAllEntries()); |
} |
TEST_F(DiskCacheBackendTest, CalculateSizeOfAllEntries) { |
@@ -1772,6 +1761,9 @@ TEST_F(DiskCacheBackendTest, MemoryOnlyCalculateSizeOfAllEntries) { |
} |
TEST_F(DiskCacheBackendTest, SimpleCacheCalculateSizeOfAllEntries) { |
+ // Use net::APP_CACHE to make size estimations deterministic via |
+ // non-optimistic writes. |
+ SetCacheType(net::APP_CACHE); |
SetSimpleCacheMode(); |
BackendCalculateSizeOfAllEntries(); |
} |