Added retry policy

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

Index: chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_mock.cc
diff --git a/chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_mock.cc b/chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_mock.cc
index b3a45e2ef0e9c15cc93c1315f838e1b9c34b71e3..4adc164993bf9276f22c7a7966a2c937003ffeb0 100644
--- a/chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_mock.cc
+++ b/chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_mock.cc
@@ -3,7 +3,9 @@
 // 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 {
 
@@ -18,4 +20,14 @@ 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