BluetoothAdapterMac::LowEnergyDeviceUpdated() Implemented

device/bluetooth/bluetooth_adapter_mac_unittest.mm

 // Copyright 2013 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/mac/sdk_forward_declarations.h"
 #include "base/memory/ref_counted.h"
 #include "base/test/test_simple_task_runner.h"
 #include "device/bluetooth/bluetooth_adapter.h"
 #include "device/bluetooth/bluetooth_adapter_mac.h"
 #include "device/bluetooth/bluetooth_discovery_session.h"
 #include "device/bluetooth/bluetooth_low_energy_device_mac.h"
 #include "device/bluetooth/test/mock_bluetooth_central_manager_mac.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/ocmock/OCMock/OCMock.h"
 
 #if defined(OS_IOS)
 #import <CoreBluetooth/CoreBluetooth.h>
 #else
 #import <IOBluetooth/IOBluetooth.h>
 #endif
 
 #import <Foundation/Foundation.h>
 
 namespace {
 // |kTestHashAddress| is the hash corresponding to identifier |kTestNSUUID|.
 NSString* const kTestNSUUID = @"00000000-1111-2222-3333-444444444444";
 const std::string kTestHashAddress = "D1:6F:E3:22:FD:5B";
+const int kTestRssi = 0;
 }  // namespace
 
 namespace device {
 
 class BluetoothAdapterMacTest : public testing::Test {
  public:
   BluetoothAdapterMacTest()
       : ui_task_runner_(new base::TestSimpleTaskRunner()),
         adapter_(new BluetoothAdapterMac()),
         adapter_mac_(static_cast<BluetoothAdapterMac*>(adapter_.get())),
         callback_count_(0),
         error_callback_count_(0) {
     adapter_mac_->InitForTest(ui_task_runner_);
   }
 
   // Helper methods for setup and access to BluetoothAdapterMacTest's members.
   void PollAdapter() { adapter_mac_->PollAdapter(); }
 
+  void LowEnergyDeviceUpdated(CBPeripheral* peripheral,
+                              NSDictionary* advertisementData,
+                              int rssi) {
+    adapter_mac_->LowEnergyDeviceUpdated(peripheral, advertisementData, rssi);
+  }
+
+  BluetoothDevice* GetDevice(const std::string& address) {
+    return adapter_->GetDevice(address);
+  }
+
   CBPeripheral* CreateMockPeripheral(NSString* identifier) {
     Class aClass = NSClassFromString(@"CBPeripheral");
     if (aClass == nil) {
       LOG(WARNING) << "CoreBluetooth not available, skipping unit test.";
       return nil;
     }
     id mock_peripheral =
         [[OCMockObject mockForClass:[CBPeripheral class]] retain];
     // isConnected deprecated on OSX SDK >= 10.9 and so BluetoothLowEnergyDevice
     // sends the state message instead.  We stub accordingly.
@@ skipping 179 matching lines @@
   EXPECT_EQ(0, [mock_central_manager_ stopScanCallCount]);
 }
 
 TEST_F(BluetoothAdapterMacTest, CheckGetPeripheralHashAddress) {
   base::scoped_nsobject<id> mock_peripheral(CreateMockPeripheral(kTestNSUUID));
   if (mock_peripheral.get() == nil)
     return;
   EXPECT_EQ(kTestHashAddress, GetHashAddress(mock_peripheral));
 }
 
+TEST_F(BluetoothAdapterMacTest, LowEnergyDeviceUpdatedNewDevice) {
+  base::scoped_nsobject<id> mock_peripheral(CreateMockPeripheral(kTestNSUUID));
+  if (mock_peripheral.get() == nil)
+    return;
+  base::scoped_nsobject<NSDictionary> advertisement_data(
+      CreateAdvertisementData());
+
+  EXPECT_EQ(0, NumDevices());
+  EXPECT_FALSE(DevicePresent(mock_peripheral));
+  LowEnergyDeviceUpdated(mock_peripheral, advertisement_data, kTestRssi);
+  EXPECT_EQ(1, NumDevices());
+  EXPECT_TRUE(DevicePresent(mock_peripheral));
+}
+
+TEST_F(BluetoothAdapterMacTest, LowEnergyDeviceUpdatedOldDevice) {
+  base::scoped_nsobject<id> mock_peripheral(CreateMockPeripheral(kTestNSUUID));
+  if (mock_peripheral.get() == nil)
+    return;
+  base::scoped_nsobject<NSDictionary> advertisement_data(
+      CreateAdvertisementData());
+
+  // Update the device for the first time and check it was correctly added to
+  // |devices_|.
+  EXPECT_EQ(0, NumDevices());
+  EXPECT_FALSE(DevicePresent(mock_peripheral));
+  LowEnergyDeviceUpdated(mock_peripheral, advertisement_data, kTestRssi);
+  EXPECT_EQ(1, NumDevices());
+  EXPECT_TRUE(DevicePresent(mock_peripheral));
+  // Search for the device by the address corresponding to |kTestNSUUID|.
+  BluetoothDeviceMac* device =
+      static_cast<BluetoothDeviceMac*>(GetDevice(kTestHashAddress));
+  base::scoped_nsobject<NSDate> first_update_time(
+      [device->GetLastUpdateTime() retain]);
+
+  // Update the device a second time. The device should be updated in
+  // |devices_| so check the time returned by GetLastUpdateTime() has increased.
+  LowEnergyDeviceUpdated(mock_peripheral, advertisement_data, kTestRssi);
+  EXPECT_EQ(1, NumDevices());
+  EXPECT_TRUE(DevicePresent(mock_peripheral));
+  device = static_cast<BluetoothDeviceMac*>(GetDevice(kTestHashAddress));
+  EXPECT_TRUE([device->GetLastUpdateTime() compare:first_update_time] ==
+              NSOrderedDescending);
+}
+
 TEST_F(BluetoothAdapterMacTest, UpdateDevicesRemovesLowEnergyDevice) {
   base::scoped_nsobject<id> mock_peripheral(CreateMockPeripheral(kTestNSUUID));
   if (mock_peripheral.get() == nil)
     return;
   base::scoped_nsobject<NSDictionary> advertisement_data(
       CreateAdvertisementData());
 
-  BluetoothLowEnergyDeviceMac* device =
-      new BluetoothLowEnergyDeviceMac(mock_peripheral, advertisement_data, 0);
+  BluetoothLowEnergyDeviceMac* device = new BluetoothLowEnergyDeviceMac(
+      mock_peripheral, advertisement_data, kTestRssi);
   SetDeviceTimeGreaterThanTimeout(device);
 
   EXPECT_EQ(0, NumDevices());
   AddLowEnergyDevice(device);
   EXPECT_EQ(1, NumDevices());
   EXPECT_TRUE(DevicePresent(mock_peripheral));
 
   // Check that object pointed to by |device| is deleted by the adapter.
   RemoveTimedOutDevices();
   EXPECT_EQ(0, NumDevices());
   EXPECT_FALSE(DevicePresent(mock_peripheral));
 }
 
 }  // namespace device