Implementing GATT connection/disconnect on OS X.

device/bluetooth/test/bluetooth_test_mac.mm

Index: device/bluetooth/test/bluetooth_test_mac.mm
diff --git a/device/bluetooth/test/bluetooth_test_mac.mm b/device/bluetooth/test/bluetooth_test_mac.mm
index d64b5f79787622c89b85492f6684c06843f3c52b..2deebb1bdb357d3b1fffe3c6b8b5505101fa4460 100644
--- a/device/bluetooth/test/bluetooth_test_mac.mm
+++ b/device/bluetooth/test/bluetooth_test_mac.mm
@@ -6,18 +6,18 @@
 
 #include <stdint.h>
 
+#include "base/mac/foundation_util.h"
 #include "base/strings/string_number_conversions.h"
 #include "build/build_config.h"
 #include "device/bluetooth/bluetooth_adapter_mac.h"
+#include "device/bluetooth/test/mock_bluetooth_cbperipheral_mac.h"
 #include "device/bluetooth/test/mock_bluetooth_central_manager_mac.h"
 #include "device/bluetooth/test/test_bluetooth_adapter_observer.h"
 #include "third_party/ocmock/OCMock/OCMock.h"
 
-#if defined(OS_IOS)
 #import <CoreBluetooth/CoreBluetooth.h>
