// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "connections/implementation/mediums/bluetooth_classic.h"

#include <memory>
#include <string>
#include <utility>

#include "gtest/gtest.h"
#include "absl/strings/string_view.h"
#include "absl/time/time.h"
#include "connections/implementation/mediums/bluetooth_radio.h"
#include "internal/platform/bluetooth_adapter.h"
#include "internal/platform/bluetooth_classic.h"
#include "internal/platform/cancellation_flag.h"
#include "internal/platform/count_down_latch.h"
#include "internal/platform/expected.h"
#include "internal/platform/feature_flags.h"
#include "internal/platform/implementation/system_clock.h"
#include "internal/platform/logging.h"
#include "internal/platform/medium_environment.h"

namespace nearby {
namespace connections {
namespace {

using FeatureFlags = FeatureFlags::Flags;

constexpr FeatureFlags kTestCases[] = {
    FeatureFlags{
        .enable_cancellation_flag = true,
    },
    FeatureFlags{
        .enable_cancellation_flag = false,
    },
};

constexpr absl::Duration kWaitDuration = absl::Milliseconds(1000);
constexpr absl::string_view kDeviceName{"Simulated BT device #1"};
constexpr absl::string_view kServiceId1{"service ID 1"};
constexpr absl::string_view kServiceId2{"service ID 2"};

class FakeBluetoothClassicMedium final : public BluetoothClassicMedium {
 public:
  explicit FakeBluetoothClassicMedium(BluetoothAdapter& adapter)
      : BluetoothClassicMedium(adapter) {}

  BluetoothSocket ConnectToService(
      BluetoothDevice& remote_device, const std::string& service_uuid,
      CancellationFlag* cancellation_flag) override {
    if (cancel_) {
      cancellation_flag->Cancel();
    }

    return BluetoothClassicMedium::ConnectToService(remote_device, service_uuid,
                                                    cancellation_flag);
  }

  void CancelDuringConnectToService() { cancel_ = true; }

 private:
  bool cancel_ = false;
};

class TestBluetoothClassic : public BluetoothClassic {
 public:
  TestBluetoothClassic(BluetoothRadio& radio,
                       std::unique_ptr<BluetoothClassicMedium> medium)
      : BluetoothClassic(radio, std::move(medium)) {}

  int connect_attempts_count(std::string service_id) {
    return service_id_to_connect_attempts_count_map_[service_id];
  }
};

class BluetoothClassicTest : public ::testing::TestWithParam<FeatureFlags> {
 protected:
  using DiscoveryCallback = BluetoothClassicMedium::DiscoveryCallback;

  void SetUp() override {
    env_.Start();
    radio_a_ = std::make_unique<BluetoothRadio>();
    radio_b_ = std::make_unique<BluetoothRadio>();
    auto medium_a = std::make_unique<FakeBluetoothClassicMedium>(
        radio_a_->GetBluetoothAdapter());
    auto medium_b = std::make_unique<FakeBluetoothClassicMedium>(
        radio_b_->GetBluetoothAdapter());
    medium_a_ = medium_a.get();
    medium_b_ = medium_b.get();
    bt_a_ =
        std::make_unique<TestBluetoothClassic>(*radio_a_, std::move(medium_a));
    bt_b_ =
        std::make_unique<TestBluetoothClassic>(*radio_b_, std::move(medium_b));
    radio_a_->GetBluetoothAdapter().SetName("Device-A");
    radio_b_->GetBluetoothAdapter().SetName("Device-B");
    radio_a_->Enable();
    radio_b_->Enable();
    env_.Sync();
  }

  void TearDown() override {
    env_.Sync(false);
    radio_a_->Disable();
    radio_b_->Disable();
    bt_a_.reset();
    bt_b_.reset();
    env_.Sync(false);
    radio_a_.reset();
    radio_b_.reset();
    env_.Stop();
  }

  MediumEnvironment& env_{MediumEnvironment::Instance()};

