| OLD | NEW |
|---|
| (Empty) |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "components/proximity_auth/ble/bluetooth_low_energy_device_whitelist.h" | |
| 6 | |
| 7 #include <algorithm> | |
| 8 #include <string> | |
| 9 #include <vector> | |
| 10 | |
| 11 #include "base/macros.h" | |
| 12 #include "components/prefs/testing_pref_service.h" | |
| 13 #include "components/proximity_auth/ble/pref_names.h" | |
| 14 #include "testing/gmock/include/gmock/gmock.h" | |
| 15 #include "testing/gtest/include/gtest/gtest.h" | |
| 16 | |
| 17 namespace proximity_auth { | |
| 18 namespace { | |
| 19 | |
| 20 const char kBluetoothAddress1[] = "11:22:33:44:55:66"; | |
| 21 const char kPublicKey1[] = "public key 1"; | |
| 22 | |
| 23 const char kBluetoothAddress2[] = "22:33:44:55:66:77"; | |
| 24 const char kPublicKey2[] = "public key 2"; | |
| 25 | |
| 26 } // namespace | |
| 27 | |
| 28 class ProximityAuthBluetoothLowEnergyDeviceWhitelistTest | |
| 29 : public testing::Test { | |
| 30 protected: | |
| 31 ProximityAuthBluetoothLowEnergyDeviceWhitelistTest() {} | |
| 32 | |
| 33 void SetUp() override { | |
| 34 BluetoothLowEnergyDeviceWhitelist::RegisterPrefs(pref_service_.registry()); | |
| 35 } | |
| 36 | |
| 37 void CheckWhitelistedDevice( | |
| 38 const std::string& bluetooth_address, | |
| 39 const std::string& public_key, | |
| 40 BluetoothLowEnergyDeviceWhitelist& device_whitelist) { | |
| 41 EXPECT_TRUE(device_whitelist.HasDeviceWithAddress(bluetooth_address)); | |
| 42 EXPECT_EQ(device_whitelist.GetDevicePublicKey(bluetooth_address), | |
| 43 public_key); | |
| 44 | |
| 45 EXPECT_TRUE(device_whitelist.HasDeviceWithPublicKey(public_key)); | |
| 46 EXPECT_EQ(device_whitelist.GetDeviceAddress(public_key), bluetooth_address); | |
| 47 } | |
| 48 | |
| 49 TestingPrefServiceSimple pref_service_; | |
| 50 | |
| 51 private: | |
| 52 DISALLOW_COPY_AND_ASSIGN(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest); | |
| 53 }; | |
| 54 | |
| 55 TEST_F(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest, RegisterPrefs) { | |
| 56 TestingPrefServiceSimple pref_service; | |
| 57 BluetoothLowEnergyDeviceWhitelist::RegisterPrefs(pref_service.registry()); | |
| 58 EXPECT_TRUE( | |
| 59 pref_service.FindPreference(prefs::kBluetoothLowEnergyDeviceWhitelist)); | |
| 60 } | |
| 61 | |
| 62 TEST_F(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest, AddDevice) { | |
| 63 BluetoothLowEnergyDeviceWhitelist device_whitelist(&pref_service_); | |
| 64 | |
| 65 device_whitelist.AddOrUpdateDevice(kBluetoothAddress1, kPublicKey1); | |
| 66 CheckWhitelistedDevice(kBluetoothAddress1, kPublicKey1, device_whitelist); | |
| 67 } | |
| 68 | |
| 69 TEST_F(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest, | |
| 70 AddDevice_TwoDevicesWithSameAddress) { | |
| 71 BluetoothLowEnergyDeviceWhitelist device_whitelist(&pref_service_); | |
| 72 | |
| 73 device_whitelist.AddOrUpdateDevice(kBluetoothAddress1, kPublicKey1); | |
| 74 CheckWhitelistedDevice(kBluetoothAddress1, kPublicKey1, device_whitelist); | |
| 75 | |
| 76 device_whitelist.AddOrUpdateDevice(kBluetoothAddress1, kPublicKey2); | |
| 77 CheckWhitelistedDevice(kBluetoothAddress1, kPublicKey2, device_whitelist); | |
| 78 | |
| 79 EXPECT_FALSE(device_whitelist.HasDeviceWithPublicKey(kPublicKey1)); | |
| 80 } | |
| 81 | |
| 82 TEST_F(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest, | |
| 83 AddDevice_TwoDevicesWithSamePublicKey) { | |
| 84 BluetoothLowEnergyDeviceWhitelist device_whitelist(&pref_service_); | |
| 85 | |
| 86 device_whitelist.AddOrUpdateDevice(kBluetoothAddress1, kPublicKey1); | |
| 87 CheckWhitelistedDevice(kBluetoothAddress1, kPublicKey1, device_whitelist); | |
| 88 | |
| 89 device_whitelist.AddOrUpdateDevice(kBluetoothAddress2, kPublicKey1); | |
| 90 CheckWhitelistedDevice(kBluetoothAddress2, kPublicKey1, device_whitelist); | |
| 91 | |
| 92 EXPECT_FALSE(device_whitelist.HasDeviceWithAddress(kBluetoothAddress1)); | |
| 93 } | |
| 94 | |
| 95 TEST_F(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest, | |
| 96 RemoveDeviceWithAddress) { | |
| 97 BluetoothLowEnergyDeviceWhitelist device_whitelist(&pref_service_); | |
| 98 | |
| 99 device_whitelist.AddOrUpdateDevice(kBluetoothAddress1, kPublicKey1); | |
| 100 CheckWhitelistedDevice(kBluetoothAddress1, kPublicKey1, device_whitelist); | |
| 101 | |
| 102 ASSERT_TRUE(device_whitelist.RemoveDeviceWithAddress(kBluetoothAddress1)); | |
| 103 EXPECT_FALSE(device_whitelist.HasDeviceWithAddress(kBluetoothAddress1)); | |
| 104 EXPECT_FALSE(device_whitelist.HasDeviceWithPublicKey(kPublicKey1)); | |
| 105 } | |
| 106 | |
| 107 TEST_F(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest, | |
| 108 RemoveDeviceWithAddress_DeviceNotPresent) { | |
| 109 BluetoothLowEnergyDeviceWhitelist device_whitelist(&pref_service_); | |
| 110 | |
| 111 device_whitelist.AddOrUpdateDevice(kBluetoothAddress1, kPublicKey1); | |
| 112 CheckWhitelistedDevice(kBluetoothAddress1, kPublicKey1, device_whitelist); | |
| 113 | |
| 114 ASSERT_FALSE(device_whitelist.RemoveDeviceWithAddress(kBluetoothAddress2)); | |
| 115 EXPECT_TRUE(device_whitelist.HasDeviceWithAddress(kBluetoothAddress1)); | |
| 116 EXPECT_TRUE(device_whitelist.HasDeviceWithPublicKey(kPublicKey1)); | |
| 117 } | |
| 118 | |
| 119 TEST_F(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest, | |
| 120 RemoveDeviceWithPublicKey) { | |
| 121 BluetoothLowEnergyDeviceWhitelist device_whitelist(&pref_service_); | |
| 122 | |
| 123 device_whitelist.AddOrUpdateDevice(kBluetoothAddress1, kPublicKey1); | |
| 124 CheckWhitelistedDevice(kBluetoothAddress1, kPublicKey1, device_whitelist); | |
| 125 | |
| 126 ASSERT_TRUE(device_whitelist.RemoveDeviceWithPublicKey(kPublicKey1)); | |
| 127 EXPECT_FALSE(device_whitelist.HasDeviceWithAddress(kBluetoothAddress1)); | |
| 128 EXPECT_FALSE(device_whitelist.HasDeviceWithPublicKey(kPublicKey1)); | |
| 129 } | |
| 130 | |
| 131 TEST_F(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest, | |
| 132 RemoveDeviceWithPublicKey_DeviceNotPresent) { | |
| 133 BluetoothLowEnergyDeviceWhitelist device_whitelist(&pref_service_); | |
| 134 | |
| 135 device_whitelist.AddOrUpdateDevice(kBluetoothAddress1, kPublicKey1); | |
| 136 CheckWhitelistedDevice(kBluetoothAddress1, kPublicKey1, device_whitelist); | |
| 137 | |
| 138 ASSERT_FALSE(device_whitelist.RemoveDeviceWithPublicKey(kPublicKey2)); | |
| 139 EXPECT_TRUE(device_whitelist.HasDeviceWithAddress(kBluetoothAddress1)); | |
| 140 EXPECT_TRUE(device_whitelist.HasDeviceWithPublicKey(kPublicKey1)); | |
| 141 } | |
| 142 | |
| 143 TEST_F(ProximityAuthBluetoothLowEnergyDeviceWhitelistTest, GetPublicKeys) { | |
| 144 BluetoothLowEnergyDeviceWhitelist device_whitelist(&pref_service_); | |
| 145 | |
| 146 device_whitelist.AddOrUpdateDevice(kBluetoothAddress1, kPublicKey1); | |
| 147 CheckWhitelistedDevice(kBluetoothAddress1, kPublicKey1, device_whitelist); | |
| 148 | |
| 149 device_whitelist.AddOrUpdateDevice(kBluetoothAddress2, kPublicKey2); | |
| 150 CheckWhitelistedDevice(kBluetoothAddress2, kPublicKey2, device_whitelist); | |
| 151 | |
| 152 std::vector<std::string> public_keys = device_whitelist.GetPublicKeys(); | |
| 153 | |
| 154 // Note: it's not guarantee that the order in |public_key| is the same as | |
| 155 // insertion, so directly comparing vectors would not work. | |
| 156 EXPECT_TRUE(public_keys.size() == 2); | |
| 157 EXPECT_TRUE(std::find(public_keys.begin(), public_keys.end(), kPublicKey1) != | |
| 158 public_keys.end()); | |
| 159 EXPECT_TRUE(std::find(public_keys.begin(), public_keys.end(), kPublicKey2) != | |
| 160 public_keys.end()); | |
| 161 } | |
| 162 | |
| 163 } // namespace proximity_auth | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 2843443003:
[EasyUnlock] Delete unused BLE code. (Closed)
