Fix update and enrollment flow

components/pairing/fake_host_pairing_controller.h

 // Copyright 2014 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 COMPONENTS_PAIRING_FAKE_HOST_PAIRING_CONTROLLER_H_
 #define COMPONENTS_PAIRING_FAKE_HOST_PAIRING_CONTROLLER_H_
 
 #include "base/macros.h"
 #include "base/observer_list.h"
 #include "base/time/time.h"
@@ skipping 15 matching lines @@
   // * code - 6 digits or empty string. Default: empty string. If strings is
   // empty, random code is generated.
   // * device_name - string. Default: "Chromebox-01".
   // * domain - string. Default: "example.com".
   explicit FakeHostPairingController(const std::string& config);
   virtual ~FakeHostPairingController();
 
   // Applies given |config| to flow.
   void ApplyConfig(const std::string& config);
 
-  // Overridden from HostPairingFlow:
+ private:
+  void ChangeStage(Stage new_stage);
+  void ChangeStageLater(Stage new_stage);
+
+  // HostPairingController:
   virtual void AddObserver(Observer* observer) override;
   virtual void RemoveObserver(Observer* observer) override;
   virtual Stage GetCurrentStage() override;
   virtual void StartPairing() override;
   virtual std::string GetDeviceName() override;
   virtual std::string GetConfirmationCode() override;
   virtual std::string GetEnrollmentDomain() override;
   virtual void OnUpdateStatusChanged(UpdateStatus update_status) override;
-  virtual void SetEnrollmentComplete(bool success) override;
+  virtual void OnEnrollmentStatusChanged(
+      EnrollmentStatus enrollment_status) override;
 
- private:
-  void ChangeStage(Stage new_stage);
-  void ChangeStageLater(Stage new_stage);
-
-  // HostPairingFlow::Observer:
+  // HostPairingController::Observer:
   virtual void PairingStageChanged(Stage new_stage) override;
   virtual void ConfigureHost(bool accepted_eula,
                              const std::string& lang,
                              const std::string& timezone,
                              bool send_reports,
                              const std::string& keyboard_layout) override;
   virtual void EnrollHost(const std::string& auth_token) override;
 
   ObserverList<Observer> observers_;
   Stage current_stage_;
   std::string device_name_;
   std::string confirmation_code_;
   base::TimeDelta async_duration_;
 
   // If this flag is true error happens on |STAGE_ENROLLING| once.
   bool enrollment_should_fail_;
 
   // Controller starts its work like if update and reboot already happened.
   bool start_after_update_;
 
   std::string enrollment_domain_;
 
   DISALLOW_COPY_AND_ASSIGN(FakeHostPairingController);
 };
 
 }  // namespace pairing_chromeos
 
 #endif  // COMPONENTS_PAIRING_FAKE_HOST_PAIRING_CONTROLLER_H_