device_policy_cros_browser_test.cc revision a1401311d1ab56c4ed0a474bd38c108f75cb0cd9 
1// Copyright (c) 2013 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/device_policy_cros_browser_test.h"
6
7#include <string>
8#include <vector>
9
10#include "base/file_util.h"
11#include "base/files/file_path.h"
12#include "base/path_service.h"
13#include "base/stl_util.h"
14#include "chrome/browser/chromeos/policy/device_policy_builder.h"
15#include "chrome/browser/chromeos/policy/enterprise_install_attributes.h"
16#include "chrome/browser/chromeos/policy/proto/install_attributes.pb.h"
17#include "chromeos/chromeos_paths.h"
18#include "chromeos/dbus/fake_dbus_thread_manager.h"
19#include "chromeos/dbus/fake_session_manager_client.h"
20#include "crypto/rsa_private_key.h"
21#include "testing/gmock/include/gmock/gmock.h"
22#include "testing/gtest/include/gtest/gtest.h"
23
24using ::testing::_;
25using ::testing::AnyNumber;
26using ::testing::Return;
27
28namespace policy {
29
30DevicePolicyCrosTestHelper::DevicePolicyCrosTestHelper() {
31  CHECK(temp_dir_.CreateUniqueTempDir());
32}
33
34DevicePolicyCrosTestHelper::~DevicePolicyCrosTestHelper() {}
35
36void DevicePolicyCrosTestHelper::MarkAsEnterpriseOwned() {
37  cryptohome::SerializedInstallAttributes install_attrs_proto;
38  cryptohome::SerializedInstallAttributes::Attribute* attribute = NULL;
39
40  attribute = install_attrs_proto.add_attributes();
41  attribute->set_name(EnterpriseInstallAttributes::kAttrEnterpriseOwned);
42  attribute->set_value("true");
43
44  attribute = install_attrs_proto.add_attributes();
45  attribute->set_name(EnterpriseInstallAttributes::kAttrEnterpriseUser);
46  attribute->set_value(device_policy_.policy_data().username());
47
48  base::FilePath install_attrs_file =
49      temp_dir_.path().AppendASCII("install_attributes.pb");
50  const std::string install_attrs_blob(
51      install_attrs_proto.SerializeAsString());
52  ASSERT_EQ(static_cast<int>(install_attrs_blob.size()),
53            base::WriteFile(install_attrs_file,
54                            install_attrs_blob.c_str(),
55                            install_attrs_blob.size()));
56  ASSERT_TRUE(PathService::Override(chromeos::FILE_INSTALL_ATTRIBUTES,
57                                    install_attrs_file));
58}
59
60void DevicePolicyCrosTestHelper::InstallOwnerKey() {
61  base::FilePath owner_key_file = temp_dir_.path().AppendASCII("owner.key");
62  std::vector<uint8> owner_key_bits;
63  ASSERT_TRUE(
64      device_policy()->GetSigningKey()->ExportPublicKey(&owner_key_bits));
65  ASSERT_EQ(base::WriteFile(
66          owner_key_file,
67          reinterpret_cast<const char*>(vector_as_array(&owner_key_bits)),
68          owner_key_bits.size()),
69      static_cast<int>(owner_key_bits.size()));
70  ASSERT_TRUE(PathService::Override(chromeos::FILE_OWNER_KEY, owner_key_file));
71}
72
73DevicePolicyCrosBrowserTest::DevicePolicyCrosBrowserTest()
74    : fake_dbus_thread_manager_(new chromeos::FakeDBusThreadManager),
75      fake_session_manager_client_(new chromeos::FakeSessionManagerClient) {
76  fake_dbus_thread_manager_->SetFakeClients();
77  fake_dbus_thread_manager_->SetSessionManagerClient(
78      scoped_ptr<chromeos::SessionManagerClient>(fake_session_manager_client_));
79}
80
81DevicePolicyCrosBrowserTest::~DevicePolicyCrosBrowserTest() {
82}
83
84void DevicePolicyCrosBrowserTest::SetUpInProcessBrowserTestFixture() {
85  chromeos::DBusThreadManager::SetInstanceForTesting(fake_dbus_thread_manager_);
86  InProcessBrowserTest::SetUpInProcessBrowserTestFixture();
87}
88
89void DevicePolicyCrosBrowserTest::TearDownInProcessBrowserTestFixture() {
90  InProcessBrowserTest::TearDownInProcessBrowserTestFixture();
91}
92
93void DevicePolicyCrosBrowserTest::MarkAsEnterpriseOwned() {
94  test_helper_.MarkAsEnterpriseOwned();
95}
96
97void DevicePolicyCrosBrowserTest::InstallOwnerKey() {
98  test_helper_.InstallOwnerKey();
99}
100
101void DevicePolicyCrosBrowserTest::RefreshDevicePolicy() {
102  // Reset the key to its original state.
103  device_policy()->SetDefaultSigningKey();
104  device_policy()->Build();
105  session_manager_client()->set_device_policy(device_policy()->GetBlob());
106  session_manager_client()->OnPropertyChangeComplete(true);
107}
108
109}  // namespace policy
110