Implemented consumer management unenrollment.

chrome/browser/chromeos/policy/consumer_unenrollment_handler_unittest.cc

+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/chromeos/policy/consumer_unenrollment_handler.h"
+
+#include "base/bind.h"
+#include "base/bind_helpers.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/thread_task_runner_handle.h"
+#include "chrome/browser/chromeos/policy/consumer_management_service.h"
+#include "chrome/browser/chromeos/policy/consumer_management_stage.h"
+#include "chrome/browser/chromeos/policy/device_cloud_policy_store_chromeos.h"
+#include "chrome/browser/chromeos/policy/enterprise_install_attributes.h"
+#include "chrome/browser/chromeos/policy/fake_consumer_management_service.h"
+#include "chrome/browser/chromeos/policy/fake_device_cloud_policy_manager.h"
+#include "chrome/browser/chromeos/settings/device_settings_service.h"
+#include "chrome/browser/chromeos/settings/device_settings_test_helper.h"
+#include "chromeos/dbus/fake_cryptohome_client.h"
+#include "components/ownership/mock_owner_key_util.h"
+#include "policy/proto/device_management_backend.pb.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace policy {
+
+class ConsumerUnenrollmentHandlerTest
+    : public chromeos::DeviceSettingsTestBase {
+ public:
+  ConsumerUnenrollmentHandlerTest()
+      : fake_service_(new FakeConsumerManagementService()),
+        fake_cryptohome_client_(new chromeos::FakeCryptohomeClient()),
+        install_attributes_(
+            new EnterpriseInstallAttributes(fake_cryptohome_client_)) {
+    // Set up FakeConsumerManagementService.
+    fake_service_->SetStatusAndStage(
+        ConsumerManagementService::STATUS_ENROLLED,
+        ConsumerManagementStage::None());
+
+  }
+
+  void SetUp() override {
+    DeviceSettingsTestBase::SetUp();
+
+    // Set up the ownership, so that we can modify device settings.
+    owner_key_util_->SetPrivateKey(device_policy_.GetSigningKey());
+    InitOwner(device_policy_.policy_data().username(), true);
+    FlushDeviceSettings();
+
+    // Set up DeviceSettingsService.
+    device_settings_service_.SetManagementSettings(
+      enterprise_management::PolicyData::CONSUMER_MANAGED,
+      "fake_request_token",
+      "fake_device_id",
+      base::Bind(&ConsumerUnenrollmentHandlerTest::OnManagementSettingsSet,
+                 base::Unretained(this)));
+    FlushDeviceSettings();
+
+    // Set up FakeDeviceCloudPolicyManager.
+    scoped_ptr<DeviceCloudPolicyStoreChromeOS> store_(
+        new DeviceCloudPolicyStoreChromeOS(
+            &device_settings_service_,
+            install_attributes_.get(),
+            base::ThreadTaskRunnerHandle::Get()));
+    fake_manager_.reset(new FakeDeviceCloudPolicyManager(
+        store_.Pass(),
+        base::ThreadTaskRunnerHandle::Get()));
+  }
+
+  void OnManagementSettingsSet() {
+    EXPECT_EQ(chromeos::DeviceSettingsService::STORE_SUCCESS,
+              device_settings_service_.status());
+  }
+
+  void RunUnenrollment() {
+    handler_.reset(new ConsumerUnenrollmentHandler(
+        &device_settings_service_,
+        fake_service_.get(),
+        fake_manager_.get()));
+    handler_->Start();
+    FlushDeviceSettings();
+  }
+
+  scoped_ptr<FakeConsumerManagementService> fake_service_;
+  chromeos::FakeCryptohomeClient* fake_cryptohome_client_;
+  scoped_ptr<EnterpriseInstallAttributes> install_attributes_;
+  scoped_ptr<FakeDeviceCloudPolicyManager> fake_manager_;
+
+  scoped_ptr<ConsumerUnenrollmentHandler> handler_;
+};
+
+TEST_F(ConsumerUnenrollmentHandlerTest, UnenrollmentSucceeds) {
+  EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage());
+
+  RunUnenrollment();
+
+  EXPECT_EQ(ConsumerManagementStage::UnenrollmentSuccess(),
+            fake_service_->GetStage());
+  EXPECT_EQ(enterprise_management::PolicyData::LOCAL_OWNER,
+            device_settings_service_.policy_data()->management_mode());
+  EXPECT_FALSE(device_settings_service_.policy_data()->has_request_token());
+  EXPECT_FALSE(device_settings_service_.policy_data()->has_device_id());
+}
+
+TEST_F(ConsumerUnenrollmentHandlerTest,
+       UnenrollmentFailsOnServerError) {
+  EXPECT_EQ(ConsumerManagementStage::None(), fake_service_->GetStage());
+  fake_manager_->set_unregister_result(false);
+
+  RunUnenrollment();
+
+  EXPECT_EQ(ConsumerManagementStage::UnenrollmentDMServerFailed(),
+            fake_service_->GetStage());
+  EXPECT_EQ(enterprise_management::PolicyData::CONSUMER_MANAGED,
+            device_settings_service_.policy_data()->management_mode());
+  EXPECT_EQ("fake_request_token",
+            device_settings_service_.policy_data()->request_token());
+  EXPECT_EQ("fake_device_id",
+            device_settings_service_.policy_data()->device_id());
+}
+
+}  // namespace policy