| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 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_connection_helper.h" | |
| 6 | |
| 7 #include "net/quic/crypto/quic_random.h" | |
| 8 #include "net/tools/quic/test_tools/mock_epoll_server.h" | |
| 9 #include "testing/gtest/include/gtest/gtest.h" | |
| 10 | |
| 11 using net::tools::test::MockEpollServer; | |
| 12 | |
| 13 namespace net { | |
| 14 namespace tools { | |
| 15 namespace test { | |
| 16 namespace { | |
| 17 | |
| 18 class TestDelegate : public QuicAlarm::Delegate { | |
| 19 public: | |
| 20 TestDelegate() : fired_(false) {} | |
| 21 | |
| 22 QuicTime OnAlarm() override { | |
| 23 fired_ = true; | |
| 24 return QuicTime::Zero(); | |
| 25 } | |
| 26 | |
| 27 bool fired() const { return fired_; } | |
| 28 | |
| 29 private: | |
| 30 bool fired_; | |
| 31 }; | |
| 32 | |
| 33 class QuicEpollConnectionHelperTest : public ::testing::Test { | |
| 34 protected: | |
| 35 QuicEpollConnectionHelperTest() : helper_(&epoll_server_) {} | |
| 36 | |
| 37 MockEpollServer epoll_server_; | |
| 38 QuicEpollConnectionHelper helper_; | |
| 39 }; | |
| 40 | |
| 41 TEST_F(QuicEpollConnectionHelperTest, GetClock) { | |
| 42 const QuicClock* clock = helper_.GetClock(); | |
| 43 QuicTime start = clock->Now(); | |
| 44 | |
| 45 QuicTime::Delta delta = QuicTime::Delta::FromMilliseconds(5); | |
| 46 epoll_server_.AdvanceBy(delta.ToMicroseconds()); | |
| 47 | |
| 48 EXPECT_EQ(start.Add(delta), clock->Now()); | |
| 49 } | |
| 50 | |
| 51 TEST_F(QuicEpollConnectionHelperTest, GetRandomGenerator) { | |
| 52 QuicRandom* random = helper_.GetRandomGenerator(); | |
| 53 EXPECT_EQ(QuicRandom::GetInstance(), random); | |
| 54 } | |
| 55 | |
| 56 TEST_F(QuicEpollConnectionHelperTest, CreateAlarm) { | |
| 57 TestDelegate* delegate = new TestDelegate(); | |
| 58 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate)); | |
| 59 | |
| 60 const QuicClock* clock = helper_.GetClock(); | |
| 61 QuicTime start = clock->Now(); | |
| 62 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); | |
| 63 alarm->Set(start.Add(delta)); | |
| 64 | |
| 65 epoll_server_.AdvanceByAndCallCallbacks(delta.ToMicroseconds()); | |
| 66 EXPECT_EQ(start.Add(delta), clock->Now()); | |
| 67 } | |
| 68 | |
| 69 TEST_F(QuicEpollConnectionHelperTest, CreateAlarmAndCancel) { | |
| 70 TestDelegate* delegate = new TestDelegate(); | |
| 71 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate)); | |
| 72 | |
| 73 const QuicClock* clock = helper_.GetClock(); | |
| 74 QuicTime start = clock->Now(); | |
| 75 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); | |
| 76 alarm->Set(start.Add(delta)); | |
| 77 alarm->Cancel(); | |
| 78 | |
| 79 epoll_server_.AdvanceByExactlyAndCallCallbacks(delta.ToMicroseconds()); | |
| 80 EXPECT_EQ(start.Add(delta), clock->Now()); | |
| 81 EXPECT_FALSE(delegate->fired()); | |
| 82 } | |
| 83 | |
| 84 TEST_F(QuicEpollConnectionHelperTest, CreateAlarmAndReset) { | |
| 85 TestDelegate* delegate = new TestDelegate(); | |
| 86 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate)); | |
| 87 | |
| 88 const QuicClock* clock = helper_.GetClock(); | |
| 89 QuicTime start = clock->Now(); | |
| 90 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); | |
| 91 alarm->Set(clock->Now().Add(delta)); | |
| 92 alarm->Cancel(); | |
| 93 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); | |
| 94 alarm->Set(clock->Now().Add(new_delta)); | |
| 95 | |
| 96 epoll_server_.AdvanceByExactlyAndCallCallbacks(delta.ToMicroseconds()); | |
| 97 EXPECT_EQ(start.Add(delta), clock->Now()); | |
| 98 EXPECT_FALSE(delegate->fired()); | |
| 99 | |
| 100 epoll_server_.AdvanceByExactlyAndCallCallbacks( | |
| 101 new_delta.Subtract(delta).ToMicroseconds()); | |
| 102 EXPECT_EQ(start.Add(new_delta), clock->Now()); | |
| 103 EXPECT_TRUE(delegate->fired()); | |
| 104 } | |
| 105 | |
| 106 TEST_F(QuicEpollConnectionHelperTest, CreateAlarmAndUpdate) { | |
| 107 TestDelegate* delegate = new TestDelegate(); | |
| 108 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate)); | |
| 109 | |
| 110 const QuicClock* clock = helper_.GetClock(); | |
| 111 QuicTime start = clock->Now(); | |
| 112 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); | |
| 113 alarm->Set(clock->Now().Add(delta)); | |
| 114 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); | |
| 115 alarm->Update(clock->Now().Add(new_delta), | |
| 116 QuicTime::Delta::FromMicroseconds(1)); | |
| 117 | |
| 118 epoll_server_.AdvanceByExactlyAndCallCallbacks(delta.ToMicroseconds()); | |
| 119 EXPECT_EQ(start.Add(delta), clock->Now()); | |
| 120 EXPECT_FALSE(delegate->fired()); | |
| 121 | |
| 122 // Move the alarm forward 1us and ensure it doesn't move forward. | |
| 123 alarm->Update(clock->Now().Add(new_delta), | |
| 124 QuicTime::Delta::FromMicroseconds(2)); | |
| 125 | |
| 126 epoll_server_.AdvanceByExactlyAndCallCallbacks( | |
| 127 new_delta.Subtract(delta).ToMicroseconds()); | |
| 128 EXPECT_EQ(start.Add(new_delta), clock->Now()); | |
| 129 EXPECT_TRUE(delegate->fired()); | |
| 130 | |
| 131 // Set the alarm via an update call. | |
| 132 new_delta = QuicTime::Delta::FromMicroseconds(5); | |
| 133 alarm->Update(clock->Now().Add(new_delta), | |
| 134 QuicTime::Delta::FromMicroseconds(1)); | |
| 135 EXPECT_TRUE(alarm->IsSet()); | |
| 136 | |
| 137 // Update it with an uninitialized time and ensure it's cancelled. | |
| 138 alarm->Update(QuicTime::Zero(), QuicTime::Delta::FromMicroseconds(1)); | |
| 139 EXPECT_FALSE(alarm->IsSet()); | |
| 140 } | |
| 141 | |
| 142 } // namespace | |
| 143 } // namespace test | |
| 144 } // namespace tools | |
| 145 } // namespace net | |
| OLD | NEW |