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

Side by Side Diff: components/proximity_auth/bluetooth_throttler_impl_unittest.cc

Issue 2561203002: Migrate weave-related classes from proximity_auth/ble to cryptauth/ble. (Closed)
Patch Set: Rebase. Created 4 years 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
OLDNEW
(Empty)
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 "components/proximity_auth/bluetooth_throttler_impl.h"
6
7 #include <utility>
8
9 #include "base/macros.h"
10 #include "base/memory/ptr_util.h"
11 #include "base/test/simple_test_tick_clock.h"
12 #include "base/time/time.h"
13 #include "components/proximity_auth/fake_connection.h"
14 #include "components/proximity_auth/wire_message.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16
17 namespace proximity_auth {
18 namespace {
19
20 class TestBluetoothThrottler : public BluetoothThrottlerImpl {
21 public:
22 explicit TestBluetoothThrottler(std::unique_ptr<base::TickClock> clock)
23 : BluetoothThrottlerImpl(std::move(clock)) {}
24 ~TestBluetoothThrottler() override {}
25
26 // Increase visibility for testing.
27 using BluetoothThrottlerImpl::GetCooldownTimeDelta;
28
29 private:
30 DISALLOW_COPY_AND_ASSIGN(TestBluetoothThrottler);
31 };
32
33 } // namespace
34
35 class ProximityAuthBluetoothThrottlerImplTest : public testing::Test {
36 public:
37 ProximityAuthBluetoothThrottlerImplTest()
38 : clock_(new base::SimpleTestTickClock),
39 throttler_(base::WrapUnique(clock_)) {
40 // The throttler treats null times as special, so start with a non-null
41 // time.
42 clock_->Advance(base::TimeDelta::FromSeconds(1));
43 }
44
45 void PerformConnectionStateTransition(Connection::Status old_status,
46 Connection::Status new_status) {
47 FakeConnection connection((cryptauth::RemoteDevice()));
48 throttler_.OnConnection(&connection);
49 static_cast<ConnectionObserver*>(&throttler_)
50 ->OnConnectionStatusChanged(&connection, old_status, new_status);
51 }
52
53 protected:
54 // The clock is owned by the |throttler_|.
55 base::SimpleTestTickClock* clock_;
56 TestBluetoothThrottler throttler_;
57 };
58
59 TEST_F(ProximityAuthBluetoothThrottlerImplTest,
60 GetDelay_FirstConnectionIsNotThrottled) {
61 EXPECT_EQ(base::TimeDelta(), throttler_.GetDelay());
62 }
63
64 TEST_F(ProximityAuthBluetoothThrottlerImplTest,
65 GetDelay_ConnectionAfterDisconnectIsThrottled) {
66 // Simulate a connection followed by a disconnection.
67 PerformConnectionStateTransition(Connection::CONNECTED,
68 Connection::DISCONNECTED);
69 EXPECT_GT(throttler_.GetDelay(), base::TimeDelta());
70 }
71
72 TEST_F(ProximityAuthBluetoothThrottlerImplTest,
73 GetDelay_ConnectionAfterIsProgressDisconnectIsThrottled) {
74 // Simulate an attempt to connect (in progress connection) followed by a
75 // disconnection.
76 PerformConnectionStateTransition(Connection::IN_PROGRESS,
77 Connection::DISCONNECTED);
78 EXPECT_GT(throttler_.GetDelay(), base::TimeDelta());
79 }
80
81 TEST_F(ProximityAuthBluetoothThrottlerImplTest,
82 GetDelay_DelayedConnectionAfterDisconnectIsNotThrottled) {
83 // Simulate a connection followed by a disconnection, then allow the cooldown
84 // period to elapse.
85 PerformConnectionStateTransition(Connection::CONNECTED,
86 Connection::DISCONNECTED);
87 clock_->Advance(throttler_.GetCooldownTimeDelta());
88 EXPECT_EQ(base::TimeDelta(), throttler_.GetDelay());
89 }
90
91 TEST_F(ProximityAuthBluetoothThrottlerImplTest,
92 GetDelay_DelayedConnectionAfterInProgressDisconnectIsNotThrottled) {
93 // Simulate an attempt to connect (in progress connection) followed by a
94 // disconnection, then allow the cooldown period to elapse.
95 PerformConnectionStateTransition(Connection::IN_PROGRESS,
96 Connection::DISCONNECTED);
97 clock_->Advance(throttler_.GetCooldownTimeDelta());
98 EXPECT_EQ(base::TimeDelta(), throttler_.GetDelay());
99 }
100
101 } // namespace proximity_auth
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698