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

chrome/browser/automation/automation_provider_observers.h

 // 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.
 
 #ifndef CHROME_BROWSER_AUTOMATION_AUTOMATION_PROVIDER_OBSERVERS_H_
 #define CHROME_BROWSER_AUTOMATION_AUTOMATION_PROVIDER_OBSERVERS_H_
 #pragma once
 
 #include <deque>
 #include <map>
@@ skipping 83 matching lines @@
 };
 
 #if defined(OS_CHROMEOS)
 // Watches for NetworkManager events. Because NetworkLibrary loads
 // asynchronously, this is used to make sure it is done before tests are run.
 class NetworkManagerInitObserver
     : public chromeos::NetworkLibrary::NetworkManagerObserver {
  public:
   explicit NetworkManagerInitObserver(AutomationProvider* automation);
   virtual ~NetworkManagerInitObserver();
+  virtual bool Init();
   virtual void OnNetworkManagerChanged(chromeos::NetworkLibrary* obj);
 
  private:
   base::WeakPtr<AutomationProvider> automation_;
 
   DISALLOW_COPY_AND_ASSIGN(NetworkManagerInitObserver);
 };
 #endif  // defined(OS_CHROMEOS)
 
 // Watches for NewTabUI page loads for performance timing purposes.
@@ skipping 581 matching lines @@
                        const NotificationDetails& details);
 
  private:
   NotificationRegistrar registrar_;
   AutomationProvider* automation_;
   IPC::Message* reply_message_;
   bool lock_screen_;
 
   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.
 class AutomationProviderBookmarkModelObserver : BookmarkModelObserver {
  public:
   AutomationProviderBookmarkModelObserver(AutomationProvider* provider,
                                           IPC::Message* reply_message,
                                           BookmarkModel* model);
   virtual ~AutomationProviderBookmarkModelObserver();
 
@@ skipping 462 matching lines @@
   // Sends the |reply_message_| to |automation_| indicating we're done.
   void RunOnUIThread();
 
   base::WeakPtr<AutomationProvider> automation_;
   scoped_ptr<IPC::Message> reply_message_;
 
   DISALLOW_COPY_AND_ASSIGN(WaitForProcessLauncherThreadToGoIdleObserver);
 };
 
 #endif  // CHROME_BROWSER_AUTOMATION_AUTOMATION_PROVIDER_OBSERVERS_H_