diff options
Diffstat (limited to 'chromium/net/disk_cache/flash/segment_unittest.cc')
-rw-r--r-- | chromium/net/disk_cache/flash/segment_unittest.cc | 152 |
1 files changed, 152 insertions, 0 deletions
diff --git a/chromium/net/disk_cache/flash/segment_unittest.cc b/chromium/net/disk_cache/flash/segment_unittest.cc new file mode 100644 index 00000000000..3f61701cbe0 --- /dev/null +++ b/chromium/net/disk_cache/flash/segment_unittest.cc @@ -0,0 +1,152 @@ +// 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 "base/memory/scoped_ptr.h" +#include "net/disk_cache/disk_cache_test_util.h" +#include "net/disk_cache/flash/segment.h" +#include "net/disk_cache/flash/storage.h" +#include "net/disk_cache/flash/flash_cache_test_base.h" +#include "net/disk_cache/flash/format.h" +#include "testing/gtest/include/gtest/gtest.h" + +namespace { + +template<int SIZE> +struct Entry { + enum { size = SIZE }; + + Entry() { CacheTestFillBuffer(data, size, false); } + + bool operator==(const Entry& rhs) const { + return std::equal(data, data + size, rhs.data); + } + + char data[size]; +}; + +const int32 kSmallEntrySize = 100; +const int32 kLargeEntrySize = disk_cache::kFlashSegmentSize / 4; + +typedef Entry<kSmallEntrySize> SmallEntry; +typedef Entry<kLargeEntrySize> LargeEntry; + +const int32 kSegmentFreeSpace = disk_cache::kFlashSegmentSize - + disk_cache::kFlashSummarySize; + +} // namespace + +TEST_F(FlashCacheTest, SegmentUserTracking) { + disk_cache::Storage storage(path_, kStorageSize); + ASSERT_TRUE(storage.Init()); + + scoped_ptr<disk_cache::Segment> segment( + new disk_cache::Segment(0, false, &storage)); + EXPECT_TRUE(segment->Init()); + + EXPECT_TRUE(segment->HasNoUsers()); + segment->AddUser(); + segment->AddUser(); + EXPECT_FALSE(segment->HasNoUsers()); + + segment->ReleaseUser(); + EXPECT_FALSE(segment->HasNoUsers()); + segment->ReleaseUser(); + EXPECT_TRUE(segment->HasNoUsers()); + + EXPECT_TRUE(segment->Close()); +} + +TEST_F(FlashCacheTest, SegmentCreateDestroy) { + disk_cache::Storage storage(path_, kStorageSize); + ASSERT_TRUE(storage.Init()); + + int32 index = 0; + scoped_ptr<disk_cache::Segment> segment( + new disk_cache::Segment(index, false, &storage)); + EXPECT_TRUE(segment->Init()); + EXPECT_TRUE(segment->Close()); + + index = kNumTestSegments - 1; + segment.reset(new disk_cache::Segment(index, false, &storage)); + EXPECT_TRUE(segment->Init()); + EXPECT_TRUE(segment->Close()); + + int32 invalid_index = kNumTestSegments; + segment.reset(new disk_cache::Segment(invalid_index, false, &storage)); + EXPECT_FALSE(segment->Init()); + + invalid_index = -1; + segment.reset(new disk_cache::Segment(invalid_index, false, &storage)); + EXPECT_FALSE(segment->Init()); +} + +TEST_F(FlashCacheTest, SegmentWriteDataReadData) { + disk_cache::Storage storage(path_, kStorageSize); + ASSERT_TRUE(storage.Init()); + + int32 index = rand() % kNumTestSegments; + scoped_ptr<disk_cache::Segment> segment( + new disk_cache::Segment(index, false, &storage)); + + EXPECT_TRUE(segment->Init()); + SmallEntry entry1; + EXPECT_TRUE(segment->CanHold(entry1.size)); + int32 offset = segment->write_offset(); + EXPECT_TRUE(segment->WriteData(entry1.data, entry1.size)); + segment->StoreOffset(offset); + EXPECT_TRUE(segment->Close()); + + segment.reset(new disk_cache::Segment(index, true, &storage)); + EXPECT_TRUE(segment->Init()); + SmallEntry entry2; + EXPECT_TRUE(segment->ReadData(entry2.data, entry2.size, offset)); + EXPECT_EQ(entry1, entry2); + EXPECT_TRUE(segment->Close()); +} + +TEST_F(FlashCacheTest, SegmentFillWithSmallEntries) { + disk_cache::Storage storage(path_, kStorageSize); + ASSERT_TRUE(storage.Init()); + + int32 index = rand() % kNumTestSegments; + scoped_ptr<disk_cache::Segment> segment( + new disk_cache::Segment(index, false, &storage)); + + EXPECT_TRUE(segment->Init()); + SmallEntry entry; + int32 num_bytes_written = 0; + while (segment->CanHold(entry.size)) { + int32 offset = segment->write_offset(); + EXPECT_TRUE(segment->WriteData(entry.data, entry.size)); + segment->StoreOffset(offset); + num_bytes_written += entry.size; + } + int32 space_left = kSegmentFreeSpace - num_bytes_written; + EXPECT_GE(space_left, entry.size); + EXPECT_EQ(segment->GetOffsets().size(), disk_cache::kFlashMaxEntryCount); + EXPECT_TRUE(segment->Close()); +} + +TEST_F(FlashCacheTest, SegmentFillWithLargeEntries) { + disk_cache::Storage storage(path_, kStorageSize); + ASSERT_TRUE(storage.Init()); + + int32 index = rand() % kNumTestSegments; + scoped_ptr<disk_cache::Segment> segment( + new disk_cache::Segment(index, false, &storage)); + + EXPECT_TRUE(segment->Init()); + scoped_ptr<LargeEntry> entry(new LargeEntry); + int32 num_bytes_written = 0; + while (segment->CanHold(entry->size)) { + int32 offset = segment->write_offset(); + EXPECT_TRUE(segment->WriteData(entry->data, entry->size)); + segment->StoreOffset(offset); + num_bytes_written += entry->size; + } + int32 space_left = kSegmentFreeSpace - num_bytes_written; + EXPECT_LT(space_left, entry->size); + EXPECT_LT(segment->GetOffsets().size(), disk_cache::kFlashMaxEntryCount); + EXPECT_TRUE(segment->Close()); +} |