| 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 "base/bind.h" | |
| 8 #include "base/bind_helpers.h" | |
| 9 #include "base/memory/scoped_ptr.h" | |
| 10 #include "base/thread_task_runner_handle.h" | |
| 11 #include "chrome/browser/chromeos/ownership/fake_owner_settings_service.h" | |
| 12 #include "chrome/browser/chromeos/policy/consumer_management_service.h" | |
| 13 #include "chrome/browser/chromeos/policy/consumer_management_stage.h" | |
| 14 #include "chrome/browser/chromeos/policy/device_cloud_policy_store_chromeos.h" | |
| 15 #include "chrome/browser/chromeos/policy/enterprise_install_attributes.h" | |
| 16 #include "chrome/browser/chromeos/policy/fake_consumer_management_service.h" | |
| 17 #include "chrome/browser/chromeos/policy/fake_device_cloud_policy_manager.h" | |
| 18 #include "chrome/browser/chromeos/settings/device_settings_service.h" | |
| 19 #include "chrome/browser/chromeos/settings/device_settings_test_helper.h" | |
| 20 #include "chrome/test/base/testing_profile.h" | |
| 21 #include "chromeos/dbus/fake_cryptohome_client.h" | |
| 22 #include "components/ownership/mock_owner_key_util.h" | |
| 23 #include "policy/proto/device_management_backend.pb.h" | |
| 24 #include "testing/gtest/include/gtest/gtest.h" | |
| 25 | |
| 26 namespace policy { | |
| 27 | |
| 28 class ConsumerUnenrollmentHandlerTest | |
| 29 : public chromeos::DeviceSettingsTestBase { | |
| 30 public: | |
| 31 ConsumerUnenrollmentHandlerTest() | |
| 32 : fake_service_(new FakeConsumerManagementService()), | |
| 33 fake_cryptohome_client_(new chromeos::FakeCryptohomeClient()), | |
| 34 install_attributes_( | |
| 35 new EnterpriseInstallAttributes(fake_cryptohome_client_)) { | |
| 36 // Set up FakeConsumerManagementService. | |
| 37 fake_service_->SetStatusAndStage( | |
| 38 ConsumerManagementService::STATUS_ENROLLED, | |
| 39 ConsumerManagementStage::None()); | |
| 40 } | |
| 41 | |
| 42 void SetUp() override { | |
| 43 DeviceSettingsTestBase::SetUp(); | |
| 44 | |
| 45 // Set up the ownership, so that we can modify device settings. | |
| 46 owner_key_util_->SetPrivateKey(device_policy_.GetSigningKey()); | |
| 47 InitOwner(device_policy_.policy_data().username(), true); | |
| 48 FlushDeviceSettings(); | |
| 49 | |
| 50 | |
| 51 // Set up FakeDeviceCloudPolicyManager. | |
| 52 scoped_ptr<DeviceCloudPolicyStoreChromeOS> store_( | |
| 53 new DeviceCloudPolicyStoreChromeOS( | |
| 54 &device_settings_service_, | |
| 55 install_attributes_.get(), | |
| 56 base::ThreadTaskRunnerHandle::Get())); | |
| 57 fake_manager_.reset(new FakeDeviceCloudPolicyManager( | |
| 58 store_.Pass(), | |
| 59 base::ThreadTaskRunnerHandle::Get())); | |
| 60 | |
| 61 // Set up FakeOwnerSettingsService. | |
| 62 fake_owner_settings_service_.reset( | |
| 63 new chromeos::FakeOwnerSettingsService( | |
| 64 profile_.get(), owner_key_util_)); | |
| 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 &device_settings_service_, | |
| 82 fake_service_.get(), | |
| 83 fake_manager_.get(), | |
| 84 fake_owner_settings_service_.get())); | |
| 85 handler_->Start(); | |
| 86 FlushDeviceSettings(); | |
| 87 } | |
| 88 | |
| 89 scoped_ptr<FakeConsumerManagementService> fake_service_; | |
| 90 chromeos::FakeCryptohomeClient* fake_cryptohome_client_; | |
| 91 scoped_ptr<EnterpriseInstallAttributes> install_attributes_; | |
| 92 scoped_ptr<FakeDeviceCloudPolicyManager> fake_manager_; | |
| 93 scoped_ptr<chromeos::FakeOwnerSettingsService> fake_owner_settings_service_; | |
| 94 | |
| 95 scoped_ptr<ConsumerUnenrollmentHandler> handler_; | |
| 96 }; | |
| 97 | |
| 98 TEST_F(ConsumerUnenrollmentHandlerTest, UnenrollmentSucceeds) { | |
| 99 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); | |
| 100 | |
| 101 RunUnenrollment(); | |
| 102 | |
| 103 EXPECT_EQ(ConsumerManagementStage::UnenrollmentSuccess(), | |
| 104 fake_service_->GetStage()); | |
| 105 const chromeos::OwnerSettingsServiceChromeOS::ManagementSettings& settings = | |
| 106 fake_owner_settings_service_->last_settings(); | |
| 107 EXPECT_EQ(policy::MANAGEMENT_MODE_LOCAL_OWNER, settings.management_mode); | |
| 108 EXPECT_EQ("", settings.request_token); | |
| 109 EXPECT_EQ("", settings.device_id); | |
| 110 } | |
| 111 | |
| 112 TEST_F(ConsumerUnenrollmentHandlerTest, | |
| 113 UnenrollmentFailsOnServerError) { | |
| 114 EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage()); | |
| 115 fake_manager_->set_unregister_result(false); | |
| 116 | |
| 117 RunUnenrollment(); | |
| 118 | |
| 119 EXPECT_EQ(ConsumerManagementStage::UnenrollmentDMServerFailed(), | |
| 120 fake_service_->GetStage()); | |
| 121 const chromeos::OwnerSettingsServiceChromeOS::ManagementSettings& settings = | |
| 122 fake_owner_settings_service_->last_settings(); | |
| 123 EXPECT_EQ(policy::MANAGEMENT_MODE_CONSUMER_MANAGED, settings.management_mode); | |
| 124 EXPECT_EQ("fake_request_token", settings.request_token); | |
| 125 EXPECT_EQ("fake_device_id", settings.device_id); | |
| 126 } | |
| 127 | |
| 128 } // namespace policy | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 814123006:
Revert of Implemented consumer management unenrollment. (Closed)
Base URL: https://chromium.googlesource.com/chromium/src.git@dcpm