| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/chromeos/policy/consumer_management_notifier.h" | 5 #include "chrome/browser/chromeos/policy/consumer_management_notifier.h" |
| 6 | 6 |
| 7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
| 8 #include "chrome/browser/browser_process.h" | 8 #include "chrome/browser/browser_process.h" |
| 9 #include "chrome/browser/browser_process_platform_part.h" | 9 #include "chrome/browser/browser_process_platform_part.h" |
| 10 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h" | 10 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h" |
| (...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 57 notification_.reset( | 57 notification_.reset( |
| 58 new ConsumerManagementNotifier(profile(), fake_service_)); | 58 new ConsumerManagementNotifier(profile(), fake_service_)); |
| 59 } | 59 } |
| 60 | 60 |
| 61 bool HasEnrollmentNotification() { | 61 bool HasEnrollmentNotification() { |
| 62 return g_browser_process->notification_ui_manager()->FindById( | 62 return g_browser_process->notification_ui_manager()->FindById( |
| 63 "consumer_management.enroll", | 63 "consumer_management.enroll", |
| 64 NotificationUIManager::GetProfileID(profile())); | 64 NotificationUIManager::GetProfileID(profile())); |
| 65 } | 65 } |
| 66 | 66 |
| 67 bool HasUnenrollmentNotification() { | |
| 68 return g_browser_process->notification_ui_manager()->FindById( | |
| 69 "consumer_management.unenroll", | |
| 70 NotificationUIManager::GetProfileID(profile())); | |
| 71 } | |
| 72 | |
| 73 FakeConsumerManagementService* fake_service_; | 67 FakeConsumerManagementService* fake_service_; |
| 74 scoped_ptr<TestingProfileManager> testing_profile_manager_; | 68 scoped_ptr<TestingProfileManager> testing_profile_manager_; |
| 75 scoped_ptr<ConsumerManagementNotifier> notification_; | 69 scoped_ptr<ConsumerManagementNotifier> notification_; |
| 76 }; | 70 }; |
| 77 | 71 |
| 78 TEST_F(ConsumerManagementNotifierTest, | 72 TEST_F(ConsumerManagementNotifierTest, ShowsNotificationWhenCreated) { |
| 79 ShowsEnrollmentNotificationWhenCreated) { | |
| 80 fake_service_->SetStatusAndStage( | 73 fake_service_->SetStatusAndStage( |
| 81 ConsumerManagementService::STATUS_UNENROLLED, | 74 ConsumerManagementService::STATUS_UNENROLLED, |
| 82 ConsumerManagementStage::EnrollmentCanceled()); | 75 ConsumerManagementStage::EnrollmentCanceled()); |
| 83 EXPECT_FALSE(HasEnrollmentNotification()); | 76 EXPECT_FALSE(HasEnrollmentNotification()); |
| 84 EXPECT_FALSE(HasUnenrollmentNotification()); | |
| 85 | 77 |
| 86 CreateConsumerManagementNotifier(); | 78 CreateConsumerManagementNotifier(); |
| 87 | 79 |
| 88 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); | 80 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); |
| 89 EXPECT_TRUE(HasEnrollmentNotification()); | 81 EXPECT_TRUE(HasEnrollmentNotification()); |
| 90 EXPECT_FALSE(HasUnenrollmentNotification()); | |
| 91 } | 82 } |
| 92 | 83 |
| 93 TEST_F(ConsumerManagementNotifierTest, | 84 TEST_F(ConsumerManagementNotifierTest, ShowsNotificationWhenStatusChanged) { |
| 94 ShowsUnenrollmentNotificationWhenCreated) { | |
| 95 fake_service_->SetStatusAndStage( | |
| 96 ConsumerManagementService::STATUS_UNENROLLED, | |
| 97 ConsumerManagementStage::UnenrollmentSuccess()); | |
| 98 EXPECT_FALSE(HasEnrollmentNotification()); | |
| 99 EXPECT_FALSE(HasUnenrollmentNotification()); | |
| 100 | |
| 101 CreateConsumerManagementNotifier(); | |
| 102 | |
| 103 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); | |
| 104 EXPECT_FALSE(HasEnrollmentNotification()); | |
| 105 EXPECT_TRUE(HasUnenrollmentNotification()); | |
| 106 } | |
| 107 | |
| 108 TEST_F(ConsumerManagementNotifierTest, | |
| 109 ShowsEnrollmentNotificationWhenStatusChanged) { | |
| 110 fake_service_->SetStatusAndStage( | 85 fake_service_->SetStatusAndStage( |
| 111 ConsumerManagementService::STATUS_ENROLLING, | 86 ConsumerManagementService::STATUS_ENROLLING, |
| 112 ConsumerManagementStage::EnrollmentOwnerStored()); | 87 ConsumerManagementStage::EnrollmentOwnerStored()); |
| 113 | 88 |
| 114 CreateConsumerManagementNotifier(); | 89 CreateConsumerManagementNotifier(); |
| 115 EXPECT_EQ(ConsumerManagementStage::EnrollmentOwnerStored(), | 90 EXPECT_EQ(ConsumerManagementStage::EnrollmentOwnerStored(), |
| 116 fake_service_->GetStage()); | 91 fake_service_->GetStage()); |
| 117 EXPECT_FALSE(HasEnrollmentNotification()); | 92 EXPECT_FALSE(HasEnrollmentNotification()); |
| 118 EXPECT_FALSE(HasUnenrollmentNotification()); | |
| 119 | 93 |
| 120 fake_service_->SetStatusAndStage( | 94 fake_service_->SetStatusAndStage( |
| 121 ConsumerManagementService::STATUS_ENROLLED, | 95 ConsumerManagementService::STATUS_ENROLLED, |
| 122 ConsumerManagementStage::EnrollmentSuccess()); | 96 ConsumerManagementStage::EnrollmentSuccess()); |
| 123 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); | 97 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); |
| 124 EXPECT_TRUE(HasEnrollmentNotification()); | 98 EXPECT_TRUE(HasEnrollmentNotification()); |
| 125 EXPECT_FALSE(HasUnenrollmentNotification()); | |
| 126 } | |
| 127 | |
| 128 TEST_F(ConsumerManagementNotifierTest, | |
| 129 ShowsUnenrollmentNotificationWhenStatusChanged) { | |
| 130 fake_service_->SetStatusAndStage( | |
| 131 ConsumerManagementService::STATUS_ENROLLED, | |
| 132 ConsumerManagementStage::None()); | |
| 133 | |
| 134 CreateConsumerManagementNotifier(); | |
| 135 EXPECT_FALSE(HasEnrollmentNotification()); | |
| 136 EXPECT_FALSE(HasUnenrollmentNotification()); | |
| 137 | |
| 138 fake_service_->SetStatusAndStage( | |
| 139 ConsumerManagementService::STATUS_UNENROLLING, | |
| 140 ConsumerManagementStage::UnenrollmentRequested()); | |
| 141 EXPECT_FALSE(HasEnrollmentNotification()); | |
| 142 EXPECT_FALSE(HasUnenrollmentNotification()); | |
| 143 | |
| 144 fake_service_->SetStatusAndStage( | |
| 145 ConsumerManagementService::STATUS_UNENROLLED, | |
| 146 ConsumerManagementStage::UnenrollmentSuccess()); | |
| 147 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); | |
| 148 EXPECT_FALSE(HasEnrollmentNotification()); | |
| 149 EXPECT_TRUE(HasUnenrollmentNotification()); | |
| 150 } | 99 } |
| 151 | 100 |
| 152 } // namespace policy | 101 } // namespace policy |
| OLD | NEW |