| // Copyright 2017 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/fido/u2f_ble_discovery.h" |
| |
| #include <string> |
| |
| #include "base/bind.h" |
| #include "base/run_loop.h" |
| #include "build/build_config.h" |
| #include "device/bluetooth/test/bluetooth_test.h" |
| #include "device/fido/mock_u2f_discovery.h" |
| #include "device/fido/u2f_ble_device.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| #if defined(OS_ANDROID) |
| #include "device/bluetooth/test/bluetooth_test_android.h" |
| #elif defined(OS_MACOSX) |
| #include "device/bluetooth/test/bluetooth_test_mac.h" |
| #elif defined(OS_WIN) |
| #include "device/bluetooth/test/bluetooth_test_win.h" |
| #elif defined(OS_CHROMEOS) || defined(OS_LINUX) |
| #include "device/bluetooth/test/bluetooth_test_bluez.h" |
| #endif |
| |
| using ::testing::_; |
| |
| namespace device { |
| |
| ACTION_P(ReturnFromAsyncCall, closure) { |
| closure.Run(); |
| } |
| |
| MATCHER_P(IdMatches, id, "") { |
| return arg->GetId() == std::string("ble:") + id; |
| } |
| |
| TEST_F(BluetoothTest, U2fBleDiscoveryNoAdapter) { |
| // We purposefully construct a temporary and provide no fake adapter, |
| // simulating cases where the discovery is destroyed before obtaining a handle |
| // to an adapter. This should be handled gracefully and not result in a crash. |
| U2fBleDiscovery discovery; |
| |
| // We don't expect any calls to the notification methods. |
| MockU2fDiscoveryObserver observer; |
| discovery.AddObserver(&observer); |
| EXPECT_CALL(observer, DiscoveryStarted(&discovery, _)).Times(0); |
| EXPECT_CALL(observer, DiscoveryStopped(&discovery, _)).Times(0); |
| EXPECT_CALL(observer, DeviceAdded(&discovery, _)).Times(0); |
| EXPECT_CALL(observer, DeviceRemoved(&discovery, _)).Times(0); |
| } |
| |
| TEST_F(BluetoothTest, U2fBleDiscoveryFindsKnownDevice) { |
| if (!PlatformSupportsLowEnergy()) { |
| LOG(WARNING) << "Low Energy Bluetooth unavailable, skipping unit test."; |
| return; |
| } |
| InitWithFakeAdapter(); |
| |
| SimulateLowEnergyDevice(4); // This device should be ignored. |
| SimulateLowEnergyDevice(7); |
| |
| U2fBleDiscovery discovery; |
| MockU2fDiscoveryObserver observer; |
| discovery.AddObserver(&observer); |
| |
| { |
| base::RunLoop run_loop; |
| auto quit = run_loop.QuitClosure(); |
| EXPECT_CALL(observer, |
| DeviceAdded(&discovery, |
| IdMatches(BluetoothTestBase::kTestDeviceAddress1))); |
| EXPECT_CALL(observer, DiscoveryStarted(&discovery, true)) |
| .WillOnce(ReturnFromAsyncCall(quit)); |
| |
| discovery.Start(); |
| run_loop.Run(); |
| } |
| |
| // TODO(crbug/763303): Delete device and check OnDeviceDeleted invocation. |
| |
| { |
| base::RunLoop run_loop; |
| auto quit = run_loop.QuitClosure(); |
| |
| EXPECT_CALL(observer, DiscoveryStopped(&discovery, true)) |
| .WillOnce(ReturnFromAsyncCall(quit)); |
| |
| discovery.Stop(); |
| run_loop.Run(); |
| } |
| } |
| |
| TEST_F(BluetoothTest, U2fBleDiscoveryFindsNewDevice) { |
| if (!PlatformSupportsLowEnergy()) { |
| LOG(WARNING) << "Low Energy Bluetooth unavailable, skipping unit test."; |
| return; |
| } |
| InitWithFakeAdapter(); |
| |
| U2fBleDiscovery discovery; |
| MockU2fDiscoveryObserver observer; |
| discovery.AddObserver(&observer); |
| |
| { |
| base::RunLoop run_loop; |
| auto quit = run_loop.QuitClosure(); |
| EXPECT_CALL(observer, DiscoveryStarted(&discovery, true)) |
| .WillOnce(ReturnFromAsyncCall(quit)); |
| |
| discovery.Start(); |
| run_loop.Run(); |
| } |
| |
| { |
| base::RunLoop run_loop; |
| auto quit = run_loop.QuitClosure(); |
| EXPECT_CALL(observer, |
| DeviceAdded(&discovery, |
| IdMatches(BluetoothTestBase::kTestDeviceAddress1))) |
| .WillOnce(ReturnFromAsyncCall(quit)); |
| |
| SimulateLowEnergyDevice(4); // This device should be ignored. |
| SimulateLowEnergyDevice(7); |
| |
| run_loop.Run(); |
| } |
| |
| // TODO(crbug/763303): Delete device and check OnDeviceDeleted invocation. |
| |
| { |
| base::RunLoop run_loop; |
| auto quit = run_loop.QuitClosure(); |
| EXPECT_CALL(observer, DiscoveryStopped(&discovery, true)) |
| .WillOnce(ReturnFromAsyncCall(quit)); |
| discovery.Stop(); |
| run_loop.Run(); |
| } |
| } |
| |
| // Simulate the scenario where the BLE device is already known at start-up time, |
| // but no service advertisements have been received from the device yet, so we |
| // do not know if it is a U2F device or not. As soon as it is discovered that |
| // the device supports the U2F service, the observer should be notified of a new |
| // U2fBleDevice. |
| TEST_F(BluetoothTest, U2fBleDiscoveryFindsUpdatedDevice) { |
| if (!PlatformSupportsLowEnergy()) { |
| LOG(WARNING) << "Low Energy Bluetooth unavailable, skipping unit test."; |
| return; |
| } |
| InitWithFakeAdapter(); |
| |
| SimulateLowEnergyDevice(3); |
| |
| U2fBleDiscovery discovery; |
| MockU2fDiscoveryObserver observer; |
| discovery.AddObserver(&observer); |
| |
| { |
| base::RunLoop run_loop; |
| auto quit = run_loop.QuitClosure(); |
| EXPECT_CALL(observer, DiscoveryStarted(&discovery, true)) |
| .WillOnce(ReturnFromAsyncCall(quit)); |
| |
| discovery.Start(); |
| run_loop.Run(); |
| |
| EXPECT_THAT(discovery.GetDevices(), ::testing::IsEmpty()); |
| } |
| |
| { |
| base::RunLoop run_loop; |
| auto quit = run_loop.QuitClosure(); |
| EXPECT_CALL(observer, |
| DeviceAdded(&discovery, |
| IdMatches(BluetoothTestBase::kTestDeviceAddress1))) |
| .WillOnce(ReturnFromAsyncCall(quit)); |
| |
| // This will update properties for device 3. |
| SimulateLowEnergyDevice(7); |
| |
| run_loop.Run(); |
| |
| const auto devices = discovery.GetDevices(); |
| ASSERT_THAT(devices, ::testing::SizeIs(1u)); |
| EXPECT_EQ(U2fBleDevice::GetId(BluetoothTestBase::kTestDeviceAddress1), |
| devices[0]->GetId()); |
| } |
| |
| // TODO(crbug/763303): Delete device and check OnDeviceDeleted invocation. |
| |
| { |
| base::RunLoop run_loop; |
| auto quit = run_loop.QuitClosure(); |
| EXPECT_CALL(observer, DiscoveryStopped(&discovery, true)) |
| .WillOnce(ReturnFromAsyncCall(quit)); |
| discovery.Stop(); |
| run_loop.Run(); |
| } |
| } |
| |
| } // namespace device |