Implement BluetoothRemoteGattCharacteristicWin::StartNotifySession and related unit tests

device/bluetooth/test/bluetooth_test_win.cc

 // 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.
 
 #include "device/bluetooth/test/bluetooth_test_win.h"
 
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/run_loop.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/test/test_pending_task.h"
 #include "base/time/time.h"
 #include "device/bluetooth/bluetooth_adapter_win.h"
 #include "device/bluetooth/bluetooth_low_energy_win.h"
 #include "device/bluetooth/bluetooth_remote_gatt_characteristic_win.h"
+#include "device/bluetooth/bluetooth_remote_gatt_descriptor_win.h"
 #include "device/bluetooth/bluetooth_remote_gatt_service_win.h"
 
 namespace {
 
 BLUETOOTH_ADDRESS CanonicalStringToBLUETOOTH_ADDRESS(
     std::string device_address) {
   BLUETOOTH_ADDRESS win_addr;
   unsigned int data[6];
   int result =
       sscanf_s(device_address.c_str(), "%02X:%02X:%02X:%02X:%02X:%02X",
@@ skipping 80 matching lines @@
   win::BluetoothClassicWrapper::SetInstanceForTest(fake_bt_classic_wrapper_);
   win::BluetoothLowEnergyWrapper::SetInstanceForTest(fake_bt_le_wrapper_);
   fake_bt_classic_wrapper_->SimulateARadio(
       base::SysUTF8ToWide(kTestAdapterName),
       CanonicalStringToBLUETOOTH_ADDRESS(kTestAdapterAddress));
 
   adapter_ = new BluetoothAdapterWin(base::Bind(
       &BluetoothTestWin::AdapterInitCallback, base::Unretained(this)));
   adapter_win_ = static_cast<BluetoothAdapterWin*>(adapter_.get());
   adapter_win_->InitForTest(nullptr, bluetooth_task_runner_);
-  adapter_win_->GetWinBluetoothTaskManager()->AddObserver(this);
   FinishPendingTasks();
 }
 
 bool BluetoothTestWin::DenyPermission() {
   return false;
 }
 
+void BluetoothTestWin::RunUntilIdle() {
+  FinishPendingTasks();
+}
+
 void BluetoothTestWin::StartLowEnergyDiscoverySession() {
   __super ::StartLowEnergyDiscoverySession();
   FinishPendingTasks();
 }
 
 BluetoothDevice* BluetoothTestWin::DiscoverLowEnergyDevice(int device_ordinal) {
   if (device_ordinal > 4 || device_ordinal < 1)
     return nullptr;
 
   std::string device_name = kTestDeviceName;
@@ skipping 177 matching lines @@
 
   RunPendingTasksUntilCallback();
 }
 
 void BluetoothTestWin::SimulateGattCharacteristicReadError(
     BluetoothGattCharacteristic* characteristic,
     BluetoothGattService::GattErrorCode error_code) {
   win::GattCharacteristic* target_characteristic =
       GetSimulatedCharacteristic(characteristic);
   CHECK(target_characteristic);
-  HRESULT hr = ERROR_SEM_TIMEOUT;
+  HRESULT hr = HRESULT_FROM_WIN32(ERROR_SEM_TIMEOUT);
   if (error_code == BluetoothGattService::GATT_ERROR_INVALID_LENGTH)
     hr = E_BLUETOOTH_ATT_INVALID_ATTRIBUTE_VALUE_LENGTH;
   fake_bt_le_wrapper_->SimulateGattCharacteristicReadError(
       target_characteristic, hr);
 
   FinishPendingTasks();
 }
 
 void BluetoothTestWin::SimulateGattCharacteristicWrite(
     BluetoothGattCharacteristic* characteristic) {
   RunPendingTasksUntilCallback();
 }
 
 void BluetoothTestWin::SimulateGattCharacteristicWriteError(
     BluetoothGattCharacteristic* characteristic,
     BluetoothGattService::GattErrorCode error_code) {
   win::GattCharacteristic* target_characteristic =
       GetSimulatedCharacteristic(characteristic);
   CHECK(target_characteristic);
-  HRESULT hr = ERROR_SEM_TIMEOUT;
+  HRESULT hr = HRESULT_FROM_WIN32(ERROR_SEM_TIMEOUT);
   if (error_code == BluetoothGattService::GATT_ERROR_INVALID_LENGTH)
     hr = E_BLUETOOTH_ATT_INVALID_ATTRIBUTE_VALUE_LENGTH;
   fake_bt_le_wrapper_->SimulateGattCharacteristicWriteError(
       target_characteristic, hr);
 
   FinishPendingTasks();
 }
 
+void BluetoothTestWin::
+    SimulateGattCharacteristicSetNotifyWillFailSynchronouslyOnce(
+        BluetoothGattCharacteristic* characteristic) {
+  win::GattCharacteristic* target_characteristic =
+      GetSimulatedCharacteristic(characteristic);
+  CHECK(target_characteristic);
+  fake_bt_le_wrapper_->SimulateGattCharacteristicSetNotifyError(
+      target_characteristic, E_BLUETOOTH_ATT_UNKNOWN_ERROR);
+}
+
 void BluetoothTestWin::DeleteDevice(BluetoothDevice* device) {
   CHECK(device);
   fake_bt_le_wrapper_->RemoveSimulatedBLEDevice(device->GetAddress());
 }
 
 void BluetoothTestWin::SimulateGattDescriptor(
     BluetoothGattCharacteristic* characteristic,
     const std::string& uuid) {
   win::GattCharacteristic* target_characteristic =
       GetSimulatedCharacteristic(characteristic);
   CHECK(target_characteristic);
   fake_bt_le_wrapper_->SimulateGattDescriptor(
       characteristic->GetService()->GetDevice()->GetAddress(),
       target_characteristic, CanonicalStringToBTH_LE_UUID(uuid));
   ForceRefreshDevice();
 }
 
-void BluetoothTestWin::OnAttemptReadGattCharacteristic() {
+void BluetoothTestWin::SimulateGattDescriptorWriteWillFailSynchronouslyOnce(
+    BluetoothGattDescriptor* descriptor) {
+  BluetoothRemoteGattDescriptorWin* win_descriptor =
+      (BluetoothRemoteGattDescriptorWin*)descriptor;
+  win::GattCharacteristic* target_characteristic =
+      GetSimulatedCharacteristic(win_descriptor->GetCharacteristic());
+  CHECK(target_characteristic);
+  win::GattDescriptor* target_descriptor =
+      fake_bt_le_wrapper_->GetSimulatedDescriptor(
+          target_characteristic,
+          std::to_string(win_descriptor->GetAttributeHandle()));
+  CHECK(target_descriptor);
+  fake_bt_le_wrapper_->SimulateGattDescriptorWriteError(
+      target_descriptor, E_BLUETOOTH_ATT_REQUEST_NOT_SUPPORTED);
+}
+
+void BluetoothTestWin::SimulateGattNotifySessionStarted(
+    BluetoothGattCharacteristic* characteristic) {
+  FinishPendingTasks();
+}
+
+void BluetoothTestWin::SimulateGattCharacteristicChanged(
+    BluetoothGattCharacteristic* characteristic,
+    const std::vector<uint8_t>& value) {
+  BluetoothGattCharacteristic* target_characteristic = characteristic;
+  if (target_characteristic == nullptr)
+    target_characteristic = remembered_characteristic_;
+  CHECK(target_characteristic);
+
+  win::GattCharacteristic* simulated_characteristic =
+      GetSimulatedCharacteristic(target_characteristic);
+  CHECK(simulated_characteristic);
+  fake_bt_le_wrapper_->SimulateGattCharacteristicValue(simulated_characteristic,
+                                                       value);
+  std::string device_address =
+      characteristic->GetService()->GetDevice()->GetAddress();
+  win::BLEDevice* target_device =
+      fake_bt_le_wrapper_->GetSimulatedBLEDevice(device_address);
+  CHECK(target_device);
+  win::GattService* parent_service =
+      GetSimulatedService(target_device, characteristic->GetService());
+  CHECK(parent_service);
+  fake_bt_le_wrapper_->SimulateCharacteristicValueChangeNotification(
+      parent_service, simulated_characteristic);
+
+  FinishPendingTasks();
+}
+
+void BluetoothTestWin::OnReadGattCharacteristicValue() {
   gatt_read_characteristic_attempts_++;
 }
 
-void BluetoothTestWin::OnAttemptWriteGattCharacteristic() {
+void BluetoothTestWin::OnWriteGattCharacteristicValue(
+    const PBTH_LE_GATT_CHARACTERISTIC_VALUE value) {
   gatt_write_characteristic_attempts_++;
-}
-
-void BluetoothTestWin::onWriteGattCharacteristicValue(
-    const PBTH_LE_GATT_CHARACTERISTIC_VALUE value) {
   last_write_value_.clear();
   for (ULONG i = 0; i < value->DataSize; i++)
     last_write_value_.push_back(value->Data[i]);
 }
 
+void BluetoothTestWin::OnWriteGattDescriptorValue(
+    const PBTH_LE_GATT_DESCRIPTOR_VALUE value) {
+  gatt_write_descriptor_attempts_++;
+  last_write_value_.clear();
+  for (ULONG i = 0; i < value->DataSize; i++)
+    last_write_value_.push_back(value->Data[i]);
+}
+
+void BluetoothTestWin::OnStartCharacteristicNotification() {
+  gatt_notify_characteristic_attempts_++;
+}
+
 win::GattService* BluetoothTestWin::GetSimulatedService(
     win::BLEDevice* device,
     BluetoothGattService* service) {
   CHECK(device);
   CHECK(service);
 
   std::vector<std::string> chain_of_att_handles;
   BluetoothRemoteGattServiceWin* win_service =
       static_cast<BluetoothRemoteGattServiceWin*>(service);
   chain_of_att_handles.insert(
@@ skipping 54 matching lines @@
 void BluetoothTestWin::ForceRefreshDevice() {
   adapter_win_->force_update_device_for_test_ = true;
   FinishPendingTasks();
 }
 
 void BluetoothTestWin::FinishPendingTasks() {
   bluetooth_task_runner_->RunPendingTasks();
   base::RunLoop().RunUntilIdle();
 }
 }