// Copyright 2021 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/p2p_cluster_pcp_handler.h"

#include <cstdint>
#include <string>
#include <tuple>

#include "gtest/gtest.h"
#include "absl/time/clock.h"
#include "absl/time/time.h"
#include "absl/types/span.h"
#include "connections/advertising_options.h"
#include "connections/connection_options.h"
#include "connections/discovery_options.h"
#include "connections/implementation/bluetooth_device_name.h"
#include "connections/implementation/bwu_manager.h"
#include "connections/implementation/client_proxy.h"
#include "connections/implementation/endpoint_channel_manager.h"
#include "connections/implementation/endpoint_manager.h"
#include "connections/implementation/flags/nearby_connections_feature_flags.h"
#include "connections/implementation/injected_bluetooth_device_store.h"
#include "connections/implementation/mediums/bluetooth_radio.h"
#include "connections/implementation/mediums/mediums.h"
#include "connections/listeners.h"
#include "connections/medium_selector.h"
#include "connections/out_of_band_connection_metadata.h"
#include "connections/status.h"
#include "connections/strategy.h"
#include "connections/v3/connection_listening_options.h"
#include "internal/flags/nearby_flags.h"
#include "internal/platform/byte_array.h"
#include "internal/platform/count_down_latch.h"
#include "internal/platform/logging.h"
#include "internal/platform/mac_address.h"
#include "internal/platform/medium_environment.h"

namespace nearby {
namespace connections {
namespace {

constexpr BooleanMediumSelector kTestCases[] = {
    BooleanMediumSelector{
        .ble = true,
    },
    BooleanMediumSelector{
        .bluetooth = true,
    },
    BooleanMediumSelector{
        .wifi_lan = true,
    },
    BooleanMediumSelector{
        .awdl = true,
    },
    BooleanMediumSelector{
        .bluetooth = true,
        .ble = true,
    },
    BooleanMediumSelector{
        .bluetooth = true,
        .wifi_lan = true,
    },
    BooleanMediumSelector{
        .ble = true,
        .wifi_lan = true,
    },
    BooleanMediumSelector{
        .bluetooth = true,
        .ble = true,
        .web_rtc = true,
        .wifi_lan = true,
    },
};

class P2pClusterPcpHandlerTest : public testing::Test {
 protected:
  void SetUp() override {
    LOG(INFO) << "SetUp: begin";
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::kEnableAwdl, true);
    SetBleExtendedAdvertisementsAvailable(true);
    SetDisableBluetoothClassicScanning(true);
    SetBleV2Enabled(true);
  }

  void SetBleExtendedAdvertisementsAvailable(bool available) {
    env_.SetBleExtendedAdvertisementsAvailable(false);
  }

  void SetDisableBluetoothClassicScanning(bool disable) {
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::
            kDisableBluetoothClassicScanning,
        disable);
  }

  void SetBleV2Enabled(bool enabled) {
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::kEnableBleV2,
        enabled);
  }

  AdvertisingOptions GetBluetoothOnlyAdvertisingOptions() {
    return AdvertisingOptions{
        {Strategy::kP2pCluster,
         BooleanMediumSelector{
             .bluetooth = true,
         }},
    };
  }

  AdvertisingOptions GetBluetoothAndBleAdvertisingOptions() {
    return AdvertisingOptions{
        {Strategy::kP2pCluster,
         BooleanMediumSelector{
             .bluetooth = true,
             .ble = true,
         }},
    };
  }

  DiscoveryOptions GetBluetoothOnlyDiscoveryOptions() {
    return DiscoveryOptions{
        {Strategy::kP2pCluster,
         BooleanMediumSelector{
             .bluetooth = true,
         }},
    };
  }

  DiscoveryOptions GetBluetoothAndBleDiscoveryOptions() {
    return DiscoveryOptions{
        {Strategy::kP2pCluster,
         BooleanMediumSelector{
             .bluetooth = true,
             .ble = true,
         }},
    };
  }

  // Returns a 6 bytes mac address from a given address.
  // address: it is in format of "01:02:03:04:05:06".
  ByteArray GetSixBytesMacAddress(MacAddress address) {
    uint8_t bytes[6];
    address.ToBytes(bytes);
    return ByteArray(reinterpret_cast<char*>(bytes), 6);
  }

  ClientProxy client_a_;
  ClientProxy client_b_;
  ClientProxy client_c_;
  std::string service_id_{"service"};
  MediumEnvironment& env_{MediumEnvironment::Instance()};
};

TEST_F(P2pClusterPcpHandlerTest, NoBluetoothDiscoveryWhenRadioIsOff) {
  env_.Start();
  Mediums mediums;
  EndpointChannelManager ecm;
  EndpointManager em(&ecm);
  BwuManager bwu(mediums, em, ecm, {}, {});
  InjectedBluetoothDeviceStore ibds;
  P2pClusterPcpHandler handler(&mediums, &em, &ecm, &bwu, ibds);
  mediums.GetBluetoothRadio().Disable();
  handler.StartDiscovery(&client_a_, service_id_,
                         GetBluetoothOnlyDiscoveryOptions(), {});
  EXPECT_FALSE(mediums.GetBluetoothClassic().IsDiscovering(service_id_));
  handler.StopDiscovery(&client_a_);

  handler.StartDiscovery(&client_a_, service_id_,
                         GetBluetoothAndBleDiscoveryOptions(), {});
  EXPECT_FALSE(mediums.GetBluetoothClassic().IsDiscovering(service_id_));
  EXPECT_TRUE(mediums.GetBleV2().IsScanning(service_id_));

  mediums.GetBluetoothRadio().Enable();
  handler.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_F(P2pClusterPcpHandlerTest,
       BluetoothCanDiscoveryWhenBluetoothDiscoveryRunning) {
  std::string endpoint_name{"endpoint_name"};

  env_.Start();
  // Enable BLE V2 extended advertisement for client_a_.
  env_.SetBleExtendedAdvertisementsAvailable(true);
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  InjectedBluetoothDeviceStore ibds_a;
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);

  // Disable BLE V2 extended advertisement for client_b_.
  env_.SetBleExtendedAdvertisementsAvailable(false);
  Mediums mediums_b;
  EndpointChannelManager ecm_b;
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {}, {});
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);
  CountDownLatch latch(1);

  EXPECT_EQ(handler_a.StartDiscovery(&client_a_, service_id_,
                                     GetBluetoothOnlyDiscoveryOptions(), {}),
            Status{Status::kSuccess});

  EXPECT_TRUE(mediums_a.GetBluetoothClassic().IsDiscovering(service_id_));

  // Start another service discovery
  EXPECT_EQ(
      handler_a.StartDiscovery(
          &client_c_, "new_service_id", GetBluetoothAndBleDiscoveryOptions(),
          {
              .endpoint_found_cb =
                  [&latch](const std::string& endpoint_id,
                           const ByteArray& endpoint_info,
                           const std::string& service_id) {
                    LOG(INFO) << "Device discovered: id=" << endpoint_id;
                    latch.CountDown();
                  },
          }),
      Status{Status::kSuccess});

  // Start Bluetooth discovery when found legacy device.
  EXPECT_EQ(
      handler_b.StartAdvertising(&client_b_, "new_service_id",
                                 GetBluetoothAndBleAdvertisingOptions(),
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});

  EXPECT_TRUE(latch.Await(absl::Milliseconds(1000)).result());

  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

