bluetooth: mac: add connected LE devices to chooser

device/bluetooth/test/mock_bluetooth_central_manager_mac.mm

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #import "device/bluetooth/test/mock_bluetooth_central_manager_mac.h"
 
+#import "base/mac/scoped_nsobject.h"
 #import "device/bluetooth/test/bluetooth_test_mac.h"
 #import "device/bluetooth/test/mock_bluetooth_cbperipheral_mac.h"
 
-@implementation MockCentralManager
+using base::scoped_nsobject;
+
+@implementation MockCentralManager {
+  scoped_nsobject<NSMutableDictionary> _connectedMockPeripheralPerServiceUUID;
+  scoped_nsobject<NSMutableArray> _retrieveConnectedPeripheralServiceUUIDs;
+}
 
 @synthesize scanForPeripheralsCallCount = _scanForPeripheralsCallCount;
 @synthesize stopScanCallCount = _stopScanCallCount;
 @synthesize delegate = _delegate;
 @synthesize state = _state;
 @synthesize bluetoothTestMac = _bluetoothTestMac;
 
+- (instancetype)init {
+  self = [super init];
+  if (self) {
+    _connectedMockPeripheralPerServiceUUID.reset(
+        [[NSMutableDictionary alloc] init]);
+    _retrieveConnectedPeripheralServiceUUIDs.reset(
+        [[NSMutableArray alloc] init]);
+  }
+  return self;
+}
+
 - (BOOL)isKindOfClass:(Class)aClass {
   if (aClass == [CBCentralManager class] ||
       [aClass isSubclassOfClass:[CBCentralManager class]]) {
     return YES;
   }
   return [super isKindOfClass:aClass];
 }
 
 - (BOOL)isMemberOfClass:(Class)aClass {
   if (aClass == [CBCentralManager class] ||
@@ skipping 18 matching lines @@
     _bluetoothTestMac->OnFakeBluetoothDeviceConnectGattCalled();
   }
 }
 
 - (void)cancelPeripheralConnection:(CBPeripheral*)peripheral {
   if (_bluetoothTestMac) {
     _bluetoothTestMac->OnFakeBluetoothGattDisconnect();
   }
 }
 
+- (NSArray*)retrieveConnectedPeripheralServiceUUIDs {
+  return [[_retrieveConnectedPeripheralServiceUUIDs.get() copy] autorelease];
+}
+
+- (NSArray*)retrieveConnectedPeripheralsWithServices:(NSArray*)services {
+  [_retrieveConnectedPeripheralServiceUUIDs.get()
+      addObjectsFromArray:[services copy]];
+  NSMutableArray* connectedPeripherals = [[NSMutableArray alloc] init];
+  for (CBUUID* uuid in services) {
+    NSSet* peripheralSet =
+        [_connectedMockPeripheralPerServiceUUID.get() objectForKey:uuid];
+    [connectedPeripherals addObjectsFromArray:peripheralSet.allObjects];
+  }
+  return connectedPeripherals;
+}
+
+- (void)setConnectedMockPeripheral:(CBPeripheral*)peripheral
+                  withServiceUUIDs:(NSSet*)serviceUUIDs {
+  for (CBUUID* uuid in serviceUUIDs) {
+    NSMutableSet* peripheralSet =
+        [_connectedMockPeripheralPerServiceUUID.get() objectForKey:uuid];
+    if (!peripheralSet) {
+      peripheralSet = [NSMutableSet set];
+      [_connectedMockPeripheralPerServiceUUID.get() setObject:peripheralSet
+                                                       forKey:uuid];
+    }
+    [peripheralSet addObject:peripheral];
+  }
+}
+
+- (void)resetRetrieveConnectedPeripheralServiceUUIDs {
+  [_retrieveConnectedPeripheralServiceUUIDs removeAllObjects];
+}
+
 @end