Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/quic_chromium_connection_helper.h" | 5 #include "net/quic/quic_chromium_alarm_factory.h" |
|
Ryan Hamilton
2016/04/20 18:44:58
Is this supposed to be part of this CL?
| |
| 6 | 6 |
| 7 #include "net/quic/test_tools/mock_clock.h" | 7 #include "net/quic/test_tools/mock_clock.h" |
| 8 #include "net/quic/test_tools/mock_random.h" | 8 #include "net/quic/test_tools/mock_random.h" |
| 9 #include "net/quic/test_tools/test_task_runner.h" | 9 #include "net/quic/test_tools/test_task_runner.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 | 11 |
| 12 namespace net { | 12 namespace net { |
| 13 namespace test { | 13 namespace test { |
| 14 namespace { | 14 namespace { |
| 15 | 15 |
| 16 class TestDelegate : public QuicAlarm::Delegate { | 16 class TestDelegate : public QuicAlarm::Delegate { |
| 17 public: | 17 public: |
| 18 TestDelegate() : fired_(false) {} | 18 TestDelegate() : fired_(false) {} |
| 19 | 19 |
| 20 void OnAlarm() override { fired_ = true; } | 20 void OnAlarm() override { fired_ = true; } |
| 21 | 21 |
| 22 bool fired() const { return fired_; } | 22 bool fired() const { return fired_; } |
| 23 void Clear() { fired_ = false; } | 23 void Clear() { fired_ = false; } |
| 24 | 24 |
| 25 private: | 25 private: |
| 26 bool fired_; | 26 bool fired_; |
| 27 }; | 27 }; |
| 28 | 28 |
| 29 class QuicChromiumConnectionHelperTest : public ::testing::Test { | 29 class QuicChromiumAlarmFactoryTest : public ::testing::Test { |
| 30 protected: | 30 protected: |
| 31 QuicChromiumConnectionHelperTest() | 31 QuicChromiumAlarmFactoryTest() |
| 32 : runner_(new TestTaskRunner(&clock_)), | 32 : runner_(new TestTaskRunner(&clock_)), |
| 33 alarm_factory_(runner_.get(), &clock_, ) {} | 33 alarm_factory_(runner_.get(), &clock_) {} |
| 34 | 34 |
| 35 scoped_refptr<TestTaskRunner> runner_; | 35 scoped_refptr<TestTaskRunner> runner_; |
| 36 QuicChromiumAlarmFactory alarm_factory_; | 36 QuicChromiumAlarmFactory alarm_factory_; |
| 37 MockClock clock_; | 37 MockClock clock_; |
| 38 }; | 38 }; |
| 39 | 39 |
| 40 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarm) { | 40 TEST_F(QuicChromiumAlarmFactoryTest, CreateAlarm) { |
| 41 TestDelegate* delegate = new TestDelegate(); | 41 TestDelegate* delegate = new TestDelegate(); |
| 42 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); | 42 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); |
| 43 | 43 |
| 44 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); | 44 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); |
| 45 alarm->Set(clock_.Now().Add(delta)); | 45 alarm->Set(clock_.Now().Add(delta)); |
| 46 | 46 |
| 47 // Verify that the alarm task has been posted. | 47 // Verify that the alarm task has been posted. |
| 48 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); | 48 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); |
| 49 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), | 49 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), |
| 50 runner_->GetPostedTasks()[0].delay); | 50 runner_->GetPostedTasks()[0].delay); |
| 51 | 51 |
| 52 runner_->RunNextTask(); | 52 runner_->RunNextTask(); |
| 53 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); | 53 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); |
| 54 EXPECT_TRUE(delegate->fired()); | 54 EXPECT_TRUE(delegate->fired()); |
| 55 } | 55 } |
| 56 | 56 |
| 57 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarmAndCancel) { | 57 TEST_F(QuicChromiumAlarmFactoryTest, CreateAlarmAndCancel) { |
| 58 TestDelegate* delegate = new TestDelegate(); | 58 TestDelegate* delegate = new TestDelegate(); |
| 59 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); | 59 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); |
| 60 | 60 |
| 61 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); | 61 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); |
| 62 alarm->Set(clock_.Now().Add(delta)); | 62 alarm->Set(clock_.Now().Add(delta)); |
| 63 alarm->Cancel(); | 63 alarm->Cancel(); |
| 64 | 64 |
| 65 // The alarm task should still be posted. | 65 // The alarm task should still be posted. |
| 66 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); | 66 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); |
| 67 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), | 67 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), |
| 68 runner_->GetPostedTasks()[0].delay); | 68 runner_->GetPostedTasks()[0].delay); |
| 69 | 69 |
| 70 runner_->RunNextTask(); | 70 runner_->RunNextTask(); |
| 71 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); | 71 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); |
| 72 EXPECT_FALSE(delegate->fired()); | 72 EXPECT_FALSE(delegate->fired()); |
| 73 } | 73 } |
| 74 | 74 |
| 75 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarmAndReset) { | 75 TEST_F(QuicChromiumAlarmFactoryTest, CreateAlarmAndReset) { |
| 76 TestDelegate* delegate = new TestDelegate(); | 76 TestDelegate* delegate = new TestDelegate(); |
| 77 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); | 77 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); |
| 78 | 78 |
| 79 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); | 79 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); |
| 80 alarm->Set(clock_.Now().Add(delta)); | 80 alarm->Set(clock_.Now().Add(delta)); |
| 81 alarm->Cancel(); | 81 alarm->Cancel(); |
| 82 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); | 82 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); |
| 83 alarm->Set(clock_.Now().Add(new_delta)); | 83 alarm->Set(clock_.Now().Add(new_delta)); |
| 84 | 84 |
| 85 // The alarm task should still be posted. | 85 // The alarm task should still be posted. |
| 86 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); | 86 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); |
| 87 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), | 87 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), |
| 88 runner_->GetPostedTasks()[0].delay); | 88 runner_->GetPostedTasks()[0].delay); |
| 89 | 89 |
| 90 runner_->RunNextTask(); | 90 runner_->RunNextTask(); |
| 91 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); | 91 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); |
| 92 EXPECT_FALSE(delegate->fired()); | 92 EXPECT_FALSE(delegate->fired()); |
| 93 | 93 |
| 94 // The alarm task should be posted again. | 94 // The alarm task should be posted again. |
| 95 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); | 95 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); |
| 96 | 96 |
| 97 runner_->RunNextTask(); | 97 runner_->RunNextTask(); |
| 98 EXPECT_EQ(QuicTime::Zero().Add(new_delta), clock_.Now()); | 98 EXPECT_EQ(QuicTime::Zero().Add(new_delta), clock_.Now()); |
| 99 EXPECT_TRUE(delegate->fired()); | 99 EXPECT_TRUE(delegate->fired()); |
| 100 } | 100 } |
| 101 | 101 |
| 102 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarmAndResetEarlier) { | 102 TEST_F(QuicChromiumAlarmFactoryTest, CreateAlarmAndResetEarlier) { |
| 103 TestDelegate* delegate = new TestDelegate(); | 103 TestDelegate* delegate = new TestDelegate(); |
| 104 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); | 104 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); |
| 105 | 105 |
| 106 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(3); | 106 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(3); |
| 107 alarm->Set(clock_.Now().Add(delta)); | 107 alarm->Set(clock_.Now().Add(delta)); |
| 108 alarm->Cancel(); | 108 alarm->Cancel(); |
| 109 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(1); | 109 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(1); |
| 110 alarm->Set(clock_.Now().Add(new_delta)); | 110 alarm->Set(clock_.Now().Add(new_delta)); |
| 111 | 111 |
| 112 // Both alarm tasks will be posted. | 112 // Both alarm tasks will be posted. |
| 113 ASSERT_EQ(2u, runner_->GetPostedTasks().size()); | 113 ASSERT_EQ(2u, runner_->GetPostedTasks().size()); |
| 114 | 114 |
| 115 // The earlier task will execute and will fire the alarm-> | 115 // The earlier task will execute and will fire the alarm-> |
| 116 runner_->RunNextTask(); | 116 runner_->RunNextTask(); |
| 117 EXPECT_EQ(QuicTime::Zero().Add(new_delta), clock_.Now()); | 117 EXPECT_EQ(QuicTime::Zero().Add(new_delta), clock_.Now()); |
| 118 EXPECT_TRUE(delegate->fired()); | 118 EXPECT_TRUE(delegate->fired()); |
| 119 delegate->Clear(); | 119 delegate->Clear(); |
| 120 | 120 |
| 121 // The latter task is still posted. | 121 // The latter task is still posted. |
| 122 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); | 122 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); |
| 123 | 123 |
| 124 // When the latter task is executed, the weak ptr will be invalid and | 124 // When the latter task is executed, the weak ptr will be invalid and |
| 125 // the alarm will not fire. | 125 // the alarm will not fire. |
| 126 runner_->RunNextTask(); | 126 runner_->RunNextTask(); |
| 127 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); | 127 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); |
| 128 EXPECT_FALSE(delegate->fired()); | 128 EXPECT_FALSE(delegate->fired()); |
| 129 } | 129 } |
| 130 | 130 |
| 131 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarmAndUpdate) { | 131 TEST_F(QuicChromiumAlarmFactoryTest, CreateAlarmAndUpdate) { |
| 132 TestDelegate* delegate = new TestDelegate(); | 132 TestDelegate* delegate = new TestDelegate(); |
| 133 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); | 133 std::unique_ptr<QuicAlarm> alarm(alarm_factory_.CreateAlarm(delegate)); |
| 134 | 134 |
| 135 QuicTime start = clock_->Now(); | 135 QuicTime start = clock_.Now(); |
| 136 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); | 136 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); |
| 137 alarm->Set(clock_->Now().Add(delta)); | 137 alarm->Set(clock_.Now().Add(delta)); |
| 138 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); | 138 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); |
| 139 alarm->Update(clock_->Now().Add(new_delta), | 139 alarm->Update(clock_.Now().Add(new_delta), |
| 140 QuicTime::Delta::FromMicroseconds(1)); | 140 QuicTime::Delta::FromMicroseconds(1)); |
| 141 | 141 |
| 142 // The alarm task should still be posted. | 142 // The alarm task should still be posted. |
| 143 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); | 143 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); |
| 144 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), | 144 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), |
| 145 runner_->GetPostedTasks()[0].delay); | 145 runner_->GetPostedTasks()[0].delay); |
| 146 | 146 |
| 147 runner_->RunNextTask(); | 147 runner_->RunNextTask(); |
| 148 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_->Now()); | 148 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); |
| 149 EXPECT_FALSE(delegate->fired()); | 149 EXPECT_FALSE(delegate->fired()); |
| 150 | 150 |
| 151 // Move the alarm forward 1us and ensure it doesn't move forward. | 151 // Move the alarm forward 1us and ensure it doesn't move forward. |
| 152 alarm->Update(clock_->Now().Add(new_delta), | 152 alarm->Update(clock_.Now().Add(new_delta), |
| 153 QuicTime::Delta::FromMicroseconds(2)); | 153 QuicTime::Delta::FromMicroseconds(2)); |
| 154 | 154 |
| 155 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); | 155 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); |
| 156 EXPECT_EQ(base::TimeDelta::FromMicroseconds( | 156 EXPECT_EQ(base::TimeDelta::FromMicroseconds( |
| 157 new_delta.Subtract(delta).ToMicroseconds()), | 157 new_delta.Subtract(delta).ToMicroseconds()), |
| 158 runner_->GetPostedTasks()[0].delay); | 158 runner_->GetPostedTasks()[0].delay); |
| 159 runner_->RunNextTask(); | 159 runner_->RunNextTask(); |
| 160 EXPECT_EQ(start.Add(new_delta), clock_->Now()); | 160 EXPECT_EQ(start.Add(new_delta), clock_.Now()); |
| 161 EXPECT_TRUE(delegate->fired()); | 161 EXPECT_TRUE(delegate->fired()); |
| 162 | 162 |
| 163 // Set the alarm via an update call. | 163 // Set the alarm via an update call. |
| 164 new_delta = QuicTime::Delta::FromMicroseconds(5); | 164 new_delta = QuicTime::Delta::FromMicroseconds(5); |
| 165 alarm->Update(clock_->Now().Add(new_delta), | 165 alarm->Update(clock_.Now().Add(new_delta), |
| 166 QuicTime::Delta::FromMicroseconds(1)); | 166 QuicTime::Delta::FromMicroseconds(1)); |
| 167 EXPECT_TRUE(alarm->IsSet()); | 167 EXPECT_TRUE(alarm->IsSet()); |
| 168 | 168 |
| 169 // Update it with an uninitialized time and ensure it's cancelled. | 169 // Update it with an uninitialized time and ensure it's cancelled. |
| 170 alarm->Update(QuicTime::Zero(), QuicTime::Delta::FromMicroseconds(1)); | 170 alarm->Update(QuicTime::Zero(), QuicTime::Delta::FromMicroseconds(1)); |
| 171 EXPECT_FALSE(alarm->IsSet()); | 171 EXPECT_FALSE(alarm->IsSet()); |
| 172 } | 172 } |
| 173 | 173 |
| 174 } // namespace | 174 } // namespace |
| 175 } // namespace test | 175 } // namespace test |
| 176 } // namespace net | 176 } // namespace net |
| OLD | NEW |