| 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_enrollment_handler_factory.h" | 5 #include "chrome/browser/chromeos/policy/consumer_enrollment_handler_factory.h" |
| 6 | 6 |
| 7 #include "chrome/browser/browser_process.h" | 7 #include "chrome/browser/browser_process.h" |
| 8 #include "chrome/browser/browser_process_platform_part.h" | 8 #include "chrome/browser/browser_process_platform_part.h" |
| 9 #include "chrome/browser/chromeos/login/users/fake_user_manager.h" | 9 #include "chrome/browser/chromeos/login/users/fake_chrome_user_manager.h" |
| 10 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h" | 10 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h" |
| 11 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h" | 11 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h" |
| 12 #include "chrome/browser/chromeos/policy/consumer_management_service.h" | 12 #include "chrome/browser/chromeos/policy/consumer_management_service.h" |
| 13 #include "chrome/browser/chromeos/policy/consumer_management_stage.h" | 13 #include "chrome/browser/chromeos/policy/consumer_management_stage.h" |
| 14 #include "chrome/browser/chromeos/policy/fake_consumer_management_service.h" | 14 #include "chrome/browser/chromeos/policy/fake_consumer_management_service.h" |
| 15 #include "chrome/test/base/testing_browser_process.h" | 15 #include "chrome/test/base/testing_browser_process.h" |
| 16 #include "chrome/test/base/testing_profile_manager.h" | 16 #include "chrome/test/base/testing_profile_manager.h" |
| 17 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 18 | 18 |
| 19 namespace { | 19 namespace { |
| 20 const char* kTestOwner = "test.owner@chromium.org.test"; | 20 const char* kTestOwner = "test.owner@chromium.org.test"; |
| 21 const char* kTestUser = "test.user@chromium.org.test"; | 21 const char* kTestUser = "test.user@chromium.org.test"; |
| 22 } | 22 } |
| 23 | 23 |
| 24 namespace policy { | 24 namespace policy { |
| 25 | 25 |
| 26 class ConsumerEnrollmentHandlerFactoryTest : public testing::Test { | 26 class ConsumerEnrollmentHandlerFactoryTest : public testing::Test { |
| 27 public: | 27 public: |
| 28 ConsumerEnrollmentHandlerFactoryTest() | 28 ConsumerEnrollmentHandlerFactoryTest() |
| 29 : fake_service_(new FakeConsumerManagementService()), | 29 : fake_service_(new FakeConsumerManagementService()), |
| 30 fake_user_manager_(new chromeos::FakeUserManager()), | 30 fake_user_manager_(new chromeos::FakeChromeUserManager()), |
| 31 scoped_user_manager_enabler_(fake_user_manager_), | 31 scoped_user_manager_enabler_(fake_user_manager_), |
| 32 testing_profile_manager_(new TestingProfileManager( | 32 testing_profile_manager_( |
| 33 TestingBrowserProcess::GetGlobal())) { | 33 new TestingProfileManager(TestingBrowserProcess::GetGlobal())) { |
| 34 // Set up FakeConsumerManagementService. | 34 // Set up FakeConsumerManagementService. |
| 35 fake_service_->SetStatusAndStage( | 35 fake_service_->SetStatusAndStage( |
| 36 ConsumerManagementService::STATUS_ENROLLING, | 36 ConsumerManagementService::STATUS_ENROLLING, |
| 37 ConsumerManagementStage::EnrollmentOwnerStored()); | 37 ConsumerManagementStage::EnrollmentOwnerStored()); |
| 38 | 38 |
| 39 // Inject fake objects. | 39 // Inject fake objects. |
| 40 BrowserPolicyConnectorChromeOS* connector = | 40 BrowserPolicyConnectorChromeOS* connector = |
| 41 g_browser_process->platform_part()->browser_policy_connector_chromeos(); | 41 g_browser_process->platform_part()->browser_policy_connector_chromeos(); |
| 42 connector->SetConsumerManagementServiceForTesting( | 42 connector->SetConsumerManagementServiceForTesting( |
| 43 make_scoped_ptr(fake_service_)); | 43 make_scoped_ptr(fake_service_)); |
| 44 | 44 |
| 45 // Set up FakeUserManager. | 45 // Set up FakeChromeUserManager. |
| 46 fake_user_manager_->AddUser(kTestOwner); | 46 fake_user_manager_->AddUser(kTestOwner); |
| 47 fake_user_manager_->AddUser(kTestUser); | 47 fake_user_manager_->AddUser(kTestUser); |
| 48 fake_user_manager_->set_owner_email(kTestOwner); | 48 fake_user_manager_->set_owner_email(kTestOwner); |
| 49 } | 49 } |
| 50 | 50 |
| 51 void SetUp() override { | 51 void SetUp() override { |
| 52 ASSERT_TRUE(testing_profile_manager_->SetUp()); | 52 ASSERT_TRUE(testing_profile_manager_->SetUp()); |
| 53 } | 53 } |
| 54 | 54 |
| 55 FakeConsumerManagementService* fake_service_; | 55 FakeConsumerManagementService* fake_service_; |
| 56 chromeos::FakeUserManager* fake_user_manager_; | 56 chromeos::FakeChromeUserManager* fake_user_manager_; |
| 57 chromeos::ScopedUserManagerEnabler scoped_user_manager_enabler_; | 57 chromeos::ScopedUserManagerEnabler scoped_user_manager_enabler_; |
| 58 scoped_ptr<TestingProfileManager> testing_profile_manager_; | 58 scoped_ptr<TestingProfileManager> testing_profile_manager_; |
| 59 }; | 59 }; |
| 60 | 60 |
| 61 TEST_F(ConsumerEnrollmentHandlerFactoryTest, ServiceIsCreated) { | 61 TEST_F(ConsumerEnrollmentHandlerFactoryTest, ServiceIsCreated) { |
| 62 Profile* profile = testing_profile_manager_->CreateTestingProfile(kTestOwner); | 62 Profile* profile = testing_profile_manager_->CreateTestingProfile(kTestOwner); |
| 63 EXPECT_TRUE(ConsumerEnrollmentHandlerFactory::GetForBrowserContext(profile)); | 63 EXPECT_TRUE(ConsumerEnrollmentHandlerFactory::GetForBrowserContext(profile)); |
| 64 } | 64 } |
| 65 | 65 |
| 66 TEST_F(ConsumerEnrollmentHandlerFactoryTest, ServiceIsNotCreatedForNonOwner) { | 66 TEST_F(ConsumerEnrollmentHandlerFactoryTest, ServiceIsNotCreatedForNonOwner) { |
| 67 Profile* profile = testing_profile_manager_->CreateTestingProfile(kTestUser); | 67 Profile* profile = testing_profile_manager_->CreateTestingProfile(kTestUser); |
| 68 EXPECT_FALSE(ConsumerEnrollmentHandlerFactory::GetForBrowserContext(profile)); | 68 EXPECT_FALSE(ConsumerEnrollmentHandlerFactory::GetForBrowserContext(profile)); |
| 69 } | 69 } |
| 70 | 70 |
| 71 TEST_F(ConsumerEnrollmentHandlerFactoryTest, | 71 TEST_F(ConsumerEnrollmentHandlerFactoryTest, |
| 72 ServiceIsNotCreatedIfItHasNothingToDo) { | 72 ServiceIsNotCreatedIfItHasNothingToDo) { |
| 73 fake_service_->SetStatusAndStage( | 73 fake_service_->SetStatusAndStage( |
| 74 ConsumerManagementService::STATUS_UNENROLLED, | 74 ConsumerManagementService::STATUS_UNENROLLED, |
| 75 ConsumerManagementStage::None()); | 75 ConsumerManagementStage::None()); |
| 76 | 76 |
| 77 Profile* profile = testing_profile_manager_->CreateTestingProfile(kTestOwner); | 77 Profile* profile = testing_profile_manager_->CreateTestingProfile(kTestOwner); |
| 78 EXPECT_FALSE(ConsumerEnrollmentHandlerFactory::GetForBrowserContext(profile)); | 78 EXPECT_FALSE(ConsumerEnrollmentHandlerFactory::GetForBrowserContext(profile)); |
| 79 } | 79 } |
| 80 | 80 |
| 81 } // namespace policy | 81 } // namespace policy |
| OLD | NEW |