Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 1 | // Copyright (c) 2013 The Chromium OS 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 "shill/scan_session.h" |
| 6 | |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 7 | #include <errno.h> |
| 8 | |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 9 | #include <limits> |
| 10 | #include <set> |
| 11 | #include <vector> |
| 12 | |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 13 | #include <base/memory/weak_ptr.h> |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 14 | #include <gmock/gmock.h> |
| 15 | #include <gtest/gtest.h> |
| 16 | |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 17 | #include "shill/mock_event_dispatcher.h" |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 18 | #include "shill/mock_netlink_manager.h" |
| 19 | #include "shill/netlink_manager.h" |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 20 | #include "shill/netlink_message_matchers.h" |
| 21 | #include "shill/nl80211_message.h" |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 22 | |
| 23 | using std::set; |
| 24 | using std::vector; |
| 25 | using testing::_; |
| 26 | using testing::ContainerEq; |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 27 | using testing::Test; |
| 28 | |
| 29 | namespace shill { |
| 30 | |
| 31 | static const uint16_t kExpectedFreq5640 = 5640; |
| 32 | static const uint16_t kExpectedFreq5600 = 5600; |
| 33 | static const uint16_t kExpectedFreq5580 = 5580; |
| 34 | static const uint16_t kExpectedFreq5560 = 5560; |
| 35 | static const uint16_t kExpectedFreq5620 = 5620; |
| 36 | |
| 37 | static WiFiProvider::FrequencyCount kConnectedFrequencies[] = { |
| 38 | WiFiProvider::FrequencyCount(kExpectedFreq5640, 40), // 40th percentile. |
| 39 | WiFiProvider::FrequencyCount(kExpectedFreq5600, 25), // 65th percentile. |
| 40 | WiFiProvider::FrequencyCount(kExpectedFreq5580, 20), // 85th percentile. |
| 41 | WiFiProvider::FrequencyCount(kExpectedFreq5560, 10), // 95th percentile. |
| 42 | WiFiProvider::FrequencyCount(kExpectedFreq5620, 5) // 100th percentile. |
| 43 | }; |
| 44 | |
| 45 | static const uint16_t kExpectedFreq2432 = 2432; |
| 46 | static const uint16_t kExpectedFreq2427 = 2427; |
| 47 | static const uint16_t kExpectedFreq2422 = 2422; |
| 48 | static const uint16_t kExpectedFreq2417 = 2417; |
| 49 | static const uint16_t kExpectedFreq2412 = 2412; |
| 50 | |
| 51 | static uint16_t kUnconnectedFrequencies[] = { |
| 52 | kExpectedFreq2432, |
| 53 | kExpectedFreq2427, |
| 54 | kExpectedFreq2422, |
| 55 | kExpectedFreq2417, |
| 56 | kExpectedFreq2412 |
| 57 | }; |
| 58 | |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 59 | static const uint16_t kNl80211FamilyId = 0x13; |
| 60 | |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 61 | class ScanSessionTest : public Test { |
| 62 | public: |
| 63 | // Test set of "all the other frequencies this device can support" in |
| 64 | // sorted order. |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 65 | ScanSessionTest() : weak_ptr_factory_(this) { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 66 | WiFiProvider::FrequencyCountList default_connected_frequencies( |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 67 | kConnectedFrequencies, |
| 68 | kConnectedFrequencies + arraysize(kConnectedFrequencies)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 69 | |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 70 | set<uint16_t> default_unconnected_frequencies( |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 71 | kUnconnectedFrequencies, |
| 72 | kUnconnectedFrequencies + arraysize(kUnconnectedFrequencies)); |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 73 | |
| 74 | BuildScanSession(default_connected_frequencies, |
| 75 | default_unconnected_frequencies); |
| 76 | } |
| 77 | |
| 78 | void BuildScanSession(const WiFiProvider::FrequencyCountList |
| 79 | &connected_frequencies, |
| 80 | const std::set<uint16_t> &unconnected_frequencies) { |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 81 | const int kArbitraryMinimum = 1; |
| 82 | const int kArbitraryMaximum = std::numeric_limits<int>::max(); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 83 | scan_session_.reset(new ScanSession(&netlink_manager_, |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 84 | &dispatcher_, |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 85 | connected_frequencies, |
| 86 | unconnected_frequencies, |
| 87 | 0, |
| 88 | ScanSession::FractionList(), |
| 89 | kArbitraryMinimum, |
| 90 | kArbitraryMaximum, |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 91 | Bind(&ScanSessionTest::OnScanError, |
Wade Guthrie | f22681f | 2013-05-31 11:46:31 -0700 | [diff] [blame] | 92 | weak_ptr_factory_.GetWeakPtr()), |
| 93 | NULL)); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 94 | } |
| 95 | |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 96 | virtual std::vector<uint16_t> GetScanFrequencies(float scan_fraction, |
| 97 | size_t min_frequencies, |
| 98 | size_t max_frequencies) { |
| 99 | return scan_session_->GetScanFrequencies(scan_fraction, min_frequencies, |
| 100 | max_frequencies); |
| 101 | } |
| 102 | ScanSession *scan_session() { return scan_session_.get(); } |
| 103 | |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 104 | void SetScanSize(size_t min_frequencies, size_t max_frequencies) { |
| 105 | scan_session_->min_frequencies_ = min_frequencies; |
| 106 | scan_session_->max_frequencies_ = max_frequencies; |
| 107 | } |
| 108 | |
| 109 | size_t GetScanFrequencyCount() { |
| 110 | return arraysize(kConnectedFrequencies) + |
| 111 | arraysize(kUnconnectedFrequencies); |
| 112 | } |
| 113 | |
| 114 | protected: |
| 115 | MOCK_METHOD0(OnScanError, void()); |
| 116 | MockNetlinkManager *netlink_manager() { return &netlink_manager_; } |
| 117 | MockEventDispatcher *dispatcher() { return &dispatcher_; } |
| 118 | |
| 119 | MockEventDispatcher dispatcher_; |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 120 | MockNetlinkManager netlink_manager_; |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 121 | scoped_ptr<ScanSession> scan_session_; |
| 122 | base::WeakPtrFactory<ScanSessionTest> weak_ptr_factory_; |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 123 | }; |
| 124 | |
| 125 | // Test that we can get a bunch of frequencies up to a specified fraction. |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 126 | TEST_F(ScanSessionTest, Fraction) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 127 | vector<uint16_t> result; |
| 128 | |
| 129 | // Get the first 83% of the connected values. |
| 130 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 131 | vector<uint16_t> expected{kExpectedFreq5640, kExpectedFreq5600, |
| 132 | kExpectedFreq5580}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 133 | result = GetScanFrequencies(.83, 1, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 134 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 135 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 136 | } |
| 137 | |
| 138 | // Get the next 4 values. |
| 139 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 140 | vector<uint16_t> expected{kExpectedFreq5560, kExpectedFreq5620, |
| 141 | kExpectedFreq2412, kExpectedFreq2417}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 142 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 1, 4); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 143 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 144 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 145 | } |
| 146 | |
| 147 | // And, get the remaining list. |
| 148 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 149 | vector<uint16_t> expected{kExpectedFreq2422, kExpectedFreq2427, |
| 150 | kExpectedFreq2432}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 151 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 20, |
| 152 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 153 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 154 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 155 | } |
| 156 | } |
| 157 | |
| 158 | // Test that we can get a bunch of frequencies up to a specified fraction, |
| 159 | // followed by another group up to a specified fraction. |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 160 | TEST_F(ScanSessionTest, TwoFractions) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 161 | vector<uint16_t> result; |
| 162 | |
| 163 | // Get the first 60% of the connected values. |
| 164 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 165 | vector<uint16_t> expected{kExpectedFreq5640, kExpectedFreq5600}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 166 | result = GetScanFrequencies(.60, 0, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 167 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 168 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 169 | } |
| 170 | |
| 171 | // Get the next 32% of the connected values. |
| 172 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 173 | vector<uint16_t> expected{kExpectedFreq5580, kExpectedFreq5560}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 174 | result = GetScanFrequencies(.32, 0, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 175 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 176 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 177 | } |
| 178 | |
| 179 | // And, get the remaining list. |
| 180 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 181 | vector<uint16_t> expected{kExpectedFreq5620, kExpectedFreq2412, |
| 182 | kExpectedFreq2417, kExpectedFreq2422, kExpectedFreq2427, |
| 183 | kExpectedFreq2432}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 184 | result = GetScanFrequencies(ScanSession::kAllFrequencies, |
| 185 | std::numeric_limits<size_t>::max(), |
| 186 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 187 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 188 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 189 | } |
| 190 | } |
| 191 | |
| 192 | // Test that we can get a bunch of frequencies up to a minimum count, even |
| 193 | // when the requested fraction has already been reached. |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 194 | TEST_F(ScanSessionTest, Min) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 195 | vector<uint16_t> result; |
| 196 | |
| 197 | // Get the first 3 previously seen values. |
| 198 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 199 | vector<uint16_t> expected{kExpectedFreq5640, kExpectedFreq5600, |
| 200 | kExpectedFreq5580}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 201 | result = GetScanFrequencies(.30, 3, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 202 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 203 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 204 | } |
| 205 | |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 206 | // Get the next value by requesting a minimum of 1. |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 207 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 208 | vector<uint16_t> expected{kExpectedFreq5560}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 209 | result = GetScanFrequencies(0.0, 1, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 210 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 211 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 212 | } |
| 213 | |
| 214 | // And, get the remaining list. |
| 215 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 216 | vector<uint16_t> expected{kExpectedFreq5620, kExpectedFreq2412, |
| 217 | kExpectedFreq2417, kExpectedFreq2422, kExpectedFreq2427, |
| 218 | kExpectedFreq2432}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 219 | result = GetScanFrequencies(ScanSession::kAllFrequencies, |
| 220 | std::numeric_limits<size_t>::max(), |
| 221 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 222 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 223 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 224 | } |
| 225 | } |
| 226 | |
| 227 | // Test that we can get up to a specified maximum number of frequencies. |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 228 | TEST_F(ScanSessionTest, Max) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 229 | vector<uint16_t> result; |
| 230 | |
| 231 | // Get the first 7 values (crosses seen/unseen boundary). |
| 232 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 233 | vector<uint16_t> expected{kExpectedFreq5640, kExpectedFreq5600, |
| 234 | kExpectedFreq5580, kExpectedFreq5560, kExpectedFreq5620, |
| 235 | kExpectedFreq2412, kExpectedFreq2417}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 236 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 1, 7); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 237 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 238 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 239 | } |
| 240 | |
| 241 | // And, get the remaining list. |
| 242 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 243 | vector<uint16_t> expected{kExpectedFreq2422, kExpectedFreq2427, |
| 244 | kExpectedFreq2432}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 245 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 20, |
| 246 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 247 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 248 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 249 | } |
| 250 | } |
| 251 | |
| 252 | // Test that we can get exactly the seen frequencies and exactly the unseen |
| 253 | // ones. |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 254 | TEST_F(ScanSessionTest, Exact) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 255 | vector<uint16_t> result; |
| 256 | |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 257 | // Get the first 5 values -- exactly on the seen/unseen border. |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 258 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 259 | vector<uint16_t> expected{kExpectedFreq5640, kExpectedFreq5600, |
| 260 | kExpectedFreq5580, kExpectedFreq5560, kExpectedFreq5620}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 261 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 5, 5); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 262 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 263 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 264 | } |
| 265 | |
| 266 | // And, get the last 5. |
| 267 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 268 | vector<uint16_t> expected{kExpectedFreq2412, kExpectedFreq2417, |
| 269 | kExpectedFreq2422, kExpectedFreq2427, kExpectedFreq2432}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 270 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 5, 5); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 271 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 272 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 273 | } |
| 274 | } |
| 275 | |
| 276 | // Test that we can get everything in one read. |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 277 | TEST_F(ScanSessionTest, AllOneRead) { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 278 | vector<uint16_t> expected{kExpectedFreq5640, kExpectedFreq5600, |
| 279 | kExpectedFreq5580, kExpectedFreq5560, kExpectedFreq5620, |
| 280 | kExpectedFreq2412, kExpectedFreq2417, kExpectedFreq2422, |
| 281 | kExpectedFreq2427, kExpectedFreq2432}; |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 282 | vector<uint16_t> result; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 283 | result = GetScanFrequencies(ScanSession::kAllFrequencies, |
| 284 | std::numeric_limits<size_t>::max(), |
| 285 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 286 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 287 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 288 | } |
| 289 | |
| 290 | // Test that we can get all the previously seen frequencies (and only the |
| 291 | // previously seen frequencies) via the requested fraction. |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 292 | TEST_F(ScanSessionTest, EverythingConnected) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 293 | vector<uint16_t> result; |
| 294 | |
| 295 | // Get the first 100% of the connected values. |
| 296 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 297 | vector<uint16_t> expected{kExpectedFreq5640, kExpectedFreq5600, |
| 298 | kExpectedFreq5580, kExpectedFreq5560, kExpectedFreq5620}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 299 | result = GetScanFrequencies(1.0, 0, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 300 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 301 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 302 | } |
| 303 | |
| 304 | // And, get the remaining list. |
| 305 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 306 | vector<uint16_t> expected{kExpectedFreq2412, kExpectedFreq2417, |
| 307 | kExpectedFreq2422, kExpectedFreq2427, kExpectedFreq2432}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 308 | result = GetScanFrequencies(ScanSession::kAllFrequencies, |
| 309 | std::numeric_limits<size_t>::max(), |
| 310 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 311 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 312 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 313 | } |
| 314 | } |
| 315 | |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 316 | TEST_F(ScanSessionTest, OnlyPreviouslySeen) { |
| 317 | // Build a scan session with only previously connected frequencies. |
| 318 | WiFiProvider::FrequencyCountList default_connected_frequencies( |
| 319 | kConnectedFrequencies, |
| 320 | kConnectedFrequencies + arraysize(kConnectedFrequencies)); |
| 321 | BuildScanSession(default_connected_frequencies, std::set<uint16_t>()); |
| 322 | |
| 323 | // Get the first 100% of the connected values. |
| 324 | vector<uint16_t> expected{kExpectedFreq5640, kExpectedFreq5600, |
| 325 | kExpectedFreq5580, kExpectedFreq5560, kExpectedFreq5620}; |
| 326 | |
| 327 | vector<uint16_t> result; |
| 328 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 1, |
| 329 | std::numeric_limits<size_t>::max()); |
| 330 | EXPECT_THAT(result, ContainerEq(expected)); |
| 331 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
| 332 | result = GetScanFrequencies(ScanSession::kAllFrequencies, |
| 333 | std::numeric_limits<size_t>::max(), |
| 334 | std::numeric_limits<size_t>::max()); |
| 335 | EXPECT_TRUE(result.empty()); |
| 336 | } |
| 337 | |
| 338 | // Verify that max works inside the list of connected frequencies. |
| 339 | TEST_F(ScanSessionTest, MaxAppliesToConnected) { |
| 340 | vector<uint16_t> result; |
| 341 | |
| 342 | { |
| 343 | vector<uint16_t> expected{kExpectedFreq5640, kExpectedFreq5600, |
| 344 | kExpectedFreq5580}; |
| 345 | |
| 346 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 1, 3); |
| 347 | EXPECT_THAT(result, ContainerEq(expected)); |
| 348 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 349 | } |
| 350 | |
| 351 | { |
| 352 | vector<uint16_t> expected{kExpectedFreq5560, kExpectedFreq5620, |
| 353 | kExpectedFreq2412}; |
| 354 | |
| 355 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 1, 3); |
| 356 | EXPECT_THAT(result, ContainerEq(expected)); |
| 357 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 358 | } |
| 359 | |
| 360 | { |
| 361 | vector<uint16_t> expected{kExpectedFreq2417, kExpectedFreq2422, |
| 362 | kExpectedFreq2427}; |
| 363 | |
| 364 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 1, 3); |
| 365 | EXPECT_THAT(result, ContainerEq(expected)); |
| 366 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 367 | } |
| 368 | |
| 369 | { |
| 370 | vector<uint16_t> expected{kExpectedFreq2432}; |
| 371 | |
| 372 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 1, 3); |
| 373 | EXPECT_THAT(result, ContainerEq(expected)); |
| 374 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
| 375 | } |
| 376 | } |
| 377 | |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 378 | // Test that we can get each value individually. |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 379 | TEST_F(ScanSessionTest, IndividualReads) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 380 | vector<uint16_t> result; |
| 381 | static const float kArbitraryFraction = 0.83; |
| 382 | |
| 383 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 384 | vector<uint16_t> expected{kExpectedFreq5640}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 385 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 386 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 387 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 388 | } |
| 389 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 390 | vector<uint16_t> expected{kExpectedFreq5600}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 391 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 392 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 393 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 394 | } |
| 395 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 396 | vector<uint16_t> expected{kExpectedFreq5580}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 397 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 398 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 399 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 400 | } |
| 401 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 402 | vector<uint16_t> expected{kExpectedFreq5560}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 403 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 404 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 405 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 406 | } |
| 407 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 408 | vector<uint16_t> expected{kExpectedFreq5620}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 409 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 410 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 411 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 412 | } |
| 413 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 414 | vector<uint16_t> expected{kExpectedFreq2412}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 415 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 416 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 417 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 418 | } |
| 419 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 420 | vector<uint16_t> expected{kExpectedFreq2417}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 421 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
| 422 | EXPECT_THAT(result, ContainerEq(expected)); |
| 423 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 424 | } |
| 425 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 426 | vector<uint16_t> expected{kExpectedFreq2422}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 427 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
| 428 | EXPECT_THAT(result, ContainerEq(expected)); |
| 429 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 430 | } |
| 431 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 432 | vector<uint16_t> expected{kExpectedFreq2427}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 433 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
| 434 | EXPECT_THAT(result, ContainerEq(expected)); |
| 435 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 436 | } |
| 437 | { |
Wade Guthrie | b86860c | 2013-09-18 15:42:56 -0700 | [diff] [blame] | 438 | vector<uint16_t> expected{kExpectedFreq2432}; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 439 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
| 440 | EXPECT_THAT(result, ContainerEq(expected)); |
| 441 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 442 | } |
| 443 | } |
| 444 | |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 445 | TEST_F(ScanSessionTest, OnTriggerScanResponse) { |
| 446 | Nl80211Message::SetMessageType(kNl80211FamilyId); |
| 447 | |
Wade Guthrie | 7347bf2 | 2013-04-30 11:21:51 -0700 | [diff] [blame] | 448 | EXPECT_CALL(*netlink_manager(), SendNl80211Message( |
| 449 | IsNl80211Command(kNl80211FamilyId, NL80211_CMD_TRIGGER_SCAN), _, _)); |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 450 | scan_session()->InitiateScan(); |
| 451 | |
| 452 | EXPECT_CALL(*this, OnScanError()); |
| 453 | NewScanResultsMessage not_supposed_to_get_this_message; |
| 454 | scan_session()->OnTriggerScanResponse(not_supposed_to_get_this_message); |
| 455 | } |
| 456 | |
| 457 | TEST_F(ScanSessionTest, ExhaustFrequencies) { |
| 458 | // Set min & max scan frequency count to 1 so each scan will be of a single |
| 459 | // frequency. |
| 460 | SetScanSize(1, 1); |
| 461 | |
| 462 | // Perform all the progressive scans until the frequencies are exhausted. |
| 463 | for (size_t i = 0; i < GetScanFrequencyCount(); ++i) { |
| 464 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | 7347bf2 | 2013-04-30 11:21:51 -0700 | [diff] [blame] | 465 | EXPECT_CALL(*netlink_manager(), SendNl80211Message( |
| 466 | IsNl80211Command(kNl80211FamilyId, NL80211_CMD_TRIGGER_SCAN), _, _)); |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 467 | scan_session()->InitiateScan(); |
| 468 | } |
| 469 | |
| 470 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | 7347bf2 | 2013-04-30 11:21:51 -0700 | [diff] [blame] | 471 | EXPECT_CALL(*netlink_manager(), SendNl80211Message( |
| 472 | IsNl80211Command(kNl80211FamilyId, NL80211_CMD_TRIGGER_SCAN), _, _)) |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 473 | .Times(0); |
| 474 | scan_session()->InitiateScan(); |
| 475 | } |
| 476 | |
| 477 | TEST_F(ScanSessionTest, OnError) { |
| 478 | Nl80211Message::SetMessageType(kNl80211FamilyId); |
| 479 | |
Wade Guthrie | 7347bf2 | 2013-04-30 11:21:51 -0700 | [diff] [blame] | 480 | EXPECT_CALL(*netlink_manager(), SendNl80211Message( |
| 481 | IsNl80211Command(kNl80211FamilyId, NL80211_CMD_TRIGGER_SCAN), _, _)); |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 482 | scan_session()->InitiateScan(); |
| 483 | |
| 484 | EXPECT_CALL(*this, OnScanError()); |
| 485 | ErrorAckMessage error_message(-EINTR); |
Wade Guthrie | 84db7ce | 2013-06-12 11:40:49 -0700 | [diff] [blame] | 486 | scan_session()->OnTriggerScanErrorResponse(NetlinkManager::kErrorFromKernel, |
| 487 | &error_message); |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 488 | } |
| 489 | |
| 490 | TEST_F(ScanSessionTest, EBusy) { |
| 491 | const size_t kSmallRetryNumber = 3; |
| 492 | Nl80211Message::SetMessageType(kNl80211FamilyId); |
| 493 | scan_session()->scan_tries_left_ = kSmallRetryNumber; |
| 494 | |
Wade Guthrie | 7347bf2 | 2013-04-30 11:21:51 -0700 | [diff] [blame] | 495 | EXPECT_CALL(*netlink_manager(), SendNl80211Message( |
| 496 | IsNl80211Command(kNl80211FamilyId, NL80211_CMD_TRIGGER_SCAN), _, _)); |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 497 | scan_session()->InitiateScan(); |
| 498 | |
| 499 | ErrorAckMessage error_message(-EBUSY); |
| 500 | for (size_t i = 0; i < kSmallRetryNumber; ++i) { |
| 501 | EXPECT_CALL(*this, OnScanError()).Times(0); |
| 502 | EXPECT_CALL(*dispatcher(), PostDelayedTask(_, _)); |
Wade Guthrie | 84db7ce | 2013-06-12 11:40:49 -0700 | [diff] [blame] | 503 | scan_session()->OnTriggerScanErrorResponse(NetlinkManager::kErrorFromKernel, |
| 504 | &error_message); |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 505 | } |
| 506 | |
| 507 | EXPECT_CALL(*this, OnScanError()); |
Wade Guthrie | 84db7ce | 2013-06-12 11:40:49 -0700 | [diff] [blame] | 508 | scan_session()->OnTriggerScanErrorResponse(NetlinkManager::kErrorFromKernel, |
| 509 | &error_message); |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 510 | } |
| 511 | |
| 512 | TEST_F(ScanSessionTest, ScanHidden) { |
| 513 | scan_session_->AddSsid(ByteString("a", 1)); |
| 514 | EXPECT_CALL(netlink_manager_, |
Wade Guthrie | 7347bf2 | 2013-04-30 11:21:51 -0700 | [diff] [blame] | 515 | SendNl80211Message(HasHiddenSSID(kNl80211FamilyId), _, _)); |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 516 | scan_session()->InitiateScan(); |
| 517 | } |
| 518 | |
| 519 | TEST_F(ScanSessionTest, ScanNoHidden) { |
| 520 | EXPECT_CALL(netlink_manager_, |
Wade Guthrie | 7347bf2 | 2013-04-30 11:21:51 -0700 | [diff] [blame] | 521 | SendNl80211Message(HasNoHiddenSSID(kNl80211FamilyId), _, _)); |
Wade Guthrie | 5a4e2ef | 2013-04-30 12:51:39 -0700 | [diff] [blame] | 522 | scan_session()->InitiateScan(); |
| 523 | } |
| 524 | |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 525 | } // namespace shill |