// Combines the bool `kEnableBleV2` as param testing but should revert it back
// if ble_v2 is done and ble will be replaced by ble_v2.
class P2pClusterPcpHandlerTestWithParam
    : public testing::TestWithParam<
          /*mediums=*/std::tuple<BooleanMediumSelector, /*ble_v2_enabled=*/bool,
                                 /*disable_bluetooth_scanning*/ bool>> {
 protected:
  void SetUp() override {
    LOG(INFO) << "SetUp: begin";
    env_.SetBleExtendedAdvertisementsAvailable(false);
    bool ble_v2_enabled = std::get<1>(GetParam());
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::kEnableBleV2,
        ble_v2_enabled);
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::kEnableAwdl, true);
    bool is_disable_bluetooth_scanning = std::get<2>(GetParam());
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::
            kDisableBluetoothClassicScanning,
        is_disable_bluetooth_scanning);
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::kEnableDct, true);
    if (advertising_options_.allowed.ble) {
      LOG(INFO) << "SetUp: BLE enabled";
    }
    if (advertising_options_.allowed.bluetooth) {
      LOG(INFO) << "SetUp: BT enabled";
    }
    if (advertising_options_.allowed.wifi_lan) {
      LOG(INFO) << "SetUp: WifiLan enabled";
    }
    if (advertising_options_.allowed.web_rtc) {
      LOG(INFO) << "SetUp: WebRTC enabled";
    }
    if (advertising_options_.allowed.awdl) {
      LOG(INFO) << "SetUp: Awdl enabled";
    }
    LOG(INFO) << "SetUp: ble v2 enabled: " << ble_v2_enabled;
    LOG(INFO) << "SetUp: is_disable_bluetooth_scanning: "
              << is_disable_bluetooth_scanning;
    LOG(INFO) << "SetUp: end";
  }

  ClientProxy client_a_;
  ClientProxy client_b_;
  std::string service_id_{"service"};
  ConnectionOptions connection_options_{
      {
          Strategy::kP2pCluster,
          std::get<0>(GetParam()),
      },
  };
  AdvertisingOptions advertising_options_{
      {
          Strategy::kP2pCluster,
          std::get<0>(GetParam()),
      },
  };
  DiscoveryOptions discovery_options_{
      {
          Strategy::kP2pCluster,
          std::get<0>(GetParam()),
      },
  };
  MediumEnvironment& env_{MediumEnvironment::Instance()};
};

