[Easy Unlock] Port the BluetoothThrottler class to native code.

components/proximity_auth/bluetooth_throttler_impl_unittest.cc

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