1 // Copyright 2015 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 "remoting/host/backoff_timer.h"
6
7 #include "base/bind.h"
8 #include "base/memory/ptr_util.h"
9 #include "base/timer/mock_timer.h"
10 #include "testing/gtest/include/gtest/gtest.h"
11
12 namespace remoting {
13
14 namespace {
15
IncrementCounter(int * counter)16 void IncrementCounter(int* counter) {
17 ++(*counter);
18 }
19
20 } // namespace
21
TEST(BackoffTimer,Basic)22 TEST(BackoffTimer, Basic) {
23 base::MockOneShotTimer* mock_timer = new base::MockOneShotTimer();
24 BackoffTimer backoff_timer;
25 backoff_timer.SetTimerForTest(base::WrapUnique(mock_timer));
26 ASSERT_FALSE(backoff_timer.IsRunning());
27
28 int counter = 0;
29 backoff_timer.Start(FROM_HERE, base::TimeDelta::FromMilliseconds(10),
30 base::TimeDelta::FromMilliseconds(50),
31 base::BindRepeating(&IncrementCounter, &counter));
32 ASSERT_TRUE(backoff_timer.IsRunning());
33 ASSERT_EQ(0, counter);
34 ASSERT_NEAR(0, mock_timer->GetCurrentDelay().InMillisecondsF(), 1);
35
36 mock_timer->Fire();
37 ASSERT_TRUE(backoff_timer.IsRunning());
38 ASSERT_EQ(1, counter);
39 EXPECT_NEAR(10, mock_timer->GetCurrentDelay().InMillisecondsF(), 1);
40
41 mock_timer->Fire();
42 ASSERT_TRUE(backoff_timer.IsRunning());
43 ASSERT_EQ(2, counter);
44 EXPECT_NEAR(20, mock_timer->GetCurrentDelay().InMillisecondsF(), 1);
45
46 mock_timer->Fire();
47 ASSERT_TRUE(backoff_timer.IsRunning());
48 ASSERT_EQ(3, counter);
49 EXPECT_NEAR(40, mock_timer->GetCurrentDelay().InMillisecondsF(), 1);
50
51 mock_timer->Fire();
52 ASSERT_TRUE(backoff_timer.IsRunning());
53 ASSERT_EQ(4, counter);
54 EXPECT_NEAR(50, mock_timer->GetCurrentDelay().InMillisecondsF(), 1);
55
56 mock_timer->Fire();
57 ASSERT_TRUE(backoff_timer.IsRunning());
58 ASSERT_EQ(5, counter);
59 EXPECT_NEAR(50, mock_timer->GetCurrentDelay().InMillisecondsF(), 1);
60
61 backoff_timer.Stop();
62 ASSERT_FALSE(backoff_timer.IsRunning());
63 }
64
65 } // namespace remoting
66