Implemented consumer management unenrollment.

chrome/browser/chromeos/policy/consumer_management_notifier_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_management_notifier.h"
 
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_process_platform_part.h"
 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h"
 #include "chrome/browser/chromeos/policy/consumer_management_service.h"
@@ skipping 45 matching lines @@
     notification_.reset(
       new ConsumerManagementNotifier(profile(), fake_service_));
   }
 
   bool HasEnrollmentNotification() {
     return g_browser_process->notification_ui_manager()->FindById(
         "consumer_management.enroll",
         NotificationUIManager::GetProfileID(profile()));
   }
 
+  bool HasUnenrollmentNotification() {
+    return g_browser_process->notification_ui_manager()->FindById(
+        "consumer_management.unenroll",
+        NotificationUIManager::GetProfileID(profile()));
+  }
+
   FakeConsumerManagementService* fake_service_;
   scoped_ptr<TestingProfileManager> testing_profile_manager_;
   scoped_ptr<ConsumerManagementNotifier> notification_;
 };
 
-TEST_F(ConsumerManagementNotifierTest, ShowsNotificationWhenCreated) {
+TEST_F(ConsumerManagementNotifierTest,
+       ShowsEnrollmentNotificationWhenCreated) {
   fake_service_->SetStatusAndStage(
       ConsumerManagementService::STATUS_UNENROLLED,
       ConsumerManagementStage(ConsumerManagementStage::ENROLLMENT_CANCELED));
   EXPECT_FALSE(HasEnrollmentNotification());

[bartfab (slow), 2014/11/28 13:25:17]

Nit: Add a check to verify that the unenrollment notification was not triggered.

[davidyu, 2014/12/01 17:05:21]

Done.

 
   CreateConsumerManagementNotifier();
 
   EXPECT_EQ(ConsumerManagementStage::NONE, fake_service_->GetStage().value());
   EXPECT_TRUE(HasEnrollmentNotification());

[bartfab (slow), 2014/11/28 13:25:17]

Nit: Add a check to verify that the unenrollment notification was not triggered.

[davidyu, 2014/12/01 17:05:21]

Done.

 }
 
-TEST_F(ConsumerManagementNotifierTest, ShowsNotificationWhenStatusChanged) {
+TEST_F(ConsumerManagementNotifierTest,
+       ShowsUnenrollmentNotificationWhenCreated) {
+  fake_service_->SetStatusAndStage(
+      ConsumerManagementService::STATUS_UNENROLLED,
+      ConsumerManagementStage(ConsumerManagementStage::UNENROLLMENT_SUCCESS));
+  EXPECT_FALSE(HasUnenrollmentNotification());

[bartfab (slow), 2014/11/28 13:25:17]

Nit: Add a check to verify that the enrollment notification was not triggered.

[davidyu, 2014/12/01 17:05:21]

Done.

+
+  CreateConsumerManagementNotifier();
+
+  EXPECT_EQ(ConsumerManagementStage::NONE, fake_service_->GetStage().value());
+  EXPECT_TRUE(HasUnenrollmentNotification());

[bartfab (slow), 2014/11/28 13:25:17]

Nit: Add a check to verify that the enrollment notification was not triggered.

[davidyu, 2014/12/01 17:05:21]

Done.

+}
+
+TEST_F(ConsumerManagementNotifierTest,
+       ShowsEnrollmentNotificationWhenStatusChanged) {
   fake_service_->SetStatusAndStage(
       ConsumerManagementService::STATUS_ENROLLING,
       ConsumerManagementStage(
           ConsumerManagementStage::ENROLLMENT_OWNER_STORED));
 
   CreateConsumerManagementNotifier();
   EXPECT_EQ(ConsumerManagementStage::ENROLLMENT_OWNER_STORED,
             fake_service_->GetStage().value());
   EXPECT_FALSE(HasEnrollmentNotification());

[bartfab (slow), 2014/11/28 13:25:17]

Nit: Add a check to verify that the unenrollment notification was not triggered.

[davidyu, 2014/12/01 17:05:21]

Done.

 
   fake_service_->SetStatusAndStage(
       ConsumerManagementService::STATUS_ENROLLED,
       ConsumerManagementStage(ConsumerManagementStage::ENROLLMENT_SUCCESS));
   EXPECT_EQ(ConsumerManagementStage::NONE, fake_service_->GetStage().value());
   EXPECT_TRUE(HasEnrollmentNotification());

[bartfab (slow), 2014/11/28 13:25:17]

Nit: Add a check to verify that the unenrollment notification was not triggered.

[davidyu, 2014/12/01 17:05:21]

Done.

 }
 
+TEST_F(ConsumerManagementNotifierTest,
+       ShowsUnenrollmentNotificationWhenStatusChanged) {
+  fake_service_->SetStatusAndStage(
+      ConsumerManagementService::STATUS_ENROLLED,
+      ConsumerManagementStage(ConsumerManagementStage::NONE));
+
+  CreateConsumerManagementNotifier();
+  EXPECT_FALSE(HasUnenrollmentNotification());

[bartfab (slow), 2014/11/28 13:25:17]

Nit: Add a check to verify that the enrollment notification was not triggered.

[davidyu, 2014/12/01 17:05:21]

Done.

+
+  fake_service_->SetStatusAndStage(
+      ConsumerManagementService::STATUS_UNENROLLING,
+      ConsumerManagementStage(ConsumerManagementStage::UNENROLLMENT_REQUESTED));
+  EXPECT_FALSE(HasUnenrollmentNotification());

[bartfab (slow), 2014/11/28 13:25:17]

Nit: Add a check to verify that the enrollment notification was not triggered.

[davidyu, 2014/12/01 17:05:21]

Done.

+
+  fake_service_->SetStatusAndStage(
+      ConsumerManagementService::STATUS_UNENROLLED,
+      ConsumerManagementStage(ConsumerManagementStage::UNENROLLMENT_SUCCESS));
+  EXPECT_EQ(ConsumerManagementStage::NONE, fake_service_->GetStage().value());
+  EXPECT_TRUE(HasUnenrollmentNotification());

[bartfab (slow), 2014/11/28 13:25:17]

Nit: Add a check to verify that the enrollment notification was not triggered.

[davidyu, 2014/12/01 17:05:21]

Done.

+}
+
 }  // namespace policy