Rewrite of the EnrollEnterpriseDevice PyAuto automation hook for WebUI.

chrome/browser/automation/automation_event_observers.h

 // 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.
 
 #ifndef CHROME_BROWSER_AUTOMATION_AUTOMATION_EVENT_OBSERVERS_H_
 #define CHROME_BROWSER_AUTOMATION_AUTOMATION_EVENT_OBSERVERS_H_
 
 #include "base/memory/scoped_ptr.h"
 #include "base/values.h"
 #include "chrome/browser/automation/automation_event_queue.h"
 #include "chrome/browser/automation/automation_provider.h"
 #include "chrome/browser/automation/automation_provider_observers.h"
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/chromeos/login/login_status_consumer.h"
 #endif  // defined(OS_CHROMEOS)
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 
 #if defined(OS_CHROMEOS)
 namespace chromeos {
+class EnterpriseEnrollmentScreenActor;
 class ExistingUserController;
 }
 #endif  // defined(OS_CHROMEOS)
 
 // AutomationEventObserver watches for a specific event, and pushes an
 // AutomationEvent into the AutomationEventQueue for each occurance.
 class AutomationEventObserver {
  public:
   explicit AutomationEventObserver(AutomationEventQueue* event_queue,
                                    bool recurring);
@@ skipping 55 matching lines @@
  public:
   LoginEventObserver(AutomationEventQueue* event_queue,
                      chromeos::ExistingUserController* controller,
                      AutomationProvider* automation);
   virtual ~LoginEventObserver();
 
   virtual void OnLoginFailure(const chromeos::LoginFailure& error) OVERRIDE;
 
   virtual void OnLoginSuccess(const std::string& username,
                               const std::string& password,
-                              bool pending_requests, bool using_oauth) OVERRIDE;
+                              bool pending_requests,
+                              bool using_oauth) OVERRIDE;
 
  private:
   chromeos::ExistingUserController* controller_;
   base::WeakPtr<AutomationProvider> automation_;
 
   void _NotifyLoginEvent(const std::string& error_string);
 
   DISALLOW_COPY_AND_ASSIGN(LoginEventObserver);
 };
 
+// Event observer that listens for enrollment to complete.
+class EnrollmentEventObserver
+    : public AutomationEventObserver,
+      public chromeos::EnterpriseEnrollmentScreenActor::Observer {
+ public:
+  EnrollmentEventObserver(
+      AutomationEventQueue* event_queue,
+      chromeos::EnterpriseEnrollmentScreenActor* enrollment_screen_actor);
+
+  virtual ~EnrollmentEventObserver();
+
+  // chromeos::EnterpriseEnrollmentScreenActor::Observer implementation.
+  virtual void OnEnrollmentComplete(
+      chromeos::EnterpriseEnrollmentScreenActor* enrollment_screen_actor,
+      bool succeeded,
+      const std::string& error_string);
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(EnrollmentEventObserver);
+};
+
 #endif  // defined(OS_CHROMEOS)
 
 #endif  // CHROME_BROWSER_AUTOMATION_AUTOMATION_EVENT_OBSERVERS_H_