Update unit tests for BluetoothDevice::GetPrimaryServicesByUUID() etc.

device/bluetooth/bluetooth_remote_gatt_service_unittest.cc

 // 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.
 
 #include "base/run_loop.h"
 #include "build/build_config.h"
 #include "device/bluetooth/bluetooth_gatt_service.h"
 #include "device/bluetooth/bluetooth_remote_gatt_characteristic.h"
 #include "device/bluetooth/test/test_bluetooth_adapter_observer.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 155 matching lines @@
   EXPECT_EQ(char_uuid3, service->GetCharacteristic(char_id3)->GetUUID());
   EXPECT_EQ(char_uuid4, service->GetCharacteristic(char_id4)->GetUUID());
 
   // GetCharacteristics & GetCharacteristic return the same object for the same
   // ID:
   EXPECT_EQ(service->GetCharacteristics()[0],
             service->GetCharacteristic(char_id1));
   EXPECT_EQ(service->GetCharacteristic(char_id1),
             service->GetCharacteristic(char_id1));
 }
+
+TEST_F(BluetoothRemoteGattServiceTest, GetCharacteristicsByUUID) {
+  if (!PlatformSupportsLowEnergy()) {
+    LOG(WARNING) << "Low Energy Bluetooth unavailable, skipping unit test.";
+    return;
+  }
+  InitWithFakeAdapter();
+  StartLowEnergyDiscoverySession();
+  BluetoothDevice* device = SimulateLowEnergyDevice(3);
+  device->CreateGattConnection(GetGattConnectionCallback(Call::EXPECTED),
+                               GetConnectErrorCallback(Call::NOT_EXPECTED));
+  SimulateGattConnection(device);
+  base::RunLoop().RunUntilIdle();
+
+  // Simulate two primary GATT services.
+  std::vector<std::string> services;
+  services.push_back("00000000-0000-1000-8000-00805f9b34fb");
+  services.push_back("01010101-0101-1000-8000-00805f9b34fb");
+  SimulateGattServicesDiscovered(device, services);
+  base::RunLoop().RunUntilIdle();
+  BluetoothRemoteGattService* service1 = device->GetGattServices()[0];
+  BluetoothRemoteGattService* service2 = device->GetGattServices()[1];
+  std::string characteristic_uuid1 = "11111111-0000-1000-8000-00805f9b34fb";
+  std::string characteristic_uuid2 = "22222222-0000-1000-8000-00805f9b34fb";
+  SimulateGattCharacteristic(service1, characteristic_uuid1,
+                             /* properties */ 0);
+  // 2 duplicate UUIDs creating 2 instances.
+  SimulateGattCharacteristic(service2, characteristic_uuid2,
+                             /* properties */ 0);
+  SimulateGattCharacteristic(service2, characteristic_uuid2,
+                             /* properties */ 0);
+
+  {
+    std::vector<BluetoothRemoteGattCharacteristic*> characteristics =
+        service1->GetCharacteristicsByUUID(BluetoothUUID(characteristic_uuid1));
+    EXPECT_EQ(1u, characteristics.size());
+    EXPECT_EQ(characteristic_uuid1,
+              characteristics[0]->GetUUID().canonical_value());
+  }
+
+  {
+    std::vector<BluetoothRemoteGattCharacteristic*> characteristics =
+        service2->GetCharacteristicsByUUID(BluetoothUUID(characteristic_uuid2));
+    EXPECT_EQ(2u, characteristics.size());
+    EXPECT_EQ(characteristic_uuid2,
+              characteristics[0]->GetUUID().canonical_value());
+    EXPECT_EQ(characteristic_uuid2,
+              characteristics[1]->GetUUID().canonical_value());
+    EXPECT_NE(characteristics[0]->GetIdentifier(),
+              characteristics[1]->GetIdentifier());
+  }
+
+  BluetoothUUID characteristic_uuid_not_exist_in_setup(
+      "33333333-0000-1000-8000-00805f9b34fb");
+  EXPECT_TRUE(
+      service1->GetCharacteristicsByUUID(characteristic_uuid_not_exist_in_setup)
+          .empty());
+  EXPECT_TRUE(
+      service2->GetCharacteristicsByUUID(characteristic_uuid_not_exist_in_setup)
+          .empty());
+}
 #endif  // defined(OS_ANDROID) || defined(OS_MACOSX) || defined(OS_WIN)
 
 #if defined(OS_MACOSX) || defined(OS_WIN)
 TEST_F(BluetoothRemoteGattServiceTest, GattCharacteristics_ObserversCalls) {
   if (!PlatformSupportsLowEnergy()) {
     LOG(WARNING) << "Low Energy Bluetooth unavailable, skipping unit test.";
     return;
   }
   InitWithFakeAdapter();
   StartLowEnergyDiscoverySession();
@@ skipping 91 matching lines @@
   std::string removed_service = service1->GetIdentifier();
   SimulateGattServiceRemoved(device->GetGattService(removed_service));
   EXPECT_EQ(1, observer.gatt_service_removed_count());
   EXPECT_EQ(1u, device->GetGattServices().size());
   EXPECT_FALSE(device->GetGattService(removed_service));
   EXPECT_EQ(device->GetGattServices()[0], service2);
 }
 #endif  // defined(OS_WIN) || defined(OS_MACOSX)
 
 }  // namespace device