Index: chrome/browser/chromeos/policy/fake_consumer_management_service.cc |
diff --git a/chrome/browser/chromeos/policy/fake_consumer_management_service.cc b/chrome/browser/chromeos/policy/fake_consumer_management_service.cc |
index 982a09bab7006bddeee164c671a42e85ee36d87f..7b12fb8bb0a59d6c9b095d47aa2b57847c0db5a6 100644 |
--- a/chrome/browser/chromeos/policy/fake_consumer_management_service.cc |
+++ b/chrome/browser/chromeos/policy/fake_consumer_management_service.cc |
@@ -9,13 +9,18 @@ namespace policy { |
FakeConsumerManagementService::FakeConsumerManagementService() |
: ConsumerManagementService(NULL, NULL), |
status_(STATUS_UNKNOWN), |
- enrollment_stage_(ENROLLMENT_STAGE_NONE), |
- enrollment_stage_before_reset_(ENROLLMENT_STAGE_NONE) { |
+ enrollment_stage_(ENROLLMENT_STAGE_NONE) { |
} |
FakeConsumerManagementService::~FakeConsumerManagementService() { |
} |
+void FakeConsumerManagementService::SetStatusAndEnrollmentStage( |
+ Status status, EnrollmentStage stage) { |
+ status_ = status; |
+ SetEnrollmentStage(stage); |
+} |
+ |
ConsumerManagementService::Status |
FakeConsumerManagementService::GetStatus() const { |
return status_; |
@@ -27,9 +32,8 @@ FakeConsumerManagementService::GetEnrollmentStage() const { |
} |
void FakeConsumerManagementService::SetEnrollmentStage(EnrollmentStage stage) { |
- if (stage == ENROLLMENT_STAGE_NONE) |
- enrollment_stage_before_reset_ = enrollment_stage_; |
enrollment_stage_ = stage; |
+ NotifyStatusChanged(); |
} |
} // namespace policy |