commit 1df7f4e6ba237549123c01882db8ec534310b87f
author Thieu Le <thieule@chromium.org> Fri Feb 10 15:21:45 2012 -0800
committer Gerrit <chrome-bot@google.com> Wed Feb 15 14:03:01 2012 -0800
tree 617d4508eb6c553684b3191540c0d8c8ea343ca9
parent 67370f6ebe6757f55249f6e87f4a96922d0c4518

shill: Add support to set physical mode.

BUG=chromium-os:24776
TEST=Unit tests, network_WiFiManager

Change-Id: Ic46404e71dbb54a06a7d668992a96df9f8db2598
Reviewed-on: https://gerrit.chromium.org/gerrit/15765
Commit-Ready: Thieu Le <thieule@chromium.org>
Reviewed-by: Thieu Le <thieule@chromium.org>
Tested-by: Thieu Le <thieule@chromium.org>

wifi_endpoint_unittest.cc

diff --git a/wifi_endpoint_unittest.cc b/wifi_endpoint_unittest.cc
index 23b5269..1e682c7 100644
--- a/wifi_endpoint_unittest.cc
+++ b/wifi_endpoint_unittest.cc
@@ -13,6 +13,7 @@
 #include <chromeos/dbus/service_constants.h>
 #include <gtest/gtest.h>
 
+#include "shill/ieee80211.h"
 #include "shill/refptr_types.h"
 #include "shill/wpa_supplicant.h"
 
@@ -69,6 +70,13 @@
     const map<string, ::DBus::Variant> &properties) {
     return WiFiEndpoint::ParseSecurity(properties);
   }
+
+  void AddIE(uint8_t type, vector<uint8_t> *ies) {
+    ies->push_back(type);           // type
+    ies->push_back(4);              // length
+    ies->insert(ies->end(), 3, 0);  // OUI
+    ies->push_back(0);              // data
+  }
 };
 
 TEST_F(WiFiEndpointTest, ParseKeyManagementMethodsEAP) {
@@ -142,4 +150,71 @@
   EXPECT_EQ("?", endpoint->ssid_string());
 }
 
+TEST_F(WiFiEndpointTest, DeterminePhyMode) {
+  {
+    map<string, ::DBus::Variant> properties;
+    vector<uint8_t> ies;
+    AddIE(IEEE_80211::kElemIdErp, &ies);
+    ::DBus::MessageIter writer =
+        properties[wpa_supplicant::kBSSPropertyIEs].writer();
+    writer << ies;
+    EXPECT_EQ(Metrics::kWiFiNetworkPhyMode11g,
+              WiFiEndpoint::DeterminePhyMode(properties, 2400));
+  }
+  {
+    map<string, ::DBus::Variant> properties;
+    vector<uint8_t> ies;
+    AddIE(IEEE_80211::kElemIdHTCap, &ies);
+    ::DBus::MessageIter writer =
+        properties[wpa_supplicant::kBSSPropertyIEs].writer();
+    writer << ies;
+    EXPECT_EQ(Metrics::kWiFiNetworkPhyMode11n,
+              WiFiEndpoint::DeterminePhyMode(properties, 2400));
+  }
+  {
+    map<string, ::DBus::Variant> properties;
+    vector<uint8_t> ies;
+    AddIE(IEEE_80211::kElemIdHTInfo, &ies);
+    ::DBus::MessageIter writer =
+        properties[wpa_supplicant::kBSSPropertyIEs].writer();
+    writer << ies;
+    EXPECT_EQ(Metrics::kWiFiNetworkPhyMode11n,
+              WiFiEndpoint::DeterminePhyMode(properties, 2400));
+  }
+  {
+    map<string, ::DBus::Variant> properties;
+    vector<uint8_t> ies;
+    AddIE(IEEE_80211::kElemIdErp, &ies);
+    AddIE(IEEE_80211::kElemIdHTCap, &ies);
+    ::DBus::MessageIter writer =
+        properties[wpa_supplicant::kBSSPropertyIEs].writer();
+    writer << ies;
+    EXPECT_EQ(Metrics::kWiFiNetworkPhyMode11n,
+              WiFiEndpoint::DeterminePhyMode(properties, 2400));
+  }
+  {
+    map<string, ::DBus::Variant> properties;
+    EXPECT_EQ(Metrics::kWiFiNetworkPhyMode11a,
+              WiFiEndpoint::DeterminePhyMode(properties, 3200));
+  }
+  {
+    map<string, ::DBus::Variant> properties;
+    vector<uint32_t> rates(1, 22000000);
+    ::DBus::MessageIter writer =
+        properties[wpa_supplicant::kBSSPropertyRates].writer();
+    writer << rates;
+    EXPECT_EQ(Metrics::kWiFiNetworkPhyMode11b,
+              WiFiEndpoint::DeterminePhyMode(properties, 2400));
+  }
+  {
+    map<string, ::DBus::Variant> properties;
+    vector<uint32_t> rates(1, 54000000);
+    ::DBus::MessageIter writer =
+        properties[wpa_supplicant::kBSSPropertyRates].writer();
+    writer << rates;
+    EXPECT_EQ(Metrics::kWiFiNetworkPhyMode11g,
+              WiFiEndpoint::DeterminePhyMode(properties, 2400));
+  }
+}
+
 }  // namespace shill