Implement BluetoothRemoteGattCharacteristicWin::StartNotifySession and related unit tests

device/bluetooth/bluetooth_low_energy_win_fake.h

 // Copyright 2016 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.
 
 #ifndef DEVICE_BLUETOOTH_BLUETOOTH_LOW_ENERGY_WIN_FAKE_H_
 #define DEVICE_BLUETOOTH_BLUETOOTH_LOW_ENERGY_WIN_FAKE_H_
 
 #include "device/bluetooth/bluetooth_low_energy_win.h"
 
 #include <set>
 #include <unordered_map>
 
 namespace device {
 namespace win {
 
 struct BLEDevice;
 struct GattService;
+struct GattServiceObserver;
 struct GattCharacteristic;
 struct GattDescriptor;
 
 // The key of BLEDevicesMap is the string of the BLE device address.
 typedef std::unordered_map<std::string, scoped_ptr<BLEDevice>> BLEDevicesMap;
 // The key of GattServicesMap, GattCharacteristicsMap and GattDescriptorsMap is
 // the string of the attribute handle.
 typedef std::unordered_map<std::string, scoped_ptr<GattService>>
     GattServicesMap;
 typedef std::unordered_map<std::string, scoped_ptr<GattCharacteristic>>
     GattCharacteristicsMap;
 typedef std::unordered_map<std::string, scoped_ptr<GattDescriptor>>
     GattDescriptorsMap;
 // The key of BLEAttributeHandleTable is the string of the BLE device address.
 typedef std::unordered_map<std::string, scoped_ptr<std::set<USHORT>>>
     BLEAttributeHandleTable;
+// The key of GattServiceObserverTable is GattServiceObserver pointer.
+// Note: The underlying data type of BLUETOOTH_GATT_EVENT_HANDLE is PVOID.
+typedef std::unordered_map<BLUETOOTH_GATT_EVENT_HANDLE,
+                           scoped_ptr<GattServiceObserver>>
+    GattServiceObserverTable;
 
 struct BLEDevice {
   BLEDevice();
   ~BLEDevice();
   scoped_ptr<BluetoothLowEnergyDeviceInfo> device_info;
   GattServicesMap primary_services;
 };
 
 struct GattService {
   GattService();
   ~GattService();
   scoped_ptr<BTH_LE_GATT_SERVICE> service_info;
   GattServicesMap included_services;
   GattCharacteristicsMap included_characteristics;
+  std::vector<BLUETOOTH_GATT_EVENT_HANDLE> observers;
 };
 
 struct GattCharacteristic {
   GattCharacteristic();
   ~GattCharacteristic();
   scoped_ptr<BTH_LE_GATT_CHARACTERISTIC> characteristic_info;
   scoped_ptr<BTH_LE_GATT_CHARACTERISTIC_VALUE> value;
   GattDescriptorsMap included_descriptors;
   std::vector<HRESULT> read_errors;
   std::vector<HRESULT> write_errors;
+  std::vector<HRESULT> notify_errors;
 };
 
 struct GattDescriptor {
   GattDescriptor();
   ~GattDescriptor();
   scoped_ptr<BTH_LE_GATT_DESCRIPTOR> descriptor_info;
   scoped_ptr<BTH_LE_GATT_DESCRIPTOR_VALUE> value;
+  std::vector<HRESULT> write_errors;
+};
+
+struct GattServiceObserver {
+  GattServiceObserver();
+  ~GattServiceObserver();
+  PFNBLUETOOTH_GATT_EVENT_CALLBACK callback;
+  PVOID context;
 };
 
 // Fake implementation of BluetoothLowEnergyWrapper. Used for BluetoothTestWin.
 class BluetoothLowEnergyWrapperFake : public BluetoothLowEnergyWrapper {
  public:
   class Observer {
    public:
     Observer() {}
     ~Observer() {}
 
-    virtual void onWriteGattCharacteristicValue(
+    virtual void OnReadGattCharacteristicValue() = 0;
+    virtual void OnWriteGattCharacteristicValue(
         const PBTH_LE_GATT_CHARACTERISTIC_VALUE value) = 0;
+    virtual void OnWriteGattDescriptorValue(
+        const PBTH_LE_GATT_DESCRIPTOR_VALUE value) = 0;
+    virtual void OnStartCharacteristicNotification() = 0;
   };
 
   BluetoothLowEnergyWrapperFake();
   ~BluetoothLowEnergyWrapperFake() override;
 
   bool IsBluetoothLowEnergySupported() override;
   bool EnumerateKnownBluetoothLowEnergyDevices(
       ScopedVector<BluetoothLowEnergyDeviceInfo>* devices,
       std::string* error) override;
   bool EnumerateKnownBluetoothLowEnergyGattServiceDevices(
@@ skipping 14 matching lines @@
       scoped_ptr<BTH_LE_GATT_DESCRIPTOR>* out_included_descriptors,
       USHORT* out_counts) override;
   HRESULT ReadCharacteristicValue(
       base::FilePath& service_path,
       const PBTH_LE_GATT_CHARACTERISTIC characteristic,
       scoped_ptr<BTH_LE_GATT_CHARACTERISTIC_VALUE>* out_value) override;
   HRESULT WriteCharacteristicValue(
       base::FilePath& service_path,
       const PBTH_LE_GATT_CHARACTERISTIC characteristic,
       PBTH_LE_GATT_CHARACTERISTIC_VALUE new_value) override;
+  HRESULT RegisterGattEvents(base::FilePath& service_path,
+                             BTH_LE_GATT_EVENT_TYPE type,
+                             PVOID event_parameter,
+                             PFNBLUETOOTH_GATT_EVENT_CALLBACK callback,
+                             PVOID context,
+                             BLUETOOTH_GATT_EVENT_HANDLE* out_handle) override;
+  HRESULT UnregisterGattEvent(
+      BLUETOOTH_GATT_EVENT_HANDLE event_handle) override;
 
   BLEDevice* SimulateBLEDevice(std::string device_name,
                                BLUETOOTH_ADDRESS device_address);
   BLEDevice* GetSimulatedBLEDevice(std::string device_address);
   void RemoveSimulatedBLEDevice(std::string device_address);
 
   // Note: |parent_service| may be nullptr to indicate a primary service.
   GattService* SimulateGattService(BLEDevice* device,
                                    GattService* parent_service,
                                    const BTH_LE_UUID& uuid);
@@ skipping 13 matching lines @@
       std::string device_address,
       GattService* parent_service,
       const BTH_LE_GATT_CHARACTERISTIC& characteristic);
   void SimulateGattCharacteriscRemove(GattService* parent_service,
                                       std::string attribute_handle);
   GattCharacteristic* GetSimulatedGattCharacteristic(
       GattService* parent_service,
       std::string attribute_handle);
   void SimulateGattCharacteristicValue(GattCharacteristic* characteristic,
                                        const std::vector<uint8_t>& value);
+  void SimulateCharacteristicValueChangeNotification(
+      GattService* parent_service,
+      GattCharacteristic* characteristic);
   void SimulateGattCharacteristicReadError(GattCharacteristic* characteristic,
                                            HRESULT error);
   void SimulateGattCharacteristicWriteError(GattCharacteristic* characteristic,
                                             HRESULT error);
+  void SimulateGattCharacteristicSetNotifyError(
+      GattCharacteristic* characteristic,
+      HRESULT error);
   void SimulateGattDescriptor(std::string device_address,
                               GattCharacteristic* characteristic,
                               const BTH_LE_UUID& uuid);
+  GattDescriptor* GetSimulatedDescriptor(
+      GattCharacteristic* parent_characteristic,
+      std::string attribute_handle);
+  void SimulateGattDescriptorWrite(GattDescriptor* descriptor,
+                                   const std::vector<uint8_t>& value);
+  void SimulateGattDescriptorWriteError(GattDescriptor* descriptor,
+                                        HRESULT error);
   void AddObserver(Observer* observer);
 
  private:
   // Get simulated characteristic by |service_path| and |characteristic| info.
   GattCharacteristic* GetSimulatedGattCharacteristic(
       base::FilePath& service_path,
       const PBTH_LE_GATT_CHARACTERISTIC characteristic);
 
   // Generate an unique attribute handle on |device_address|.
   USHORT GenerateAUniqueAttributeHandle(std::string device_address);
@@ skipping 14 matching lines @@
 
   // Extract service attribute handles from the |path| generated by
   // GenerateGattServiceDevicePath.
   std::vector<std::string> ExtractServiceAttributeHandlesFromDevicePath(
       base::string16 path);
 
   // The canonical BLE device address string format is the
   // BluetoothDevice::CanonicalizeAddress.
   std::string BluetoothAddressToCanonicalString(const BLUETOOTH_ADDRESS& btha);
 
+  // Get client |characteristic| configuration descriptor which has standard
+  // UUID = 00002902-0000-1000-8000-00805f9b34fb.
+  GattDescriptor* GetCCCDescriptor(GattCharacteristic* characteristic);
+
   // Table to store allocated attribute handle for a device.
   BLEAttributeHandleTable attribute_handle_table_;
   BLEDevicesMap simulated_devices_;
   Observer* observer_;
+  GattServiceObserverTable gatt_service_observers_;
 };
 
 }  // namespace win
 }  // namespace device
 
 #endif  // DEVICE_BLUETOOTH_BLUETOOTH_LOW_ENERGY_WIN_FAKE_H_