Reland: Bluetooth: Add service data unit tests for chromeos

device/bluetooth/bluetooth_device_unittest.cc

 // Copyright 2014 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/bluetooth_device.h"
 
 #include <stddef.h>
 
 #include "base/macros.h"
 #include "base/run_loop.h"
@@ skipping 149 matching lines @@
     return;
   }
   InitWithFakeAdapter();
   StartLowEnergyDiscoverySession();
   BluetoothDevice* device = SimulateLowEnergyDevice(3);
   ASSERT_TRUE(device);
   UUIDSet uuids = device->GetUUIDs();
   EXPECT_EQ(0u, uuids.size());
 }
 
-#if defined(OS_MACOSX)
+#if defined(OS_MACOSX) || defined(OS_CHROMEOS) || defined(OS_LINUX)
 // TODO(ortuno): Enable on Android once it supports Service Data.
 // http://crbug.com/639408
 TEST_F(BluetoothTest, GetServiceDataUUIDs_GetServiceDataForUUID) {
   if (!PlatformSupportsLowEnergy()) {
     LOG(WARNING) << "Low Energy Bluetooth unavailable, skipping unit test.";
     return;
   }
   InitWithFakeAdapter();
+
+#if !defined(OS_LINUX) && !defined(OS_CHROMEOS)
+  // TODO(crbug.com/706043): Remove #if once StartLowEnergyDiscoverySession is
+  // implemented for bluez.
   StartLowEnergyDiscoverySession();
+#endif  // !defined(OS_LINUX) && !defined(OS_CHROMEOS)
+
+  // Receive Advertisement with empty service data.
+  BluetoothDevice* device1 = SimulateLowEnergyDevice(4);
+  EXPECT_TRUE(device1->GetServiceData().empty());
+  EXPECT_TRUE(device1->GetServiceDataUUIDs().empty());
 
   // Receive Advertisement with service data.
-  BluetoothDevice* device = SimulateLowEnergyDevice(1);
+  BluetoothDevice* device2 = SimulateLowEnergyDevice(1);
 
   EXPECT_EQ(ServiceDataMap({{BluetoothUUID(kTestUUIDHeartRate), {1}}}),
-            device->GetServiceData());
+            device2->GetServiceData());
   EXPECT_EQ(UUIDSet({BluetoothUUID(kTestUUIDHeartRate)}),
-            device->GetServiceDataUUIDs());
+            device2->GetServiceDataUUIDs());
   EXPECT_EQ(std::vector<uint8_t>({1}),
-            *device->GetServiceDataForUUID(BluetoothUUID(kTestUUIDHeartRate)));
+            *device2->GetServiceDataForUUID(BluetoothUUID(kTestUUIDHeartRate)));
 
   // Receive Advertisement with no service data.
   SimulateLowEnergyDevice(3);
 
-  EXPECT_TRUE(device->GetServiceData().empty());
-  EXPECT_TRUE(device->GetServiceDataUUIDs().empty());
+// TODO(crbug.com/707039): Remove #if once the BlueZ caching behavior is
+// changed.
+#if defined(OS_CHROMEOS) || defined(OS_LINUX)
+  // On ChromeOS and Linux, BlueZ persists all service data meaning if
+  // a device stops advertising service data for a UUID, BlueZ will
+  // still return the cached value for that UUID.
+  EXPECT_EQ(ServiceDataMap({{BluetoothUUID(kTestUUIDHeartRate), {1}}}),
+            device2->GetServiceData());
+  EXPECT_EQ(UUIDSet({BluetoothUUID(kTestUUIDHeartRate)}),
+            device2->GetServiceDataUUIDs());
+  EXPECT_EQ(std::vector<uint8_t>({1}),
+            *device2->GetServiceDataForUUID(BluetoothUUID(kTestUUIDHeartRate)));
+#else
+  EXPECT_TRUE(device2->GetServiceData().empty());
+  EXPECT_TRUE(device2->GetServiceDataUUIDs().empty());
   EXPECT_EQ(nullptr,
-            device->GetServiceDataForUUID(BluetoothUUID(kTestUUIDHeartRate)));
+            device2->GetServiceDataForUUID(BluetoothUUID(kTestUUIDHeartRate)));
+#endif
 
   // Receive Advertisement with new service data.
   SimulateLowEnergyDevice(2);
 
