| 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.h" | 5 #include "chrome/browser/chromeos/policy/consumer_enrollment_handler.h" |
| 6 | 6 |
| 7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
| 8 #include "base/run_loop.h" | 8 #include "base/run_loop.h" |
| 9 #include "chrome/browser/browser_process.h" | 9 #include "chrome/browser/browser_process.h" |
| 10 #include "chrome/browser/browser_process_platform_part.h" | 10 #include "chrome/browser/browser_process_platform_part.h" |
| 11 #include "chrome/browser/chromeos/login/users/fake_user_manager.h" | 11 #include "chrome/browser/chromeos/login/users/fake_chrome_user_manager.h" |
| 12 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h" | 12 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h" |
| 13 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h" | 13 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h" |
| 14 #include "chrome/browser/chromeos/policy/consumer_management_service.h" | 14 #include "chrome/browser/chromeos/policy/consumer_management_service.h" |
| 15 #include "chrome/browser/chromeos/policy/consumer_management_stage.h" | 15 #include "chrome/browser/chromeos/policy/consumer_management_stage.h" |
| 16 #include "chrome/browser/chromeos/policy/enrollment_status_chromeos.h" | 16 #include "chrome/browser/chromeos/policy/enrollment_status_chromeos.h" |
| 17 #include "chrome/browser/chromeos/policy/fake_consumer_management_service.h" | 17 #include "chrome/browser/chromeos/policy/fake_consumer_management_service.h" |
| 18 #include "chrome/browser/chromeos/policy/fake_device_cloud_policy_initializer.h" | 18 #include "chrome/browser/chromeos/policy/fake_device_cloud_policy_initializer.h" |
| 19 #include "chrome/browser/signin/fake_profile_oauth2_token_service.h" | 19 #include "chrome/browser/signin/fake_profile_oauth2_token_service.h" |
| 20 #include "chrome/browser/signin/fake_profile_oauth2_token_service_builder.h" | 20 #include "chrome/browser/signin/fake_profile_oauth2_token_service_builder.h" |
| 21 #include "chrome/browser/signin/profile_oauth2_token_service_factory.h" | 21 #include "chrome/browser/signin/profile_oauth2_token_service_factory.h" |
| (...skipping 11 matching lines...) Expand all Loading... |
| 33 const char* kTestUser = "test.user@chromium.org.test"; | 33 const char* kTestUser = "test.user@chromium.org.test"; |
| 34 } | 34 } |
| 35 | 35 |
| 36 namespace policy { | 36 namespace policy { |
| 37 | 37 |
| 38 class ConsumerEnrollmentHandlerTest : public testing::Test { | 38 class ConsumerEnrollmentHandlerTest : public testing::Test { |
| 39 public: | 39 public: |
| 40 ConsumerEnrollmentHandlerTest() | 40 ConsumerEnrollmentHandlerTest() |
| 41 : fake_service_(new FakeConsumerManagementService()), | 41 : fake_service_(new FakeConsumerManagementService()), |
| 42 fake_initializer_(new FakeDeviceCloudPolicyInitializer()), | 42 fake_initializer_(new FakeDeviceCloudPolicyInitializer()), |
| 43 fake_user_manager_(new chromeos::FakeUserManager()), | 43 fake_user_manager_(new chromeos::FakeChromeUserManager()), |
| 44 scoped_user_manager_enabler_(fake_user_manager_), | 44 scoped_user_manager_enabler_(fake_user_manager_), |
| 45 testing_profile_manager_(new TestingProfileManager( | 45 testing_profile_manager_( |
| 46 TestingBrowserProcess::GetGlobal())) { | 46 new TestingProfileManager(TestingBrowserProcess::GetGlobal())) { |
| 47 // Set up FakeConsumerManagementService. | 47 // Set up FakeConsumerManagementService. |
| 48 fake_service_->SetStatusAndStage( | 48 fake_service_->SetStatusAndStage( |
| 49 ConsumerManagementService::STATUS_ENROLLING, | 49 ConsumerManagementService::STATUS_ENROLLING, |
| 50 ConsumerManagementStage::EnrollmentOwnerStored()); | 50 ConsumerManagementStage::EnrollmentOwnerStored()); |
| 51 | 51 |
| 52 // Inject fake objects. | 52 // Inject fake objects. |
| 53 BrowserPolicyConnectorChromeOS* connector = | 53 BrowserPolicyConnectorChromeOS* connector = |
| 54 g_browser_process->platform_part()->browser_policy_connector_chromeos(); | 54 g_browser_process->platform_part()->browser_policy_connector_chromeos(); |
| 55 connector->SetConsumerManagementServiceForTesting( | 55 connector->SetConsumerManagementServiceForTesting( |
| 56 make_scoped_ptr(fake_service_)); | 56 make_scoped_ptr(fake_service_)); |
| 57 connector->SetDeviceCloudPolicyInitializerForTesting( | 57 connector->SetDeviceCloudPolicyInitializerForTesting( |
| 58 make_scoped_ptr(fake_initializer_)); | 58 make_scoped_ptr(fake_initializer_)); |
| 59 | 59 |
| 60 // Set up FakeUserManager. | 60 // Set up FakeChromeUserManager. |
| 61 fake_user_manager_->AddUser(kTestOwner); | 61 fake_user_manager_->AddUser(kTestOwner); |
| 62 fake_user_manager_->AddUser(kTestUser); | 62 fake_user_manager_->AddUser(kTestUser); |
| 63 fake_user_manager_->set_owner_email(kTestOwner); | 63 fake_user_manager_->set_owner_email(kTestOwner); |
| 64 } | 64 } |
| 65 | 65 |
| 66 void SetUp() override { | 66 void SetUp() override { |
| 67 ASSERT_TRUE(testing_profile_manager_->SetUp()); | 67 ASSERT_TRUE(testing_profile_manager_->SetUp()); |
| 68 profile_ = testing_profile_manager_->CreateTestingProfile(kTestUser); | 68 profile_ = testing_profile_manager_->CreateTestingProfile(kTestUser); |
| 69 | 69 |
| 70 // Set up FakeProfileOAuth2TokenService and issue a fake refresh token. | 70 // Set up FakeProfileOAuth2TokenService and issue a fake refresh token. |
| (...skipping 14 matching lines...) Expand all Loading... |
| 85 | 85 |
| 86 void RunEnrollmentTest() { | 86 void RunEnrollmentTest() { |
| 87 handler_.reset( | 87 handler_.reset( |
| 88 new ConsumerEnrollmentHandler(profile_, fake_service_, NULL)); | 88 new ConsumerEnrollmentHandler(profile_, fake_service_, NULL)); |
| 89 base::RunLoop().RunUntilIdle(); | 89 base::RunLoop().RunUntilIdle(); |
| 90 } | 90 } |
| 91 | 91 |
| 92 content::TestBrowserThreadBundle thread_bundle; | 92 content::TestBrowserThreadBundle thread_bundle; |
| 93 FakeConsumerManagementService* fake_service_; | 93 FakeConsumerManagementService* fake_service_; |
| 94 FakeDeviceCloudPolicyInitializer* fake_initializer_; | 94 FakeDeviceCloudPolicyInitializer* fake_initializer_; |
| 95 chromeos::FakeUserManager* fake_user_manager_; | 95 chromeos::FakeChromeUserManager* fake_user_manager_; |
| 96 chromeos::ScopedUserManagerEnabler scoped_user_manager_enabler_; | 96 chromeos::ScopedUserManagerEnabler scoped_user_manager_enabler_; |
| 97 scoped_ptr<TestingProfileManager> testing_profile_manager_; | 97 scoped_ptr<TestingProfileManager> testing_profile_manager_; |
| 98 Profile* profile_; | 98 Profile* profile_; |
| 99 scoped_ptr<ConsumerEnrollmentHandler> handler_; | 99 scoped_ptr<ConsumerEnrollmentHandler> handler_; |
| 100 }; | 100 }; |
| 101 | 101 |
| 102 TEST_F(ConsumerEnrollmentHandlerTest, EnrollsSuccessfully) { | 102 TEST_F(ConsumerEnrollmentHandlerTest, EnrollsSuccessfully) { |
| 103 EXPECT_FALSE(fake_initializer_->was_start_enrollment_called()); | 103 EXPECT_FALSE(fake_initializer_->was_start_enrollment_called()); |
| 104 | 104 |
| 105 RunEnrollmentTest(); | 105 RunEnrollmentTest(); |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 139 EnrollmentStatus::STATUS_REGISTRATION_FAILED)); | 139 EnrollmentStatus::STATUS_REGISTRATION_FAILED)); |
| 140 | 140 |
| 141 RunEnrollmentTest(); | 141 RunEnrollmentTest(); |
| 142 | 142 |
| 143 EXPECT_TRUE(fake_initializer_->was_start_enrollment_called()); | 143 EXPECT_TRUE(fake_initializer_->was_start_enrollment_called()); |
| 144 EXPECT_EQ(ConsumerManagementStage::EnrollmentDMServerFailed(), | 144 EXPECT_EQ(ConsumerManagementStage::EnrollmentDMServerFailed(), |
| 145 fake_service_->GetStage()); | 145 fake_service_->GetStage()); |
| 146 } | 146 } |
| 147 | 147 |
| 148 } // namespace policy | 148 } // namespace policy |
| OLD | NEW |