| // Copyright 2015 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "components/proximity_auth/bluetooth_throttler_impl.h" |
| |
| #include "base/test/simple_test_tick_clock.h" |
| #include "base/time/time.h" |
| #include "components/proximity_auth/wire_message.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace proximity_auth { |
| namespace { |
| |
| class StubConnection : public Connection { |
| public: |
| StubConnection() : Connection(RemoteDevice()) {} |
| ~StubConnection() override {} |
| |
| void Connect() override {} |
| void Disconnect() override {} |
| void SendMessageImpl(scoped_ptr<WireMessage> message) override {} |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(StubConnection); |
| }; |
| |
| class TestBluetoothThrottler : public BluetoothThrottlerImpl { |
| public: |
| explicit TestBluetoothThrottler(scoped_ptr<base::TickClock> clock) |
| : BluetoothThrottlerImpl(clock.Pass()) {} |
| ~TestBluetoothThrottler() override {} |
| |
| // Increase visibility for testing. |
| using BluetoothThrottlerImpl::GetCooldownTimeDelta; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(TestBluetoothThrottler); |
| }; |
| |
| } // namespace |
| |
| class ProximityAuthBluetoothThrottlerImplTest : public testing::Test { |
| public: |
| ProximityAuthBluetoothThrottlerImplTest() |
| : clock_(new base::SimpleTestTickClock), |
| throttler_(make_scoped_ptr(clock_)) { |
| // The throttler treats null times as special, so start with a non-null |
| // time. |
| clock_->Advance(base::TimeDelta::FromSeconds(1)); |
| } |
| |
| protected: |
| // The clock is owned by the |throttler_|. |
| base::SimpleTestTickClock* clock_; |
| TestBluetoothThrottler throttler_; |
| }; |
| |
| TEST_F(ProximityAuthBluetoothThrottlerImplTest, |
| GetDelay_FirstConnectionIsNotThrottled) { |
| EXPECT_EQ(base::TimeDelta(), throttler_.GetDelay()); |
| } |
| |
| TEST_F(ProximityAuthBluetoothThrottlerImplTest, |
| GetDelay_ConnectionAfterDisconnectIsThrottled) { |
| // Simulate a connection followed by a disconnection. |
| StubConnection connection; |
| throttler_.OnConnection(&connection); |
| static_cast<ConnectionObserver*>(&throttler_) |
| ->OnConnectionStatusChanged(&connection, Connection::CONNECTED, |
| Connection::DISCONNECTED); |
| EXPECT_GT(throttler_.GetDelay(), base::TimeDelta()); |
| } |
| |
| TEST_F(ProximityAuthBluetoothThrottlerImplTest, |
| GetDelay_DelayedConnectionAfterDisconnectIsNotThrottled) { |
| // Simulate a connection followed by a disconnection, then allow the cooldown |
| // period to elapse. |
| StubConnection connection; |
| throttler_.OnConnection(&connection); |
| static_cast<ConnectionObserver*>(&throttler_) |
| ->OnConnectionStatusChanged(&connection, Connection::CONNECTED, |
| Connection::DISCONNECTED); |
| clock_->Advance(throttler_.GetCooldownTimeDelta()); |
| EXPECT_EQ(base::TimeDelta(), throttler_.GetDelay()); |
| } |
| |
| } // namespace proximity_auth |