  std::unique_ptr<BluetoothRadio> radio_a_;
  std::unique_ptr<BluetoothRadio> radio_b_;
  FakeBluetoothClassicMedium* medium_a_ = nullptr;
  FakeBluetoothClassicMedium* medium_b_ = nullptr;
  std::unique_ptr<TestBluetoothClassic> bt_a_;
  std::unique_ptr<TestBluetoothClassic> bt_b_;
};

TEST_P(BluetoothClassicTest, CanNotTurnOnDiscoverability) {
  FeatureFlags feature_flags = GetParam();
  env_.SetFeatureFlags(feature_flags);

  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothClassic& bt_client = *bt_a_;

  // Cannot turn on discoverability to an empty device name.
  EXPECT_FALSE(bt_client.TurnOnDiscoverability(""));

  // Cannot turn on discoverability when radio is disabled.
  radio_for_client.Disable();
  EXPECT_FALSE(bt_client.TurnOnDiscoverability(std::string(kDeviceName)));
  radio_for_client.Enable();

  // Cannot connect when discovery is running.
  EXPECT_TRUE(bt_client.TurnOnDiscoverability(std::string(kDeviceName)));
  env_.Sync();
  EXPECT_FALSE(bt_client.TurnOnDiscoverability(std::string(kDeviceName)));
}

TEST_P(BluetoothClassicTest, CanNotConnect) {
  FeatureFlags feature_flags = GetParam();
  env_.SetFeatureFlags(feature_flags);

  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothClassic& bt_client = *bt_a_;

  // Cannot connect to an empty service id.
  CancellationFlag flag;
  BluetoothDevice discovered_device;
  ErrorOr<BluetoothSocket> socket_for_client_result =
      bt_client.Connect(discovered_device, "", &flag);

  EXPECT_TRUE(socket_for_client_result.has_error());

  // Cannot connect when radio is disabled.
  radio_for_client.Disable();
  socket_for_client_result =
      bt_client.Connect(discovered_device, std::string(kServiceId1), &flag);
  EXPECT_TRUE(socket_for_client_result.has_error());
  radio_for_client.Enable();

  // Cannot connect when adapter is disabled.
  radio_for_client.GetBluetoothAdapter().SetStatus(
      BluetoothAdapter::Status::kDisabled);
  socket_for_client_result =
      bt_client.Connect(discovered_device, std::string(kServiceId1), &flag);
  EXPECT_TRUE(socket_for_client_result.has_error());
}

TEST_P(BluetoothClassicTest, CannotStartAcceptingConnections) {
  FeatureFlags feature_flags = GetParam();
  env_.SetFeatureFlags(feature_flags);

  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothClassic& bt_client = *bt_a_;

  // Cannot start accepting connections to an empty service ID.
  EXPECT_FALSE(bt_client.StartAcceptingConnections(
      "", [&](const std::string& service_id, BluetoothSocket socket) {}));

  // Cannot start accepting connections  when radio is disabled.
  radio_for_client.Disable();
  EXPECT_FALSE(bt_client.StartAcceptingConnections(
      std::string(kServiceId1),
      [&](const std::string& service_id, BluetoothSocket socket) {}));
  radio_for_client.Enable();

  // Cannot start accepting connections  when it is already accepting.
  EXPECT_FALSE(bt_client.IsAcceptingConnections(std::string(kServiceId1)));
  EXPECT_TRUE(bt_client.StartAcceptingConnections(
      std::string(kServiceId1),
      [&](const std::string& service_id, BluetoothSocket socket) {}));
  EXPECT_TRUE(bt_client.IsAcceptingConnections(std::string(kServiceId1)));
  env_.Sync();
  EXPECT_FALSE(bt_client.StartAcceptingConnections(
      std::string(kServiceId1),
      [&](const std::string& service_id, BluetoothSocket socket) {}));
}

TEST_P(BluetoothClassicTest, CannotStopAcceptingConnections) {
  FeatureFlags feature_flags = GetParam();
  env_.SetFeatureFlags(feature_flags);

  BluetoothClassic& bt_client = *bt_a_;

  // Cannot stop accepting connections to an empty service ID.
  EXPECT_FALSE(bt_client.StopAcceptingConnections(""));

  // Cannot stop accepting connections  when service ID is not accepting.
  EXPECT_FALSE(bt_client.StopAcceptingConnections(std::string(kServiceId1)));
}

TEST_P(BluetoothClassicTest, CannotStartDiscovery) {
  FeatureFlags feature_flags = GetParam();
  env_.SetFeatureFlags(feature_flags);

  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothClassic& bt_client = *bt_a_;

  // Cannot start discovery when service ID is empty.
  EXPECT_FALSE(bt_client.StartDiscovery("", {}));

  // Cannot start discovery when radio is disabled.
  radio_for_client.Disable();
  EXPECT_FALSE(bt_client.StartDiscovery(std::string(kServiceId1), {}));
  radio_for_client.Enable();

  // Cannot  start discovery when it is already discovering.
  EXPECT_TRUE(bt_client.StartDiscovery(std::string(kServiceId1), {}));
  EXPECT_FALSE(bt_client.StartDiscovery(std::string(kServiceId1), {}));
}

TEST_P(BluetoothClassicTest, CanConnect) {
  FeatureFlags feature_flags = GetParam();
  env_.SetFeatureFlags(feature_flags);

  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothRadio& radio_for_server = *radio_b_;
  BluetoothClassic& bt_client = *bt_a_;
  BluetoothClassic& bt_server = *bt_b_;

  EXPECT_TRUE(radio_for_client.IsEnabled());
  EXPECT_TRUE(radio_for_server.IsEnabled());

  EXPECT_TRUE(bt_server.TurnOnDiscoverability(std::string(kDeviceName)));
  EXPECT_EQ(radio_for_server.GetBluetoothAdapter().GetName(),
            std::string(kDeviceName));
  CountDownLatch latch(1);
  BluetoothDevice discovered_device;
  EXPECT_TRUE(bt_client.StartDiscovery(
      std::string(kServiceId1),
      {
          .device_discovered_cb =
              [&latch, &discovered_device](BluetoothDevice& device) {
                discovered_device = device;
                LOG(INFO) << "Discovered device=" << device.GetName()
                          << ", impl=" << &device.GetImpl();
                latch.CountDown();
              },
      }));
  EXPECT_TRUE(latch.Await(kWaitDuration).result());
  EXPECT_TRUE(bt_server.TurnOffDiscoverability());
  ASSERT_TRUE(discovered_device.IsValid());
  BluetoothSocket socket_for_server;
  CountDownLatch accept_latch(1);
  EXPECT_TRUE(bt_server.StartAcceptingConnections(
      std::string(kServiceId1),
      [&](const std::string& service_id, BluetoothSocket socket) {
        socket_for_server = std::move(socket);
        accept_latch.CountDown();
      }));
  CancellationFlag flag;
  ErrorOr<BluetoothSocket> socket_for_client_result =
      bt_client.Connect(discovered_device, std::string(kServiceId1), &flag);
  EXPECT_TRUE(accept_latch.Await(kWaitDuration).result());
  EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId1)));
  EXPECT_TRUE(socket_for_server.IsValid());
  EXPECT_TRUE(socket_for_client_result.has_value());
  EXPECT_TRUE(socket_for_client_result.value().IsValid());
  EXPECT_TRUE(socket_for_server.GetRemoteDevice().IsValid());
  EXPECT_TRUE(socket_for_client_result.value().GetRemoteDevice().IsValid());
}