TEST_P(P2pClusterPcpHandlerTestWithParam, CanConstructOne) {
  env_.Start();
  Mediums mediums;
  EndpointChannelManager ecm;
  EndpointManager em(&ecm);
  BwuManager bwu(mediums, em, ecm, {}, {});
  InjectedBluetoothDeviceStore ibds;
  P2pClusterPcpHandler handler(&mediums, &em, &ecm, &bwu, ibds);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanConstructMultiple) {
  env_.Start();
  Mediums mediums_a;
  Mediums mediums_b;
  EndpointChannelManager ecm_a;
  EndpointChannelManager ecm_b;
  EndpointManager em_a(&ecm_a);
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanAdvertise) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_, advertising_options_,
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  handler_a.StopAdvertising(&client_a_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, AdvertiseForLegacyDeviceWithBt) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_, advertising_options_,
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  // advertising for legacy device depends on both BT and BLE V2 enabled.
  if (std::get<0>(GetParam()).bluetooth && std::get<1>(GetParam())) {
    EXPECT_TRUE(mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
  } else {
    EXPECT_FALSE(
        mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
  }
  handler_a.StopAdvertising(&client_a_);
  if (std::get<0>(GetParam()).bluetooth && std::get<1>(GetParam())) {
    EXPECT_FALSE(
        mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
  }
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanUpdateAdvertisingOptions) {
  bool ble_v2_enabled = std::get<1>(GetParam());
  if (!ble_v2_enabled) {
    // Just don't run the test if ble_v2 is disabled.
    return;
  }
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  // Reset medium states.
  mediums_a.GetBluetoothClassic().TurnOffDiscoverability();
  ASSERT_FALSE(mediums_a.GetBluetoothClassic().TurnOffDiscoverability());
  mediums_a.GetBluetoothClassic().StopAcceptingConnections(service_id_);
  ASSERT_FALSE(
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  mediums_a.GetWifiLan().StopAcceptingConnections(service_id_);
  ASSERT_FALSE(mediums_a.GetWifiLan().IsAcceptingConnections(service_id_));
  mediums_a.GetWifiLan().StopAdvertising(service_id_);
  ASSERT_FALSE(mediums_a.GetWifiLan().IsAdvertising(service_id_));
  mediums_a.GetBleV2().StopAcceptingConnections(service_id_);
  ASSERT_FALSE(mediums_a.GetBleV2().IsAcceptingConnections(service_id_));
  mediums_a.GetBleV2().StopAdvertising(service_id_);
  ASSERT_FALSE(mediums_a.GetBleV2().IsAdvertising(service_id_));
  BooleanMediumSelector enabled = advertising_options_.allowed;
  if (ble_v2_enabled && enabled.bluetooth) {
    EXPECT_FALSE(
        mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
  }
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_, advertising_options_,
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  EXPECT_EQ(enabled.ble, mediums_a.GetBleV2().IsAdvertising(service_id_));
  EXPECT_EQ(enabled.wifi_lan,
            mediums_a.GetWifiLan().IsAdvertising(service_id_));
  EXPECT_EQ(
      enabled.bluetooth || enabled.ble,
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  EXPECT_EQ(enabled.bluetooth,
            mediums_a.GetBluetoothClassic().TurnOffDiscoverability());
  if (ble_v2_enabled && enabled.bluetooth) {
    EXPECT_TRUE(mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
  }
  // Turn discoverability back on
  mediums_a.GetBluetoothClassic().TurnOnDiscoverability(service_id_);
  AdvertisingOptions new_options{
      {
          Strategy::kP2pCluster,
          enabled,
      },
      false,  // auto_upgrade_bandwidth
      false,  // enforce_topology_constraints
      true,   // low_power
  };
  EXPECT_EQ(
      handler_a.UpdateAdvertisingOptions(&client_a_, service_id_, new_options),
      Status{Status::kSuccess});
  if (ble_v2_enabled) {
    EXPECT_EQ(enabled.ble, mediums_a.GetBleV2().IsAdvertising(service_id_));
    // Low power won't restart BT, nor BLE advertising for legacy device.
    if (enabled.bluetooth) {
      EXPECT_FALSE(
          mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
    }
  } else {
    EXPECT_EQ(enabled.ble, mediums_a.GetBle().IsAdvertising(service_id_));
  }
  EXPECT_FALSE(mediums_a.GetWifiLan().IsAdvertising(service_id_));
  EXPECT_FALSE(mediums_a.GetBluetoothClassic().TurnOffDiscoverability());
  EXPECT_FALSE(
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  handler_a.StopAdvertising(&client_a_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam,
       CanUpdateAdvertisingOptionsNoLowPower) {
  bool ble_v2_enabled = std::get<1>(GetParam());
  if (!ble_v2_enabled) {
    // Just don't run the test if ble_v2 is disabled.
    return;
  }
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  BooleanMediumSelector enabled = advertising_options_.allowed;
  // Reset medium states.
  mediums_a.GetBluetoothClassic().TurnOffDiscoverability();
  ASSERT_FALSE(mediums_a.GetBluetoothClassic().TurnOffDiscoverability());
  mediums_a.GetBluetoothClassic().StopAcceptingConnections(service_id_);
  ASSERT_FALSE(
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  mediums_a.GetWifiLan().StopAcceptingConnections(service_id_);
  ASSERT_FALSE(mediums_a.GetWifiLan().IsAcceptingConnections(service_id_));
  mediums_a.GetWifiLan().StopAdvertising(service_id_);
  ASSERT_FALSE(mediums_a.GetWifiLan().IsAdvertising(service_id_));
  mediums_a.GetBleV2().StopAcceptingConnections(service_id_);
  ASSERT_FALSE(mediums_a.GetBleV2().IsAcceptingConnections(service_id_));
  mediums_a.GetBleV2().StopAdvertising(service_id_);
  ASSERT_FALSE(mediums_a.GetBleV2().IsAdvertising(service_id_));
  AdvertisingOptions old_options{
      {
          Strategy::kP2pCluster,
          enabled,
      },
      false,  // auto_upgrade_bandwidth
      false,  // enforce_topology_constraints
      true,   // low_power
      false,  // enable_bluetooth_listening
  };
  if (ble_v2_enabled && enabled.bluetooth) {
    EXPECT_FALSE(
        mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
  }
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_, old_options,
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  EXPECT_EQ(enabled.ble, mediums_a.GetBleV2().IsAdvertising(service_id_));
  EXPECT_EQ(enabled.wifi_lan,
            mediums_a.GetWifiLan().IsAdvertising(service_id_));
  EXPECT_EQ(
      enabled.bluetooth,
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  if (ble_v2_enabled && enabled.bluetooth) {
    EXPECT_TRUE(mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
  }
  EXPECT_EQ(enabled.bluetooth,
            mediums_a.GetBluetoothClassic().TurnOffDiscoverability());
  EXPECT_EQ(handler_a.UpdateAdvertisingOptions(&client_a_, service_id_,
                                               advertising_options_),
            Status{Status::kSuccess});
  if (ble_v2_enabled) {
    EXPECT_EQ(enabled.ble, mediums_a.GetBleV2().IsAdvertising(service_id_));
    if (enabled.bluetooth) {
      EXPECT_TRUE(
          mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
    }
  } else {
    EXPECT_EQ(enabled.ble, mediums_a.GetBle().IsAdvertising(service_id_));
  }
  EXPECT_EQ(enabled.wifi_lan,
            mediums_a.GetWifiLan().IsAdvertising(service_id_));
  EXPECT_EQ(enabled.bluetooth,
            mediums_a.GetBluetoothClassic().TurnOffDiscoverability());
  EXPECT_EQ(
      enabled.bluetooth || enabled.ble,
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  handler_a.StopAdvertising(&client_a_);
  if (ble_v2_enabled && enabled.bluetooth) {
    EXPECT_FALSE(
        mediums_a.GetBleV2().IsAdvertisingForLegacyDevice(service_id_));
  }
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanDiscover) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  Mediums mediums_b;
  EndpointChannelManager ecm_a;
  EndpointChannelManager ecm_b;
  EndpointManager em_a(&ecm_a);
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);
  CountDownLatch latch(1);
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_, advertising_options_,
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  EXPECT_EQ(handler_b.StartDiscovery(
                &client_b_, service_id_, discovery_options_,
                {
                    .endpoint_found_cb =
                        [&latch](const std::string& endpoint_id,
                                 const ByteArray& endpoint_info,
                                 const std::string& service_id) {
                          LOG(INFO) << "Device discovered: id=" << endpoint_id;
                          latch.CountDown();
                        },
                }),
            Status{Status::kSuccess});
  EXPECT_TRUE(latch.Await(absl::Milliseconds(1000)).result());
  // We discovered endpoint over one medium. Before we finish the test, we have
  // to stop discovery for other mediums that may be still ongoing.
  handler_b.StopDiscovery(&client_b_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanDiscoverLegacy) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  Mediums mediums_b;
  EndpointChannelManager ecm_a;
  EndpointChannelManager ecm_b;
  EndpointManager em_a(&ecm_a);
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);
  CountDownLatch latch(1);
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_, advertising_options_,
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  EXPECT_EQ(handler_b.StartDiscovery(
                &client_b_, service_id_, discovery_options_,
                {
                    .endpoint_found_cb =
                        [&latch](const std::string& endpoint_id,
                                 const ByteArray& endpoint_info,
                                 const std::string& service_id) {
                          LOG(INFO) << "Device discovered: id=" << endpoint_id;
                          latch.CountDown();
                        },
                }),
            Status{Status::kSuccess});
  // advertising for legacy device depends on both BT and BLE V2 enabled.
  //  if (std::get<0>(GetParam()).bluetooth && std::get<1>(GetParam())) {
  EXPECT_TRUE(latch.Await(absl::Milliseconds(1000)).result());
  /*  } else {
      EXPECT_FALSE(latch.Await(absl::Milliseconds(1000)).result());
    }*/
  // We discovered endpoint over one medium. Before we finish the test, we have
  // to stop discovery for other mediums that may be still ongoing.
  handler_b.StopDiscovery(&client_b_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, PauseBluetoothClassicDiscovery) {
  // Skip the case which not disable bluetooth scanning.
  if (!std::get<2>(GetParam()) || !std::get<1>(GetParam()) ||
      !advertising_options_.allowed.bluetooth ||
      !advertising_options_.allowed.ble) {
    return;
  }

  env_.SetBleExtendedAdvertisementsAvailable(true);
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);

  EXPECT_EQ(
      handler_a.StartDiscovery(&client_a_, service_id_, discovery_options_, {}),
      Status{Status::kSuccess});

  EXPECT_TRUE(mediums_a.GetBleV2().IsScanning(service_id_));
  EXPECT_FALSE(mediums_a.GetBluetoothClassic().IsDiscovering(service_id_));
  // Before we finish the test, we have to stop discovery for other mediums that
  // may be still ongoing.
  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, ResumeBluetoothClassicDiscovery) {
  // Skip the case which not disable bluetooth scanning.
  if (!std::get<2>(GetParam()) || !std::get<1>(GetParam()) ||
      !advertising_options_.allowed.bluetooth ||
      !advertising_options_.allowed.ble) {
    return;
  }

  std::string endpoint_name{"endpoint_name"};

  env_.Start();
  // Enable BLE V2 extended advertisement for client_a_.
  env_.SetBleExtendedAdvertisementsAvailable(true);
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  InjectedBluetoothDeviceStore ibds_a;
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);

  // Disable BLE V2 extended advertisement for client_b_.
  env_.SetBleExtendedAdvertisementsAvailable(false);
  Mediums mediums_b;
  EndpointChannelManager ecm_b;
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {}, {});
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);
  CountDownLatch latch(1);

  EXPECT_EQ(handler_a.StartDiscovery(
                &client_a_, service_id_, discovery_options_,
                {
                    .endpoint_found_cb =
                        [&latch](const std::string& endpoint_id,
                                 const ByteArray& endpoint_info,
                                 const std::string& service_id) {
                          LOG(INFO) << "Device discovered: id=" << endpoint_id;
                          latch.CountDown();
                        },
                }),
            Status{Status::kSuccess});

  EXPECT_TRUE(mediums_a.GetBleV2().IsScanning(service_id_));
  EXPECT_FALSE(mediums_a.GetBluetoothClassic().IsDiscovering(service_id_));

  EXPECT_EQ(
      handler_b.StartAdvertising(&client_b_, service_id_, advertising_options_,
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});

  EXPECT_TRUE(latch.Await(absl::Milliseconds(1000)).result());
  absl::SleepFor(absl::Milliseconds(100));

  EXPECT_TRUE(mediums_a.GetBleV2().IsScanning(service_id_));
  EXPECT_TRUE(mediums_a.GetBluetoothClassic().IsDiscovering(service_id_));

  // Before we finish the test, we have to stop discovery for other mediums that
  // may be still ongoing.
  handler_b.StopAdvertising(&client_b_);
  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanBluetoothDiscoverChangeName) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  Mediums mediums_b;
  EndpointChannelManager ecm_a;
  EndpointChannelManager ecm_b;
  EndpointManager em_a(&ecm_a);
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);
  // For the first time we find the device with the old name.
  CountDownLatch first_found_latch(1);
  // For the second time we "find" the device with the new name.
  CountDownLatch second_found_latch(1);
  // For when the name is changed.
  CountDownLatch lost_latch(1);
  AdvertisingOptions advertising_options = {
      {
          Strategy::kP2pCluster,
          BooleanMediumSelector{
              .bluetooth = true,
          },
      },
  };

  DiscoveryOptions discovery_options = {
      {
          Strategy::kP2pCluster,
          BooleanMediumSelector{
              .bluetooth = true,
          },
      },
  };
  bool first = false;
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_, advertising_options,
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  EXPECT_EQ(handler_b.StartDiscovery(
                &client_b_, service_id_, discovery_options,
                {
                    .endpoint_found_cb =
                        [&](const std::string& endpoint_id,
                            const ByteArray& endpoint_info,
                            const std::string& service_id) {
                          LOG(INFO) << "Device discovered: id=" << endpoint_id;
                          if (!first) {
                            first_found_latch.CountDown();
                            first = true;
                          } else {
                            second_found_latch.CountDown();
                          }
                        },
                    .endpoint_lost_cb =
                        [&](const std::string& id) {
                          LOG(INFO) << "Device lost: id=" << id;
                          lost_latch.CountDown();
                        },
                }),
            Status{Status::kSuccess});
  ASSERT_TRUE(mediums_a.GetBluetoothRadio().IsAdapterValid());
  BluetoothDeviceName name(
      mediums_a.GetBluetoothRadio().GetBluetoothAdapter().GetName());
  BluetoothDeviceName new_name(name.GetVersion(), name.GetPcp(),
                               name.GetEndpointId(), name.GetServiceIdHash(),
                               ByteArray("BT Device A"), name.GetUwbAddress(),
                               name.GetWebRtcState());
  EXPECT_TRUE(first_found_latch.Await().Ok());
  mediums_a.GetBluetoothRadio().GetBluetoothAdapter().SetName(
      std::string(new_name));
  EXPECT_TRUE(second_found_latch.Await().Ok());
  EXPECT_TRUE(lost_latch.Await().Ok());
  handler_b.StopDiscovery(&client_b_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanUpdateDiscoveryOptions) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  EXPECT_EQ(
      handler_a.StartDiscovery(&client_a_, service_id_, discovery_options_, {}),
      Status{Status::kSuccess});
  BooleanMediumSelector enabled = std::get<0>(GetParam());
  bool ble_v2_enabled = std::get<1>(GetParam());
  if (ble_v2_enabled) {
    EXPECT_EQ(enabled.ble, mediums_a.GetBleV2().IsScanning(service_id_));
  } else {
    EXPECT_EQ(enabled.ble, mediums_a.GetBle().IsScanning(service_id_));
  }
  EXPECT_EQ(enabled.wifi_lan,
            mediums_a.GetWifiLan().IsDiscovering(service_id_));
  DiscoveryOptions new_options{
      {
          Strategy::kP2pCluster,
          std::get<0>(GetParam()),
      },
      false,  // auto_upgrade_bandwidth
      false,  // enforce_topology_constraints
      false,  // is_out_of_band_connection
      "",     // fast_advertisement_service_uuid
      true,   // low_power
  };
  EXPECT_EQ(
      handler_a.UpdateDiscoveryOptions(&client_a_, service_id_, new_options),
      Status{Status::kSuccess});
  if (ble_v2_enabled) {
    EXPECT_EQ(enabled.ble, mediums_a.GetBleV2().IsScanning(service_id_));
  } else {
    EXPECT_EQ(enabled.ble, mediums_a.GetBle().IsScanning(service_id_));
  }
  EXPECT_FALSE(mediums_a.GetWifiLan().IsDiscovering(service_id_));
  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanUpdateDiscoveryOptionsNoLowPower) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  BooleanMediumSelector old_enabled = discovery_options_.allowed;
  BooleanMediumSelector new_enabled = old_enabled;
  if (!new_enabled.ble) {
    new_enabled.ble = true;
  }
  // Reset medium states.
  mediums_a.GetBluetoothClassic().TurnOffDiscoverability();
  ASSERT_FALSE(mediums_a.GetBluetoothClassic().TurnOffDiscoverability());
  mediums_a.GetWifiLan().StopDiscovery(service_id_);
  ASSERT_FALSE(mediums_a.GetWifiLan().IsDiscovering(service_id_));
  if (std::get<1>(GetParam())) {
    mediums_a.GetBleV2().StopScanning(service_id_);
    ASSERT_FALSE(mediums_a.GetBleV2().IsScanning(service_id_));
  } else {
    mediums_a.GetBle().StopScanning(service_id_);
    ASSERT_FALSE(mediums_a.GetBle().IsScanning(service_id_));
  }
  DiscoveryOptions old_options = discovery_options_;
  old_options.low_power = true;
  old_options.allowed = old_enabled;
  DiscoveryOptions new_options = discovery_options_;
  new_options.allowed = new_enabled;
  // Start discovery
  EXPECT_EQ(handler_a.StartDiscovery(&client_a_, service_id_, old_options, {}),
            Status{Status::kSuccess});
  if (std::get<1>(GetParam())) {
    EXPECT_EQ(old_enabled.ble, mediums_a.GetBleV2().IsScanning(service_id_));
  } else {
    EXPECT_EQ(old_enabled.ble, mediums_a.GetBle().IsScanning(service_id_));
  }
  EXPECT_EQ(old_enabled.wifi_lan,
            mediums_a.GetWifiLan().IsDiscovering(service_id_));
  EXPECT_EQ(old_enabled.bluetooth,
            mediums_a.GetBluetoothClassic().StopDiscovery(service_id_));
  LOG(INFO) << "started discovery";
  // Update discovery options
  EXPECT_TRUE(
      handler_a.UpdateDiscoveryOptions(&client_a_, service_id_, new_options)
          .Ok());
  LOG(INFO) << "updated discovery options";
  if (std::get<1>(GetParam())) {
    EXPECT_EQ(new_enabled.ble, mediums_a.GetBleV2().IsScanning(service_id_));
  } else {
    EXPECT_EQ(new_enabled.ble, mediums_a.GetBle().IsScanning(service_id_));
  }
  EXPECT_EQ(new_enabled.wifi_lan,
            mediums_a.GetWifiLan().IsDiscovering(service_id_));
  EXPECT_EQ(new_enabled.bluetooth,
            mediums_a.GetBluetoothClassic().StopDiscovery(service_id_));
  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam,
       UpdateDiscoveryOptionsSkipMediumRestart) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  BooleanMediumSelector enabled = discovery_options_.allowed;
  // Reset medium states.
  mediums_a.GetBluetoothClassic().TurnOffDiscoverability();
  ASSERT_FALSE(mediums_a.GetBluetoothClassic().TurnOffDiscoverability());
  mediums_a.GetWifiLan().StopDiscovery(service_id_);
  ASSERT_FALSE(mediums_a.GetWifiLan().IsDiscovering(service_id_));
  if (std::get<1>(GetParam())) {
    mediums_a.GetBleV2().StopScanning(service_id_);
    ASSERT_FALSE(mediums_a.GetBleV2().IsScanning(service_id_));
  } else {
    mediums_a.GetBle().StopScanning(service_id_);
    ASSERT_FALSE(mediums_a.GetBle().IsScanning(service_id_));
  }
  // Start discovery
  EXPECT_EQ(
      handler_a.StartDiscovery(&client_a_, service_id_, discovery_options_, {}),
      Status{Status::kSuccess});
  if (std::get<1>(GetParam())) {
    EXPECT_EQ(enabled.ble, mediums_a.GetBleV2().IsScanning(service_id_));
  } else {
    EXPECT_EQ(enabled.ble, mediums_a.GetBle().IsScanning(service_id_));
  }
  EXPECT_EQ(enabled.wifi_lan,
            mediums_a.GetWifiLan().IsDiscovering(service_id_));
  EXPECT_EQ(enabled.bluetooth,
            mediums_a.GetBluetoothClassic().StopDiscovery(service_id_));
  // Update discovery options
  auto result = handler_a.UpdateDiscoveryOptions(&client_a_, service_id_,
                                                 discovery_options_);
  EXPECT_TRUE(result.Ok());
  LOG(INFO) << "updated discovery options";
  if (std::get<1>(GetParam())) {
    EXPECT_EQ(enabled.ble, mediums_a.GetBleV2().IsScanning(service_id_));
  } else {
    EXPECT_EQ(enabled.ble, mediums_a.GetBle().IsScanning(service_id_));
  }
  EXPECT_EQ(enabled.wifi_lan,
            mediums_a.GetWifiLan().IsDiscovering(service_id_));
  // We didn't restart the medium.
  EXPECT_FALSE(mediums_a.GetBluetoothClassic().StopDiscovery(service_id_));
  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanConnect) {
  env_.Start();
  std::string endpoint_name_a{"endpoint_name"};
  Mediums mediums_a;
  Mediums mediums_b;
  BluetoothRadio& radio_a = mediums_a.GetBluetoothRadio();
  BluetoothRadio& radio_b = mediums_b.GetBluetoothRadio();
  radio_a.GetBluetoothAdapter().SetName("BT Device A");
  radio_b.GetBluetoothAdapter().SetName("BT Device B");
  EndpointChannelManager ecm_a;
  EndpointChannelManager ecm_b;
  EndpointManager em_a(&ecm_a);
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {},
                   {.allow_upgrade_to = {.bluetooth = true}});
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {},
                   {.allow_upgrade_to = {.bluetooth = true}});
  InjectedBluetoothDeviceStore ibds_a;
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);
  CountDownLatch discover_latch(1);
  CountDownLatch connect_latch(2);
  struct DiscoveredInfo {
    std::string endpoint_id;
    ByteArray endpoint_info;
    std::string service_id;
  } discovered;
  EXPECT_EQ(
      handler_a.StartAdvertising(
          &client_a_, service_id_, advertising_options_,
          {
              .endpoint_info = ByteArray{endpoint_name_a},
              .listener =
                  {
                      .initiated_cb =
                          [&connect_latch](const std::string& endpoint_id,
                                           const ConnectionResponseInfo& info) {
                            LOG(INFO)
                                << "StartAdvertising: initiated_cb called";
                            connect_latch.CountDown();
                          },
                  },
          }),
      Status{Status::kSuccess});
  EXPECT_EQ(handler_b.StartDiscovery(
                &client_b_, service_id_, discovery_options_,
                {
                    .endpoint_found_cb =
                        [&discover_latch, &discovered](
                            const std::string& endpoint_id,
                            const ByteArray& endpoint_info,
                            const std::string& service_id) {
                          LOG(INFO) << "Device discovered: id=" << endpoint_id
                                    << ", endpoint_info="
                                    << std::string{endpoint_info};
                          discovered = {
                              .endpoint_id = endpoint_id,
                              .endpoint_info = endpoint_info,
                              .service_id = service_id,
                          };
                          discover_latch.CountDown();
                        },
                }),
            Status{Status::kSuccess});

  EXPECT_TRUE(discover_latch.Await(absl::Milliseconds(1000)).result());
  EXPECT_EQ(endpoint_name_a, std::string{discovered.endpoint_info});

  const std::string kBssid = "34:36:3B:C7:8C:71";
  const std::int32_t kFreq = 5200;
  constexpr char kIp4Bytes[] = {(char)192, (char)168, (char)1, (char)37, 0};

  connection_options_.connection_info.supports_5_ghz = true;
  connection_options_.connection_info.bssid = kBssid;
  connection_options_.connection_info.ap_frequency = kFreq;
  connection_options_.connection_info.ip_address.resize(4);
  connection_options_.connection_info.ip_address = kIp4Bytes;

  client_b_.AddCancellationFlag(discovered.endpoint_id);
  handler_b.RequestConnection(
      &client_b_, discovered.endpoint_id,
      {.endpoint_info = discovered.endpoint_info,
       .listener =
           {
               .initiated_cb =
                   [&connect_latch](const std::string& endpoint_id,
                                    const ConnectionResponseInfo& info) {
                     LOG(INFO) << "RequestConnection: initiated_cb called";
                     connect_latch.CountDown();
                   },
           }},
      connection_options_);
  std::string client_b_local_endpoint = client_b_.GetLocalEndpointId();

  EXPECT_TRUE(connect_latch.Await(absl::Milliseconds(1000)).result());
  EXPECT_TRUE(client_b_.Is5GHzSupported(discovered.endpoint_id));
  EXPECT_EQ(client_b_.GetBssid(discovered.endpoint_id), kBssid);
  EXPECT_EQ(client_b_.GetApFrequency(discovered.endpoint_id), kFreq);
  EXPECT_EQ(client_b_.GetIPAddress(discovered.endpoint_id),
            std::string(kIp4Bytes));
  // When connection is established, EndpointManager will setup KeepAliveManager
  // loop. When it fails, the connection will be dismantled. Since this a unit
  // test, KeepAliveManager won't be really up. The disconnection may happen
  // before the following check, which cause the check fail. So we check the
  // connection status first.
  if (client_b_.IsConnectedToEndpoint(discovered.endpoint_id)) {
    EXPECT_EQ(client_a_.Is5GHzSupported(client_b_local_endpoint),
              mediums_b.GetWifi().GetCapability().supports_5_ghz);
    EXPECT_EQ(client_a_.GetBssid(client_b_local_endpoint),
              mediums_b.GetWifi().GetInformation().bssid);
    EXPECT_EQ(client_a_.GetApFrequency(client_b_local_endpoint),
              mediums_b.GetWifi().GetInformation().ap_frequency);
    EXPECT_EQ(client_a_.GetIPAddress(client_b_local_endpoint),
              mediums_b.GetWifi().GetInformation().ip_address_4_bytes);
  }

  handler_a.StopAdvertising(&client_a_);
  handler_b.StopDiscovery(&client_b_);
  bwu_a.Shutdown();
  bwu_b.Shutdown();
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanConnectWithDctEnabled) {
  env_.Start();
  ByteArray endpoint_info_a{
      "\x22\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x0b"
      "\x54\x65\x73\x74\x20\x64\x65\x76\x69\x63\x65",
      29};
  ClientProxy client_a;
  ClientProxy client_b;

  Mediums mediums_a;
  Mediums mediums_b;
  BluetoothRadio& radio_a = mediums_a.GetBluetoothRadio();
  BluetoothRadio& radio_b = mediums_b.GetBluetoothRadio();
  radio_a.GetBluetoothAdapter().SetName("BT Device A");
  radio_b.GetBluetoothAdapter().SetName("BT Device B");
  EndpointChannelManager ecm_a;
  EndpointChannelManager ecm_b;
  EndpointManager em_a(&ecm_a);
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {},
                   {.allow_upgrade_to = {.bluetooth = true}});
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {},
                   {.allow_upgrade_to = {.bluetooth = true}});
  InjectedBluetoothDeviceStore ibds_a;
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);
  CountDownLatch discover_latch(1);
  CountDownLatch connect_latch(2);
  struct DiscoveredInfo {
    std::string endpoint_id;
    ByteArray endpoint_info;
    std::string service_id;
  } discovered;
  EXPECT_EQ(
      handler_a.StartAdvertising(
          &client_a, service_id_, advertising_options_,
          {
              .endpoint_info = endpoint_info_a,
              .listener =
                  {
                      .initiated_cb =
                          [&connect_latch](const std::string& endpoint_id,
                                           const ConnectionResponseInfo& info) {
                            LOG(INFO)
                                << "StartAdvertising: initiated_cb called";
                            connect_latch.CountDown();
                          },
                  },
          }),
      Status{Status::kSuccess});
  EXPECT_EQ(handler_b.StartDiscovery(
                &client_b, service_id_, discovery_options_,
                {
                    .endpoint_found_cb =
                        [&discover_latch, &discovered](
                            const std::string& endpoint_id,
                            const ByteArray& endpoint_info,
                            const std::string& service_id) {
                          LOG(INFO) << "Device discovered: id=" << endpoint_id
                                    << ", endpoint_info="
                                    << std::string{endpoint_info};
                          discovered = {
                              .endpoint_id = endpoint_id,
                              .endpoint_info = endpoint_info,
                              .service_id = service_id,
                          };
                          discover_latch.CountDown();
                        },
                }),
            Status{Status::kSuccess});

  EXPECT_TRUE(discover_latch.Await(absl::Milliseconds(1000)).result());
  EXPECT_EQ(endpoint_info_a, discovered.endpoint_info);

  const std::string kBssid = "34:36:3B:C7:8C:71";
  const std::int32_t kFreq = 5200;
  constexpr char kIp4Bytes[] = {(char)192, (char)168, (char)1, (char)37, 0};

  connection_options_.connection_info.supports_5_ghz = true;
  connection_options_.connection_info.bssid = kBssid;
  connection_options_.connection_info.ap_frequency = kFreq;
  connection_options_.connection_info.ip_address.resize(4);
  connection_options_.connection_info.ip_address = kIp4Bytes;

  client_b.AddCancellationFlag(discovered.endpoint_id);
  handler_b.RequestConnection(
      &client_b, discovered.endpoint_id,
      {.endpoint_info = discovered.endpoint_info,
       .listener =
           {
               .initiated_cb =
                   [&connect_latch](const std::string& endpoint_id,
                                    const ConnectionResponseInfo& info) {
                     LOG(INFO) << "RequestConnection: initiated_cb called";
                     connect_latch.CountDown();
                   },
           }},
      connection_options_);
  std::string client_b_local_endpoint = client_b.GetLocalEndpointId();

  EXPECT_TRUE(connect_latch.Await(absl::Milliseconds(1000)).result());
  EXPECT_TRUE(client_b.Is5GHzSupported(discovered.endpoint_id));
  EXPECT_EQ(client_b.GetBssid(discovered.endpoint_id), kBssid);
  EXPECT_EQ(client_b.GetApFrequency(discovered.endpoint_id), kFreq);
  EXPECT_EQ(client_b.GetIPAddress(discovered.endpoint_id),
            std::string(kIp4Bytes));
  // When connection is established, EndpointManager will setup KeepAliveManager
  // loop. When it fails, the connection will be dismantled. Since this a unit
  // test, KeepAliveManager won't be really up. The disconnection may happen
  // before the following check, which cause the check fail. So we check the
  // connection status first.
  if (client_b.IsConnectedToEndpoint(discovered.endpoint_id)) {
    EXPECT_EQ(client_a.Is5GHzSupported(client_b_local_endpoint),
              mediums_b.GetWifi().GetCapability().supports_5_ghz);
    EXPECT_EQ(client_a.GetBssid(client_b_local_endpoint),
              mediums_b.GetWifi().GetInformation().bssid);
    EXPECT_EQ(client_a.GetApFrequency(client_b_local_endpoint),
              mediums_b.GetWifi().GetInformation().ap_frequency);
    EXPECT_EQ(client_a.GetIPAddress(client_b_local_endpoint),
              mediums_b.GetWifi().GetInformation().ip_address_4_bytes);
  }

  handler_a.StopAdvertising(&client_a);
  handler_b.StopDiscovery(&client_b);
  bwu_a.Shutdown();
  bwu_b.Shutdown();
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam,
       CanStartListeningForIncomingConnections) {
  env_.Start();
  std::string endpoint_name_a{"endpoint_name"};
  Mediums mediums_a;
  BluetoothRadio& radio_a = mediums_a.GetBluetoothRadio();
  radio_a.GetBluetoothAdapter().SetName("BT Device A");
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {},
                   {.allow_upgrade_to = {.bluetooth = true}});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  v3::ConnectionListeningOptions v3_options{
      .strategy = Strategy::kP2pCluster,
      .enable_ble_listening = true,
      .enable_bluetooth_listening = true,
      .enable_wlan_listening = true,
  };
  // make sure mediums are not accepting before calling handler.
  ASSERT_FALSE(
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  ASSERT_FALSE(mediums_a.GetWifiLan().IsAcceptingConnections(service_id_));
  if (std::get<1>(GetParam())) {
    ASSERT_FALSE(mediums_a.GetBleV2().IsAcceptingConnections(service_id_));
  } else {
    ASSERT_FALSE(mediums_a.GetBle().IsAcceptingConnections(service_id_));
  }
  // call handler.
  auto result = handler_a.StartListeningForIncomingConnections(
      &client_a_, service_id_, v3_options, {});
  // now check to make sure we are in fact accepting connections.
  EXPECT_TRUE(
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  EXPECT_TRUE(mediums_a.GetWifiLan().IsAcceptingConnections(service_id_));
  if (std::get<1>(GetParam())) {
    EXPECT_TRUE(mediums_a.GetBleV2().IsAcceptingConnections(service_id_));
  } else {
    EXPECT_TRUE(mediums_a.GetBle().IsAcceptingConnections(service_id_));
  }
  EXPECT_EQ(result.second.size(), 3);
  ASSERT_TRUE(client_a_.IsListeningForIncomingConnections());
  EXPECT_EQ(client_a_.GetListeningForIncomingConnectionsServiceId(),
            service_id_);
  EXPECT_EQ(client_a_.GetListeningOptions().enable_ble_listening,
            v3_options.enable_ble_listening);
  EXPECT_EQ(client_a_.GetListeningOptions().enable_bluetooth_listening,
            v3_options.enable_bluetooth_listening);
  EXPECT_EQ(client_a_.GetListeningOptions().enable_wlan_listening,
            v3_options.enable_wlan_listening);
  EXPECT_EQ(client_a_.GetListeningOptions().strategy, v3_options.strategy);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam,
       CanStopListeningForIncomingConnections) {
  env_.Start();
  std::string endpoint_name_a{"endpoint_name"};
  Mediums mediums_a;
  BluetoothRadio& radio_a = mediums_a.GetBluetoothRadio();
  radio_a.GetBluetoothAdapter().SetName("BT Device A");
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {},
                   {.allow_upgrade_to = {.bluetooth = true}});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  v3::ConnectionListeningOptions v3_options{
      .strategy = Strategy::kP2pCluster,
      .enable_ble_listening = true,
      .enable_bluetooth_listening = true,
      .enable_wlan_listening = true,
  };
  // make sure mediums are not accepting before calling handler.
  ASSERT_FALSE(
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  ASSERT_FALSE(mediums_a.GetWifiLan().IsAcceptingConnections(service_id_));
  if (std::get<1>(GetParam())) {
    ASSERT_FALSE(mediums_a.GetBleV2().IsAcceptingConnections(service_id_));
  } else {
    ASSERT_FALSE(mediums_a.GetBle().IsAcceptingConnections(service_id_));
  }
  // call handler.
  auto result = handler_a.StartListeningForIncomingConnections(
      &client_a_, service_id_, v3_options, {});
  // now check to make sure we are in fact accepting connections.
  ASSERT_TRUE(
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  ASSERT_TRUE(mediums_a.GetWifiLan().IsAcceptingConnections(service_id_));
  if (std::get<1>(GetParam())) {
    ASSERT_TRUE(mediums_a.GetBleV2().IsAcceptingConnections(service_id_));
  } else {
    ASSERT_TRUE(mediums_a.GetBle().IsAcceptingConnections(service_id_));
  }
  // stop.
  handler_a.StopListeningForIncomingConnections(&client_a_);
  EXPECT_FALSE(
      mediums_a.GetBluetoothClassic().IsAcceptingConnections(service_id_));
  EXPECT_FALSE(mediums_a.GetWifiLan().IsAcceptingConnections(service_id_));
  if (std::get<1>(GetParam())) {
    EXPECT_FALSE(mediums_a.GetBleV2().IsAcceptingConnections(service_id_));
  } else {
    EXPECT_FALSE(mediums_a.GetBle().IsAcceptingConnections(service_id_));
  }
  env_.Stop();
}

TEST_F(P2pClusterPcpHandlerTest, CanAwdlDiscovery) {
  std::string endpoint_name{"endpoint_name"};

  env_.Start();
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  InjectedBluetoothDeviceStore ibds_a;
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);

  EXPECT_EQ(handler_a.StartDiscovery(&client_a_, service_id_,
                                     DiscoveryOptions{
                                         {Strategy::kP2pCluster,
                                          BooleanMediumSelector{
                                              .awdl = true,
                                          }},
                                     },
                                     {}),
            Status{Status::kSuccess});

  EXPECT_TRUE(mediums_a.GetAwdl().IsDiscovering(service_id_));

  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_F(P2pClusterPcpHandlerTest, CanAwdlWifiLanDiscovery) {
  std::string endpoint_name{"endpoint_name"};

  env_.Start();
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  InjectedBluetoothDeviceStore ibds_a;
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);

  EXPECT_EQ(handler_a.StartDiscovery(&client_a_, service_id_,
                                     DiscoveryOptions{
                                         {Strategy::kP2pCluster,
                                          BooleanMediumSelector{
                                              .wifi_lan = true,
                                              .awdl = true,
                                          }},
                                     },
                                     {}),
            Status{Status::kSuccess});

  EXPECT_TRUE(mediums_a.GetAwdl().IsDiscovering(service_id_));
  EXPECT_TRUE(mediums_a.GetWifiLan().IsDiscovering(service_id_));

  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_F(P2pClusterPcpHandlerTest, CanAwdlAdvertise) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_,
                                 AdvertisingOptions{{Strategy::kP2pCluster,
                                                     BooleanMediumSelector{
                                                         .awdl = true,
                                                     }}},
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  EXPECT_TRUE(mediums_a.GetAwdl().IsAdvertising(service_id_));
  EXPECT_FALSE(mediums_a.GetWifiLan().IsAdvertising(service_id_));
  handler_a.StopAdvertising(&client_a_);
  env_.Stop();
}

TEST_F(P2pClusterPcpHandlerTest, CanAwdlWifiLanAdvertise) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_,
                                 AdvertisingOptions{{Strategy::kP2pCluster,
                                                     BooleanMediumSelector{
                                                         .wifi_lan = true,
                                                         .awdl = true,
                                                     }}},
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  EXPECT_TRUE(mediums_a.GetAwdl().IsAdvertising(service_id_));
  EXPECT_TRUE(mediums_a.GetWifiLan().IsAdvertising(service_id_));
  handler_a.StopAdvertising(&client_a_);
  env_.Stop();
}

