Add WiFi SSID getter to NetworkChangeNotifier.

net/base/address_tracker_linux_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/memory/scoped_ptr.h"
 #include "net/base/address_tracker_linux.h"
 
 #include <linux/if.h>
 
 #include <vector>
 
 #include "base/bind.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 #ifndef IFA_F_HOMEADDRESS
 #define IFA_F_HOMEADDRESS 0x10
 #endif
 
 namespace net {
 namespace internal {
 namespace {
 
 const int kTestInterfaceTun = 123;
+const char kWiFiSSID[] = "TestWiFi";
+const char kInterfaceWithNoSSID[] = "wlan999";
 
 char* TestGetInterfaceName(int interface_index, char* buf) {
   if (interface_index == kTestInterfaceTun)
     return strncpy(buf, "tun0", IFNAMSIZ);
   return strncpy(buf, "eth0", IFNAMSIZ);
 }
 
+std::string TestGetInterfaceSSID(const std::string& ifname) {
+  return (ifname == kInterfaceWithNoSSID) ? "" : kWiFiSSID;
+}
+
 }  // namespace
 
 typedef std::vector<char> Buffer;
 
 class AddressTrackerLinuxTest : public testing::Test {
  protected:
   AddressTrackerLinuxTest() {}
 
   void InitializeAddressTracker(bool tracking) {
     if (tracking) {
@@ skipping 504 matching lines @@
   EXPECT_TRUE(HandleLinkMessage(buffer));
   EXPECT_EQ(1u, GetOnlineLinks().count(0));
   EXPECT_EQ(1u, GetOnlineLinks().size());
 }
 
 TEST_F(AddressTrackerLinuxTest, NonTrackingModeInit) {
   AddressTrackerLinux tracker;
   tracker.Init();
 }
 
+TEST_F(AddressTrackerLinuxTest, ConnectionSSIDFromInterfaceList) {
+  NetworkInterfaceList list;
+  EXPECT_EQ(std::string(), AddressTrackerLinux::ConnectionSSIDFromInterfaceList(
+                               list, TestGetInterfaceSSID));
+
+  NetworkInterface interface1;
+  interface1.name = "wlan0";
+  interface1.type = NetworkChangeNotifier::CONNECTION_WIFI;
+  list.push_back(interface1);
+  ASSERT_EQ(1u, list.size());
+  EXPECT_EQ(std::string(kWiFiSSID),
+            AddressTrackerLinux::ConnectionSSIDFromInterfaceList(
+                list, TestGetInterfaceSSID));
+
+  NetworkInterface interface2;
+  interface2.name = "wlan1";
+  interface2.type = NetworkChangeNotifier::CONNECTION_WIFI;
+  list.push_back(interface2);
+  ASSERT_EQ(2u, list.size());
+  EXPECT_EQ(std::string(kWiFiSSID),
+            AddressTrackerLinux::ConnectionSSIDFromInterfaceList(
+                list, TestGetInterfaceSSID));
+
+  NetworkInterface interface3;
+  interface3.name = kInterfaceWithNoSSID;
+  interface3.type = NetworkChangeNotifier::CONNECTION_WIFI;
+  list.push_back(interface3);
+  ASSERT_EQ(3u, list.size());
+  EXPECT_EQ(std::string(), AddressTrackerLinux::ConnectionSSIDFromInterfaceList(
+                               list, TestGetInterfaceSSID));
+
+  list.pop_back();
+  NetworkInterface interface4;
+  interface4.name = "eth0";
+  interface4.type = NetworkChangeNotifier::CONNECTION_ETHERNET;
+  list.push_back(interface4);
+  ASSERT_EQ(3u, list.size());
+  EXPECT_EQ(std::string(), AddressTrackerLinux::ConnectionSSIDFromInterfaceList(
+                               list, TestGetInterfaceSSID));
+}
+
 }  // namespace
 
 }  // namespace internal
 }  // namespace net