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 | |
| 7 | #include <limits> |
| 8 | #include <set> |
| 9 | #include <vector> |
| 10 | |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 11 | #include <gmock/gmock.h> |
| 12 | #include <gtest/gtest.h> |
| 13 | |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 14 | #include "shill/mock_netlink_manager.h" |
| 15 | #include "shill/netlink_manager.h" |
| 16 | |
| 17 | using std::set; |
| 18 | using std::vector; |
| 19 | using testing::_; |
| 20 | using testing::ContainerEq; |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 21 | using testing::Test; |
| 22 | |
| 23 | namespace shill { |
| 24 | |
| 25 | static const uint16_t kExpectedFreq5640 = 5640; |
| 26 | static const uint16_t kExpectedFreq5600 = 5600; |
| 27 | static const uint16_t kExpectedFreq5580 = 5580; |
| 28 | static const uint16_t kExpectedFreq5560 = 5560; |
| 29 | static const uint16_t kExpectedFreq5620 = 5620; |
| 30 | |
| 31 | static WiFiProvider::FrequencyCount kConnectedFrequencies[] = { |
| 32 | WiFiProvider::FrequencyCount(kExpectedFreq5640, 40), // 40th percentile. |
| 33 | WiFiProvider::FrequencyCount(kExpectedFreq5600, 25), // 65th percentile. |
| 34 | WiFiProvider::FrequencyCount(kExpectedFreq5580, 20), // 85th percentile. |
| 35 | WiFiProvider::FrequencyCount(kExpectedFreq5560, 10), // 95th percentile. |
| 36 | WiFiProvider::FrequencyCount(kExpectedFreq5620, 5) // 100th percentile. |
| 37 | }; |
| 38 | |
| 39 | static const uint16_t kExpectedFreq2432 = 2432; |
| 40 | static const uint16_t kExpectedFreq2427 = 2427; |
| 41 | static const uint16_t kExpectedFreq2422 = 2422; |
| 42 | static const uint16_t kExpectedFreq2417 = 2417; |
| 43 | static const uint16_t kExpectedFreq2412 = 2412; |
| 44 | |
| 45 | static uint16_t kUnconnectedFrequencies[] = { |
| 46 | kExpectedFreq2432, |
| 47 | kExpectedFreq2427, |
| 48 | kExpectedFreq2422, |
| 49 | kExpectedFreq2417, |
| 50 | kExpectedFreq2412 |
| 51 | }; |
| 52 | |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 53 | class ScanSessionTest : public Test { |
| 54 | public: |
| 55 | // Test set of "all the other frequencies this device can support" in |
| 56 | // sorted order. |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 57 | ScanSessionTest() { |
| 58 | WiFiProvider::FrequencyCountList connected_frequencies( |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 59 | kConnectedFrequencies, |
| 60 | kConnectedFrequencies + arraysize(kConnectedFrequencies)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 61 | |
| 62 | set<uint16_t> unconnected_frequencies( |
| 63 | kUnconnectedFrequencies, |
| 64 | kUnconnectedFrequencies + arraysize(kUnconnectedFrequencies)); |
| 65 | const int kArbitraryMinimum = 1; |
| 66 | const int kArbitraryMaximum = std::numeric_limits<int>::max(); |
| 67 | ScanSession::OnScanFailed null_error_handler; |
| 68 | scan_session_.reset(new ScanSession(&netlink_manager_, |
| 69 | NULL, |
| 70 | connected_frequencies, |
| 71 | unconnected_frequencies, |
| 72 | 0, |
| 73 | ScanSession::FractionList(), |
| 74 | kArbitraryMinimum, |
| 75 | kArbitraryMaximum, |
| 76 | null_error_handler)); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 77 | } |
| 78 | |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 79 | virtual std::vector<uint16_t> GetScanFrequencies(float scan_fraction, |
| 80 | size_t min_frequencies, |
| 81 | size_t max_frequencies) { |
| 82 | return scan_session_->GetScanFrequencies(scan_fraction, min_frequencies, |
| 83 | max_frequencies); |
| 84 | } |
| 85 | ScanSession *scan_session() { return scan_session_.get(); } |
| 86 | |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 87 | private: |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 88 | scoped_ptr<ScanSession> scan_session_; |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 89 | MockNetlinkManager netlink_manager_; |
| 90 | }; |
| 91 | |
| 92 | // Test that we can get a bunch of frequencies up to a specified fraction. |
| 93 | TEST_F(ScanSessionTest, FractionTest) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 94 | vector<uint16_t> result; |
| 95 | |
| 96 | // Get the first 83% of the connected values. |
| 97 | { |
| 98 | vector<uint16_t> expected; |
| 99 | expected.push_back(kExpectedFreq5640); |
| 100 | expected.push_back(kExpectedFreq5600); |
| 101 | expected.push_back(kExpectedFreq5580); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 102 | result = GetScanFrequencies(.83, 1, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 103 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 104 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 105 | } |
| 106 | |
| 107 | // Get the next 4 values. |
| 108 | { |
| 109 | vector<uint16_t> expected; |
| 110 | expected.push_back(kExpectedFreq5560); |
| 111 | expected.push_back(kExpectedFreq5620); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 112 | expected.push_back(kExpectedFreq2412); |
| 113 | expected.push_back(kExpectedFreq2417); |
| 114 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 1, 4); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 115 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 116 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 117 | } |
| 118 | |
| 119 | // And, get the remaining list. |
| 120 | { |
| 121 | vector<uint16_t> expected; |
| 122 | expected.push_back(kExpectedFreq2422); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 123 | expected.push_back(kExpectedFreq2427); |
| 124 | expected.push_back(kExpectedFreq2432); |
| 125 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 20, |
| 126 | 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_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 129 | } |
| 130 | } |
| 131 | |
| 132 | // Test that we can get a bunch of frequencies up to a specified fraction, |
| 133 | // followed by another group up to a specified fraction. |
| 134 | TEST_F(ScanSessionTest, TwoFractionsTest) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 135 | vector<uint16_t> result; |
| 136 | |
| 137 | // Get the first 60% of the connected values. |
| 138 | { |
| 139 | vector<uint16_t> expected; |
| 140 | expected.push_back(kExpectedFreq5640); |
| 141 | expected.push_back(kExpectedFreq5600); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 142 | result = GetScanFrequencies(.60, 0, std::numeric_limits<size_t>::max()); |
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 | // Get the next 32% of the connected values. |
| 148 | { |
| 149 | vector<uint16_t> expected; |
| 150 | expected.push_back(kExpectedFreq5580); |
| 151 | expected.push_back(kExpectedFreq5560); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 152 | result = GetScanFrequencies(.32, 0, 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_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 155 | } |
| 156 | |
| 157 | // And, get the remaining list. |
| 158 | { |
| 159 | vector<uint16_t> expected; |
| 160 | expected.push_back(kExpectedFreq5620); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 161 | expected.push_back(kExpectedFreq2412); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 162 | expected.push_back(kExpectedFreq2417); |
| 163 | expected.push_back(kExpectedFreq2422); |
| 164 | expected.push_back(kExpectedFreq2427); |
| 165 | expected.push_back(kExpectedFreq2432); |
| 166 | result = GetScanFrequencies(ScanSession::kAllFrequencies, |
| 167 | std::numeric_limits<size_t>::max(), |
| 168 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 169 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 170 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 171 | } |
| 172 | } |
| 173 | |
| 174 | // Test that we can get a bunch of frequencies up to a minimum count, even |
| 175 | // when the requested fraction has already been reached. |
| 176 | TEST_F(ScanSessionTest, MinTest) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 177 | vector<uint16_t> result; |
| 178 | |
| 179 | // Get the first 3 previously seen values. |
| 180 | { |
| 181 | vector<uint16_t> expected; |
| 182 | expected.push_back(kExpectedFreq5640); |
| 183 | expected.push_back(kExpectedFreq5600); |
| 184 | expected.push_back(kExpectedFreq5580); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 185 | result = GetScanFrequencies(.30, 3, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 186 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 187 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 188 | } |
| 189 | |
| 190 | // Get the next value by requensting a minimum of 1. |
| 191 | { |
| 192 | vector<uint16_t> expected; |
| 193 | expected.push_back(kExpectedFreq5560); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 194 | result = GetScanFrequencies(0.0, 1, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 195 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 196 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 197 | } |
| 198 | |
| 199 | // And, get the remaining list. |
| 200 | { |
| 201 | vector<uint16_t> expected; |
| 202 | expected.push_back(kExpectedFreq5620); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 203 | expected.push_back(kExpectedFreq2412); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 204 | expected.push_back(kExpectedFreq2417); |
| 205 | expected.push_back(kExpectedFreq2422); |
| 206 | expected.push_back(kExpectedFreq2427); |
| 207 | expected.push_back(kExpectedFreq2432); |
| 208 | result = GetScanFrequencies(ScanSession::kAllFrequencies, |
| 209 | std::numeric_limits<size_t>::max(), |
| 210 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 211 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 212 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 213 | } |
| 214 | } |
| 215 | |
| 216 | // Test that we can get up to a specified maximum number of frequencies. |
| 217 | TEST_F(ScanSessionTest, MaxTest) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 218 | vector<uint16_t> result; |
| 219 | |
| 220 | // Get the first 7 values (crosses seen/unseen boundary). |
| 221 | { |
| 222 | vector<uint16_t> expected; |
| 223 | expected.push_back(kExpectedFreq5640); |
| 224 | expected.push_back(kExpectedFreq5600); |
| 225 | expected.push_back(kExpectedFreq5580); |
| 226 | expected.push_back(kExpectedFreq5560); |
| 227 | expected.push_back(kExpectedFreq5620); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 228 | expected.push_back(kExpectedFreq2412); |
| 229 | expected.push_back(kExpectedFreq2417); |
| 230 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 1, 7); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 231 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 232 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 233 | } |
| 234 | |
| 235 | // And, get the remaining list. |
| 236 | { |
| 237 | vector<uint16_t> expected; |
| 238 | expected.push_back(kExpectedFreq2422); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 239 | expected.push_back(kExpectedFreq2427); |
| 240 | expected.push_back(kExpectedFreq2432); |
| 241 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 20, |
| 242 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 243 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 244 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 245 | } |
| 246 | } |
| 247 | |
| 248 | // Test that we can get exactly the seen frequencies and exactly the unseen |
| 249 | // ones. |
| 250 | TEST_F(ScanSessionTest, ExactTest) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 251 | vector<uint16_t> result; |
| 252 | |
| 253 | // Get the first 5 values -- exectly on the seen/unseen border. |
| 254 | { |
| 255 | vector<uint16_t> expected; |
| 256 | expected.push_back(kExpectedFreq5640); |
| 257 | expected.push_back(kExpectedFreq5600); |
| 258 | expected.push_back(kExpectedFreq5580); |
| 259 | expected.push_back(kExpectedFreq5560); |
| 260 | expected.push_back(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 | { |
| 268 | vector<uint16_t> expected; |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 269 | expected.push_back(kExpectedFreq2412); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 270 | expected.push_back(kExpectedFreq2417); |
| 271 | expected.push_back(kExpectedFreq2422); |
| 272 | expected.push_back(kExpectedFreq2427); |
| 273 | expected.push_back(kExpectedFreq2432); |
| 274 | result = GetScanFrequencies(ScanSession::kAllFrequencies, 5, 5); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 275 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 276 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 277 | } |
| 278 | } |
| 279 | |
| 280 | // Test that we can get everything in one read. |
| 281 | TEST_F(ScanSessionTest, AllOneReadTest) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 282 | |
| 283 | vector<uint16_t> expected; |
| 284 | expected.push_back(kExpectedFreq5640); |
| 285 | expected.push_back(kExpectedFreq5600); |
| 286 | expected.push_back(kExpectedFreq5580); |
| 287 | expected.push_back(kExpectedFreq5560); |
| 288 | expected.push_back(kExpectedFreq5620); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 289 | expected.push_back(kExpectedFreq2412); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 290 | expected.push_back(kExpectedFreq2417); |
| 291 | expected.push_back(kExpectedFreq2422); |
| 292 | expected.push_back(kExpectedFreq2427); |
| 293 | expected.push_back(kExpectedFreq2432); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 294 | vector<uint16_t> result; |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 295 | result = GetScanFrequencies(ScanSession::kAllFrequencies, |
| 296 | std::numeric_limits<size_t>::max(), |
| 297 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 298 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 299 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 300 | } |
| 301 | |
| 302 | // Test that we can get all the previously seen frequencies (and only the |
| 303 | // previously seen frequencies) via the requested fraction. |
| 304 | TEST_F(ScanSessionTest, EverythingFractionTest) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 305 | vector<uint16_t> result; |
| 306 | |
| 307 | // Get the first 100% of the connected values. |
| 308 | { |
| 309 | vector<uint16_t> expected; |
| 310 | expected.push_back(kExpectedFreq5640); |
| 311 | expected.push_back(kExpectedFreq5600); |
| 312 | expected.push_back(kExpectedFreq5580); |
| 313 | expected.push_back(kExpectedFreq5560); |
| 314 | expected.push_back(kExpectedFreq5620); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 315 | result = GetScanFrequencies(1.0, 0, std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 316 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 317 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 318 | } |
| 319 | |
| 320 | // And, get the remaining list. |
| 321 | { |
| 322 | vector<uint16_t> expected; |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 323 | expected.push_back(kExpectedFreq2412); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 324 | expected.push_back(kExpectedFreq2417); |
| 325 | expected.push_back(kExpectedFreq2422); |
| 326 | expected.push_back(kExpectedFreq2427); |
| 327 | expected.push_back(kExpectedFreq2432); |
| 328 | result = GetScanFrequencies(ScanSession::kAllFrequencies, |
| 329 | std::numeric_limits<size_t>::max(), |
| 330 | std::numeric_limits<size_t>::max()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 331 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 332 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 333 | } |
| 334 | } |
| 335 | |
| 336 | // Test that we can get each value individually. |
| 337 | TEST_F(ScanSessionTest, IndividualReadsTest) { |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 338 | vector<uint16_t> result; |
| 339 | static const float kArbitraryFraction = 0.83; |
| 340 | |
| 341 | { |
| 342 | vector<uint16_t> expected; |
| 343 | expected.push_back(kExpectedFreq5640); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 344 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 345 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 346 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 347 | } |
| 348 | { |
| 349 | vector<uint16_t> expected; |
| 350 | expected.push_back(kExpectedFreq5600); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 351 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 352 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 353 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 354 | } |
| 355 | { |
| 356 | vector<uint16_t> expected; |
| 357 | expected.push_back(kExpectedFreq5580); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 358 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 359 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 360 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 361 | } |
| 362 | { |
| 363 | vector<uint16_t> expected; |
| 364 | expected.push_back(kExpectedFreq5560); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 365 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 366 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 367 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 368 | } |
| 369 | { |
| 370 | vector<uint16_t> expected; |
| 371 | expected.push_back(kExpectedFreq5620); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 372 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 373 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 374 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 375 | } |
| 376 | { |
| 377 | vector<uint16_t> expected; |
| 378 | expected.push_back(kExpectedFreq2412); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 379 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 380 | EXPECT_THAT(result, ContainerEq(expected)); |
Wade Guthrie | b9c3feb | 2013-04-25 16:31:19 -0700 | [diff] [blame] | 381 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 382 | } |
| 383 | { |
| 384 | vector<uint16_t> expected; |
| 385 | expected.push_back(kExpectedFreq2417); |
| 386 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
| 387 | EXPECT_THAT(result, ContainerEq(expected)); |
| 388 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 389 | } |
| 390 | { |
| 391 | vector<uint16_t> expected; |
| 392 | expected.push_back(kExpectedFreq2422); |
| 393 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
| 394 | EXPECT_THAT(result, ContainerEq(expected)); |
| 395 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 396 | } |
| 397 | { |
| 398 | vector<uint16_t> expected; |
| 399 | expected.push_back(kExpectedFreq2427); |
| 400 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
| 401 | EXPECT_THAT(result, ContainerEq(expected)); |
| 402 | EXPECT_TRUE(scan_session()->HasMoreFrequencies()); |
| 403 | } |
| 404 | { |
| 405 | vector<uint16_t> expected; |
| 406 | expected.push_back(kExpectedFreq2432); |
| 407 | result = GetScanFrequencies(kArbitraryFraction, 1, 1); |
| 408 | EXPECT_THAT(result, ContainerEq(expected)); |
| 409 | EXPECT_FALSE(scan_session()->HasMoreFrequencies()); |
Wade Guthrie | a60a11c | 2013-04-12 17:47:34 -0700 | [diff] [blame] | 410 | } |
| 411 | } |
| 412 | |
| 413 | } // namespace shill |