-#else  // !defined(OS_IOS)
-#import <IOBluetooth/IOBluetooth.h>
-#endif  // defined(OS_IOS)
+
+using base::mac::ObjCCast;
 
 namespace device {
 
@@ -63,7 +63,9 @@ const std::string BluetoothTestMac::kTestPeripheralUUID2 =
 
 BluetoothTestMac::BluetoothTestMac() {}
 
-BluetoothTestMac::~BluetoothTestMac() {}
+BluetoothTestMac::~BluetoothTestMac() {
+  [mock_central_manager_ release];
+}
 
 void BluetoothTestMac::SetUp() {}
 
@@ -83,9 +85,10 @@ void BluetoothTestMac::InitWithoutDefaultAdapter() {
   adapter_ = adapter_mac_;
 
   if (BluetoothAdapterMac::IsLowEnergyAvailable()) {
-    id low_energy_central_manager = [[MockCentralManager alloc] init];
-    [low_energy_central_manager setState:CBCentralManagerStateUnsupported];
-    adapter_mac_->SetCentralManagerForTesting(low_energy_central_manager);
+    mock_central_manager_ = [[MockCentralManager alloc] init];
+    [mock_central_manager_ setBluetoothTestMac:this];
+    [mock_central_manager_ setState:CBCentralManagerStateUnsupported];
+    adapter_mac_->SetCentralManagerForTesting((id)mock_central_manager_);
   }
 }
 
@@ -97,9 +100,10 @@ void BluetoothTestMac::InitWithFakeAdapter() {
   adapter_ = adapter_mac_;
 
   if (BluetoothAdapterMac::IsLowEnergyAvailable()) {
-    id low_energy_central_manager = [[MockCentralManager alloc] init];
-    [low_energy_central_manager setState:CBCentralManagerStatePoweredOn];
-    adapter_mac_->SetCentralManagerForTesting(low_energy_central_manager);
+    mock_central_manager_ = [[MockCentralManager alloc] init];
+    mock_central_manager_.bluetoothTestMac = this;
+    [mock_central_manager_ setState:CBCentralManagerStatePoweredOn];
+    adapter_mac_->SetCentralManagerForTesting((id)mock_central_manager_);
   }
 }
 
@@ -151,12 +155,17 @@ BluetoothDevice* BluetoothTestMac::DiscoverLowEnergyDevice(int device_ordinal) {
       break;
     }
     case 3: {
-      CBPeripheral* peripheral = CreateMockPeripheral(
-          [NSString stringWithUTF8String:kTestPeripheralUUID1.c_str()]);
-      NSString* name =
-          [NSString stringWithUTF8String:kTestDeviceNameEmpty.c_str()];
-      NSArray* uuids = nil;
-      NSDictionary* advertisement_data = CreateAdvertisementData(name, uuids);
+      base::scoped_nsobject<NSString> uuid_string(
+          [[NSString alloc] initWithUTF8String:kTestPeripheralUUID1.c_str()]);
+      base::scoped_nsobject<NSUUID> identifier(
+          [[NSUUID alloc] initWithUUIDString:uuid_string]);
+      base::scoped_nsobject<CBPeripheral> peripheral;
+      peripheral.reset((CBPeripheral*)[[MockCBPeripheral alloc]
+          initWithIdentifier:identifier]);
+      base::scoped_nsobject<NSString> name(
+          [[NSString alloc] initWithUTF8String:kTestDeviceNameEmpty.c_str()]);
+      base::scoped_nsobject<NSDictionary> advertisement_data(
+          [CreateAdvertisementData(name, nil) retain]);
       [central_manager_delegate centralManager:central_manager
                          didDiscoverPeripheral:peripheral
                              advertisementData:advertisement_data
@@ -180,6 +189,56 @@ BluetoothDevice* BluetoothTestMac::DiscoverLowEnergyDevice(int device_ordinal) {
   return observer.last_device();
 }
 
+void BluetoothTestMac::SimulateGattConnection(BluetoothDevice* device) {
+  BluetoothLowEnergyDeviceMac* lowEnergyDeviceMac =
+      static_cast<BluetoothLowEnergyDeviceMac*>(device);
+  CBPeripheral* peripheral = lowEnergyDeviceMac->GetPeripheral();
+  MockCBPeripheral* mockPeripheral = (MockCBPeripheral*)peripheral;

[msarda, 2016/03/02 10:31:48]

Here and everywhere in this file.

Sorry for this cast mess, I think my comments were not clear. If you cannot use
ObjcCast, prefer using static_cast.

[jlebel, 2016/03/02 15:34:17]

Done.

+  [mockPeripheral setState:CBPeripheralStateConnected];
+  CBCentralManager* centralManager = (CBCentralManager*)mock_central_manager_;
+  [centralManager.delegate centralManager:centralManager
+                     didConnectPeripheral:peripheral];
+}
+
+void BluetoothTestMac::SimulateGattDisconnection(BluetoothDevice* device) {
+  BluetoothLowEnergyDeviceMac* lowEnergyDeviceMac =
+      static_cast<BluetoothLowEnergyDeviceMac*>(device);
+  CBPeripheral* peripheral = lowEnergyDeviceMac->GetPeripheral();
+  MockCBPeripheral* mockPeripheral = (MockCBPeripheral*)peripheral;
+  [mockPeripheral setState:CBPeripheralStateDisconnected];
+  CBCentralManager* centralManager = (CBCentralManager*)mock_central_manager_;
+  [centralManager.delegate centralManager:centralManager
+                  didDisconnectPeripheral:peripheral
+                                    error:nil];
+}
+
+void BluetoothTestMac::SimulateGattConnectionError(
+    BluetoothDevice* device,
+    BluetoothDevice::ConnectErrorCode errorCode) {
+  BluetoothLowEnergyDeviceMac* lowEnergyDeviceMac =
+      static_cast<BluetoothLowEnergyDeviceMac*>(device);
+  CBPeripheral* peripheral = lowEnergyDeviceMac->GetPeripheral();
+  MockCBPeripheral* mockPeripheral = (MockCBPeripheral*)peripheral;
+  [mockPeripheral setState:CBPeripheralStateDisconnected];
+  CBCentralManager* centralManager = (CBCentralManager*)mock_central_manager_;
+  // TODO(http://crbug.com/585894): Need to convert the connect error code into
+  // NSError
+  NSError* error = [NSError errorWithDomain:@"BluetoothDevice::ConnectErrorCode"
+                                       code:-1
+                                   userInfo:nil];
+  [centralManager.delegate centralManager:centralManager
+               didFailToConnectPeripheral:peripheral
+                                    error:error];
+}
+
+void BluetoothTestMac::OnFakeBluetoothDeviceConnectGattCalled() {
+  gatt_connection_attempts_++;
+}
+
+void BluetoothTestMac::OnFakeBluetoothGattDisconnect() {
+  gatt_disconnection_attempts_++;
+}
+
 // Utility function for generating new (CBUUID, address) pairs where CBUUID
 // hashes to address. For use when adding a new device address to the testing
 // suite because CoreBluetooth peripherals have CBUUIDs in place of addresses,