TEST_P(P2pClusterPcpHandlerTestWithParam, CanUpdateAwdlDiscoveryOptions) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  discovery_options_.allowed.wifi_lan = true;
  discovery_options_.allowed.awdl = false;
  EXPECT_EQ(
      handler_a.StartDiscovery(&client_a_, service_id_, discovery_options_, {}),
      Status{Status::kSuccess});
  EXPECT_TRUE(mediums_a.GetWifiLan().IsDiscovering(service_id_));
  EXPECT_FALSE(mediums_a.GetAwdl().IsDiscovering(service_id_));
  discovery_options_.allowed.wifi_lan = false;
  discovery_options_.allowed.awdl = true;
  EXPECT_EQ(handler_a.UpdateDiscoveryOptions(&client_a_, service_id_,
                                             discovery_options_),
            Status{Status::kSuccess});
  EXPECT_FALSE(mediums_a.GetWifiLan().IsDiscovering(service_id_));
  EXPECT_TRUE(mediums_a.GetAwdl().IsDiscovering(service_id_));
  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_F(P2pClusterPcpHandlerTest, CanUpdateAwdlAdvertisingOptions) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  AdvertisingOptions advertising_options{
      {Strategy::kP2pCluster, BooleanMediumSelector{}}};
  advertising_options.allowed.wifi_lan = true;
  EXPECT_EQ(
      handler_a.StartAdvertising(&client_a_, service_id_, advertising_options,
                                 {.endpoint_info = ByteArray{endpoint_name}}),
      Status{Status::kSuccess});
  EXPECT_TRUE(mediums_a.GetWifiLan().IsAdvertising(service_id_));
  EXPECT_FALSE(mediums_a.GetAwdl().IsAdvertising(service_id_));
  advertising_options.allowed.wifi_lan = false;
  advertising_options.allowed.awdl = true;
  EXPECT_EQ(handler_a.UpdateAdvertisingOptions(&client_a_, service_id_,
                                               advertising_options),
            Status{Status::kSuccess});
  EXPECT_FALSE(mediums_a.GetWifiLan().IsAdvertising(service_id_));
  EXPECT_TRUE(mediums_a.GetAwdl().IsAdvertising(service_id_));
  handler_a.StopAdvertising(&client_a_);
  env_.Stop();
}

