// 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