PyAuto automation hooks: blocking wifi connect, disconnect, and network scan.

chrome/browser/automation/testing_automation_provider_chromeos.cc

 // Copyright (c) 2011 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 "chrome/browser/automation/testing_automation_provider.h"
 
 #include "base/values.h"
 #include "chrome/browser/automation/automation_provider_json.h"
 #include "chrome/browser/automation/automation_provider_observers.h"
 #include "chrome/browser/chromeos/cros/cros_library.h"
@@ skipping 48 matching lines @@
   new LoginManagerObserver(this, reply_message);
   controller->LoginAsGuest();
 }
 
 void TestingAutomationProvider::Login(DictionaryValue* args,
                                       IPC::Message* reply_message) {
   std::string username, password;
   if (!args->GetString("username", &username) ||
       !args->GetString("password", &password)) {
     AutomationJSONReply(this, reply_message).SendError(
-        "Invalid or missing args");
+        "Invalid or missing args.");
     return;
   }
 
   chromeos::ExistingUserController* controller =
       chromeos::ExistingUserController::current_controller();
   // Set up an observer (it will delete itself).
   new LoginManagerObserver(this, reply_message);
   controller->Login(username, password);
 }
 
@@ skipping 109 matching lines @@
       item->SetString("roaming_state",
                       cellular_networks[i]->GetRoamingStateString());
       items->Append(item);
     }
     return_value->Set("cellular_networks", items);
   }
 
   reply.SendSuccess(return_value.get());
 }
 
+void TestingAutomationProvider::NetworkScan(DictionaryValue* args,
+                                            IPC::Message* reply_message) {
+  AutomationJSONReply reply(this, reply_message);
+
+  if (!CrosLibrary::Get()->EnsureLoaded()) {
+    reply.SendError("Could not load cros library.");
+    return;
+  }
+
+  NetworkLibrary* network_library = CrosLibrary::Get()->GetNetworkLibrary();
+  network_library->RequestNetworkScan();
+
+  // Set up an observer (it will delete itself).
+  new NetworkScanObserver(this, reply_message);
+}
+
 void TestingAutomationProvider::ConnectToWifiNetwork(
     DictionaryValue* args, IPC::Message* reply_message) {
   AutomationJSONReply reply(this, reply_message);
   std::string service_path, password, identity, certpath;
   if (!args->GetString("service_path", &service_path) ||
       !args->GetString("password", &password) ||
       !args->GetString("identity", &identity) ||
       !args->GetString("certpath", &certpath)) {
-    reply.SendError("Invalid or missing args");
+    reply.SendError("Invalid or missing args.");
     return;
   }
 
   if (!CrosLibrary::Get()->EnsureLoaded()) {
     reply.SendError("Could not load cros library.");
     return;
   }
 
   NetworkLibrary* network_library = CrosLibrary::Get()->GetNetworkLibrary();
   chromeos::WifiNetwork* wifi =
       network_library->FindWifiNetworkByPath(service_path);
   if (!wifi) {
-    reply.SendError("Failed to connect");
+    reply.SendError("No network found with specified service path.");
     return;
   }
   if (!password.empty())
     wifi->SetPassphrase(password);
   if (!identity.empty())
     wifi->SetIdentity(identity);
   if (!certpath.empty())
     wifi->SetCertPath(certpath);
-  network_library->ConnectToWifiNetwork(service_path);
 
-  // TODO(stevenjb): Observe the network library and check for a successful
-  // connection.
+  // Set up an observer (it will delete itself).
+  new NetworkConnectObserver(this, reply_message, service_path);
 
+  network_library->ConnectToWifiNetwork(wifi);
+  network_library->RequestNetworkScan();
+}
+
+void TestingAutomationProvider::DisconnectFromWifiNetwork(
+    DictionaryValue* args, IPC::Message* reply_message) {
+  AutomationJSONReply reply(this, reply_message);
+  std::string service_path;
+  if (!args->GetString("service_path", &service_path)) {
+    reply.SendError("Invalid or missing args.");
+    return;
+  }
+
+  if (!CrosLibrary::Get()->EnsureLoaded()) {
+    reply.SendError("Could not load cros library.");
+    return;
+  }
+
+  NetworkLibrary* network_library = CrosLibrary::Get()->GetNetworkLibrary();
+  chromeos::WifiNetwork* wifi =
+      network_library->FindWifiNetworkByPath(service_path);
+  if (!wifi) {
+    reply.SendError("No network found with specified service path.");
+    return;
+  }
+
+  network_library->DisconnectFromWirelessNetwork(wifi);
   reply.SendSuccess(NULL);
 }