TEST_P(BluetoothClassicTest, CanCancelBeforeConnect) {
  FeatureFlags feature_flags = GetParam();
  env_.SetFeatureFlags(feature_flags);

  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothRadio& radio_for_server = *radio_b_;
  TestBluetoothClassic& bt_client = *bt_a_;
  TestBluetoothClassic& bt_server = *bt_b_;

  EXPECT_TRUE(radio_for_client.IsEnabled());
  EXPECT_TRUE(radio_for_server.IsEnabled());

  EXPECT_TRUE(bt_server.TurnOnDiscoverability(std::string(kDeviceName)));
  EXPECT_EQ(radio_for_server.GetBluetoothAdapter().GetName(),
            std::string(kDeviceName));
  CountDownLatch latch(1);
  BluetoothDevice discovered_device;
  EXPECT_TRUE(bt_client.StartDiscovery(
      std::string(kServiceId1),
      {
          .device_discovered_cb =
              [&latch, &discovered_device](BluetoothDevice& device) {
                LOG(INFO) << "Discovered device=" << device.GetName()
                          << ", impl=" << &device.GetImpl();
                discovered_device = device;
                latch.CountDown();
              },
      }));
  EXPECT_TRUE(latch.Await(kWaitDuration).result());
  EXPECT_TRUE(bt_server.TurnOffDiscoverability());
  ASSERT_TRUE(discovered_device.IsValid());
  BluetoothSocket socket_for_server;
  CountDownLatch accept_latch(1);
  EXPECT_TRUE(bt_server.StartAcceptingConnections(
      std::string(kServiceId1),
      [&](const std::string& service_id, BluetoothSocket socket) {
        socket_for_server = std::move(socket);
        accept_latch.CountDown();
      }));
  CancellationFlag flag(true);
  ErrorOr<BluetoothSocket> socket_for_client_result =
      bt_client.Connect(discovered_device, std::string(kServiceId1), &flag);
  // If FeatureFlag is disabled, Cancelled is false as no-op.
  if (!feature_flags.enable_cancellation_flag) {
    EXPECT_TRUE(accept_latch.Await(kWaitDuration).result());
    EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId1)));
    EXPECT_TRUE(socket_for_server.IsValid());
    EXPECT_TRUE(socket_for_client_result.has_value());
    EXPECT_TRUE(socket_for_client_result.value().IsValid());
    EXPECT_TRUE(socket_for_server.GetRemoteDevice().IsValid());
    EXPECT_TRUE(socket_for_client_result.value().GetRemoteDevice().IsValid());
  } else {
    EXPECT_FALSE(accept_latch.Await(kWaitDuration).result());
    EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId1)));
    EXPECT_FALSE(socket_for_server.IsValid());
    EXPECT_TRUE(socket_for_client_result.has_error());

    // Expect an invalid socket from stopping during the first attempt to
    // connect, because `Connect` returned immediately when it checked for
    // cancellation.
    EXPECT_EQ(1, bt_client.connect_attempts_count(std::string(kServiceId1)));
  }
}

