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