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 |