TEST_P(BluetoothClassicTest, CanCancelDuringConnect) {
  FeatureFlags feature_flags = GetParam();
  env_.SetFeatureFlags(feature_flags);

  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothRadio& radio_for_server = *radio_b_;
  TestBluetoothClassic& bt_client = *bt_a_;
  TestBluetoothClassic& bt_server = *bt_b_;

  // Simulate the flag being cancelled during connection attempt.
  medium_a_->CancelDuringConnectToService();

  EXPECT_TRUE(radio_for_client.IsEnabled());
  EXPECT_TRUE(radio_for_server.IsEnabled());

  EXPECT_TRUE(bt_server.TurnOnDiscoverability(std::string(kDeviceName)));
  EXPECT_EQ(radio_for_server.GetBluetoothAdapter().GetName(),
            std::string(kDeviceName));
  CountDownLatch latch(1);
  BluetoothDevice discovered_device;
  EXPECT_TRUE(bt_client.StartDiscovery(
      std::string(kServiceId1),
      {
          .device_discovered_cb =
              [&latch, &discovered_device](BluetoothDevice& device) {
                discovered_device = device;
                LOG(INFO) << "Discovered device=" << device.GetName()
                          << ", impl=" << &device.GetImpl();
                latch.CountDown();
              },
      }));
  EXPECT_TRUE(latch.Await(kWaitDuration).result());
  EXPECT_TRUE(bt_server.TurnOffDiscoverability());
  ASSERT_TRUE(discovered_device.IsValid());
  BluetoothSocket socket_for_server;
  CountDownLatch accept_latch(1);
  EXPECT_TRUE(bt_server.StartAcceptingConnections(
      std::string(kServiceId1),
      [&](const std::string& service_id, BluetoothSocket socket) {
        socket_for_server = std::move(socket);
        accept_latch.CountDown();
      }));
  CancellationFlag flag;
  ErrorOr<BluetoothSocket> socket_for_client_result =
      bt_client.Connect(discovered_device, std::string(kServiceId1), &flag);
  // If FeatureFlag is disabled, Cancelled is false as no-op.
  if (!feature_flags.enable_cancellation_flag) {
    EXPECT_TRUE(accept_latch.Await(kWaitDuration).result());
    EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId1)));
    EXPECT_TRUE(socket_for_server.IsValid());
    EXPECT_TRUE(socket_for_client_result.has_value());
    EXPECT_TRUE(socket_for_client_result.value().IsValid());
    EXPECT_TRUE(socket_for_server.GetRemoteDevice().IsValid());
    EXPECT_TRUE(socket_for_client_result.value().GetRemoteDevice().IsValid());
  } else {
    EXPECT_FALSE(accept_latch.Await(kWaitDuration).result());
    EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId1)));
    EXPECT_FALSE(socket_for_server.IsValid());
    EXPECT_TRUE(socket_for_client_result.has_error());

    // Since the flag was cancelled during the initial `AttemptToConnect`,
    // except only one attempt instead of the usual three, because the
    // cancellation flag should short-circuit the lengthy connection attempts
    // during shutdown. Because of the way the iteration happens, the check for
    // is cancelled happens after the counter has already been incremented, but
    // before the attempt actually occurs.
    EXPECT_EQ(2, bt_client.connect_attempts_count(std::string(kServiceId1)));
  }
}

