Move the notification part out of ConsumerEnrollmentHandler so that it can be reused for unenrollme…

chrome/browser/chromeos/policy/consumer_enrollment_handler_factory_unittest.cc

 // 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.
 
 #include "chrome/browser/chromeos/policy/consumer_enrollment_handler_factory.h"
 
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_process_platform_part.h"
 #include "chrome/browser/chromeos/login/users/fake_user_manager.h"
 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h"
@@ skipping 13 matching lines @@
 
 class ConsumerEnrollmentHandlerFactoryTest : public testing::Test {
  public:
   ConsumerEnrollmentHandlerFactoryTest()
       : fake_service_(new FakeConsumerManagementService()),
         fake_user_manager_(new chromeos::FakeUserManager()),
         scoped_user_manager_enabler_(fake_user_manager_),
         testing_profile_manager_(new TestingProfileManager(
             TestingBrowserProcess::GetGlobal())) {
     // Set up FakeConsumerManagementService.
-    fake_service_->set_status(ConsumerManagementService::STATUS_ENROLLING);
-    fake_service_->SetEnrollmentStage(
+    fake_service_->SetStatusAndEnrollmentStage(
+        ConsumerManagementService::STATUS_ENROLLING,
         ConsumerManagementService::ENROLLMENT_STAGE_OWNER_STORED);
 
     // Inject fake objects.
     BrowserPolicyConnectorChromeOS* connector =
         g_browser_process->platform_part()->browser_policy_connector_chromeos();
     connector->SetConsumerManagementServiceForTesting(
         make_scoped_ptr(fake_service_));
 
     // Set up FakeUserManager.
     fake_user_manager_->AddUser(kTestOwner);
@@ skipping 16 matching lines @@
   EXPECT_TRUE(ConsumerEnrollmentHandlerFactory::GetForBrowserContext(profile));
 }
 
 TEST_F(ConsumerEnrollmentHandlerFactoryTest, ServiceIsNotCreatedForNonOwner) {
   Profile* profile = testing_profile_manager_->CreateTestingProfile(kTestUser);
   EXPECT_FALSE(ConsumerEnrollmentHandlerFactory::GetForBrowserContext(profile));
 }
 
 TEST_F(ConsumerEnrollmentHandlerFactoryTest,
        ServiceIsNotCreatedIfItHasNothingToDo) {
-  fake_service_->set_status(ConsumerManagementService::STATUS_UNENROLLED);
-  fake_service_->SetEnrollmentStage(
+  fake_service_->SetStatusAndEnrollmentStage(
+      ConsumerManagementService::STATUS_UNENROLLED,
       ConsumerManagementService::ENROLLMENT_STAGE_NONE);
 
   Profile* profile = testing_profile_manager_->CreateTestingProfile(kTestOwner);
   EXPECT_FALSE(ConsumerEnrollmentHandlerFactory::GetForBrowserContext(profile));
 }
 
 }  // namespace policy