| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/proximity_auth/ble/bluetooth_low_energy_connection.h" | 5 #include "components/proximity_auth/ble/bluetooth_low_energy_connection.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 | 8 |
| 9 #include <memory> | 9 #include <memory> |
| 10 #include <utility> | 10 #include <utility> |
| (...skipping 121 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 132 remote_device_(CreateLERemoteDeviceForTest()), | 132 remote_device_(CreateLERemoteDeviceForTest()), |
| 133 service_uuid_(device::BluetoothUUID(kServiceUUID)), | 133 service_uuid_(device::BluetoothUUID(kServiceUUID)), |
| 134 to_peripheral_char_uuid_(device::BluetoothUUID(kToPeripheralCharUUID)), | 134 to_peripheral_char_uuid_(device::BluetoothUUID(kToPeripheralCharUUID)), |
| 135 from_peripheral_char_uuid_( | 135 from_peripheral_char_uuid_( |
| 136 device::BluetoothUUID(kFromPeripheralCharUUID)), | 136 device::BluetoothUUID(kFromPeripheralCharUUID)), |
| 137 notify_session_alias_(NULL), | 137 notify_session_alias_(NULL), |
| 138 bluetooth_throttler_(new NiceMock<MockBluetoothThrottler>), | 138 bluetooth_throttler_(new NiceMock<MockBluetoothThrottler>), |
| 139 task_runner_(new base::TestSimpleTaskRunner) {} | 139 task_runner_(new base::TestSimpleTaskRunner) {} |
| 140 | 140 |
| 141 void SetUp() override { | 141 void SetUp() override { |
| 142 device_ = base::WrapUnique(new NiceMock<device::MockBluetoothDevice>( | 142 device_ = base::MakeUnique<NiceMock<device::MockBluetoothDevice>>( |
| 143 adapter_.get(), 0, kTestRemoteDeviceName, | 143 adapter_.get(), 0, kTestRemoteDeviceName, |
| 144 kTestRemoteDeviceBluetoothAddress, false, false)); | 144 kTestRemoteDeviceBluetoothAddress, false, false); |
| 145 | 145 |
| 146 service_ = base::WrapUnique(new NiceMock<device::MockBluetoothGattService>( | 146 service_ = base::MakeUnique<NiceMock<device::MockBluetoothGattService>>( |
| 147 device_.get(), kServiceID, service_uuid_, true, false)); | 147 device_.get(), kServiceID, service_uuid_, true, false); |
| 148 to_peripheral_char_ = | 148 to_peripheral_char_ = |
| 149 base::WrapUnique(new NiceMock<device::MockBluetoothGattCharacteristic>( | 149 base::MakeUnique<NiceMock<device::MockBluetoothGattCharacteristic>>( |
| 150 service_.get(), kToPeripheralCharID, to_peripheral_char_uuid_, | 150 service_.get(), kToPeripheralCharID, to_peripheral_char_uuid_, |
| 151 false, kCharacteristicProperties, | 151 false, kCharacteristicProperties, |
| 152 device::BluetoothRemoteGattCharacteristic::PERMISSION_NONE)); | 152 device::BluetoothRemoteGattCharacteristic::PERMISSION_NONE); |
| 153 | 153 |
| 154 from_peripheral_char_ = | 154 from_peripheral_char_ = |
| 155 base::WrapUnique(new NiceMock<device::MockBluetoothGattCharacteristic>( | 155 base::MakeUnique<NiceMock<device::MockBluetoothGattCharacteristic>>( |
| 156 service_.get(), kFromPeripheralCharID, from_peripheral_char_uuid_, | 156 service_.get(), kFromPeripheralCharID, from_peripheral_char_uuid_, |
| 157 false, kCharacteristicProperties, | 157 false, kCharacteristicProperties, |
| 158 device::BluetoothRemoteGattCharacteristic::PERMISSION_NONE)); | 158 device::BluetoothRemoteGattCharacteristic::PERMISSION_NONE); |
| 159 | 159 |
| 160 device::BluetoothAdapterFactory::SetAdapterForTesting(adapter_); | 160 device::BluetoothAdapterFactory::SetAdapterForTesting(adapter_); |
| 161 | 161 |
| 162 std::vector<const device::BluetoothDevice*> devices; | 162 std::vector<const device::BluetoothDevice*> devices; |
| 163 devices.push_back(device_.get()); | 163 devices.push_back(device_.get()); |
| 164 ON_CALL(*adapter_, GetDevices()).WillByDefault(Return(devices)); | 164 ON_CALL(*adapter_, GetDevices()).WillByDefault(Return(devices)); |
| 165 ON_CALL(*adapter_, GetDevice(kTestRemoteDeviceBluetoothAddress)) | 165 ON_CALL(*adapter_, GetDevice(kTestRemoteDeviceBluetoothAddress)) |
| 166 .WillByDefault(Return(device_.get())); | 166 .WillByDefault(Return(device_.get())); |
| 167 ON_CALL(*device_, GetGattService(kServiceID)) | 167 ON_CALL(*device_, GetGattService(kServiceID)) |
| 168 .WillByDefault(Return(service_.get())); | 168 .WillByDefault(Return(service_.get())); |
| (...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 213 // Preparing |connection| to run |create_gatt_connection_success_callback_|. | 213 // Preparing |connection| to run |create_gatt_connection_success_callback_|. |
| 214 EXPECT_FALSE(create_gatt_connection_error_callback_.is_null()); | 214 EXPECT_FALSE(create_gatt_connection_error_callback_.is_null()); |
| 215 ASSERT_FALSE(create_gatt_connection_success_callback_.is_null()); | 215 ASSERT_FALSE(create_gatt_connection_success_callback_.is_null()); |
| 216 EXPECT_CALL(*connection, CreateCharacteristicsFinder(_, _)) | 216 EXPECT_CALL(*connection, CreateCharacteristicsFinder(_, _)) |
| 217 .WillOnce(DoAll( | 217 .WillOnce(DoAll( |
| 218 SaveArg<0>(&characteristics_finder_success_callback_), | 218 SaveArg<0>(&characteristics_finder_success_callback_), |
| 219 SaveArg<1>(&characteristics_finder_error_callback_), | 219 SaveArg<1>(&characteristics_finder_error_callback_), |
| 220 Return(new NiceMock<MockBluetoothLowEnergyCharacteristicsFinder>))); | 220 Return(new NiceMock<MockBluetoothLowEnergyCharacteristicsFinder>))); |
| 221 | 221 |
| 222 create_gatt_connection_success_callback_.Run( | 222 create_gatt_connection_success_callback_.Run( |
| 223 base::WrapUnique(new NiceMock<device::MockBluetoothGattConnection>( | 223 base::MakeUnique<NiceMock<device::MockBluetoothGattConnection>>( |
| 224 adapter_, kTestRemoteDeviceBluetoothAddress))); | 224 adapter_, kTestRemoteDeviceBluetoothAddress)); |
| 225 | 225 |
| 226 EXPECT_EQ(connection->sub_status(), | 226 EXPECT_EQ(connection->sub_status(), |
| 227 BluetoothLowEnergyConnection::SubStatus::WAITING_CHARACTERISTICS); | 227 BluetoothLowEnergyConnection::SubStatus::WAITING_CHARACTERISTICS); |
| 228 EXPECT_EQ(connection->status(), Connection::IN_PROGRESS); | 228 EXPECT_EQ(connection->status(), Connection::IN_PROGRESS); |
| 229 } | 229 } |
| 230 | 230 |
| 231 // Transitions |connection| from WAITING_CHARACTERISTICS to | 231 // Transitions |connection| from WAITING_CHARACTERISTICS to |
| 232 // WAITING_NOTIFY_SESSION state. | 232 // WAITING_NOTIFY_SESSION state. |
| 233 void CharacteristicsFound(MockBluetoothLowEnergyConnection* connection) { | 233 void CharacteristicsFound(MockBluetoothLowEnergyConnection* connection) { |
| 234 EXPECT_CALL(*from_peripheral_char_, StartNotifySession(_, _)) | 234 EXPECT_CALL(*from_peripheral_char_, StartNotifySession(_, _)) |
| (...skipping 356 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 591 EXPECT_CALL(*to_peripheral_char_, WriteRemoteCharacteristic(_, _, _)) | 591 EXPECT_CALL(*to_peripheral_char_, WriteRemoteCharacteristic(_, _, _)) |
| 592 .WillOnce( | 592 .WillOnce( |
| 593 DoAll(SaveArg<0>(&last_value_written_on_to_peripheral_char_), | 593 DoAll(SaveArg<0>(&last_value_written_on_to_peripheral_char_), |
| 594 SaveArg<1>(&write_remote_characteristic_success_callback_), | 594 SaveArg<1>(&write_remote_characteristic_success_callback_), |
| 595 SaveArg<2>(&write_remote_characteristic_error_callback_))); | 595 SaveArg<2>(&write_remote_characteristic_error_callback_))); |
| 596 | 596 |
| 597 // Message (bytes) that is going to be sent. | 597 // Message (bytes) that is going to be sent. |
| 598 int message_size = 100; | 598 int message_size = 100; |
| 599 std::string message(message_size, 'A'); | 599 std::string message(message_size, 'A'); |
| 600 message[0] = 'B'; | 600 message[0] = 'B'; |
| 601 connection->SendMessage(base::WrapUnique(new FakeWireMessage(message))); | 601 connection->SendMessage(base::MakeUnique<FakeWireMessage>(message)); |
| 602 | 602 |
| 603 // Expecting that |kSendSignal| + |message_size| + |message| was written. | 603 // Expecting that |kSendSignal| + |message_size| + |message| was written. |
| 604 EXPECT_EQ(last_value_written_on_to_peripheral_char_, | 604 EXPECT_EQ(last_value_written_on_to_peripheral_char_, |
| 605 CreateFirstCharacteristicValue(message, message.size())); | 605 CreateFirstCharacteristicValue(message, message.size())); |
| 606 EXPECT_CALL(*connection, OnDidSendMessage(_, _)); | 606 EXPECT_CALL(*connection, OnDidSendMessage(_, _)); |
| 607 | 607 |
| 608 RunWriteCharacteristicSuccessCallback(); | 608 RunWriteCharacteristicSuccessCallback(); |
| 609 } | 609 } |
| 610 | 610 |
| 611 TEST_F(ProximityAuthBluetoothLowEnergyConnectionTest, | 611 TEST_F(ProximityAuthBluetoothLowEnergyConnectionTest, |
| 612 SendMessage_LagerThanCharacteristicSize) { | 612 SendMessage_LagerThanCharacteristicSize) { |
| 613 std::unique_ptr<MockBluetoothLowEnergyConnection> connection( | 613 std::unique_ptr<MockBluetoothLowEnergyConnection> connection( |
| 614 CreateConnection()); | 614 CreateConnection()); |
| 615 InitializeConnection(connection.get()); | 615 InitializeConnection(connection.get()); |
| 616 | 616 |
| 617 // Expecting a first call of WriteRemoteCharacteristic, after SendMessage is | 617 // Expecting a first call of WriteRemoteCharacteristic, after SendMessage is |
| 618 // called. | 618 // called. |
| 619 EXPECT_CALL(*to_peripheral_char_, WriteRemoteCharacteristic(_, _, _)) | 619 EXPECT_CALL(*to_peripheral_char_, WriteRemoteCharacteristic(_, _, _)) |
| 620 .WillOnce( | 620 .WillOnce( |
| 621 DoAll(SaveArg<0>(&last_value_written_on_to_peripheral_char_), | 621 DoAll(SaveArg<0>(&last_value_written_on_to_peripheral_char_), |
| 622 SaveArg<1>(&write_remote_characteristic_success_callback_), | 622 SaveArg<1>(&write_remote_characteristic_success_callback_), |
| 623 SaveArg<2>(&write_remote_characteristic_error_callback_))); | 623 SaveArg<2>(&write_remote_characteristic_error_callback_))); |
| 624 | 624 |
| 625 // Message (bytes) that is going to be sent. | 625 // Message (bytes) that is going to be sent. |
| 626 int message_size = 600; | 626 int message_size = 600; |
| 627 std::string message(message_size, 'A'); | 627 std::string message(message_size, 'A'); |
| 628 message[0] = 'B'; | 628 message[0] = 'B'; |
| 629 connection->SendMessage(base::WrapUnique(new FakeWireMessage(message))); | 629 connection->SendMessage(base::MakeUnique<FakeWireMessage>(message)); |
| 630 | 630 |
| 631 // Expecting that |kSendSignal| + |message_size| was written in the first 8 | 631 // Expecting that |kSendSignal| + |message_size| was written in the first 8 |
| 632 // bytes. | 632 // bytes. |
| 633 std::vector<uint8_t> prefix( | 633 std::vector<uint8_t> prefix( |
| 634 last_value_written_on_to_peripheral_char_.begin(), | 634 last_value_written_on_to_peripheral_char_.begin(), |
| 635 last_value_written_on_to_peripheral_char_.begin() + 8); | 635 last_value_written_on_to_peripheral_char_.begin() + 8); |
| 636 EXPECT_EQ(prefix, CreateSendSignalWithSize(message_size)); | 636 EXPECT_EQ(prefix, CreateSendSignalWithSize(message_size)); |
| 637 std::vector<uint8_t> bytes_received( | 637 std::vector<uint8_t> bytes_received( |
| 638 last_value_written_on_to_peripheral_char_.begin() + 8, | 638 last_value_written_on_to_peripheral_char_.begin() + 8, |
| 639 last_value_written_on_to_peripheral_char_.end()); | 639 last_value_written_on_to_peripheral_char_.end()); |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 685 ASSERT_FALSE(create_gatt_connection_success_callback_.is_null()); | 685 ASSERT_FALSE(create_gatt_connection_success_callback_.is_null()); |
| 686 | 686 |
| 687 // Preparing |connection| to run |create_gatt_connection_success_callback_|. | 687 // Preparing |connection| to run |create_gatt_connection_success_callback_|. |
| 688 EXPECT_CALL(*connection, CreateCharacteristicsFinder(_, _)) | 688 EXPECT_CALL(*connection, CreateCharacteristicsFinder(_, _)) |
| 689 .WillOnce(DoAll( | 689 .WillOnce(DoAll( |
| 690 SaveArg<0>(&characteristics_finder_success_callback_), | 690 SaveArg<0>(&characteristics_finder_success_callback_), |
| 691 SaveArg<1>(&characteristics_finder_error_callback_), | 691 SaveArg<1>(&characteristics_finder_error_callback_), |
| 692 Return(new NiceMock<MockBluetoothLowEnergyCharacteristicsFinder>))); | 692 Return(new NiceMock<MockBluetoothLowEnergyCharacteristicsFinder>))); |
| 693 | 693 |
| 694 create_gatt_connection_success_callback_.Run( | 694 create_gatt_connection_success_callback_.Run( |
| 695 base::WrapUnique(new NiceMock<device::MockBluetoothGattConnection>( | 695 base::MakeUnique<NiceMock<device::MockBluetoothGattConnection>>( |
| 696 adapter_, kTestRemoteDeviceBluetoothAddress))); | 696 adapter_, kTestRemoteDeviceBluetoothAddress)); |
| 697 | 697 |
| 698 CharacteristicsFound(connection.get()); | 698 CharacteristicsFound(connection.get()); |
| 699 NotifySessionStarted(connection.get()); | 699 NotifySessionStarted(connection.get()); |
| 700 ResponseSignalReceived(connection.get()); | 700 ResponseSignalReceived(connection.get()); |
| 701 } | 701 } |
| 702 | 702 |
| 703 } // namespace proximity_auth | 703 } // namespace proximity_auth |
| OLD | NEW |