TEST_F(P2pClusterPcpHandlerTest, FailedToInjectEndpointWithoutDiscovery) {
  env_.Start();
  std::string endpoint_name{"endpoint_name"};
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);

  ByteArray remote_bluetooth_mac_address("\x01\x02\x03\x04\x05\x06");
  MacAddress mac_address;
  MacAddress::FromBytes(absl::MakeSpan(reinterpret_cast<const uint8_t*>(
                                           remote_bluetooth_mac_address.data()),
                                       remote_bluetooth_mac_address.size()),
                        mac_address);
  OutOfBandConnectionMetadata metadata = {
      .medium = location::nearby::proto::connections::Medium::BLUETOOTH,
      .endpoint_id = "ABCD",
      .endpoint_info = ByteArray("endpoint_info"),
      .remote_bluetooth_mac_address = remote_bluetooth_mac_address,
  };

  handler_a.InjectEndpoint(&client_a_, service_id_, metadata);
  env_.Sync();

  EXPECT_FALSE(ibds_a.IsInjectedDevice(mac_address));
  env_.Stop();
}

TEST_F(P2pClusterPcpHandlerTest, CanInjectEndpoint) {
  env_.Start();
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);

  DiscoveryOptions discovery_options{
      {Strategy::kP2pCluster,
       BooleanMediumSelector{
           .bluetooth = true,
       }},
      /* auto_upgrade_bandwidth= */ false,
      /* enforce_topology_constraints= */ false,
      /* is_out_of_band_connection= */ true,
  };

  CountDownLatch found_latch(1);
  std::string found_endpoint_id;

  EXPECT_EQ(
      handler_a.StartDiscovery(&client_a_, service_id_, discovery_options,
                               {
                                   .endpoint_found_cb =
                                       [&](const std::string& endpoint_id,
                                           const ByteArray& endpoint_info,
                                           const std::string& service_id) {
                                         found_endpoint_id = endpoint_id;
                                         found_latch.CountDown();
                                       },
                               }),
      Status{Status::kSuccess});

  std::string endpoint_id = "ABCD";
  std::string endpoint_info_name = "endpoint_info";
  ByteArray endpoint_info(endpoint_info_name);

  OutOfBandConnectionMetadata metadata = {
      .medium = location::nearby::proto::connections::Medium::BLUETOOTH,
      .endpoint_id = endpoint_id,
      .endpoint_info = endpoint_info,
      .remote_bluetooth_mac_address = GetSixBytesMacAddress(
          mediums_a.GetBluetoothRadio().GetBluetoothAdapter().GetAddress()),
  };

  handler_a.InjectEndpoint(&client_a_, service_id_, metadata);

  EXPECT_TRUE(found_latch.Await(absl::Milliseconds(1000)).result());
  EXPECT_EQ(found_endpoint_id, endpoint_id);
  EXPECT_TRUE(ibds_a.IsInjectedDevice(
      mediums_a.GetBluetoothRadio().GetBluetoothAdapter().GetAddress()));

  handler_a.StopDiscovery(&client_a_);
  env_.Stop();
}