TEST_P(BluetoothClassicTest, CanCancelDuringConnect_MultipleEndpoints) {
  FeatureFlags feature_flags = GetParam();
  env_.SetFeatureFlags(feature_flags);

  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothRadio& radio_for_server = *radio_b_;
  TestBluetoothClassic& bt_client = *bt_a_;
  TestBluetoothClassic& bt_server = *bt_b_;
  EXPECT_TRUE(radio_for_client.IsEnabled());
  EXPECT_TRUE(radio_for_server.IsEnabled());

  EXPECT_TRUE(bt_server.TurnOnDiscoverability(std::string(kDeviceName)));
  EXPECT_EQ(radio_for_server.GetBluetoothAdapter().GetName(),
            std::string(kDeviceName));
  CountDownLatch latch(1);
  BluetoothDevice discovered_device;
  EXPECT_TRUE(bt_client.StartDiscovery(
      std::string(kServiceId1),
      {
          .device_discovered_cb =
              [&latch, &discovered_device](BluetoothDevice& device) {
                discovered_device = device;
                LOG(INFO) << "Discovered device=" << device.GetName()
                          << ", impl=" << &device.GetImpl();
                latch.CountDown();
              },
      }));
  EXPECT_TRUE(latch.Await(kWaitDuration).result());
  EXPECT_TRUE(bt_server.TurnOffDiscoverability());
  ASSERT_TRUE(discovered_device.IsValid());
  BluetoothSocket socket_for_server1;
  BluetoothSocket socket_for_server2;
  CountDownLatch accept_latch(1);

  EXPECT_TRUE(bt_server.StartAcceptingConnections(
      std::string(kServiceId1),
      [&](const std::string& service_id, BluetoothSocket socket) {
        socket_for_server1 = std::move(socket);
        accept_latch.CountDown();
      }));
  CancellationFlag flag;
  ErrorOr<BluetoothSocket> socket_for_client1_result =
      bt_client.Connect(discovered_device, std::string(kServiceId1), &flag);

  // Simulate the flag being cancelled during connection attempt to a different
  // endpoint.
  medium_a_->CancelDuringConnectToService();
  EXPECT_TRUE(accept_latch.Await(kWaitDuration).result());

  CountDownLatch accept_latch2(1);
  EXPECT_TRUE(bt_server.StartAcceptingConnections(
      std::string(kServiceId2),
      [&](const std::string& service_id, BluetoothSocket socket) {
        socket_for_server2 = std::move(socket);
        accept_latch2.CountDown();
      }));

  CancellationFlag flag2;
  ErrorOr<BluetoothSocket> socket_for_client2_result =
      bt_client.Connect(discovered_device, std::string(kServiceId2), &flag2);

  // If FeatureFlag is disabled, Cancelled is false as no-op.
  if (!feature_flags.enable_cancellation_flag) {
    EXPECT_TRUE(accept_latch2.Await(kWaitDuration).result());
    EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId1)));
    EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId2)));
    EXPECT_TRUE(socket_for_server1.IsValid());
    EXPECT_TRUE(socket_for_server2.IsValid());
    EXPECT_TRUE(socket_for_client1_result.has_value());
    EXPECT_TRUE(socket_for_client1_result.value().IsValid());
    EXPECT_TRUE(socket_for_client2_result.has_value());
    EXPECT_TRUE(socket_for_client2_result.value().IsValid());
    EXPECT_TRUE(socket_for_server1.GetRemoteDevice().IsValid());
    EXPECT_TRUE(socket_for_server2.GetRemoteDevice().IsValid());
    EXPECT_TRUE(socket_for_client1_result.value().GetRemoteDevice().IsValid());
    EXPECT_TRUE(socket_for_client2_result.value().GetRemoteDevice().IsValid());
  } else {
    EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId1)));
    EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId2)));
    EXPECT_TRUE(socket_for_client1_result.has_value());
    EXPECT_TRUE(socket_for_client1_result.value().IsValid());
    EXPECT_TRUE(socket_for_client2_result.has_error());

    // Since the flag was cancelled during the initial `AttemptToConnect`,
    // except only one attempt instead of the usual three, because the
    // cancellation flag should short-circuit the lengthy connection attempts
    // during shutdown. Because of the way the iteration happens, the check for
    // is cancelled happens after the counter has already been incremented, but
    // before the attempt actually occurs.
    EXPECT_EQ(2, bt_client.connect_attempts_count(std::string(kServiceId2)));

    // With the first service name, we expect one attempt count since it
    // succeeded.
    EXPECT_EQ(1, bt_client.connect_attempts_count(std::string(kServiceId1)));
  }
}

