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/base/backoff_entry.h" | |
6 #include "testing/gtest/include/gtest/gtest.h" | |
7 | |
8 namespace { | |
9 | |
10 using base::TimeDelta; | |
11 using base::TimeTicks; | |
12 using net::BackoffEntry; | |
13 | |
14 BackoffEntry::Policy base_policy = { 0, 1000, 2.0, 0.0, 20000, 2000, false }; | |
15 | |
16 class TestBackoffEntry : public BackoffEntry { | |
17 public: | |
18 explicit TestBackoffEntry(const Policy* const policy) | |
19 : BackoffEntry(policy), | |
20 now_(TimeTicks()) { | |
21 // Work around initialization in constructor not picking up | |
22 // fake time. | |
23 SetCustomReleaseTime(TimeTicks()); | |
24 } | |
25 | |
26 ~TestBackoffEntry() override {} | |
27 | |
28 TimeTicks ImplGetTimeNow() const override { return now_; } | |
29 | |
30 void set_now(const TimeTicks& now) { | |
31 now_ = now; | |
32 } | |
33 | |
34 private: | |
35 TimeTicks now_; | |
36 | |
37 DISALLOW_COPY_AND_ASSIGN(TestBackoffEntry); | |
38 }; | |
39 | |
40 TEST(BackoffEntryTest, BaseTest) { | |
41 TestBackoffEntry entry(&base_policy); | |
42 EXPECT_FALSE(entry.ShouldRejectRequest()); | |
43 EXPECT_EQ(TimeDelta(), entry.GetTimeUntilRelease()); | |
44 | |
45 entry.InformOfRequest(false); | |
46 EXPECT_TRUE(entry.ShouldRejectRequest()); | |
47 EXPECT_EQ(TimeDelta::FromMilliseconds(1000), entry.GetTimeUntilRelease()); | |
48 } | |
49 | |
50 TEST(BackoffEntryTest, CanDiscardNeverExpires) { | |
51 BackoffEntry::Policy never_expires_policy = base_policy; | |
52 never_expires_policy.entry_lifetime_ms = -1; | |
53 TestBackoffEntry never_expires(&never_expires_policy); | |
54 EXPECT_FALSE(never_expires.CanDiscard()); | |
55 never_expires.set_now(TimeTicks() + TimeDelta::FromDays(100)); | |
56 EXPECT_FALSE(never_expires.CanDiscard()); | |
57 } | |
58 | |
59 TEST(BackoffEntryTest, CanDiscard) { | |
60 TestBackoffEntry entry(&base_policy); | |
61 // Because lifetime is non-zero, we shouldn't be able to discard yet. | |
62 EXPECT_FALSE(entry.CanDiscard()); | |
63 | |
64 // Test the "being used" case. | |
65 entry.InformOfRequest(false); | |
66 EXPECT_FALSE(entry.CanDiscard()); | |
67 | |
68 // Test the case where there are errors but we can time out. | |
69 entry.set_now( | |
70 entry.GetReleaseTime() + TimeDelta::FromMilliseconds(1)); | |
71 EXPECT_FALSE(entry.CanDiscard()); | |
72 entry.set_now(entry.GetReleaseTime() + TimeDelta::FromMilliseconds( | |
73 base_policy.maximum_backoff_ms + 1)); | |
74 EXPECT_TRUE(entry.CanDiscard()); | |
75 | |
76 // Test the final case (no errors, dependent only on specified lifetime). | |
77 entry.set_now(entry.GetReleaseTime() + TimeDelta::FromMilliseconds( | |
78 base_policy.entry_lifetime_ms - 1)); | |
79 entry.InformOfRequest(true); | |
80 EXPECT_FALSE(entry.CanDiscard()); | |
81 entry.set_now(entry.GetReleaseTime() + TimeDelta::FromMilliseconds( | |
82 base_policy.entry_lifetime_ms)); | |
83 EXPECT_TRUE(entry.CanDiscard()); | |
84 } | |
85 | |
86 TEST(BackoffEntryTest, CanDiscardAlwaysDelay) { | |
87 BackoffEntry::Policy always_delay_policy = base_policy; | |
88 always_delay_policy.always_use_initial_delay = true; | |
89 always_delay_policy.entry_lifetime_ms = 0; | |
90 | |
91 TestBackoffEntry entry(&always_delay_policy); | |
92 | |
93 // Because lifetime is non-zero, we shouldn't be able to discard yet. | |
94 entry.set_now(entry.GetReleaseTime() + TimeDelta::FromMilliseconds(2000)); | |
95 EXPECT_TRUE(entry.CanDiscard()); | |
96 | |
97 // Even with no failures, we wait until the delay before we allow discard. | |
98 entry.InformOfRequest(true); | |
99 EXPECT_FALSE(entry.CanDiscard()); | |
100 | |
101 // Wait until the delay expires, and we can discard the entry again. | |
102 entry.set_now(entry.GetReleaseTime() + TimeDelta::FromMilliseconds(1000)); | |
103 EXPECT_TRUE(entry.CanDiscard()); | |
104 } | |
105 | |
106 TEST(BackoffEntryTest, CanDiscardNotStored) { | |
107 BackoffEntry::Policy no_store_policy = base_policy; | |
108 no_store_policy.entry_lifetime_ms = 0; | |
109 TestBackoffEntry not_stored(&no_store_policy); | |
110 EXPECT_TRUE(not_stored.CanDiscard()); | |
111 } | |
112 | |
113 TEST(BackoffEntryTest, ShouldIgnoreFirstTwo) { | |
114 BackoffEntry::Policy lenient_policy = base_policy; | |
115 lenient_policy.num_errors_to_ignore = 2; | |
116 | |
117 BackoffEntry entry(&lenient_policy); | |
118 | |
119 entry.InformOfRequest(false); | |
120 EXPECT_FALSE(entry.ShouldRejectRequest()); | |
121 | |
122 entry.InformOfRequest(false); | |
123 EXPECT_FALSE(entry.ShouldRejectRequest()); | |
124 | |
125 entry.InformOfRequest(false); | |
126 EXPECT_TRUE(entry.ShouldRejectRequest()); | |
127 } | |
128 | |
129 TEST(BackoffEntryTest, ReleaseTimeCalculation) { | |
130 TestBackoffEntry entry(&base_policy); | |
131 | |
132 // With zero errors, should return "now". | |
133 TimeTicks result = entry.GetReleaseTime(); | |
134 EXPECT_EQ(entry.ImplGetTimeNow(), result); | |
135 | |
136 // 1 error. | |
137 entry.InformOfRequest(false); | |
138 result = entry.GetReleaseTime(); | |
139 EXPECT_EQ(entry.ImplGetTimeNow() + TimeDelta::FromMilliseconds(1000), result); | |
140 EXPECT_EQ(TimeDelta::FromMilliseconds(1000), entry.GetTimeUntilRelease()); | |
141 | |
142 // 2 errors. | |
143 entry.InformOfRequest(false); | |
144 result = entry.GetReleaseTime(); | |
145 EXPECT_EQ(entry.ImplGetTimeNow() + TimeDelta::FromMilliseconds(2000), result); | |
146 EXPECT_EQ(TimeDelta::FromMilliseconds(2000), entry.GetTimeUntilRelease()); | |
147 | |
148 // 3 errors. | |
149 entry.InformOfRequest(false); | |
150 result = entry.GetReleaseTime(); | |
151 EXPECT_EQ(entry.ImplGetTimeNow() + TimeDelta::FromMilliseconds(4000), result); | |
152 EXPECT_EQ(TimeDelta::FromMilliseconds(4000), entry.GetTimeUntilRelease()); | |
153 | |
154 // 6 errors (to check it doesn't pass maximum). | |
155 entry.InformOfRequest(false); | |
156 entry.InformOfRequest(false); | |
157 entry.InformOfRequest(false); | |
158 result = entry.GetReleaseTime(); | |
159 EXPECT_EQ( | |
160 entry.ImplGetTimeNow() + TimeDelta::FromMilliseconds(20000), result); | |
161 } | |
162 | |
163 TEST(BackoffEntryTest, ReleaseTimeCalculationAlwaysDelay) { | |
164 BackoffEntry::Policy always_delay_policy = base_policy; | |
165 always_delay_policy.always_use_initial_delay = true; | |
166 always_delay_policy.num_errors_to_ignore = 2; | |
167 | |
168 TestBackoffEntry entry(&always_delay_policy); | |
169 | |
170 // With previous requests, should return "now". | |
171 TimeTicks result = entry.GetReleaseTime(); | |
172 EXPECT_EQ(TimeDelta(), entry.GetTimeUntilRelease()); | |
173 | |
174 // 1 error. | |
175 entry.InformOfRequest(false); | |
176 EXPECT_EQ(TimeDelta::FromMilliseconds(1000), entry.GetTimeUntilRelease()); | |
177 | |
178 // 2 errors. | |
179 entry.InformOfRequest(false); | |
180 EXPECT_EQ(TimeDelta::FromMilliseconds(1000), entry.GetTimeUntilRelease()); | |
181 | |
182 // 3 errors, exponential backoff starts. | |
183 entry.InformOfRequest(false); | |
184 EXPECT_EQ(TimeDelta::FromMilliseconds(2000), entry.GetTimeUntilRelease()); | |
185 | |
186 // 4 errors. | |
187 entry.InformOfRequest(false); | |
188 EXPECT_EQ(TimeDelta::FromMilliseconds(4000), entry.GetTimeUntilRelease()); | |
189 | |
190 // 8 errors (to check it doesn't pass maximum). | |
191 entry.InformOfRequest(false); | |
192 entry.InformOfRequest(false); | |
193 entry.InformOfRequest(false); | |
194 entry.InformOfRequest(false); | |
195 result = entry.GetReleaseTime(); | |
196 EXPECT_EQ(TimeDelta::FromMilliseconds(20000), entry.GetTimeUntilRelease()); | |
197 } | |
198 | |
199 TEST(BackoffEntryTest, ReleaseTimeCalculationWithJitter) { | |
200 for (int i = 0; i < 10; ++i) { | |
201 BackoffEntry::Policy jittery_policy = base_policy; | |
202 jittery_policy.jitter_factor = 0.2; | |
203 | |
204 TestBackoffEntry entry(&jittery_policy); | |
205 | |
206 entry.InformOfRequest(false); | |
207 entry.InformOfRequest(false); | |
208 entry.InformOfRequest(false); | |
209 TimeTicks result = entry.GetReleaseTime(); | |
210 EXPECT_LE( | |
211 entry.ImplGetTimeNow() + TimeDelta::FromMilliseconds(3200), result); | |
212 EXPECT_GE( | |
213 entry.ImplGetTimeNow() + TimeDelta::FromMilliseconds(4000), result); | |
214 } | |
215 } | |
216 | |
217 TEST(BackoffEntryTest, FailureThenSuccess) { | |
218 TestBackoffEntry entry(&base_policy); | |
219 | |
220 // Failure count 1, establishes horizon. | |
221 entry.InformOfRequest(false); | |
222 TimeTicks release_time = entry.GetReleaseTime(); | |
223 EXPECT_EQ(TimeTicks() + TimeDelta::FromMilliseconds(1000), release_time); | |
224 | |
225 // Success, failure count 0, should not advance past | |
226 // the horizon that was already set. | |
227 entry.set_now(release_time - TimeDelta::FromMilliseconds(200)); | |
228 entry.InformOfRequest(true); | |
229 EXPECT_EQ(release_time, entry.GetReleaseTime()); | |
230 | |
231 // Failure, failure count 1. | |
232 entry.InformOfRequest(false); | |
233 EXPECT_EQ(release_time + TimeDelta::FromMilliseconds(800), | |
234 entry.GetReleaseTime()); | |
235 } | |
236 | |
237 TEST(BackoffEntryTest, FailureThenSuccessAlwaysDelay) { | |
238 BackoffEntry::Policy always_delay_policy = base_policy; | |
239 always_delay_policy.always_use_initial_delay = true; | |
240 always_delay_policy.num_errors_to_ignore = 1; | |
241 | |
242 TestBackoffEntry entry(&always_delay_policy); | |
243 | |
244 // Failure count 1. | |
245 entry.InformOfRequest(false); | |
246 EXPECT_EQ(TimeDelta::FromMilliseconds(1000), entry.GetTimeUntilRelease()); | |
247 | |
248 // Failure count 2. | |
249 entry.InformOfRequest(false); | |
250 EXPECT_EQ(TimeDelta::FromMilliseconds(2000), entry.GetTimeUntilRelease()); | |
251 entry.set_now(entry.GetReleaseTime() + TimeDelta::FromMilliseconds(2000)); | |
252 | |
253 // Success. We should go back to the original delay. | |
254 entry.InformOfRequest(true); | |
255 EXPECT_EQ(TimeDelta::FromMilliseconds(1000), entry.GetTimeUntilRelease()); | |
256 | |
257 // Failure count reaches 2 again. We should increase the delay once more. | |
258 entry.InformOfRequest(false); | |
259 EXPECT_EQ(TimeDelta::FromMilliseconds(2000), entry.GetTimeUntilRelease()); | |
260 entry.set_now(entry.GetReleaseTime() + TimeDelta::FromMilliseconds(2000)); | |
261 } | |
262 | |
263 TEST(BackoffEntryTest, RetainCustomHorizon) { | |
264 TestBackoffEntry custom(&base_policy); | |
265 TimeTicks custom_horizon = TimeTicks() + TimeDelta::FromDays(3); | |
266 custom.SetCustomReleaseTime(custom_horizon); | |
267 custom.InformOfRequest(false); | |
268 custom.InformOfRequest(true); | |
269 custom.set_now(TimeTicks() + TimeDelta::FromDays(2)); | |
270 custom.InformOfRequest(false); | |
271 custom.InformOfRequest(true); | |
272 EXPECT_EQ(custom_horizon, custom.GetReleaseTime()); | |
273 | |
274 // Now check that once we are at or past the custom horizon, | |
275 // we get normal behavior. | |
276 custom.set_now(TimeTicks() + TimeDelta::FromDays(3)); | |
277 custom.InformOfRequest(false); | |
278 EXPECT_EQ( | |
279 TimeTicks() + TimeDelta::FromDays(3) + TimeDelta::FromMilliseconds(1000), | |
280 custom.GetReleaseTime()); | |
281 } | |
282 | |
283 TEST(BackoffEntryTest, RetainCustomHorizonWhenInitialErrorsIgnored) { | |
284 // Regression test for a bug discovered during code review. | |
285 BackoffEntry::Policy lenient_policy = base_policy; | |
286 lenient_policy.num_errors_to_ignore = 1; | |
287 TestBackoffEntry custom(&lenient_policy); | |
288 TimeTicks custom_horizon = TimeTicks() + TimeDelta::FromDays(3); | |
289 custom.SetCustomReleaseTime(custom_horizon); | |
290 custom.InformOfRequest(false); // This must not reset the horizon. | |
291 EXPECT_EQ(custom_horizon, custom.GetReleaseTime()); | |
292 } | |
293 | |
294 TEST(BackoffEntryTest, OverflowProtection) { | |
295 BackoffEntry::Policy large_multiply_policy = base_policy; | |
296 large_multiply_policy.multiply_factor = 256; | |
297 TestBackoffEntry custom(&large_multiply_policy); | |
298 | |
299 // Trigger enough failures such that more than 11 bits of exponent are used | |
300 // to represent the exponential backoff intermediate values. Given a multiply | |
301 // factor of 256 (2^8), 129 iterations is enough: 2^(8*(129-1)) = 2^1024. | |
302 for (int i = 0; i < 129; ++i) { | |
303 custom.set_now(custom.ImplGetTimeNow() + custom.GetTimeUntilRelease()); | |
304 custom.InformOfRequest(false); | |
305 ASSERT_TRUE(custom.ShouldRejectRequest()); | |
306 } | |
307 | |
308 // Max delay should still be respected. | |
309 EXPECT_EQ(20000, custom.GetTimeUntilRelease().InMilliseconds()); | |
310 } | |
311 | |
312 } // namespace | |
OLD | NEW |