device_policy_cros_browser_test.cc revision 3551c9c881056c480085172ff9840cab31610854 
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 <vector>
8
9#include "base/file_util.h"
10#include "base/files/file_path.h"
11#include "base/path_service.h"
12#include "base/stl_util.h"
13#include "chrome/browser/chromeos/policy/cros_enterprise_test_utils.h"
14#include "chrome/browser/chromeos/policy/device_policy_builder.h"
15#include "chrome/browser/policy/proto/chromeos/chrome_device_policy.pb.h"
16#include "chromeos/chromeos_paths.h"
17#include "chromeos/dbus/fake_session_manager_client.h"
18#include "chromeos/dbus/mock_dbus_thread_manager_without_gmock.h"
19#include "crypto/rsa_private_key.h"
20#include "testing/gmock/include/gmock/gmock.h"
21#include "testing/gtest/include/gtest/gtest.h"
22
23using ::testing::_;
24using ::testing::AnyNumber;
25using ::testing::Return;
26
27namespace policy {
28
29void DevicePolicyCrosBrowserTest::MarkAsEnterpriseOwned() {
30  ASSERT_TRUE(temp_dir_.IsValid());
31  test_utils::MarkAsEnterpriseOwned(DevicePolicyBuilder::kFakeUsername,
32                                    temp_dir_.path());
33}
34
35DevicePolicyCrosBrowserTest::DevicePolicyCrosBrowserTest()
36    : mock_dbus_thread_manager_(
37        new chromeos::MockDBusThreadManagerWithoutGMock) {
38}
39
40DevicePolicyCrosBrowserTest::~DevicePolicyCrosBrowserTest() {
41}
42
43void DevicePolicyCrosBrowserTest::SetUpInProcessBrowserTestFixture() {
44  chromeos::DBusThreadManager::InitializeForTesting(mock_dbus_thread_manager_);
45  InProcessBrowserTest::SetUpInProcessBrowserTestFixture();
46}
47
48void DevicePolicyCrosBrowserTest::InstallOwnerKey() {
49  ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
50  base::FilePath owner_key_file = temp_dir_.path().AppendASCII("owner.key");
51  std::vector<uint8> owner_key_bits;
52  ASSERT_TRUE(
53      device_policy()->GetSigningKey()->ExportPublicKey(&owner_key_bits));
54  ASSERT_EQ(
55      file_util::WriteFile(
56          owner_key_file,
57          reinterpret_cast<const char*>(vector_as_array(&owner_key_bits)),
58          owner_key_bits.size()),
59      static_cast<int>(owner_key_bits.size()));
60  ASSERT_TRUE(PathService::Override(chromeos::FILE_OWNER_KEY, owner_key_file));
61}
62
63void DevicePolicyCrosBrowserTest::RefreshDevicePolicy() {
64  // Reset the key to its original state.
65  device_policy_.SetDefaultSigningKey();
66  device_policy_.Build();
67  session_manager_client()->set_device_policy(device_policy_.GetBlob());
68  session_manager_client()->OnPropertyChangeComplete(true);
69}
70
71void DevicePolicyCrosBrowserTest::TearDownInProcessBrowserTestFixture() {
72  InProcessBrowserTest::TearDownInProcessBrowserTestFixture();
73  chromeos::DBusThreadManager::Shutdown();
74}
75
76}  // namespace policy
77