| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "chrome/browser/chromeos/policy/consumer_unenrollment_handler.h" | |
| 6 | |
| 7 #include <memory> | |
| 8 #include <utility> | |
| 9 | |
| 10 #include "base/bind.h" | |
| 11 #include "base/bind_helpers.h" | |
| 12 #include "base/threading/thread_task_runner_handle.h" | |
| 13 #include "chrome/browser/chromeos/ownership/fake_owner_settings_service.h" | |
| 14 #include "chrome/browser/chromeos/policy/consumer_management_service.h" | |
| 15 #include "chrome/browser/chromeos/policy/consumer_management_stage.h" | |
| 16 #include "chrome/browser/chromeos/policy/device_cloud_policy_store_chromeos.h" | |
| 17 #include "chrome/browser/chromeos/policy/enterprise_install_attributes.h" | |
| 18 #include "chrome/browser/chromeos/policy/fake_consumer_management_service.h" | |
| 19 #include "chrome/browser/chromeos/policy/fake_device_cloud_policy_manager.h" | |
| 20 #include "chrome/browser/chromeos/settings/device_settings_service.h" | |
| 21 #include "chrome/browser/chromeos/settings/device_settings_test_helper.h" | |
| 22 #include "chrome/test/base/testing_profile.h" | |
| 23 #include "chromeos/dbus/fake_cryptohome_client.h" | |
| 24 #include "components/ownership/mock_owner_key_util.h" | |
| 25 #include "policy/proto/device_management_backend.pb.h" | |
| 26 #include "testing/gtest/include/gtest/gtest.h" | |
| 27 | |
| 28 namespace policy { | |
| 29 | |
| 30 class ConsumerUnenrollmentHandlerTest | |
| 31 : public chromeos::DeviceSettingsTestBase { | |
| 32 public: | |
| 33 ConsumerUnenrollmentHandlerTest() | |
| 34 : fake_service_(new FakeConsumerManagementService()), | |
| 35 fake_cryptohome_client_(new chromeos::FakeCryptohomeClient()), | |
| 36 install_attributes_( | |
| 37 new EnterpriseInstallAttributes(fake_cryptohome_client_.get())) { | |
| 38 // Set up FakeConsumerManagementService. | |
| 39 fake_service_->SetStatusAndStage( | |
| 40 ConsumerManagementService::STATUS_ENROLLED, | |
| 41 ConsumerManagementStage::None()); | |
| 42 } | |
| 43 | |
| 44 void SetUp() override { | |
| 45 DeviceSettingsTestBase::SetUp(); | |
| 46 | |
| 47 // Set up the ownership, so that we can modify device settings. | |
| 48 owner_key_util_->SetPrivateKey(device_policy_.GetSigningKey()); | |
| 49 InitOwner(AccountId::FromUserEmail(device_policy_.policy_data().username()), | |
| 50 true); | |
| 51 FlushDeviceSettings(); | |
| 52 | |
| 53 | |
| 54 // Set up FakeDeviceCloudPolicyManager. | |
| 55 std::unique_ptr<DeviceCloudPolicyStoreChromeOS> store_( | |
| 56 new DeviceCloudPolicyStoreChromeOS( | |
| 57 &device_settings_service_, install_attributes_.get(), | |
| 58 base::ThreadTaskRunnerHandle::Get())); | |
| 59 fake_manager_.reset(new FakeDeviceCloudPolicyManager( | |
| 60 std::move(store_), base::ThreadTaskRunnerHandle::Get())); | |
| 61 | |
| 62 // Set up FakeOwnerSettingsService. | |
| 63 fake_owner_settings_service_.reset(new chromeos::FakeOwnerSettingsService( | |
| 64 profile_.get(), owner_key_util_, nullptr)); | |
| 65 chromeos::OwnerSettingsServiceChromeOS::ManagementSettings settings; | |
| 66 settings.management_mode = policy::MANAGEMENT_MODE_CONSUMER_MANAGED; | |
| 67 settings.request_token = "fake_request_token"; | |
| 68 settings.device_id = "fake_device_id"; | |
| 69 fake_owner_settings_service_->SetManagementSettings( | |
| 70 settings, | |
| 71 base::Bind(&ConsumerUnenrollmentHandlerTest::OnManagementSettingsSet, | |
| 72 base::Unretained(this))); | |
| 73 } | |
| 74 | |
| 75 void OnManagementSettingsSet(bool success) { | |
| 76 EXPECT_TRUE(success); | |
| 77 } | |
| 78 | |
| 79 void RunUnenrollment() { | |
| 80 handler_.reset(new ConsumerUnenrollmentHandler( | |
| 81 fake_service_.get(), fake_manager_.get(), | |
| 82 fake_owner_settings_service_.get())); | |
| 83 handler_->Start(); | |
| 84 FlushDeviceSettings(); | |
| 85 } | |
| 86 | |
| 87 std::unique_ptr<FakeConsumerManagementService> fake_service_; | |
| 88 std::unique_ptr<chromeos::FakeCryptohomeClient> fake_cryptohome_client_; | |
| 89 std::unique_ptr<EnterpriseInstallAttributes> install_attributes_; | |
| 90 std::unique_ptr<FakeDeviceCloudPolicyManager> fake_manager_; | |
| 91 std::unique_ptr<chromeos::FakeOwnerSettingsService> | |
| 92 fake_owner_settings_service_; | |
| 93 | |
| 94 std::unique_ptr<ConsumerUnenrollmentHandler> handler_; | |
| 95 }; | |
| 96 | |
| 97 TEST_F(ConsumerUnenrollmentHandlerTest, UnenrollmentSucceeds) { | |
| 98 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); | |
| 99 | |
| 100 RunUnenrollment(); | |
| 101 | |
| 102 EXPECT_EQ(ConsumerManagementStage::UnenrollmentSuccess(), | |
| 103 fake_service_->GetStage()); | |
| 104 const chromeos::OwnerSettingsServiceChromeOS::ManagementSettings& settings = | |
| 105 fake_owner_settings_service_->last_settings(); | |
| 106 EXPECT_EQ(policy::MANAGEMENT_MODE_LOCAL_OWNER, settings.management_mode); | |
| 107 EXPECT_EQ("", settings.request_token); | |
| 108 EXPECT_EQ("", settings.device_id); | |
| 109 } | |
| 110 | |
| 111 TEST_F(ConsumerUnenrollmentHandlerTest, | |
| 112 UnenrollmentFailsOnServerError) { | |
| 113 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); | |
| 114 fake_manager_->set_unregister_result(false); | |
| 115 | |
| 116 RunUnenrollment(); | |
| 117 | |
| 118 EXPECT_EQ(ConsumerManagementStage::UnenrollmentDMServerFailed(), | |
| 119 fake_service_->GetStage()); | |
| 120 const chromeos::OwnerSettingsServiceChromeOS::ManagementSettings& settings = | |
| 121 fake_owner_settings_service_->last_settings(); | |
| 122 EXPECT_EQ(policy::MANAGEMENT_MODE_CONSUMER_MANAGED, settings.management_mode); | |
| 123 EXPECT_EQ("fake_request_token", settings.request_token); | |
| 124 EXPECT_EQ("fake_device_id", settings.device_id); | |
| 125 } | |
| 126 | |
| 127 } // namespace policy | |
| OLD | NEW |