OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "google_apis/gcm/engine/heartbeat_manager.h" | 5 #include "google_apis/gcm/engine/heartbeat_manager.h" |
6 | 6 |
| 7 #include <memory> |
| 8 |
7 #include "base/bind.h" | 9 #include "base/bind.h" |
8 #include "base/bind_helpers.h" | 10 #include "base/bind_helpers.h" |
9 #include "base/memory/scoped_ptr.h" | 11 #include "base/memory/ptr_util.h" |
10 #include "base/test/test_simple_task_runner.h" | 12 #include "base/test/test_simple_task_runner.h" |
11 #include "base/thread_task_runner_handle.h" | 13 #include "base/thread_task_runner_handle.h" |
12 #include "base/time/time.h" | 14 #include "base/time/time.h" |
13 #include "base/timer/timer.h" | 15 #include "base/timer/timer.h" |
14 #include "google_apis/gcm/protocol/mcs.pb.h" | 16 #include "google_apis/gcm/protocol/mcs.pb.h" |
15 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
16 | 18 |
17 namespace gcm { | 19 namespace gcm { |
18 | 20 |
19 namespace { | 21 namespace { |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
54 int reconnects_triggered() const { return reconnects_triggered_; } | 56 int reconnects_triggered() const { return reconnects_triggered_; } |
55 | 57 |
56 // Starts the heartbeat manager. | 58 // Starts the heartbeat manager. |
57 void StartManager(); | 59 void StartManager(); |
58 | 60 |
59 private: | 61 private: |
60 // Helper functions for verifying heartbeat manager effects. | 62 // Helper functions for verifying heartbeat manager effects. |
61 void SendHeartbeatClosure(); | 63 void SendHeartbeatClosure(); |
62 void TriggerReconnectClosure(ConnectionFactory::ConnectionResetReason reason); | 64 void TriggerReconnectClosure(ConnectionFactory::ConnectionResetReason reason); |
63 | 65 |
64 scoped_ptr<TestHeartbeatManager> manager_; | 66 std::unique_ptr<TestHeartbeatManager> manager_; |
65 | 67 |
66 int heartbeats_sent_; | 68 int heartbeats_sent_; |
67 int reconnects_triggered_; | 69 int reconnects_triggered_; |
68 | 70 |
69 scoped_refptr<base::TestSimpleTaskRunner> task_runner_; | 71 scoped_refptr<base::TestSimpleTaskRunner> task_runner_; |
70 base::ThreadTaskRunnerHandle task_runner_handle_; | 72 base::ThreadTaskRunnerHandle task_runner_handle_; |
71 }; | 73 }; |
72 | 74 |
73 HeartbeatManagerTest::HeartbeatManagerTest() | 75 HeartbeatManagerTest::HeartbeatManagerTest() |
74 : manager_(new TestHeartbeatManager()), | 76 : manager_(new TestHeartbeatManager()), |
(...skipping 100 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
175 | 177 |
176 EXPECT_LE(manager()->GetNextHeartbeatTime() - base::TimeTicks::Now(), | 178 EXPECT_LE(manager()->GetNextHeartbeatTime() - base::TimeTicks::Now(), |
177 base::TimeDelta::FromMilliseconds(kIntervalMs)); | 179 base::TimeDelta::FromMilliseconds(kIntervalMs)); |
178 EXPECT_NE(heartbeat, manager()->GetNextHeartbeatTime()); | 180 EXPECT_NE(heartbeat, manager()->GetNextHeartbeatTime()); |
179 } | 181 } |
180 | 182 |
181 // Updating the timer used for heartbeats before starting should not start the | 183 // Updating the timer used for heartbeats before starting should not start the |
182 // timer. | 184 // timer. |
183 TEST_F(HeartbeatManagerTest, UpdateTimerBeforeStart) { | 185 TEST_F(HeartbeatManagerTest, UpdateTimerBeforeStart) { |
184 manager()->UpdateHeartbeatTimer( | 186 manager()->UpdateHeartbeatTimer( |
185 make_scoped_ptr(new base::Timer(true, false))); | 187 base::WrapUnique(new base::Timer(true, false))); |
186 EXPECT_TRUE(manager()->GetNextHeartbeatTime().is_null()); | 188 EXPECT_TRUE(manager()->GetNextHeartbeatTime().is_null()); |
187 } | 189 } |
188 | 190 |
189 // Updating the timer used for heartbeats after starting should restart the | 191 // Updating the timer used for heartbeats after starting should restart the |
190 // timer but not increase the heartbeat time by more than a millisecond. | 192 // timer but not increase the heartbeat time by more than a millisecond. |
191 TEST_F(HeartbeatManagerTest, UpdateTimerAfterStart) { | 193 TEST_F(HeartbeatManagerTest, UpdateTimerAfterStart) { |
192 StartManager(); | 194 StartManager(); |
193 base::TimeTicks heartbeat = manager()->GetNextHeartbeatTime(); | 195 base::TimeTicks heartbeat = manager()->GetNextHeartbeatTime(); |
194 | 196 |
195 manager()->UpdateHeartbeatTimer( | 197 manager()->UpdateHeartbeatTimer( |
196 make_scoped_ptr(new base::Timer(true, false))); | 198 base::WrapUnique(new base::Timer(true, false))); |
197 EXPECT_LT(manager()->GetNextHeartbeatTime() - heartbeat, | 199 EXPECT_LT(manager()->GetNextHeartbeatTime() - heartbeat, |
198 base::TimeDelta::FromMilliseconds(5)); | 200 base::TimeDelta::FromMilliseconds(5)); |
199 } | 201 } |
200 | 202 |
201 // Stopping the manager should reset the heartbeat timer. | 203 // Stopping the manager should reset the heartbeat timer. |
202 TEST_F(HeartbeatManagerTest, Stop) { | 204 TEST_F(HeartbeatManagerTest, Stop) { |
203 StartManager(); | 205 StartManager(); |
204 EXPECT_GT(manager()->GetNextHeartbeatTime(), base::TimeTicks::Now()); | 206 EXPECT_GT(manager()->GetNextHeartbeatTime(), base::TimeTicks::Now()); |
205 | 207 |
206 manager()->Stop(); | 208 manager()->Stop(); |
(...skipping 99 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
306 heartbeat = manager()->GetNextHeartbeatTime(); | 308 heartbeat = manager()->GetNextHeartbeatTime(); |
307 EXPECT_GT(heartbeat - base::TimeTicks::Now(), | 309 EXPECT_GT(heartbeat - base::TimeTicks::Now(), |
308 base::TimeDelta::FromMilliseconds(kDefaultAckIntervalMs)); | 310 base::TimeDelta::FromMilliseconds(kDefaultAckIntervalMs)); |
309 EXPECT_LE(heartbeat - base::TimeTicks::Now(), | 311 EXPECT_LE(heartbeat - base::TimeTicks::Now(), |
310 base::TimeDelta::FromMilliseconds(kCustomIntervalMs)); | 312 base::TimeDelta::FromMilliseconds(kCustomIntervalMs)); |
311 } | 313 } |
312 | 314 |
313 } // namespace | 315 } // namespace |
314 | 316 |
315 } // namespace gcm | 317 } // namespace gcm |
OLD | NEW |