INSTANTIATE_TEST_SUITE_P(ParametrisedBluetoothClassicTest, BluetoothClassicTest,
                         ::testing::ValuesIn(kTestCases));

TEST_F(BluetoothClassicTest, CanConstructValidObject) {
  EXPECT_TRUE(bt_a_->IsMediumValid());
  EXPECT_TRUE(bt_a_->IsAdapterValid());
  EXPECT_TRUE(bt_a_->IsAvailable());
  EXPECT_TRUE(bt_b_->IsMediumValid());
  EXPECT_TRUE(bt_b_->IsAdapterValid());
  EXPECT_TRUE(bt_b_->IsAvailable());
  EXPECT_NE(&radio_a_->GetBluetoothAdapter(), &radio_b_->GetBluetoothAdapter());
}

TEST_F(BluetoothClassicTest, CanStartAdvertising) {
  EXPECT_TRUE(bt_a_->TurnOnDiscoverability(std::string(kDeviceName)));
  EXPECT_EQ(radio_a_->GetBluetoothAdapter().GetName(), kDeviceName);
}

TEST_F(BluetoothClassicTest, CanStopAdvertising) {
  EXPECT_TRUE(bt_a_->TurnOnDiscoverability(std::string(kDeviceName)));
  EXPECT_EQ(radio_a_->GetBluetoothAdapter().GetName(), kDeviceName);
  EXPECT_TRUE(bt_a_->TurnOffDiscoverability());
}

TEST_F(BluetoothClassicTest, CanStartDiscovery) {
  EXPECT_TRUE(bt_a_->TurnOnDiscoverability(std::string(kDeviceName)));
  EXPECT_EQ(radio_a_->GetBluetoothAdapter().GetName(), kDeviceName);
  CountDownLatch latch(1);
  EXPECT_TRUE(bt_b_->StartDiscovery(
      std::string(kServiceId1),
      {
          .device_discovered_cb =
              [&latch](BluetoothDevice& device) { latch.CountDown(); },
      }));
  EXPECT_TRUE(latch.Await(kWaitDuration).result());
  EXPECT_TRUE(bt_a_->TurnOffDiscoverability());
}

TEST_F(BluetoothClassicTest, CanStopDiscovery) {
  CountDownLatch latch(1);
  EXPECT_TRUE(bt_a_->StartDiscovery(
      std::string(kServiceId1),
      {
          .device_discovered_cb =
              [&latch](BluetoothDevice& device) { latch.CountDown(); },
      }));
  EXPECT_FALSE(latch.Await(kWaitDuration).result());
  EXPECT_TRUE(bt_a_->StopDiscovery(std::string(kServiceId1)));
}

TEST_F(BluetoothClassicTest, CanDiscoverDeviceChanges) {
  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothRadio& radio_for_server = *radio_b_;
  BluetoothClassic& bt_client = *bt_a_;
  BluetoothClassic& bt_server = *bt_b_;

  EXPECT_TRUE(radio_for_client.IsEnabled());
  EXPECT_TRUE(radio_for_server.IsEnabled());

  EXPECT_TRUE(bt_server.TurnOnDiscoverability(std::string(kDeviceName)));
  EXPECT_EQ(radio_for_server.GetBluetoothAdapter().GetName(), kDeviceName);
  CountDownLatch discovered_latch(1);
  CountDownLatch rename_latch(1);
  CountDownLatch lost_latch(1);
  BluetoothDevice discovered_device;
  EXPECT_TRUE(bt_client.StartDiscovery(
      std::string(kServiceId1),
      {
          .device_discovered_cb =
              [&discovered_latch, &discovered_device](BluetoothDevice& device) {
                discovered_device = device;
                LOG(INFO) << "Discovered device=" << device.GetName()
                          << ", impl=" << &device.GetImpl();
                discovered_latch.CountDown();
              },
          .device_name_changed_cb =
              [&rename_latch, &discovered_device](BluetoothDevice& device) {
                discovered_device = device;
                LOG(INFO) << "Rename device=" << device.GetName()
                          << ", impl=" << &device.GetImpl();
                rename_latch.CountDown();
              },
          .device_lost_cb =
              [&lost_latch, &discovered_device](BluetoothDevice& device) {
                discovered_device = device;
                LOG(INFO) << "Lost device=" << device.GetName()
                          << ", impl=" << &device.GetImpl();
                lost_latch.CountDown();
              },
      }));
  EXPECT_TRUE(discovered_latch.Await(kWaitDuration).result());
  EXPECT_TRUE(radio_for_server.GetBluetoothAdapter().SetName("new_name"));
  EXPECT_TRUE(rename_latch.Await(kWaitDuration).result());
  EXPECT_TRUE(radio_for_server.Disable());
  EXPECT_TRUE(lost_latch.Await(kWaitDuration).result());
  EXPECT_TRUE(bt_client.StopDiscovery(std::string(kServiceId1)));
}

