device_policy_cros_browser_test.cc revision 2385ea399aae016c0806a4f9ef3c9cfe3d2a39df 
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/files/scoped_temp_dir.h"
13#include "base/path_service.h"
14#include "base/stl_util.h"
15#include "chrome/browser/chromeos/policy/device_policy_builder.h"
16#include "chrome/browser/chromeos/policy/enterprise_install_attributes.h"
17#include "chrome/browser/policy/proto/chromeos/install_attributes.pb.h"
18#include "chromeos/chromeos_paths.h"
19#include "chromeos/dbus/mock_dbus_thread_manager_without_gmock.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
30void DevicePolicyCrosBrowserTest::MarkAsEnterpriseOwned() {
31  cryptohome::SerializedInstallAttributes install_attrs_proto;
32  cryptohome::SerializedInstallAttributes::Attribute* attribute = NULL;
33
34  attribute = install_attrs_proto.add_attributes();
35  attribute->set_name(EnterpriseInstallAttributes::kAttrEnterpriseOwned);
36  attribute->set_value("true");
37
38  attribute = install_attrs_proto.add_attributes();
39  attribute->set_name(EnterpriseInstallAttributes::kAttrEnterpriseUser);
40  attribute->set_value(DevicePolicyBuilder::kFakeUsername);
41
42  base::FilePath install_attrs_file =
43      temp_dir_.path().AppendASCII("install_attributes.pb");
44  const std::string install_attrs_blob(
45      install_attrs_proto.SerializeAsString());
46  ASSERT_EQ(static_cast<int>(install_attrs_blob.size()),
47            file_util::WriteFile(install_attrs_file,
48                                 install_attrs_blob.c_str(),
49                                 install_attrs_blob.size()));
50  ASSERT_TRUE(PathService::Override(chromeos::FILE_INSTALL_ATTRIBUTES,
51                                    install_attrs_file));
52}
53
54DevicePolicyCrosBrowserTest::DevicePolicyCrosBrowserTest()
55    : mock_dbus_thread_manager_(
56        new chromeos::MockDBusThreadManagerWithoutGMock) {
57}
58
59DevicePolicyCrosBrowserTest::~DevicePolicyCrosBrowserTest() {
60}
61
62void DevicePolicyCrosBrowserTest::SetUpInProcessBrowserTestFixture() {
63  chromeos::DBusThreadManager::InitializeForTesting(mock_dbus_thread_manager_);
64  CrosInProcessBrowserTest::SetUpInProcessBrowserTestFixture();
65}
66
67void DevicePolicyCrosBrowserTest::InstallOwnerKey() {
68  ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
69  base::FilePath owner_key_file = temp_dir_.path().AppendASCII("owner.key");
70  std::vector<uint8> owner_key_bits;
71  ASSERT_TRUE(
72      device_policy()->GetSigningKey()->ExportPublicKey(&owner_key_bits));
73  ASSERT_EQ(
74      file_util::WriteFile(
75          owner_key_file,
76          reinterpret_cast<const char*>(vector_as_array(&owner_key_bits)),
77          owner_key_bits.size()),
78      static_cast<int>(owner_key_bits.size()));
79  ASSERT_TRUE(PathService::Override(chromeos::FILE_OWNER_KEY, owner_key_file));
80}
81
82void DevicePolicyCrosBrowserTest::RefreshDevicePolicy() {
83  // Reset the key to its original state.
84  device_policy_.SetDefaultSigningKey();
85  device_policy_.Build();
86  session_manager_client()->set_device_policy(device_policy_.GetBlob());
87  session_manager_client()->OnPropertyChangeComplete(true);
88}
89
90void DevicePolicyCrosBrowserTest::TearDownInProcessBrowserTestFixture() {
91  CrosInProcessBrowserTest::TearDownInProcessBrowserTestFixture();
92  chromeos::DBusThreadManager::Shutdown();
93}
94
95}  // namespace policy
96