OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "net/tools/quic/quic_epoll_alarm_factory.h" |
| 6 #include "net/tools/quic/test_tools/mock_epoll_server.h" |
| 7 #include "testing/base/public/gmock.h" |
| 8 #include "testing/base/public/gunit.h" |
| 9 |
| 10 namespace net { |
| 11 namespace test { |
| 12 namespace { |
| 13 |
| 14 // The boolean parameter denotes whether or not to use an arena. |
| 15 class QuicEpollAlarmFactoryTest : public ::testing::TestWithParam<bool> { |
| 16 protected: |
| 17 QuicEpollAlarmFactoryTest() |
| 18 : clock_(&epoll_server_), alarm_factory_(&epoll_server_) {} |
| 19 |
| 20 QuicConnectionArena* GetArenaParam() { |
| 21 return GetParam() ? &arena_ : nullptr; |
| 22 } |
| 23 |
| 24 const QuicEpollClock clock_; |
| 25 QuicEpollAlarmFactory alarm_factory_; |
| 26 test::MockEpollServer epoll_server_; |
| 27 QuicConnectionArena arena_; |
| 28 }; |
| 29 |
| 30 TEST_P(QuicEpollAlarmFactoryTest, CreateAlarm) { |
| 31 QuicArenaScopedPtr<TestDelegate> delegate = |
| 32 QuicArenaScopedPtr<TestDelegate>(new TestDelegate()); |
| 33 QuicArenaScopedPtr<QuicAlarm> alarm( |
| 34 alarm_factory_.CreateAlarm(std::move(delegate), GetArenaParam())); |
| 35 |
| 36 QuicTime start = clock_.Now(); |
| 37 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); |
| 38 alarm->Set(start.Add(delta)); |
| 39 |
| 40 epoll_server_.AdvanceByAndWaitForEventsAndExecuteCallbacks( |
| 41 delta.ToMicroseconds()); |
| 42 EXPECT_EQ(start.Add(delta), clock_.Now()); |
| 43 } |
| 44 |
| 45 TEST_P(QuicEpollAlarmFactoryTest, CreateAlarmAndCancel) { |
| 46 QuicArenaScopedPtr<TestDelegate> delegate = |
| 47 QuicArenaScopedPtr<TestDelegate>(new TestDelegate()); |
| 48 TestDelegate* unowned_delegate = delegate.get(); |
| 49 QuicArenaScopedPtr<QuicAlarm> alarm( |
| 50 alarm_factory_.CreateAlarm(std::move(delegate), GetArenaParam())); |
| 51 |
| 52 QuicTime start = clock_.Now(); |
| 53 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); |
| 54 alarm->Set(start.Add(delta)); |
| 55 alarm->Cancel(); |
| 56 |
| 57 epoll_server_.AdvanceByExactlyAndCallCallbacks(delta.ToMicroseconds()); |
| 58 EXPECT_EQ(start.Add(delta), clock_.Now()); |
| 59 EXPECT_FALSE(unowned_delegate->fired()); |
| 60 } |
| 61 |
| 62 TEST_P(QuicEpollAlarmFactoryTest, CreateAlarmAndReset) { |
| 63 QuicArenaScopedPtr<TestDelegate> delegate = |
| 64 QuicArenaScopedPtr<TestDelegate>(new TestDelegate()); |
| 65 TestDelegate* unowned_delegate = delegate.get(); |
| 66 QuicArenaScopedPtr<QuicAlarm> alarm( |
| 67 alarm_factory_.CreateAlarm(std::move(delegate), GetArenaParam())); |
| 68 |
| 69 QuicTime start = clock_.Now(); |
| 70 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); |
| 71 alarm->Set(clock_.Now().Add(delta)); |
| 72 alarm->Cancel(); |
| 73 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); |
| 74 alarm->Set(clock_.Now().Add(new_delta)); |
| 75 |
| 76 epoll_server_.AdvanceByExactlyAndCallCallbacks(delta.ToMicroseconds()); |
| 77 EXPECT_EQ(start.Add(delta), clock_.Now()); |
| 78 EXPECT_FALSE(unowned_delegate->fired()); |
| 79 |
| 80 epoll_server_.AdvanceByExactlyAndCallCallbacks( |
| 81 new_delta.Subtract(delta).ToMicroseconds()); |
| 82 EXPECT_EQ(start.Add(new_delta), clock_.Now()); |
| 83 EXPECT_TRUE(unowned_delegate->fired()); |
| 84 } |
| 85 |
| 86 TEST_P(QuicEpollAlarmFactoryTest, CreateAlarmAndUpdate) { |
| 87 QuicArenaScopedPtr<TestDelegate> delegate = |
| 88 QuicArenaScopedPtr<TestDelegate>(new TestDelegate()); |
| 89 TestDelegate* unowned_delegate = delegate.get(); |
| 90 QuicArenaScopedPtr<QuicAlarm> alarm( |
| 91 alarm_factory_.CreateAlarm(std::move(delegate), GetArenaParam())); |
| 92 |
| 93 QuicTime start = clock_.Now(); |
| 94 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); |
| 95 alarm->Set(clock_.Now().Add(delta)); |
| 96 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); |
| 97 alarm->Update(clock_.Now().Add(new_delta), |
| 98 QuicTime::Delta::FromMicroseconds(1)); |
| 99 |
| 100 epoll_server_.AdvanceByExactlyAndCallCallbacks(delta.ToMicroseconds()); |
| 101 EXPECT_EQ(start.Add(delta), clock_.Now()); |
| 102 EXPECT_FALSE(unowned_delegate->fired()); |
| 103 |
| 104 // Move the alarm forward 1us and ensure it doesn't move forward. |
| 105 alarm->Update(clock_.Now().Add(new_delta), |
| 106 QuicTime::Delta::FromMicroseconds(2)); |
| 107 |
| 108 epoll_server_.AdvanceByExactlyAndCallCallbacks( |
| 109 new_delta.Subtract(delta).ToMicroseconds()); |
| 110 EXPECT_EQ(start.Add(new_delta), clock_.Now()); |
| 111 EXPECT_TRUE(unowned_delegate->fired()); |
| 112 |
| 113 // Set the alarm via an update call. |
| 114 new_delta = QuicTime::Delta::FromMicroseconds(5); |
| 115 alarm->Update(clock_.Now().Add(new_delta), |
| 116 QuicTime::Delta::FromMicroseconds(1)); |
| 117 EXPECT_TRUE(alarm->IsSet()); |
| 118 |
| 119 // Update it with an uninitialized time and ensure it's cancelled. |
| 120 alarm->Update(QuicTime::Zero(), QuicTime::Delta::FromMicroseconds(1)); |
| 121 EXPECT_FALSE(alarm->IsSet()); |
| 122 } |
| 123 |
| 124 } // namespace |
| 125 } // namespace test |
| 126 } // namespace net |
OLD | NEW |