| // Copyright (c) 2011 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/basictypes.h" |
| #include "base/macros.h" |
| #include "base/message_loop/message_loop.h" |
| #include "base/test/simple_test_tick_clock.h" |
| #include "components/copresence/timed_map.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace { |
| |
| struct Value { |
| Value() : value(0) {} |
| explicit Value(int new_value) : value(new_value) {} |
| |
| int value; |
| }; |
| |
| } // namespace |
| |
| class TimedMapTest : public testing::Test { |
| public: |
| using Map = copresence::TimedMap<int, Value>; |
| |
| TimedMapTest() {} |
| |
| private: |
| // Exists since the timer needs a message loop. |
| base::MessageLoop message_loop_; |
| |
| DISALLOW_COPY_AND_ASSIGN(TimedMapTest); |
| }; |
| |
| TEST_F(TimedMapTest, Basic) { |
| Map map(base::TimeDelta::FromSeconds(9999), 3); |
| |
| EXPECT_FALSE(map.HasKey(0)); |
| EXPECT_EQ(0, map.GetValue(0).value); |
| |
| map.Add(0x1337, Value(0x7331)); |
| EXPECT_TRUE(map.HasKey(0x1337)); |
| EXPECT_EQ(0x7331, map.GetValue(0x1337).value); |
| |
| map.Add(0xbaad, Value(0xf00d)); |
| EXPECT_TRUE(map.HasKey(0xbaad)); |
| EXPECT_EQ(0xf00d, map.GetValue(0xbaad).value); |
| EXPECT_EQ(0x7331, map.GetValue(0x1337).value); |
| |
| map.Add(0x1234, Value(0x5678)); |
| EXPECT_TRUE(map.HasKey(0x1234)); |
| EXPECT_TRUE(map.HasKey(0xbaad)); |
| EXPECT_TRUE(map.HasKey(0x1337)); |
| |
| EXPECT_EQ(0x5678, map.GetValue(0x1234).value); |
| EXPECT_EQ(0xf00d, map.GetValue(0xbaad).value); |
| EXPECT_EQ(0x7331, map.GetValue(0x1337).value); |
| } |
| |
| TEST_F(TimedMapTest, ValueReplacement) { |
| Map map(base::TimeDelta::FromSeconds(9999), 10); |
| |
| map.Add(0x1337, Value(0x7331)); |
| EXPECT_TRUE(map.HasKey(0x1337)); |
| EXPECT_EQ(0x7331, map.GetValue(0x1337).value); |
| |
| map.Add(0xbaad, Value(0xf00d)); |
| EXPECT_TRUE(map.HasKey(0xbaad)); |
| EXPECT_EQ(0xf00d, map.GetValue(0xbaad).value); |
| |
| map.Add(0x1337, Value(0xd00d)); |
| EXPECT_TRUE(map.HasKey(0x1337)); |
| EXPECT_EQ(0xd00d, map.GetValue(0x1337).value); |
| } |
| |
| TEST_F(TimedMapTest, SizeEvict) { |
| Map two_element_map(base::TimeDelta::FromSeconds(9999), 2); |
| |
| two_element_map.Add(0x1337, Value(0x7331)); |
| EXPECT_TRUE(two_element_map.HasKey(0x1337)); |
| EXPECT_EQ(0x7331, two_element_map.GetValue(0x1337).value); |
| |
| two_element_map.Add(0xbaad, Value(0xf00d)); |
| EXPECT_TRUE(two_element_map.HasKey(0xbaad)); |
| EXPECT_EQ(0xf00d, two_element_map.GetValue(0xbaad).value); |
| |
| two_element_map.Add(0x1234, Value(0x5678)); |
| EXPECT_TRUE(two_element_map.HasKey(0x1234)); |
| EXPECT_EQ(0xf00d, two_element_map.GetValue(0xbaad).value); |
| |
| EXPECT_FALSE(two_element_map.HasKey(0x1337)); |
| EXPECT_EQ(0, two_element_map.GetValue(0x1337).value); |
| } |
| |
| TEST_F(TimedMapTest, TimedEvict) { |
| const int kLargeTimeValueSeconds = 9999; |
| Map map(base::TimeDelta::FromSeconds(kLargeTimeValueSeconds), 2); |
| |
| // The map takes ownership of the clock, but we retain a pointer. |
| base::SimpleTestTickClock* clock = new base::SimpleTestTickClock; |
| map.set_clock_for_testing(make_scoped_ptr<base::TickClock>(clock)); |
| |
| // Add value at T=0. |
| map.Add(0x1337, Value(0x7331)); |
| EXPECT_TRUE(map.HasKey(0x1337)); |
| EXPECT_EQ(0x7331, map.GetValue(0x1337).value); |
| |
| // Add value at T=kLargeTimeValueSeconds-1. |
| clock->Advance(base::TimeDelta::FromSeconds(kLargeTimeValueSeconds - 1)); |
| map.Add(0xbaad, Value(0xf00d)); |
| |
| // Check values at T=kLargeTimeValueSeconds-1. |
| EXPECT_TRUE(map.HasKey(0xbaad)); |
| EXPECT_EQ(0xf00d, map.GetValue(0xbaad).value); |
| EXPECT_TRUE(map.HasKey(0x1337)); |
| EXPECT_EQ(0x7331, map.GetValue(0x1337).value); |
| |
| // Check values at T=kLargeTimeValueSeconds. |
| clock->Advance(base::TimeDelta::FromSeconds(1)); |
| EXPECT_FALSE(map.HasKey(0x1337)); |
| EXPECT_EQ(0, map.GetValue(0x1337).value); |
| EXPECT_TRUE(map.HasKey(0xbaad)); |
| EXPECT_EQ(0xf00d, map.GetValue(0xbaad).value); |
| |
| // Check values at T=2*kLargeTimeValueSeconds |
| clock->Advance(base::TimeDelta::FromSeconds(kLargeTimeValueSeconds)); |
| EXPECT_FALSE(map.HasKey(0xbaad)); |
| EXPECT_EQ(0, map.GetValue(0xbaad).value); |
| } |