Index: content/shell/browser/layout_test/layout_test_bluetooth_adapter_provider.cc |
diff --git a/content/shell/browser/layout_test/layout_test_bluetooth_adapter_provider.cc b/content/shell/browser/layout_test/layout_test_bluetooth_adapter_provider.cc |
index fd7957d662ec4a6334bf242b03e248a0455aa953..8c9423f426dfcfd34df4530e02e4a569de9c5cc7 100644 |
--- a/content/shell/browser/layout_test/layout_test_bluetooth_adapter_provider.cc |
+++ b/content/shell/browser/layout_test/layout_test_bluetooth_adapter_provider.cc |
@@ -69,6 +69,8 @@ const char kGenericAccessServiceUUID[] = "1800"; |
const char kGlucoseServiceUUID[] = "1808"; |
const char kHeartRateServiceUUID[] = "180d"; |
const char kHumanInterfaceDeviceServiceUUID[] = "1812"; |
+const char kRequestDisconnectionServiceUUID[] = |
+ "01d7d889-7451-419f-aeb8-d65e7b9277af"; |
const char kTxPowerServiceUUID[] = "1804"; |
// Characteristics: |
const char kBlacklistExcludeReadsCharacteristicUUID[] = |
@@ -156,6 +158,8 @@ LayoutTestBluetoothAdapterProvider::GetBluetoothAdapter( |
return GetMissingCharacteristicHeartRateAdapter(); |
if (fake_adapter_name == "HeartRateAdapter") |
return GetHeartRateAdapter(); |
+ if (fake_adapter_name == "DisconnectingHeartRateAdapter") |
+ return GetDisconnectingHeartRateAdapter(); |
if (fake_adapter_name == "BlacklistTestAdapter") |
return GetBlacklistTestAdapter(); |
if (fake_adapter_name == "FailingConnectionsAdapter") |
@@ -422,6 +426,42 @@ LayoutTestBluetoothAdapterProvider::GetHeartRateAdapter() { |
device->AddMockService(GetGenericAccessService(device.get())); |
device->AddMockService(GetHeartRateService(adapter.get(), device.get())); |
+ |
+ adapter->AddMockDevice(std::move(device)); |
+ |
+ return adapter; |
+} |
+ |
+// static |
+scoped_refptr<NiceMockBluetoothAdapter> |
+LayoutTestBluetoothAdapterProvider::GetDisconnectingHeartRateAdapter() { |
+ scoped_refptr<NiceMockBluetoothAdapter> adapter(GetEmptyAdapter()); |
+ NiceMockBluetoothAdapter* adapter_ptr = adapter.get(); |
+ std::unique_ptr<NiceMockBluetoothDevice> device( |
+ GetHeartRateDevice(adapter.get())); |
+ NiceMockBluetoothDevice* device_ptr = device.get(); |
+ |
+ // TODO(ortuno): Implement the rest of the service's characteristics |
+ // See: http://crbug.com/529975 |
+ |
+ device->AddMockService(GetGenericAccessService(device.get())); |
+ device->AddMockService(GetHeartRateService(adapter.get(), device.get())); |
+ |
+ // Set up a service to disconnect the device when its characteristics are |
+ // discovered. |
+ std::unique_ptr<NiceMockBluetoothGattService> disconnection_service = |
+ GetBaseGATTService(device_ptr, kRequestDisconnectionServiceUUID); |
+ ON_CALL(*disconnection_service, GetCharacteristics()) |
+ .WillByDefault(Invoke([adapter_ptr, device_ptr]() { |
+ LOG(ERROR) << "Got characteristics"; |
+ device_ptr->SetConnected(false); |
+ FOR_EACH_OBSERVER(BluetoothAdapter::Observer, |
+ adapter_ptr->GetObservers(), |
+ DeviceChanged(adapter_ptr, device_ptr)); |
+ return std::vector<device::BluetoothRemoteGattCharacteristic*>(); |
+ })); |
+ device->AddMockService(std::move(disconnection_service)); |
+ |
adapter->AddMockDevice(std::move(device)); |
return adapter; |