TEST_F(P2pClusterPcpHandlerTest, CanConnectToInjectedEndpoint) {
  env_.Start();
  // Setup handler_a (advertiser)
  Mediums mediums_a;
  EndpointChannelManager ecm_a;
  EndpointManager em_a(&ecm_a);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {}, {});
  InjectedBluetoothDeviceStore ibds_a;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  mediums_a.GetBluetoothRadio().GetBluetoothAdapter().SetName("Device A");

  // Setup handler_b (discoverer)
  Mediums mediums_b;
  EndpointChannelManager ecm_b;
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {}, {});
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);
  mediums_b.GetBluetoothRadio().GetBluetoothAdapter().SetName("Device B");

  CountDownLatch found_latch(1);
  CountDownLatch connect_latch(2);

  std::string discovered_endpoint_id;
  ByteArray discovered_endpoint_info;

  // 1. Advertiser starts advertising
  std::string endpoint_info_name = "Advertiser Info";
  EXPECT_EQ(handler_a.StartAdvertising(
                &client_a_, service_id_, GetBluetoothOnlyAdvertisingOptions(),
                {
                    .endpoint_info = ByteArray{endpoint_info_name},
                    .listener =
                        {
                            .initiated_cb =
                                [&](const std::string& endpoint_id,
                                    const ConnectionResponseInfo& info) {
                                  connect_latch.CountDown();
                                },
                        },
                }),
            Status{Status::kSuccess});

  // 2. Discoverer starts out-of-band discovery
  DiscoveryOptions discovery_options{
      {Strategy::kP2pCluster,
       BooleanMediumSelector{
           .bluetooth = true,
       }},
      /* auto_upgrade_bandwidth= */ false,
      /* enforce_topology_constraints= */ false,
      /* is_out_of_band_connection= */ true,
  };

  EXPECT_EQ(handler_b.StartDiscovery(
                &client_b_, service_id_, discovery_options,
                {
                    .endpoint_found_cb =
                        [&](const std::string& endpoint_id,
                            const ByteArray& endpoint_info,
                            const std::string& service_id) {
                          discovered_endpoint_id = endpoint_id;
                          discovered_endpoint_info = endpoint_info;
                          found_latch.CountDown();
                        },
                }),
            Status{Status::kSuccess});

  // 3. Inject the endpoint into the discoverer
  OutOfBandConnectionMetadata metadata = {
      .medium = location::nearby::proto::connections::Medium::BLUETOOTH,
      .endpoint_id = client_a_.GetLocalEndpointId(),
      .endpoint_info = ByteArray{endpoint_info_name},
      .remote_bluetooth_mac_address = GetSixBytesMacAddress(
          mediums_a.GetBluetoothRadio().GetBluetoothAdapter().GetAddress()),
  };

  handler_b.InjectEndpoint(&client_b_, service_id_, metadata);

  EXPECT_TRUE(found_latch.Await(absl::Milliseconds(1000)).result());
  EXPECT_EQ(discovered_endpoint_id, client_a_.GetLocalEndpointId());

  // 4. Discoverer requests connection to the injected endpoint
  client_b_.AddCancellationFlag(discovered_endpoint_id);
  handler_b.RequestConnection(
      &client_b_, discovered_endpoint_id,
      {.endpoint_info = discovered_endpoint_info,
       .listener =
           {
               .initiated_cb =
                   [&](const std::string& endpoint_id,
                       const ConnectionResponseInfo& info) {
                     connect_latch.CountDown();
                   },
           }},
      {});

  EXPECT_TRUE(connect_latch.Await(absl::Milliseconds(2000)).result());

  handler_a.StopAdvertising(&client_a_);
  handler_b.StopDiscovery(&client_b_);
  env_.Stop();
}

