Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(186)

Side by Side Diff: net/quic/quic_chromium_connection_helper_test.cc

Issue 1581283002: Rename chromium-specific QUIC classes. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: add missing Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « net/quic/quic_chromium_connection_helper.cc ('k') | net/quic/quic_connection_helper.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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_connection_helper.h" 5 #include "net/quic/quic_chromium_connection_helper.h"
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 QuicTime OnAlarm() override { 20 QuicTime OnAlarm() override {
21 fired_ = true; 21 fired_ = true;
22 return QuicTime::Zero(); 22 return QuicTime::Zero();
23 } 23 }
24 24
25 bool fired() const { return fired_; } 25 bool fired() const { return fired_; }
26 void Clear() { fired_ = false; } 26 void Clear() { fired_ = false; }
27 27
28 private: 28 private:
29 bool fired_; 29 bool fired_;
30 }; 30 };
31 31
32 class QuicConnectionHelperTest : public ::testing::Test { 32 class QuicChromiumConnectionHelperTest : public ::testing::Test {
33 protected: 33 protected:
34 QuicConnectionHelperTest() 34 QuicChromiumConnectionHelperTest()
35 : runner_(new TestTaskRunner(&clock_)), 35 : runner_(new TestTaskRunner(&clock_)),
36 helper_(runner_.get(), &clock_, &random_generator_) {} 36 helper_(runner_.get(), &clock_, &random_generator_) {}
37 37
38 scoped_refptr<TestTaskRunner> runner_; 38 scoped_refptr<TestTaskRunner> runner_;
39 QuicConnectionHelper helper_; 39 QuicChromiumConnectionHelper helper_;
40 MockClock clock_; 40 MockClock clock_;
41 MockRandom random_generator_; 41 MockRandom random_generator_;
42 }; 42 };
43 43
44 TEST_F(QuicConnectionHelperTest, GetClock) { 44 TEST_F(QuicChromiumConnectionHelperTest, GetClock) {
45 EXPECT_EQ(&clock_, helper_.GetClock()); 45 EXPECT_EQ(&clock_, helper_.GetClock());
46 } 46 }
47 47
48 TEST_F(QuicConnectionHelperTest, GetRandomGenerator) { 48 TEST_F(QuicChromiumConnectionHelperTest, GetRandomGenerator) {
49 EXPECT_EQ(&random_generator_, helper_.GetRandomGenerator()); 49 EXPECT_EQ(&random_generator_, helper_.GetRandomGenerator());
50 } 50 }
51 51
52 TEST_F(QuicConnectionHelperTest, CreateAlarm) { 52 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarm) {
53 TestDelegate* delegate = new TestDelegate(); 53 TestDelegate* delegate = new TestDelegate();
54 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate)); 54 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate));
55 55
56 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); 56 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1);
57 alarm->Set(clock_.Now().Add(delta)); 57 alarm->Set(clock_.Now().Add(delta));
58 58
59 // Verify that the alarm task has been posted. 59 // Verify that the alarm task has been posted.
60 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); 60 ASSERT_EQ(1u, runner_->GetPostedTasks().size());
61 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), 61 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()),
62 runner_->GetPostedTasks()[0].delay); 62 runner_->GetPostedTasks()[0].delay);
63 63
64 runner_->RunNextTask(); 64 runner_->RunNextTask();
65 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); 65 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now());
66 EXPECT_TRUE(delegate->fired()); 66 EXPECT_TRUE(delegate->fired());
67 } 67 }
68 68
69 TEST_F(QuicConnectionHelperTest, CreateAlarmAndCancel) { 69 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarmAndCancel) {
70 TestDelegate* delegate = new TestDelegate(); 70 TestDelegate* delegate = new TestDelegate();
71 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate)); 71 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate));
72 72
73 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); 73 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1);
74 alarm->Set(clock_.Now().Add(delta)); 74 alarm->Set(clock_.Now().Add(delta));
75 alarm->Cancel(); 75 alarm->Cancel();
76 76
77 // The alarm task should still be posted. 77 // The alarm task should still be posted.
78 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); 78 ASSERT_EQ(1u, runner_->GetPostedTasks().size());
79 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), 79 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()),
80 runner_->GetPostedTasks()[0].delay); 80 runner_->GetPostedTasks()[0].delay);
81 81
82 runner_->RunNextTask(); 82 runner_->RunNextTask();
83 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); 83 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now());
84 EXPECT_FALSE(delegate->fired()); 84 EXPECT_FALSE(delegate->fired());
85 } 85 }
86 86
87 TEST_F(QuicConnectionHelperTest, CreateAlarmAndReset) { 87 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarmAndReset) {
88 TestDelegate* delegate = new TestDelegate(); 88 TestDelegate* delegate = new TestDelegate();
89 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate)); 89 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate));
90 90
91 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); 91 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1);
92 alarm->Set(clock_.Now().Add(delta)); 92 alarm->Set(clock_.Now().Add(delta));
93 alarm->Cancel(); 93 alarm->Cancel();
94 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); 94 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3);
95 alarm->Set(clock_.Now().Add(new_delta)); 95 alarm->Set(clock_.Now().Add(new_delta));
96 96
97 // The alarm task should still be posted. 97 // The alarm task should still be posted.
98 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); 98 ASSERT_EQ(1u, runner_->GetPostedTasks().size());
99 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()), 99 EXPECT_EQ(base::TimeDelta::FromMicroseconds(delta.ToMicroseconds()),
100 runner_->GetPostedTasks()[0].delay); 100 runner_->GetPostedTasks()[0].delay);
101 101
102 runner_->RunNextTask(); 102 runner_->RunNextTask();
103 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); 103 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now());
104 EXPECT_FALSE(delegate->fired()); 104 EXPECT_FALSE(delegate->fired());
105 105
106 // The alarm task should be posted again. 106 // The alarm task should be posted again.
107 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); 107 ASSERT_EQ(1u, runner_->GetPostedTasks().size());
108 108
109 runner_->RunNextTask(); 109 runner_->RunNextTask();
110 EXPECT_EQ(QuicTime::Zero().Add(new_delta), clock_.Now()); 110 EXPECT_EQ(QuicTime::Zero().Add(new_delta), clock_.Now());
111 EXPECT_TRUE(delegate->fired()); 111 EXPECT_TRUE(delegate->fired());
112 } 112 }
113 113
114 TEST_F(QuicConnectionHelperTest, CreateAlarmAndResetEarlier) { 114 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarmAndResetEarlier) {
115 TestDelegate* delegate = new TestDelegate(); 115 TestDelegate* delegate = new TestDelegate();
116 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate)); 116 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate));
117 117
118 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(3); 118 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(3);
119 alarm->Set(clock_.Now().Add(delta)); 119 alarm->Set(clock_.Now().Add(delta));
120 alarm->Cancel(); 120 alarm->Cancel();
121 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(1); 121 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(1);
122 alarm->Set(clock_.Now().Add(new_delta)); 122 alarm->Set(clock_.Now().Add(new_delta));
123 123
124 // Both alarm tasks will be posted. 124 // Both alarm tasks will be posted.
125 ASSERT_EQ(2u, runner_->GetPostedTasks().size()); 125 ASSERT_EQ(2u, runner_->GetPostedTasks().size());
126 126
127 // The earlier task will execute and will fire the alarm-> 127 // The earlier task will execute and will fire the alarm->
128 runner_->RunNextTask(); 128 runner_->RunNextTask();
129 EXPECT_EQ(QuicTime::Zero().Add(new_delta), clock_.Now()); 129 EXPECT_EQ(QuicTime::Zero().Add(new_delta), clock_.Now());
130 EXPECT_TRUE(delegate->fired()); 130 EXPECT_TRUE(delegate->fired());
131 delegate->Clear(); 131 delegate->Clear();
132 132
133 // The latter task is still posted. 133 // The latter task is still posted.
134 ASSERT_EQ(1u, runner_->GetPostedTasks().size()); 134 ASSERT_EQ(1u, runner_->GetPostedTasks().size());
135 135
136 // When the latter task is executed, the weak ptr will be invalid and 136 // When the latter task is executed, the weak ptr will be invalid and
137 // the alarm will not fire. 137 // the alarm will not fire.
138 runner_->RunNextTask(); 138 runner_->RunNextTask();
139 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now()); 139 EXPECT_EQ(QuicTime::Zero().Add(delta), clock_.Now());
140 EXPECT_FALSE(delegate->fired()); 140 EXPECT_FALSE(delegate->fired());
141 } 141 }
142 142
143 TEST_F(QuicConnectionHelperTest, CreateAlarmAndUpdate) { 143 TEST_F(QuicChromiumConnectionHelperTest, CreateAlarmAndUpdate) {
144 TestDelegate* delegate = new TestDelegate(); 144 TestDelegate* delegate = new TestDelegate();
145 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate)); 145 scoped_ptr<QuicAlarm> alarm(helper_.CreateAlarm(delegate));
146 146
147 const QuicClock* clock = helper_.GetClock(); 147 const QuicClock* clock = helper_.GetClock();
148 QuicTime start = clock->Now(); 148 QuicTime start = clock->Now();
149 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1); 149 QuicTime::Delta delta = QuicTime::Delta::FromMicroseconds(1);
150 alarm->Set(clock->Now().Add(delta)); 150 alarm->Set(clock->Now().Add(delta));
151 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3); 151 QuicTime::Delta new_delta = QuicTime::Delta::FromMicroseconds(3);
152 alarm->Update(clock->Now().Add(new_delta), 152 alarm->Update(clock->Now().Add(new_delta),
153 QuicTime::Delta::FromMicroseconds(1)); 153 QuicTime::Delta::FromMicroseconds(1));
(...skipping 26 matching lines...) Expand all
180 EXPECT_TRUE(alarm->IsSet()); 180 EXPECT_TRUE(alarm->IsSet());
181 181
182 // Update it with an uninitialized time and ensure it's cancelled. 182 // Update it with an uninitialized time and ensure it's cancelled.
183 alarm->Update(QuicTime::Zero(), QuicTime::Delta::FromMicroseconds(1)); 183 alarm->Update(QuicTime::Zero(), QuicTime::Delta::FromMicroseconds(1));
184 EXPECT_FALSE(alarm->IsSet()); 184 EXPECT_FALSE(alarm->IsSet());
185 } 185 }
186 186
187 } // namespace 187 } // namespace
188 } // namespace test 188 } // namespace test
189 } // namespace net 189 } // namespace net
OLDNEW
« no previous file with comments | « net/quic/quic_chromium_connection_helper.cc ('k') | net/quic/quic_connection_helper.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698