Added retry policy

chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_mock.cc

 // Copyright 2015 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/chromeos/login/enrollment/enterprise_enrollment_helper_mock.h"
+#include "chrome/browser/chromeos/login/enrollment/enrollment_screen.h"
 #include "chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_impl.h"
+#include "chrome/browser/chromeos/policy/enrollment_status_chromeos.h"
 
 namespace chromeos {
 
 EnterpriseEnrollmentHelperMock::EnterpriseEnrollmentHelperMock(
     EnrollmentStatusConsumer* status_consumer)
     : EnterpriseEnrollmentHelper(status_consumer) {}
 
 EnterpriseEnrollmentHelperMock::~EnterpriseEnrollmentHelperMock() {}
 
 EnterpriseEnrollmentHelper::EnrollmentStatusConsumer*
 EnterpriseEnrollmentHelperMock::status_consumer() {
   return EnterpriseEnrollmentHelper::status_consumer();
 }
 
+void EnterpriseEnrollmentHelperMock::SimulateEnrollmentSuccess() {
+  static_cast<EnrollmentScreen*>(status_consumer())
+      ->ShowEnrollmentStatusOnSuccess();
+}
+
+void EnterpriseEnrollmentHelperMock::SimulateEnrollmentFailure() {
+  status_consumer()->OnEnrollmentError(policy::EnrollmentStatus::ForStatus(
+      policy::EnrollmentStatus::REGISTRATION_FAILED));
+}
+
 }  // namespace chromeos