-  EXPECT_EQ(ServiceDataMap({{BluetoothUUID(kTestUUIDHeartRate), {2}},
-                            {BluetoothUUID(kTestUUIDImmediateAlert), {0}}}),
-            device->GetServiceData());
+  EXPECT_EQ(ServiceDataMap(
+                {{BluetoothUUID(kTestUUIDHeartRate), std::vector<uint8_t>({})},
+                 {BluetoothUUID(kTestUUIDImmediateAlert), {0, 2}}}),
+            device2->GetServiceData());
   EXPECT_EQ(UUIDSet({BluetoothUUID(kTestUUIDHeartRate),
                      BluetoothUUID(kTestUUIDImmediateAlert)}),
-            device->GetServiceDataUUIDs());
-  EXPECT_EQ(std::vector<uint8_t>({2}),
-            *device->GetServiceDataForUUID(BluetoothUUID(kTestUUIDHeartRate)));
+            device2->GetServiceDataUUIDs());
+  EXPECT_EQ(std::vector<uint8_t>({}),
+            *device2->GetServiceDataForUUID(BluetoothUUID(kTestUUIDHeartRate)));
   EXPECT_EQ(
-      std::vector<uint8_t>({0}),
-      *device->GetServiceDataForUUID(BluetoothUUID(kTestUUIDImmediateAlert)));
+      std::vector<uint8_t>({0, 2}),
+      *device2->GetServiceDataForUUID(BluetoothUUID(kTestUUIDImmediateAlert)));
 
+#if !defined(OS_LINUX) && !defined(OS_CHROMEOS)
+  // TODO(crbug.com/706043): Remove #if once StartLowEnergyDiscoverySession is
+  // implemented for bluez.
   // Stop discovery.
   discovery_sessions_[0]->Stop(GetCallback(Call::EXPECTED),
                                GetErrorCallback(Call::NOT_EXPECTED));
   ASSERT_FALSE(adapter_->IsDiscovering());
   ASSERT_FALSE(discovery_sessions_[0]->IsActive());
 
-  EXPECT_TRUE(device->GetServiceData().empty());
-  EXPECT_TRUE(device->GetServiceDataUUIDs().empty());
+  EXPECT_TRUE(device2->GetServiceData().empty());
+  EXPECT_TRUE(device2->GetServiceDataUUIDs().empty());
   EXPECT_EQ(nullptr,
-            device->GetServiceDataForUUID(BluetoothUUID(kTestUUIDHeartRate)));
-  EXPECT_EQ(nullptr, device->GetServiceDataForUUID(
+            device2->GetServiceDataForUUID(BluetoothUUID(kTestUUIDHeartRate)));
+  EXPECT_EQ(nullptr, device2->GetServiceDataForUUID(
                          BluetoothUUID(kTestUUIDImmediateAlert)));
+#endif  // !defined(OS_LINUX) && !defined(OS_CHROMEOS)
 }
