blob: 8c6b5b87a74a8926fb9f0902019916e6b31872b5 [file] [log] [blame]
// Copyright (c) 2016 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 "quic/core/quic_lru_cache.h"
#include "quic/platform/api/quic_test.h"
namespace quic {
namespace test {
namespace {
struct CachedItem {
explicit CachedItem(uint32_t new_value) : value(new_value) {}
uint32_t value;
TEST(QuicLRUCacheTest, InsertAndLookup) {
QuicLRUCache<int, CachedItem> cache(5);
EXPECT_EQ(cache.end(), cache.Lookup(1));
EXPECT_EQ(0u, cache.Size());
EXPECT_EQ(5u, cache.MaxSize());
// Check that item 1 was properly inserted.
std::unique_ptr<CachedItem> item1(new CachedItem(11));
cache.Insert(1, std::move(item1));
EXPECT_EQ(1u, cache.Size());
EXPECT_EQ(11u, cache.Lookup(1)->second->value);
// Check that item 2 overrides item 1.
std::unique_ptr<CachedItem> item2(new CachedItem(12));
cache.Insert(1, std::move(item2));
EXPECT_EQ(1u, cache.Size());
EXPECT_EQ(12u, cache.Lookup(1)->second->value);
std::unique_ptr<CachedItem> item3(new CachedItem(13));
cache.Insert(3, std::move(item3));
EXPECT_EQ(2u, cache.Size());
auto iter = cache.Lookup(3);
ASSERT_NE(cache.end(), iter);
EXPECT_EQ(13u, iter->second->value);
ASSERT_EQ(cache.end(), cache.Lookup(3));
EXPECT_EQ(1u, cache.Size());
// No memory leakage.
EXPECT_EQ(0u, cache.Size());
TEST(QuicLRUCacheTest, Eviction) {
QuicLRUCache<int, CachedItem> cache(3);
for (size_t i = 1; i <= 4; ++i) {
std::unique_ptr<CachedItem> item(new CachedItem(10 + i));
cache.Insert(i, std::move(item));
EXPECT_EQ(3u, cache.Size());
EXPECT_EQ(3u, cache.MaxSize());
// Make sure item 1 is evicted.
EXPECT_EQ(cache.end(), cache.Lookup(1));
EXPECT_EQ(14u, cache.Lookup(4)->second->value);
EXPECT_EQ(12u, cache.Lookup(2)->second->value);
std::unique_ptr<CachedItem> item5(new CachedItem(15));
cache.Insert(5, std::move(item5));
// Make sure item 3 is evicted.
EXPECT_EQ(cache.end(), cache.Lookup(3));
EXPECT_EQ(15u, cache.Lookup(5)->second->value);
// No memory leakage.
EXPECT_EQ(0u, cache.Size());
} // namespace
} // namespace test
} // namespace quic