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

chrome/browser/automation/automation_provider_observers.h

Index: chrome/browser/automation/automation_provider_observers.h
diff --git a/chrome/browser/automation/automation_provider_observers.h b/chrome/browser/automation/automation_provider_observers.h
index 2eff551ee352445e39f5b96023c8bd68d6d537fb..4d9d2636775a8ac76c601dca83337fa8e52197e7 100644
--- a/chrome/browser/automation/automation_provider_observers.h
+++ b/chrome/browser/automation/automation_provider_observers.h
@@ -101,6 +101,7 @@ class NetworkManagerInitObserver
  public:
   explicit NetworkManagerInitObserver(AutomationProvider* automation);
   virtual ~NetworkManagerInitObserver();
+  virtual bool Init();
   virtual void OnNetworkManagerChanged(chromeos::NetworkLibrary* obj);
 
  private:
@@ -702,6 +703,46 @@ class ScreenLockUnlockObserver : public NotificationObserver {
 
   DISALLOW_COPY_AND_ASSIGN(ScreenLockUnlockObserver);
 };
+
+class NetworkScanObserver
+    : public chromeos::NetworkLibrary::NetworkManagerObserver {
+ public:
+  NetworkScanObserver(AutomationProvider* automation,
+                      IPC::Message* reply_message);
+
+  virtual ~NetworkScanObserver();
+
+  // NetworkLibrary::NetworkManagerObserver implementation.
+  virtual void OnNetworkManagerChanged(chromeos::NetworkLibrary* obj);
+
+ private:
+  AutomationProvider* automation_;
+  IPC::Message* reply_message_;
+
+  DISALLOW_COPY_AND_ASSIGN(NetworkScanObserver);
+};
+
+// Waits for a connection success or failure for the specified
+// network and returns the status to the automation provider.
+class NetworkConnectObserver
+    : public chromeos::NetworkLibrary::NetworkManagerObserver {
+ public:
+  NetworkConnectObserver(AutomationProvider* automation,
+                         IPC::Message* reply_message,
+                         const std::string& service_path);
+
+  virtual ~NetworkConnectObserver();
+
+  // NetworkLibrary::NetworkManagerObserver implementation.
+  virtual void OnNetworkManagerChanged(chromeos::NetworkLibrary* obj);
+
+ private:
+  AutomationProvider* automation_;
+  IPC::Message* reply_message_;
+  std::string service_path_;
+
+  DISALLOW_COPY_AND_ASSIGN(NetworkConnectObserver);
+};
 #endif  // defined(OS_CHROMEOS)
 
 // Waits for the bookmark model to load.