-#endif  // defined(OS_MACOSX)
+#endif  // defined(OS_MACOSX) || defined(OS_CHROMEOS) || defined(OS_LINUX)
 
 #if defined(OS_ANDROID) || defined(OS_MACOSX)
 // Tests that the Advertisement Data fields are correctly updated during
 // discovery.
 TEST_F(BluetoothTest, AdvertisementData_Discovery) {
   if (!PlatformSupportsLowEnergy()) {
     LOG(WARNING) << "Low Energy Bluetooth unavailable, skipping unit test.";
     return;
   }
   InitWithFakeAdapter();
@@ skipping 49 matching lines @@
   SimulateLowEnergyDevice(2);
   EXPECT_EQ(2, observer.device_changed_count());
 
   EXPECT_EQ(ToInt8(TestRSSI::LOWER), device->GetInquiryRSSI().value());
   EXPECT_EQ(UUIDSet({BluetoothUUID(kTestUUIDImmediateAlert),
                      BluetoothUUID(kTestUUIDLinkLoss)}),
             device->GetUUIDs());
 #if defined(OS_MACOSX)
   // TODO(ortuno): Enable on Android once it supports Service Data.
   // http://crbug.com/639408
-  EXPECT_EQ(ServiceDataMap({{BluetoothUUID(kTestUUIDHeartRate), {2}},
-                            {BluetoothUUID(kTestUUIDImmediateAlert), {0}}}),
+  EXPECT_EQ(ServiceDataMap(
+                {{BluetoothUUID(kTestUUIDHeartRate), std::vector<uint8_t>({})},
+                 {BluetoothUUID(kTestUUIDImmediateAlert), {0, 2}}}),
             device->GetServiceData());
 #endif  // defined(OS_MACOSX)
   EXPECT_EQ(ToInt8(TestTxPower::LOWER), device->GetInquiryTxPower().value());
 
   // Stop discovery session, should notify of device changed.
   //  - GetInquiryRSSI: Should return nullopt because we are no longer
   //    discovering.
   //  - GetUUIDs: Should not return any UUIDs.
   //  - GetServiceData: Should return empty map.
   //  - GetInquiryTxPower: Should return nullopt because we are no longer
@@ skipping 295 matching lines @@
 
   EXPECT_EQ(3, observer.device_changed_count());
   EXPECT_EQ(ToInt8(TestRSSI::LOWER), device->GetInquiryRSSI().value());
   EXPECT_EQ(UUIDSet({BluetoothUUID(kTestUUIDLinkLoss),
                      BluetoothUUID(kTestUUIDImmediateAlert),
                      BluetoothUUID(kTestUUIDHeartRate)}),
             device->GetUUIDs());
 #if defined(OS_MACOSX)
   // TODO(ortuno): Enable on Android once it supports Service Data.
   // http://crbug.com/639408
-  EXPECT_EQ(ServiceDataMap({{BluetoothUUID(kTestUUIDHeartRate), {2}},
-                            {BluetoothUUID(kTestUUIDImmediateAlert), {0}}}),
+  EXPECT_EQ(ServiceDataMap(
+                {{BluetoothUUID(kTestUUIDHeartRate), std::vector<uint8_t>({})},
+                 {BluetoothUUID(kTestUUIDImmediateAlert), {0, 2}}}),
             device->GetServiceData());
 #endif  // defined(OS_MACOSX)
   EXPECT_EQ(ToInt8(TestTxPower::LOWER), device->GetInquiryTxPower().value());
 
   // Stop discovery session, should notify of device changed.
   //  - GetInquiryRSSI: Should return nullopt because we are no longer
   //    discovering.
   //  - GetUUIDs: Should only return Service UUIDs.
   //  - GetServiceData: Should return an empty map since we are no longer
   //    discovering.
@@ skipping 84 matching lines @@
   device = SimulateLowEnergyDevice(2);
 
   EXPECT_EQ(1, observer.device_changed_count());
   EXPECT_EQ(ToInt8(TestRSSI::LOWER), device->GetInquiryRSSI().value());
   EXPECT_EQ(UUIDSet({BluetoothUUID(kTestUUIDLinkLoss),
                      BluetoothUUID(kTestUUIDImmediateAlert)}),
             device->GetUUIDs());
 #if defined(OS_MACOSX)
   // TODO(ortuno): Enable on Android once it supports Service Data.
   // http://crbug.com/639408
-  EXPECT_EQ(ServiceDataMap({{BluetoothUUID(kTestUUIDHeartRate), {2}},
-                            {BluetoothUUID(kTestUUIDImmediateAlert), {0}}}),
+  EXPECT_EQ(ServiceDataMap(
+                {{BluetoothUUID(kTestUUIDHeartRate), std::vector<uint8_t>({})},
+                 {BluetoothUUID(kTestUUIDImmediateAlert), {0, 2}}}),
             device->GetServiceData());
 #endif  // defined(OS_MACOSX)
   EXPECT_EQ(ToInt8(TestTxPower::LOWER), device->GetInquiryTxPower().value());
 
   // Discover Services, should notify of device changed.
   //  - GetUUIDs: Should return Advertised UUIDs and Service UUIDs.
   std::vector<std::string> services;
   services.push_back(kTestUUIDHeartRate);
   SimulateGattServicesDiscovered(device, services);
   base::RunLoop().RunUntilIdle();
@@ skipping 17 matching lines @@
 
   EXPECT_EQ(3, observer.device_changed_count());
   EXPECT_EQ(ToInt8(TestRSSI::LOWER), device->GetInquiryRSSI().value());
   EXPECT_EQ(UUIDSet({BluetoothUUID(kTestUUIDLinkLoss),
                      BluetoothUUID(kTestUUIDImmediateAlert)}),
             device->GetUUIDs());
 
 #if defined(OS_MACOSX)
   // TODO(ortuno): Enable on Android once it supports Service Data.
   // http://crbug.com/639408
-  EXPECT_EQ(ServiceDataMap({{BluetoothUUID(kTestUUIDHeartRate), {2}},
-                            {BluetoothUUID(kTestUUIDImmediateAlert), {0}}}),
+  EXPECT_EQ(ServiceDataMap(
+                {{BluetoothUUID(kTestUUIDHeartRate), std::vector<uint8_t>({})},
+                 {BluetoothUUID(kTestUUIDImmediateAlert), {0, 2}}}),
             device->GetServiceData());
 #endif  // defined(OS_MACOSX)
   EXPECT_EQ(ToInt8(TestTxPower::LOWER), device->GetInquiryTxPower().value());
 
   // Receive Advertisement with new UUIDs, should notify of device changed.
   //  - GetInquiryRSSI: Should return last packet's rssi.
   //  - GetUUIDs: Should return only new Advertised UUIDs.
   //  - GetServiceData: Should return only new advertised Service Data.
   //  - GetInquiryTxPower: Should return the last packet's advertised Tx Power.
   device = SimulateLowEnergyDevice(1);
@@ skipping 842 matching lines @@
                   .empty());
   EXPECT_TRUE(device_
                   ->GetCharacteristicsByUUID(
                       service_instance_id2,
                       BluetoothUUID(characteristic_uuid_not_exist_in_setup))
                   .empty());
 }
 #endif  // defined(OS_ANDROID) || defined(OS_MACOSX)
 
 }  // namespace device