Rewrite of the EnrollEnterpriseDevice PyAuto automation hook for WebUI.

chrome/browser/automation/automation_event_observers_chromeos.cc

 // 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.
 
 #include "chrome/browser/automation/automation_event_observers.h"
 #include "chrome/browser/browser_process.h"
+#include "chrome/browser/chromeos/login/enrollment/enterprise_enrollment_screen_actor.h"
 #include "chrome/browser/chromeos/login/existing_user_controller.h"
 
 LoginEventObserver::LoginEventObserver(
     AutomationEventQueue* event_queue,
     chromeos::ExistingUserController* controller,
     AutomationProvider* automation)
     : AutomationEventObserver(event_queue, false),
       controller_(controller),
       automation_(automation->AsWeakPtr()) {
   controller_->set_login_status_consumer(this);
@@ skipping 20 matching lines @@
 
 void LoginEventObserver::_NotifyLoginEvent(const std::string& error_string) {
   DictionaryValue* dict = new DictionaryValue;
   dict->SetString("type", "login_event");
   if (error_string.length())
     dict->SetString("error_string", error_string);
   NotifyEvent(dict);
   controller_->set_login_status_consumer(NULL);
   RemoveIfDone();
 }
+
+EnrollmentEventObserver::EnrollmentEventObserver(
+    AutomationEventQueue* event_queue,
+    chromeos::EnterpriseEnrollmentScreenActor* enrollment_screen_actor)
+    : AutomationEventObserver(event_queue, false) {
+  enrollment_screen_actor->AddObserver(this);
+}
+
+EnrollmentEventObserver::~EnrollmentEventObserver() {}
+
+void EnrollmentEventObserver::OnEnrollmentComplete(
+    chromeos::EnterpriseEnrollmentScreenActor* enrollment_screen_actor,
+    bool succeeded,
+    const std::string& error_string) {
+  DictionaryValue* dict = new DictionaryValue;
+  enrollment_screen_actor->RemoveObserver(this);
+  if (!succeeded)
+    dict->SetString("error_string", error_string);
+  NotifyEvent(dict);
+  RemoveIfDone();
+}