chrome/browser/chromeos/policy/user_policy_test_helper.cc

// Copyright 2015 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/user_policy_test_helper.h"

#include <utility>

#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/json/json_writer.h"
#include "base/run_loop.h"
#include "base/values.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/chromeos/policy/user_cloud_policy_manager_chromeos.h"
#include "chrome/browser/chromeos/policy/user_policy_manager_factory_chromeos.h"
#include "chrome/browser/policy/profile_policy_connector.h"
#include "chrome/browser/policy/profile_policy_connector_factory.h"
#include "chrome/browser/policy/test/local_policy_test_server.h"
#include "chrome/browser/profiles/profile.h"
#include "components/policy/core/browser/browser_policy_connector.h"
#include "components/policy/core/common/cloud/cloud_policy_client.h"
#include "components/policy/core/common/cloud/cloud_policy_constants.h"
#include "components/policy/core/common/cloud/cloud_policy_core.h"
#include "components/policy/core/common/policy_service.h"
#include "components/policy/core/common/policy_switches.h"
#include "components/policy/proto/device_management_backend.pb.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

namespace policy {

namespace {

std::string BuildPolicy(const base::DictionaryValue& mandatory,
                        const base::DictionaryValue& recommended,
                        const std::string& policyType,
                        const std::string& account_id) {
  std::unique_ptr<base::DictionaryValue> policy_type_dict(
      new base::DictionaryValue);
  policy_type_dict->SetWithoutPathExpansion("mandatory",
                                            mandatory.CreateDeepCopy());
  policy_type_dict->SetWithoutPathExpansion("recommended",
                                            recommended.CreateDeepCopy());

  std::unique_ptr<base::ListValue> managed_users_list(new base::ListValue);
  managed_users_list->AppendString("*");

  base::DictionaryValue root_dict;
  root_dict.SetWithoutPathExpansion(policyType, std::move(policy_type_dict));
  root_dict.SetWithoutPathExpansion("managed_users",
                                    std::move(managed_users_list));
  root_dict.SetKey("policy_user", base::Value(account_id));
  root_dict.SetKey("current_key_index", base::Value(0));

  std::string json_policy;
  base::JSONWriter::WriteWithOptions(
      root_dict, base::JSONWriter::OPTIONS_PRETTY_PRINT, &json_policy);
  return json_policy;
}

}  // namespace

UserPolicyTestHelper::UserPolicyTestHelper(const std::string& account_id)
    : account_id_(account_id) {
}

UserPolicyTestHelper::~UserPolicyTestHelper() {
}

void UserPolicyTestHelper::Init(
    const base::DictionaryValue& mandatory_policy,
    const base::DictionaryValue& recommended_policy) {
  ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
  WritePolicyFile(mandatory_policy, recommended_policy);

  test_server_.reset(new LocalPolicyTestServer(PolicyFilePath()));
  ASSERT_TRUE(test_server_->Start());
}

void UserPolicyTestHelper::UpdateCommandLine(
    base::CommandLine* command_line) const {
  command_line->AppendSwitchASCII(policy::switches::kDeviceManagementUrl,
                                  test_server_->GetServiceURL().spec());
}

void UserPolicyTestHelper::WaitForInitialPolicy(Profile* profile) {
  BrowserPolicyConnector* const connector =
      g_browser_process->browser_policy_connector();
  connector->ScheduleServiceInitialization(0);

  UserCloudPolicyManagerChromeOS* const policy_manager =
      UserPolicyManagerFactoryChromeOS::GetCloudPolicyManagerForProfile(
          profile);

  // Give a bogus OAuth token to the |policy_manager|. This should make its
  // CloudPolicyClient fetch the DMToken.
  ASSERT_FALSE(policy_manager->core()->client()->is_registered());
  const enterprise_management::DeviceRegisterRequest::Type registration_type =
      enterprise_management::DeviceRegisterRequest::USER;
  policy_manager->core()->client()->Register(
      registration_type,
      enterprise_management::DeviceRegisterRequest::FLAVOR_USER_REGISTRATION,
      enterprise_management::LicenseType::UNDEFINED, "bogus", std::string(),
      std::string(), std::string());

  policy::ProfilePolicyConnector* const profile_connector =
      policy::ProfilePolicyConnectorFactory::GetForBrowserContext(profile);
  policy::PolicyService* const policy_service =
      profile_connector->policy_service();

  base::RunLoop run_loop;
  policy_service->RefreshPolicies(run_loop.QuitClosure());
  run_loop.Run();
}

void UserPolicyTestHelper::UpdatePolicy(
    const base::DictionaryValue& mandatory_policy,
    const base::DictionaryValue& recommended_policy,
    Profile* profile) {
  WritePolicyFile(mandatory_policy, recommended_policy);

  policy::ProfilePolicyConnector* const profile_connector =
      policy::ProfilePolicyConnectorFactory::GetForBrowserContext(profile);
  policy::PolicyService* const policy_service =
      profile_connector->policy_service();

  base::RunLoop run_loop;
  policy_service->RefreshPolicies(run_loop.QuitClosure());
  run_loop.Run();
}

void UserPolicyTestHelper::DeletePolicyFile() {
  base::DeleteFile(PolicyFilePath(), false);
}

void UserPolicyTestHelper::WritePolicyFile(
    const base::DictionaryValue& mandatory,
    const base::DictionaryValue& recommended) {
  const std::string policy = BuildPolicy(
      mandatory, recommended, dm_protocol::kChromeUserPolicyType, account_id_);
  const int bytes_written =
      base::WriteFile(PolicyFilePath(), policy.data(), policy.size());
  ASSERT_EQ(static_cast<int>(policy.size()), bytes_written);
}

base::FilePath UserPolicyTestHelper::PolicyFilePath() const {
  return temp_dir_.GetPath().AppendASCII("policy.json");
}

}  // namespace policy