Implemented consumer management unenrollment.

chrome/browser/chromeos/policy/consumer_management_notifier_unittest.cc

Index: chrome/browser/chromeos/policy/consumer_management_notifier_unittest.cc
diff --git a/chrome/browser/chromeos/policy/consumer_management_notifier_unittest.cc b/chrome/browser/chromeos/policy/consumer_management_notifier_unittest.cc
index 899855dccca5272ad6a60fd7ae82452d03771403..af9bdb41a7ae16e8cddf11d8a5ff811d4637f0e5 100644
--- a/chrome/browser/chromeos/policy/consumer_management_notifier_unittest.cc
+++ b/chrome/browser/chromeos/policy/consumer_management_notifier_unittest.cc
@@ -63,12 +63,19 @@ class ConsumerManagementNotifierTest : public BrowserWithTestWindowTest {
         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));
@@ -80,7 +87,21 @@ TEST_F(ConsumerManagementNotifierTest, ShowsNotificationWhenCreated) {
   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(
@@ -98,4 +119,25 @@ TEST_F(ConsumerManagementNotifierTest, ShowsNotificationWhenStatusChanged) {
   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