mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-15 07:06:11 -04:00
699 lines
27 KiB
C++
699 lines
27 KiB
C++
// 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
|