WebDriver extension support in TestingAutomationProvider.

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 1626 matching lines @@
  private:
   content::NotificationRegistrar registrar_;
   base::WeakPtr<AutomationProvider> automation_;
   scoped_ptr<IPC::Message> reply_message_;
   int event_type_;
   int count_;
 
   DISALLOW_COPY_AND_ASSIGN(InputEventAckNotificationObserver);
 };
 
-// Allows the automation provider to wait for all the tabs to finish any
-// pending loads. This only waits for tabs that exist at the observer's
-// creation. Will send a message on construction if no tabs are loading
-// currently.
-class AllTabsStoppedLoadingObserver : public TabEventObserver,
+// Allows the automation provider to wait for all render views to finish any
+// pending loads. This wait is only guaranteed for views that exist at the
+// observer's creation. Will send a message on construction if no views are
+// loading currently.

[dennis_jeffrey, 2011/12/06 00:49:08]

nit: 'loading currently' --> 'currently loading'

[kkania, 2011/12/12 17:10:06]

Done.

+class AllViewsStoppedLoadingObserver : public TabEventObserver,
                                       public content::NotificationObserver {

[dennis_jeffrey, 2011/12/06 00:49:08]

indent by 1 more space

[kkania, 2011/12/12 17:10:06]

Done.

  public:
-  AllTabsStoppedLoadingObserver(AutomationProvider* automation,
-                                IPC::Message* reply_message);
-  virtual ~AllTabsStoppedLoadingObserver();
+  AllViewsStoppedLoadingObserver(
+      AutomationProvider* automation,
+      IPC::Message* reply_message,
+      ExtensionProcessManager* extension_process_manager);
+  virtual ~AllViewsStoppedLoadingObserver();
 
   // TabEventObserver implementation.
   virtual void OnFirstPendingLoad(TabContents* tab_contents) OVERRIDE;
   virtual void OnNoMorePendingLoads(TabContents* tab_contents) OVERRIDE;
 
   // content::NotificationObserver implementation.
   virtual void Observe(int type,
                        const content::NotificationSource& source,
                        const content::NotificationDetails& details) OVERRIDE;
 
  private:
   typedef std::set<TabContents*> TabSet;
 
   // Checks if there are no pending loads. If none, it will send an automation
   // relpy and delete itself.
   void CheckIfNoMorePendingLoads();
 
-  TabSet pending_tabs_;
-  content::NotificationRegistrar registrar_;
   base::WeakPtr<AutomationProvider> automation_;
   scoped_ptr<IPC::Message> reply_message_;
+  ExtensionProcessManager* extension_process_manager_;
+  content::NotificationRegistrar registrar_;
+  TabSet pending_tabs_;
 
-  DISALLOW_COPY_AND_ASSIGN(AllTabsStoppedLoadingObserver);
+  DISALLOW_COPY_AND_ASSIGN(AllViewsStoppedLoadingObserver);
 };
 
 // Observer used to listen for new tab creation to complete.
 class NewTabObserver : public content::NotificationObserver {
  public:
   NewTabObserver(AutomationProvider* automation, IPC::Message* reply_message);
 
   virtual void Observe(int type,
                        const content::NotificationSource& source,
                        const content::NotificationDetails& details) OVERRIDE;
@@ skipping 142 matching lines @@
 
   base::WeakPtr<AutomationProvider> automation_;
   scoped_ptr<IPC::Message> reply_message_;
   std::vector<policy::ConfigurationPolicyObserverRegistrar*> registrars_;
 
   DISALLOW_COPY_AND_ASSIGN(PolicyUpdatesObserver);
 };
 
 #endif  // defined(ENABLE_CONFIGURATION_POLICY)
 
+// Waits for an extension popup to appear and load.
+class ExtensionPopupObserver : public content::NotificationObserver {
+ public:
+  ExtensionPopupObserver(
+      AutomationProvider* automation,
+      IPC::Message* reply_message,
+      const std::string& extension_id);
+  ~ExtensionPopupObserver();
+
+  virtual void Observe(int type,
+                       const content::NotificationSource& source,
+                       const content::NotificationDetails& details) OVERRIDE;
+
+ private:
+  base::WeakPtr<AutomationProvider> automation_;
+  scoped_ptr<IPC::Message> reply_message_;
+  std::string extension_id_;
+  content::NotificationRegistrar registrar_;
+
+  DISALLOW_COPY_AND_ASSIGN(ExtensionPopupObserver);
+};
+
 #endif  // CHROME_BROWSER_AUTOMATION_AUTOMATION_PROVIDER_OBSERVERS_H_