Implement a method to get the total size of a cache entry

net/disk_cache/entry_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/files/file.h"
 #include "base/files/file_util.h"
@@ skipping 3689 matching lines @@
   disk_cache::Entry* entry = NULL;
   const std::string key("the key");
   const int kSize = 100;
   scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(kSize));
   scoped_refptr<net::IOBuffer> buffer_read(new net::IOBuffer(kSize));
   CacheTestFillBuffer(buffer->data(), kSize, false);
 
   ASSERT_THAT(CreateEntry(key, &entry), IsOk());
   EXPECT_TRUE(entry);
 
+  long initial_entry_size = entry->GetEntrySize();
+  DCHECK_GE(initial_entry_size, static_cast<long>(key.size()));
+
   // Write something into stream0.
   EXPECT_EQ(kSize, WriteData(entry, 0, 0, buffer.get(), kSize, false));
   EXPECT_EQ(kSize, ReadData(entry, 0, 0, buffer_read.get(), kSize));
   EXPECT_EQ(0, memcmp(buffer->data(), buffer_read->data(), kSize));
+  DCHECK_EQ(initial_entry_size + kSize, entry->GetEntrySize());
   entry->Close();
 
   // Extend stream1.
   ASSERT_THAT(OpenEntry(key, &entry), IsOk());
   int stream1_size = 100;
   EXPECT_EQ(0, WriteData(entry, 1, stream1_size, buffer.get(), 0, false));
   EXPECT_EQ(stream1_size, entry->GetDataSize(1));
+  DCHECK_EQ(initial_entry_size + kSize + stream1_size, entry->GetEntrySize());
   entry->Close();
 
   // Check that stream0 data has not been modified and that the EOF record for
   // stream 0 contains a crc.
   // The entry needs to be reopened before checking the crc: Open will perform
   // the synchronization with the previous Close. This ensures the EOF records
   // have been written to disk before we attempt to read them independently.
   ASSERT_THAT(OpenEntry(key, &entry), IsOk());
   base::FilePath entry_file0_path = cache_path_.AppendASCII(
       disk_cache::simple_util::GetFilenameFromKeyAndFileIndex(key, 0));
@@ skipping 15 matching lines @@
               disk_cache::SimpleFileEOF::FLAG_HAS_CRC32);
 
   buffer_read = new net::IOBuffer(kSize);
   EXPECT_EQ(kSize, ReadData(entry, 0, 0, buffer_read.get(), kSize));
   EXPECT_EQ(0, memcmp(buffer->data(), buffer_read->data(), kSize));
 
   // Shrink stream1.
   stream1_size = 50;
   EXPECT_EQ(0, WriteData(entry, 1, stream1_size, buffer.get(), 0, true));
   EXPECT_EQ(stream1_size, entry->GetDataSize(1));
+  DCHECK_EQ(initial_entry_size + kSize + stream1_size, entry->GetEntrySize());
   entry->Close();
 
   // Check that stream0 data has not been modified.
   buffer_read = new net::IOBuffer(kSize);
   ASSERT_THAT(OpenEntry(key, &entry), IsOk());
   EXPECT_EQ(kSize, ReadData(entry, 0, 0, buffer_read.get(), kSize));
   EXPECT_EQ(0, memcmp(buffer->data(), buffer_read->data(), kSize));
   entry->Close();
   entry = NULL;
 }
@@ skipping 519 matching lines @@
   entry->Close();
 
   base::RunLoop().RunUntilIdle();
   disk_cache::SimpleBackendImpl::FlushWorkerPoolForTesting();
   base::RunLoop().RunUntilIdle();
 
   EXPECT_TRUE(
       disk_cache::simple_util::CorruptStream0LengthFromEntry(key, cache_path_));
   EXPECT_NE(net::OK, OpenEntry(key, &entry));
 }