TEST_F(BluetoothClassicTest, CanStartAcceptingConnections) {
  BluetoothRadio& radio_for_client = *radio_a_;
  BluetoothRadio& radio_for_server = *radio_b_;
  BluetoothClassic& bt_client = *bt_a_;
  BluetoothClassic& bt_server = *bt_b_;

  EXPECT_TRUE(radio_for_client.IsEnabled());
  EXPECT_TRUE(radio_for_server.IsEnabled());

  EXPECT_TRUE(bt_server.TurnOnDiscoverability(std::string(kDeviceName)));
  EXPECT_EQ(radio_for_server.GetBluetoothAdapter().GetName(), kDeviceName);
  CountDownLatch latch(1);
  BluetoothDevice discovered_device;
  EXPECT_TRUE(bt_client.StartDiscovery(
      std::string(kServiceId1),
      {
          .device_discovered_cb =
              [&latch, &discovered_device](BluetoothDevice& device) {
                discovered_device = device;
                LOG(INFO) << "Discovered device=" << device.GetName()
                          << ",impl=" << &device.GetImpl();
                latch.CountDown();
              },
      }));
  EXPECT_TRUE(latch.Await(kWaitDuration).result());
  EXPECT_TRUE(bt_server.TurnOffDiscoverability());
  EXPECT_TRUE(discovered_device.IsValid());
  EXPECT_TRUE(
      bt_server.StartAcceptingConnections(std::string(kServiceId1), {}));
  // Allow StartAcceptingConnections do something, before stopping it.
  // This is best effort, because no callbacks are invoked in this scenario.
  SystemClock::Sleep(kWaitDuration);
  EXPECT_TRUE(bt_server.StopAcceptingConnections(std::string(kServiceId1)));
  EXPECT_TRUE(bt_client.StopDiscovery(std::string(kServiceId1)));
}

TEST_F(BluetoothClassicTest, CheckDiscoveryingStatus) {
  BluetoothClassic& bluetooth_classic = *bt_a_;

  EXPECT_FALSE(bluetooth_classic.IsDiscovering(std::string(kServiceId1)));
  EXPECT_TRUE(bluetooth_classic.StartDiscovery(
      std::string(kServiceId1),
      {
          .device_discovered_cb = [](BluetoothDevice& device) {},
      }));
  EXPECT_TRUE(bluetooth_classic.IsDiscovering(std::string(kServiceId1)));
  EXPECT_TRUE(bluetooth_classic.StopDiscovery(std::string(kServiceId1)));
  EXPECT_FALSE(bluetooth_classic.IsDiscovering(std::string(kServiceId1)));
  EXPECT_FALSE(bluetooth_classic.StopDiscovery(std::string(kServiceId1)));
}

TEST_F(BluetoothClassicTest, GetMacAddress) {
  EXPECT_TRUE(bt_a_->GetAddress().IsSet());
  radio_a_->Disable();
  EXPECT_FALSE(bt_a_->GetAddress().IsSet());
}

TEST_F(BluetoothClassicTest, GetRemoteDevice) {
  EXPECT_EQ(bt_a_->GetRemoteDevice(radio_b_->GetBluetoothAdapter().GetAddress())
                .GetAddress(),
            radio_b_->GetBluetoothAdapter().GetAddress());
  radio_a_->Disable();
  EXPECT_FALSE(
      bt_a_->GetRemoteDevice(radio_b_->GetBluetoothAdapter().GetAddress())
          .IsValid());
}

}  // namespace
}  // namespace connections
}  // namespace nearby