class P2pLostHandlerTestWithParam : public testing::TestWithParam<bool> {
 protected:
  void SetUp() override {
    LOG(INFO) << "SetUp: begin";
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::kEnableAwdl, true);
    env_.SetBleExtendedAdvertisementsAvailable(true);
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::
            kDisableBluetoothClassicScanning,
        false);
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::kEnableBleV2,
        true);
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::kEnableInstantOnLost,
        GetParam());
    NearbyFlags::GetInstance().OverrideBoolFlagValue(
        config_package_nearby::nearby_connections_feature::
            kEnableAdvertisingForInstantOnLost,
        true);
  }

  AdvertisingOptions GetBleOnlyAdvertisingOptions() {
    return AdvertisingOptions{
        {Strategy::kP2pCluster,
         BooleanMediumSelector{
             .ble = true,
         }},
    };
  }

  DiscoveryOptions GetBleOnlyDiscoveryOptions() {
    return DiscoveryOptions{
        {Strategy::kP2pCluster,
         BooleanMediumSelector{
             .ble = true,
         }},
    };
  }

  ClientProxy client_a_;
  ClientProxy client_b_;
  std::string service_id_{"service"};
  MediumEnvironment& env_{MediumEnvironment::Instance()};
};


TEST_P(P2pLostHandlerTestWithParam, CanConnectWithInstantLostEnabled) {
  env_.Start();

  std::string endpoint_name_a{"endpoint_name"};
  Mediums mediums_a;
  Mediums mediums_b;
  BluetoothRadio& radio_a = mediums_a.GetBluetoothRadio();
  BluetoothRadio& radio_b = mediums_b.GetBluetoothRadio();
  radio_a.GetBluetoothAdapter().SetName("BT Device A");
  radio_b.GetBluetoothAdapter().SetName("BT Device B");
  EndpointChannelManager ecm_a;
  EndpointChannelManager ecm_b;
  EndpointManager em_a(&ecm_a);
  EndpointManager em_b(&ecm_b);
  BwuManager bwu_a(mediums_a, em_a, ecm_a, {},
                   {.allow_upgrade_to = {.bluetooth = true}});
  BwuManager bwu_b(mediums_b, em_b, ecm_b, {},
                   {.allow_upgrade_to = {.bluetooth = true}});
  InjectedBluetoothDeviceStore ibds_a;
  InjectedBluetoothDeviceStore ibds_b;
  P2pClusterPcpHandler handler_a(&mediums_a, &em_a, &ecm_a, &bwu_a, ibds_a);
  P2pClusterPcpHandler handler_b(&mediums_b, &em_b, &ecm_b, &bwu_b, ibds_b);

  CountDownLatch discover_latch(1);
  CountDownLatch connect_latch(2);
  CountDownLatch lost_latch(1);

  struct DiscoveredInfo {
    std::string endpoint_id;
    ByteArray endpoint_info;
    std::string service_id;
  } discovered;

  EXPECT_EQ(handler_b.StartDiscovery(
                &client_b_, service_id_, GetBleOnlyDiscoveryOptions(),
                {
                    .endpoint_found_cb =
                        [&discover_latch, &discovered](
                            const std::string& endpoint_id,
                            const ByteArray& endpoint_info,
                            const std::string& service_id) {
                          LOG(INFO) << "Device discovered: id=" << endpoint_id
                                    << ", endpoint_info="
                                    << std::string{endpoint_info};
                          discovered = {
                              .endpoint_id = endpoint_id,
                              .endpoint_info = endpoint_info,
                              .service_id = service_id,
                          };
                          discover_latch.CountDown();
                        },
                    .endpoint_lost_cb =
                        [&lost_latch](const std::string& endpoint_id) {
                          LOG(INFO) << "Device lost: id=" << endpoint_id;
                          lost_latch.CountDown();
                        },
                }),
            Status{Status::kSuccess});

  EXPECT_EQ(
      handler_a.StartAdvertising(
          &client_a_, service_id_, GetBleOnlyAdvertisingOptions(),
          {
              .endpoint_info = ByteArray{endpoint_name_a},
              .listener =
                  {
                      .initiated_cb =
                          [&connect_latch](const std::string& endpoint_id,
                                           const ConnectionResponseInfo& info) {
                            LOG(INFO)
                                << "StartAdvertising: initiated_cb called";
                            connect_latch.CountDown();
                          },
                  },
          }),
      Status{Status::kSuccess});

  EXPECT_TRUE(discover_latch.Await(absl::Milliseconds(1000)).result());
  EXPECT_EQ(endpoint_name_a, std::string{discovered.endpoint_info});

  client_b_.AddCancellationFlag(discovered.endpoint_id);
  handler_b.RequestConnection(
      &client_b_, discovered.endpoint_id,
      {.endpoint_info = discovered.endpoint_info,
       .listener =
           {
               .initiated_cb =
                   [&connect_latch](const std::string& endpoint_id,
                                    const ConnectionResponseInfo& info) {
                     LOG(INFO) << "RequestConnection: initiated_cb called";
                     connect_latch.CountDown();
                   },
           }},
      {});
  std::string client_b_local_endpoint = client_b_.GetLocalEndpointId();

  EXPECT_TRUE(connect_latch.Await(absl::Milliseconds(1000)).result());

  handler_a.StopAdvertising(&client_a_);
  EXPECT_TRUE(lost_latch.Await(absl::Milliseconds(3000)).result());
  handler_b.StopDiscovery(&client_b_);
  bwu_a.Shutdown();
  bwu_b.Shutdown();
  env_.Stop();
  env_.SetBleExtendedAdvertisementsAvailable(false);
}

INSTANTIATE_TEST_SUITE_P(
    ParametrisedPcpHandlerTest, P2pClusterPcpHandlerTestWithParam,
    ::testing::Combine(/*mediums=*/::testing::ValuesIn(kTestCases),
                       /*ble_v2_enabled=*/::testing::Bool(),
                       /*disable_bluetooth_scanning=*/::testing::Bool()));
INSTANTIATE_TEST_SUITE_P(ParametrisedP2pLostHandlerTest,
                         P2pLostHandlerTestWithParam, testing::Bool());

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