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 |